mirror of
https://github.com/soimort/you-get.git
synced 2025-02-11 20:52:31 +03:00
56 lines
1.4 KiB
Python
56 lines
1.4 KiB
Python
|
#!/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
|
||
|
|
|
||
|
|
#
|
||
|
|
#params ={"-y":"playlist","-q":None,}
|
||
|
|
#if Only Key ,Value should be None
|
||
|
|
#-r -o should not be included in params
|
||
|
|
|
||
|
|
def download_rtmpdump_stream(url, title, ext,params={},output_dir='.'):
|
||
|
|
filename = '%s.%s' % (title, ext)
|
||
|
|
filepath = os.path.join(output_dir, filename)
|
||
|
|
|
||
|
|
cmdline = [RTMPDUMP, '-r']
|
||
|
|
cmdline.append(url)
|
||
|
|
cmdline.append('-o')
|
||
|
|
cmdline.append(filepath)
|
||
|
|
|
||
|
|
for key in params.keys():
|
||
|
|
cmdline.append(key)
|
||
|
|
if params[key]!=None:
|
||
|
|
cmdline.append(params[key])
|
||
|
|
|
||
|
|
# cmdline.append('-y')
|
||
|
|
# cmdline.append(playpath)
|
||
|
|
print("Call rtmpdump:\n"+" ".join(cmdline)+"\n")
|
||
|
|
subprocess.call(cmdline)
|
||
|
|
return
|
||
|
|
|
||
|
|
#
|
||
|
|
#To be refactor
|
||
|
|
#
|
||
|
|
def play_rtmpdump_stream(player, url, params={}):
|
||
|
|
cmdline="rtmpdump -r '%s' "%url
|
||
|
|
for key in params.keys():
|
||
|
|
cmdline+=key+" "+params[key] if params[key]!=None else ""+" "
|
||
|
|
cmdline+=" -o - | %s -"%player
|
||
|
|
print(cmdline)
|
||
|
|
os.system(cmdline)
|
||
|
|
# os.system("rtmpdump -r '%s' -y '%s' -o - | %s -" % (url, playpath, player))
|
||
|
|
return
|