Saverr/svtplay-dl
1
0
Fork 0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-11-27 21:54:17 +01:00
Code Issues Projects Releases Wiki Activity

raw: refactor filename part

Browse Source
This commit is contained in:
Johan Andersson 2015-01-28 20:57:37 +01:00
parent 68854fc53a
commit a1fdefa0ca
1 changed files with 14 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
lib/svtplay_dl/service/raw.py
View File

@ -17,26 +17,27 @@ class Raw(Service):
if self.exclude(options):
return
extention = False
filename = os.path.basename(self.url[:self.url.rfind("/")-1])
if options.output and os.path.isdir(options.output):
options.output = "%s/%s" % (os.path.dirname(options.output), filename)
extention = True
elif options.output is None:
options.output = "%s" % filename
extention = True
if self.url.find(".f4m") > 0:
filename = os.path.basename(self.url[:self.url.rfind("/")-1])
if options.output and os.path.isdir(options.output):
filename = "%s/%s" % (os.path.dirname(options.output), filename)
options.output = "%s.ts" % filename
elif options.output is None:
options.output = "%s.flv" % filename
if extention:
options.output = "%s.flv" % options.output
streams = hdsparse(copy.copy(options), self.url)
if streams:
for n in list(streams.keys()):
yield streams[n]
if self.url.find(".m3u8") > 0:
streams = hlsparse(self.url)
filename = os.path.basename(self.url[:self.url.rfind("/")-1])
if options.output and os.path.isdir(options.output):
filename = "%s/%s" % (os.path.dirname(options.output), filename)
options.output = "%s.ts" % filename
elif options.output is None:
options.output = "%s.ts" % filename
if extention:
options.output = "%s.ts" % options.output
for n in list(streams.keys()):
yield HLS(copy.copy(options), streams[n], n)