#!/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