1
0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-12-11 12:24:18 +01:00
svtplay-dl/lib/svtplay_dl/service/svt.py
2023-03-16 22:23:02 +01:00

47 lines
1.7 KiB
Python

import codecs
import copy
import json
import re
from svtplay_dl.error import ServiceError
from svtplay_dl.service.svtplay import Svtplay
from svtplay_dl.subtitle import subtitle_probe
class Svt(Svtplay):
supported_domains = ["svt.se", "www.svt.se"]
def get(self):
vid = None
data = self.get_urldata()
match = re.search("n.urqlState=(.*);", data)
if not match:
match = re.search(r"stateData = JSON.parse\(\"(.*)\"\)\<\/script", data)
if not match:
yield ServiceError("Cant find video info.")
return
janson = json.loads(codecs.escape_decode(match.group(1))[0].decode("utf-8"))
if janson["recipe"]["content"]["data"]["videoClips"]:
vid = janson["recipe"]["content"]["data"]["videoClips"][0]["id"]
else:
vid = janson["recipe"]["content"]["data"]["videoEpisodes"][0]["id"]
res = self.http.get(f"https://api.svt.se/videoplayer-api/video/{vid}")
else:
janson = json.loads(match.group(1))
for key in list(janson.keys()):
janson2 = json.loads(janson[key]["data"])
if "page" in janson2 and "topMedia" in janson2["page"]:
vid = janson2["page"]["topMedia"]["svtId"]
if not vid:
yield ServiceError("Can't find any videos")
return
res = self.http.get(f"https://api.svt.se/video/{vid}")
janson = res.json()
if "subtitleReferences" in janson:
for i in janson["subtitleReferences"]:
if "url" in i:
yield from subtitle_probe(copy.copy(self.config), i["url"], output=self.output)
yield from self._get_video(janson)