1
0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-11-24 12:15:40 +01:00
svtplay-dl/lib/svtplay_dl/service/svtplay.py

181 lines
6.6 KiB
Python
Raw Normal View History

2013-03-02 21:26:28 +01:00
# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
from __future__ import absolute_import
import re
import os
import xml.etree.ElementTree as ET
2014-06-07 20:43:40 +02:00
import copy
import hashlib
from svtplay_dl.log import log
from svtplay_dl.service import Service, OpenGraphThumbMixin
from svtplay_dl.utils import filenamify, ensure_unicode
from svtplay_dl.utils.urllib import urlparse, urljoin
2014-04-27 20:48:13 +02:00
from svtplay_dl.fetcher.hds import hdsparse
from svtplay_dl.fetcher.hls import hlsparse
from svtplay_dl.fetcher.http import HTTP
from svtplay_dl.subtitle import subtitle
from svtplay_dl.error import ServiceError
2015-09-15 20:10:32 +02:00
class Svtplay(Service, OpenGraphThumbMixin):
supported_domains = ['svtplay.se', 'svt.se', 'beta.svtplay.se', 'svtflow.se']
def get(self):
old = False
parse = urlparse(self.url)
if parse.netloc == "www.svtplay.se" or parse.netloc == "svtplay.se":
if parse.path[:6] != "/video":
yield ServiceError("This mode is not supported anymore. need the url with the video")
return
match = re.search('data-video-id="([^"]+)"', self.get_urldata())
if not match:
match = re.search("/video/([0-9]+)/", parse.path)
if not match:
yield ServiceError("Cant find video id for this video")
return
vid = match.group(1)
if re.match("^[0-9]+$", vid):
old = True
2015-10-29 18:08:25 +01:00
url = "http://www.svt.se/videoplayer-api/video/%s" % vid
data = self.http.request("get", url)
2015-08-30 11:27:31 +02:00
if data.status_code == 404:
2015-10-29 18:08:25 +01:00
yield ServiceError("Can't get the json file for %s" % url)
2015-08-30 11:27:31 +02:00
return
data = data.json()
if "live" in data:
self.options.live = data["live"]
if old:
params = {"output": "json"}
dataj = self.http.request("get", self.url, params=params).json()
else:
dataj = data
if self.options.output_auto:
self.options.service = "svtplay"
self.options.output = self.outputfilename(dataj, self.options.output, ensure_unicode(self.get_urldata()))
if self.exclude(self.options):
2015-09-06 23:04:48 +02:00
yield ServiceError("Excluding video")
return
if "subtitleReferences" in data:
for i in data["subtitleReferences"]:
if i["format"] == "websrt":
yield subtitle(copy.copy(self.options), "wrst", i["url"])
if old and dataj["video"]["subtitleReferences"]:
try:
suburl = dataj["video"]["subtitleReferences"][0]["url"]
except KeyError:
pass
if suburl and len(suburl) > 0:
yield subtitle(copy.copy(self.options), "wrst", suburl)
if self.options.force_subtitle:
return
if len(data["videoReferences"]) == 0:
yield ServiceError("Media doesn't have any associated videos (yet?)")
return
for i in data["videoReferences"]:
if i["format"] == "hls" or i["format"] == "ios":
streams = hlsparse(self.options, self.http.request("get", i["url"]), i["url"])
if streams:
for n in list(streams.keys()):
yield streams[n]
if i["format"] == "hds" or i["format"] == "flash":
match = re.search(r"\/se\/secure\/", i["url"])
if not match:
streams = hdsparse(self.options, self.http.request("get", i["url"], params={"hdcore": "3.7.0"}), i["url"])
if streams:
for n in list(streams.keys()):
yield streams[n]
def find_all_episodes(self, options):
match = re.search(r'<link rel="alternate" type="application/rss\+xml" [^>]*href="([^"]+)"',
self.get_urldata())
if match is None:
match = re.findall(r'a class="play[^"]+"\s+href="(/video[^"]+)"', self.get_urldata())
if not match:
log.error("Couldn't retrieve episode list")
return
episodes = [urljoin("http://www.svtplay.se", x) for x in match]
else:
2015-09-01 23:53:13 +02:00
data = self.http.request("get", match.group(1)).content
xml = ET.XML(data)
episodes = [x.text for x in xml.findall(".//item/link")]
episodes_new = []
n = 1
for i in episodes:
episodes_new.append(i)
if n == options.all_last:
break
n += 1
return sorted(episodes_new)
def outputfilename(self, data, filename, raw):
directory = os.path.dirname(filename)
if "statistics" in data:
name = data["statistics"]["folderStructure"]
if name.find(".") > 0:
name = name[:name.find(".")]
match = re.search("^arkiv-", name)
if match:
name = name.replace("arkiv-", "")
name = name.replace("-", ".")
other = filenamify(data["context"]["title"])
id = data["videoId"]
else:
name = data["programTitle"]
if name.find(".") > 0:
name = name[:name.find(".")]
name = name.replace(" - ", ".")
other = filenamify(data["episodeTitle"])
id = hashlib.sha256(data["programVersionId"]).hexdigest()[:7]
season = self.seasoninfo(raw)
if season:
title = "%s.%s.%s-%s-svtplay" % (name, season, other, id)
else:
title = "%s.%s-%s-svtplay" % (name, other, id)
title = filenamify(title)
if len(directory):
output = os.path.join(directory, title)
else:
output = title
return output
def seasoninfo(self, data):
match = re.search(r'play_video-area-aside__sub-title">([^<]+)<span', data)
if match:
line = match.group(1)
else:
match = re.search(r'data-title="([^"]+)"', data)
if match:
line = match.group(1)
else:
return None
line = re.sub(" +", "", match.group(1)).replace('\n', '')
match = re.search(r"(song(\d+)-)?Avsnitt(\d+)", line)
if match:
if match.group(2) is None:
season = 1
else:
season = int(match.group(2))
if season < 10:
season = "0%s" % season
episode = int(match.group(3))
if episode < 10:
episode = "0%s" % episode
return "S%sE%s" % (season, episode)
else:
return None