mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-28 06:04:17 +01:00
187 lines
7.2 KiB
Python
187 lines
7.2 KiB
Python
# 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 copy
|
|
import os
|
|
import hashlib
|
|
from svtplay_dl.service import Service, OpenGraphThumbMixin
|
|
from svtplay_dl.error import ServiceError
|
|
from svtplay_dl.log import log
|
|
from svtplay_dl.fetcher.hds import hdsparse
|
|
from svtplay_dl.fetcher.hls import hlsparse
|
|
from svtplay_dl.fetcher.dash import dashparse
|
|
from svtplay_dl.utils import ensure_unicode, filenamify, is_py2, decode_html_entities
|
|
from svtplay_dl.subtitle import subtitle
|
|
from svtplay_dl.utils.urllib import urlparse, parse_qs
|
|
|
|
|
|
class OppetArkiv(Service, OpenGraphThumbMixin):
|
|
supported_domains = ['oppetarkiv.se']
|
|
|
|
def get(self):
|
|
vid = self.find_video_id()
|
|
if vid is None:
|
|
yield ServiceError("Cant find video id for this video")
|
|
return
|
|
|
|
url = "http://api.svt.se/videoplayer-api/video/{0}".format(vid)
|
|
data = self.http.request("get", url)
|
|
if data.status_code == 404:
|
|
yield ServiceError("Can't get the json file for {0}".format(url))
|
|
return
|
|
|
|
data = data.json()
|
|
if "live" in data:
|
|
self.options.live = data["live"]
|
|
|
|
if self.options.output_auto:
|
|
self.options.service = "svtplay"
|
|
self.options.output = self.outputfilename(data, self.options.output, ensure_unicode(self.get_urldata()))
|
|
|
|
if self.exclude():
|
|
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 len(data["videoReferences"]) == 0:
|
|
yield ServiceError("Media doesn't have any associated videos (yet?)")
|
|
return
|
|
|
|
for i in data["videoReferences"]:
|
|
parse = urlparse(i["url"])
|
|
query = parse_qs(parse.query)
|
|
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 "alt" in query and len(query["alt"]) > 0:
|
|
alt = self.http.get(query["alt"][0])
|
|
if alt:
|
|
streams = hlsparse(self.options, self.http.request("get", alt.request.url), alt.request.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]
|
|
if "alt" in query and len(query["alt"]) > 0:
|
|
alt = self.http.get(query["alt"][0])
|
|
if alt:
|
|
streams = hdsparse(self.options,
|
|
self.http.request("get", alt.request.url, params={"hdcore": "3.7.0"}),
|
|
alt.request.url)
|
|
if streams:
|
|
for n in list(streams.keys()):
|
|
yield streams[n]
|
|
if i["format"] == "dash264" or i["format"] == "dashhbbtv":
|
|
streams = dashparse(self.options, self.http.request("get", i["url"]), i["url"])
|
|
if streams:
|
|
for n in list(streams.keys()):
|
|
yield streams[n]
|
|
|
|
if "alt" in query and len(query["alt"]) > 0:
|
|
alt = self.http.get(query["alt"][0])
|
|
if alt:
|
|
streams = dashparse(self.options, self.http.request("get", alt.request.url), alt.request.url)
|
|
if streams:
|
|
for n in list(streams.keys()):
|
|
yield streams[n]
|
|
|
|
def find_video_id(self):
|
|
match = re.search('data-video-id="([^"]+)"', self.get_urldata())
|
|
if match:
|
|
return match.group(1)
|
|
return None
|
|
|
|
def find_all_episodes(self, options):
|
|
page = 1
|
|
data = self.get_urldata()
|
|
match = re.search(r'"/etikett/titel/([^"/]+)', data)
|
|
if match is None:
|
|
match = re.search(r'"http://www.oppetarkiv.se/etikett/titel/([^/]+)/', self.url)
|
|
if match is None:
|
|
log.error("Couldn't find title")
|
|
return
|
|
program = match.group(1)
|
|
episodes = []
|
|
|
|
n = 0
|
|
if self.options.all_last > 0:
|
|
sort = "tid_fallande"
|
|
else:
|
|
sort = "tid_stigande"
|
|
|
|
while True:
|
|
url = "http://www.oppetarkiv.se/etikett/titel/{0}/?sida={1}&sort={2}&embed=true".format(program, page, sort)
|
|
data = self.http.request("get", url)
|
|
if data.status_code == 404:
|
|
break
|
|
|
|
data = data.text
|
|
regex = re.compile(r'href="(/video/[^"]+)"')
|
|
for match in regex.finditer(data):
|
|
if n == self.options.all_last:
|
|
break
|
|
episodes.append("http://www.oppetarkiv.se{0}".format(match.group(1)))
|
|
n += 1
|
|
page += 1
|
|
|
|
return episodes
|
|
|
|
def outputfilename(self, data, filename, raw):
|
|
directory = os.path.dirname(filename)
|
|
if is_py2:
|
|
id = hashlib.sha256(data["programVersionId"]).hexdigest()[:7]
|
|
else:
|
|
id = hashlib.sha256(data["programVersionId"].encode("utf-8")).hexdigest()[:7]
|
|
|
|
datatitle = re.search('data-title="([^"]+)"', self.get_urldata())
|
|
if not datatitle:
|
|
return None
|
|
datat = decode_html_entities(datatitle.group(1))
|
|
name = self.name(datat)
|
|
episode = self.seasoninfo(datat)
|
|
if is_py2:
|
|
name = name.encode("utf8")
|
|
if episode:
|
|
title = "{0}.{1}-{2}-svtplay".format(name, episode, id)
|
|
else:
|
|
title = "{0}-{1}-svtplay".format(name, id)
|
|
title = filenamify(title)
|
|
if len(directory):
|
|
output = os.path.join(directory, title)
|
|
else:
|
|
output = title
|
|
return output
|
|
|
|
def seasoninfo(self, data):
|
|
episode = None
|
|
match = re.search("S.song (\d+) - Avsnitt (\d+)", data)
|
|
if match:
|
|
episode = "s{0:02d}e{1:02d}".format(int(match.group(1)), int(match.group(2)))
|
|
else:
|
|
match = re.search("Avsnitt (\d+)", data)
|
|
if match:
|
|
episode = "e{0:02d}".format(int(match.group(1)))
|
|
return episode
|
|
|
|
def name(selfs, data):
|
|
if data.find(" - S.song") > 0:
|
|
title = data[:data.find(" - S.song")]
|
|
else:
|
|
if data.find(" - Avsnitt") > 0:
|
|
title = data[:data.find(" - Avsnitt")]
|
|
else:
|
|
title = data
|
|
return title
|