mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-30 15:14:14 +01:00
97 lines
3.5 KiB
Python
97 lines
3.5 KiB
Python
# ex:ts=4:sw=4:sts=4:et
|
|
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
|
|
import json
|
|
import logging
|
|
import re
|
|
from urllib.parse import urlparse
|
|
|
|
from svtplay_dl.error import ServiceError
|
|
from svtplay_dl.service.svtplay import Svtplay
|
|
|
|
|
|
class Barnkanalen(Svtplay):
|
|
supported_domains = ["svt.se"]
|
|
supported_path = "/barnkanalen"
|
|
info_search_expr = r"<script id=\"__NEXT_DATA__\" type=\"application\/json\">({.+})<\/script>"
|
|
|
|
@classmethod
|
|
def handles(cls, url):
|
|
urlp = urlparse(url)
|
|
|
|
correctpath = urlp.path.startswith(cls.supported_path)
|
|
if urlp.netloc in cls.supported_domains and correctpath:
|
|
return True
|
|
|
|
# For every listed domain, try with www. subdomain as well.
|
|
if urlp.netloc in ["www." + x for x in cls.supported_domains] and correctpath:
|
|
return True
|
|
|
|
return False
|
|
|
|
def get(self):
|
|
data = self.get_urldata()
|
|
match = re.search(self.info_search_expr, data)
|
|
if not match:
|
|
yield ServiceError("Can't find video info.")
|
|
return
|
|
|
|
janson = json.loads(match.group(1))
|
|
if "episodeId" not in janson["query"]:
|
|
yield ServiceError("need a url with one to be able to download")
|
|
return
|
|
|
|
vid = janson["query"]["episodeId"]
|
|
title = janson["props"]["pageProps"]["initialState"]["featuredTitle"]["name"]
|
|
seasonnr = None
|
|
episodenr = None
|
|
episodename = None
|
|
for season in janson["props"]["pageProps"]["initialState"]["featuredTitle"]["associatedContent"]:
|
|
tmp_season = season["name"]
|
|
for episode in season["items"]:
|
|
if "variants" in episode["item"]:
|
|
svtID = episode["item"]["variants"][0]
|
|
else:
|
|
svtID = episode["item"]
|
|
if vid == svtID["svtId"]:
|
|
match = re.search(r"S.song (\d+)", tmp_season)
|
|
if match:
|
|
seasonnr = match.group(1)
|
|
match = re.search(r"^(\d+)\. (.*)$", episode["item"]["name"])
|
|
if match:
|
|
episodenr = match.group(1)
|
|
episodename = match.group(2)
|
|
else:
|
|
episodename = episode["item"]["name"]
|
|
break
|
|
|
|
self.output["title"] = title
|
|
self.output["id"] = vid.lower()
|
|
self.output["season"] = seasonnr
|
|
self.output["episode"] = episodenr
|
|
self.output["episodename"] = episodename
|
|
res = self.http.get(f"https://api.svt.se/video/{vid}")
|
|
janson = res.json()
|
|
yield from self._get_video(janson)
|
|
|
|
def find_all_episodes(self, config):
|
|
episodes = []
|
|
data = self.get_urldata()
|
|
match = re.search(self.info_search_expr, data)
|
|
if not match:
|
|
logging.error("Can't find video info.")
|
|
return episodes
|
|
|
|
janson = json.loads(match.group(1))
|
|
title = janson["query"]["titleSlug"]
|
|
|
|
for season in janson["props"]["pageProps"]["initialState"]["featuredTitle"]["associatedContent"]:
|
|
for episode in season["items"]:
|
|
if "variants" in episode["item"]:
|
|
svtID = episode["item"]["variants"][0]
|
|
else:
|
|
svtID = episode["item"]
|
|
episodes.append(f"https://www.svt.se/barnkanalen/barnplay/{title}/{svtID['svtId']}/")
|
|
if config.get("all_last") > 0:
|
|
return episodes[-config.get("all_last") :]
|
|
return episodes
|