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

svtplay: Add season and episode info in the filename

Browse Source
This commit is contained in:
Johan Andersson 2014-12-28 13:57:50 +01:00
parent d152e02d06
commit 4519897cfe
1 changed files with 38 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
lib/svtplay_dl/service/svtplay.py
View File

@ -59,19 +59,8 @@ class Svtplay(Service, OpenGraphThumbMixin):
options.live = False
if options.output_auto:
directory = os.path.dirname(options.output)
options.service = "svtplay"
name = data["statistics"]["folderStructure"]
if name.find(".") > 0:
title = "%s-%s-%s-%s" % (name[:name.find(".")], data["statistics"]["title"], data["videoId"], options.service)
else:
title = "%s-%s-%s-%s" % (name, data["statistics"]["title"], data["videoId"], options.service)
title = filenamify(title)
if len(directory):
options.output = "%s/%s" % (directory, title)
else:
options.output = title
options.output = outputfilename(data, options.output, self.get_urldata()[1])
if self.exclude(options):
return
@ -134,3 +123,40 @@ class Svtplay(Service, OpenGraphThumbMixin):
break
n += 1
return sorted(episodes_new)
def outputfilename(data, filename, raw):
directory = os.path.dirname(filename)
name = data["statistics"]["folderStructure"]
if name.find(".") > 0:
name = name[:name.find(".")]
name = name.replace("-", ".")
season = seasoninfo(raw)
other = data["statistics"]["title"].replace("-",".")
if season:
title = "%s.%s.%s-%s-svtplay" % (name, season, other, data["videoId"])
else:
title = "%s.%s.%s-svtplay" % (name, other, data["videoId"])
title = filenamify(title)
if len(directory):
output = "%s/%s" % (directory, title)
else:
output = title
return output
def seasoninfo(line):
match = re.search(r'play_video-area-aside__sub-title">([^<]+)<span', line)
if match:
line = re.sub(" +", "", match.group(1)).replace('\n','')
match = re.search(r"(\w+(\d+)-)?Avsnitt(\d+)", line)
if match:
if match.group(2) is None:
season = 1
else:
season = match.group(2)
return "S%sE%s" % (season, match.group(3))
else:
return None
else:
return None