Initial upload.
[svtplaydump.git] / svtplaydump.py
1 #!/usr/bin/env python
2 #
3 # (C) Copyright 2010 Mikael Frykholm <mikael@frykholm.com>
4 #
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU General Public License as published by
7 # the Free Software Foundation, either version 3 of the License, or
8 # (at your option) any later version.
9 #
10 # This program is distributed in the hope that it will be useful,
11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 # GNU General Public License for more details.
14 #
15 # You should have received a copy of the GNU General Public License
16 # along with this program. If not, see <http://www.gnu.org/licenses/>
17 #
18 # Changelog:
19 # 0.2 added python 2.4 urlparse compatibility
20 # 0.1 initial release
21
22 from BeautifulSoup import BeautifulSoup
23 from subprocess import *
24 try:
25 import urlparse
26 except ImportError:
27 pass
28 import urllib2
29 try:
30 import urllib2.urlparse as urlparse
31 except ImportError:
32 pass
33 import sys
34
35 def main(argv=None):
36 if argv is None:
37 argv=sys.argv
38 soup = BeautifulSoup(urllib2.urlopen(argv[1]).read())
39 flashvars = urlparse.parse_qs(soup.find("param", {"name":"flashvars",'value':True})['value'])
40 title = None
41 try:
42 title = soup.find("div","info").ul.li.h2.string
43 except:
44 title = "unnamed"
45 if 'dynamicStreams' in flashvars:
46 url = flashvars['dynamicStreams'][0].split('url:')[1].split('.mp4,')[0] +'.mp4'
47 filename = title+".mp4"
48 print Popen(["rtmpdump",u"-o"+filename,"-r", url], stdout=PIPE).communicate()[0]
49 if 'pathflv' in flashvars:
50 rtmp = flashvars['pathflv'][0]
51 filename = title+".flv"
52 print Popen(["mplayer","-dumpstream","-dumpfile",filename, rtmp], stdout=PIPE).communicate()[0]
53 else:
54 print "Could not find any streams"
55 return
56
57 if __name__ == "__main__":
58 sys.exit(main())