add rtmpdump wrapper

This commit is contained in:
Mort Yao 2014-04-26 15:17:45 +02:00
parent 0d19e034dd
commit b3d10e32a9
2 changed files with 32 additions and 0 deletions

View File

@ -3,3 +3,4 @@
from .join_flv import concat_flv from .join_flv import concat_flv
from .join_mp4 import concat_mp4 from .join_mp4 import concat_mp4
from .ffmpeg import * from .ffmpeg import *
from .rtmpdump import *

View File

@ -0,0 +1,31 @@
#!/usr/bin/env python
import os.path
import subprocess
def get_usable_rtmpdump(cmd):
try:
p = subprocess.Popen([cmd], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
out, err = p.communicate()
return cmd
except:
return None
RTMPDUMP = get_usable_rtmpdump('rtmpdump')
def has_rtmpdump_installed():
return RTMPDUMP is not None
def download_rtmpdump_stream(url, playpath, title, ext, output_dir='.'):
filename = '%s.%s' % (title, ext)
filepath = os.path.join(output_dir, filename)
params = [RTMPDUMP, '-r']
params.append(url)
params.append('-y')
params.append(playpath)
params.append('-o')
params.append(filepath)
subprocess.call(params)
return