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

pokemon: make it work with the website

Browse Source
This commit is contained in:
Johan Andersson 2021-09-08 10:16:00 +02:00
parent 404373ee30
commit 1737a162e2
1 changed files with 18 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
lib/svtplay_dl/service/pokemon.py
View File

@ -8,31 +8,32 @@ from svtplay_dl.service import Service
class Pokemon(Service, OpenGraphThumbMixin):
supported_domains = ["pokemon.com"]
supported_domains = ["watch.pokemon.com"]
def get(self):
data = self.get_urldata()
parse = urlparse(self.url)
match = re.search(r"^/([a-z]{2})/", parse.path)
if not match:
yield ServiceError("Cant county code")
if parse.fragment == "":
yield ServiceError("need the whole url")
return
res = self.http.get(f"http://www.pokemon.com/api/pokemontv/channels?region={match.group(1)}")
janson = res.json()
match = re.search('data-video-season="([0-9]+)"', data)
season = match.group(1)
match = re.search('data-video-episode="([0-9]+)"', data)
episode = match.group(1)
match = re.search(r"id=([a-f0-9]+)\&", parse.fragment)
if not match:
yield ServiceError("Cant find the ID in the url")
return
res = self.http.get("https://www.pokemon.com/api/pokemontv/v2/channels/us/")
janson = res.json()
stream = None
for i in janson:
for n in i["media"]:
if season == n["season"] and episode == n["episode"]:
stream = n["stream_url"]
if n["id"] == match.group(1):
stream = n
break
self.output["title"] = "pokemon"
self.output["season"] = season
self.output["episode"] = episode
self.output["season"] = stream["season"]
self.output["episode"] = stream["episode"]
self.output["episodename"] = stream["title"]
yield from hlsparse(self.config, self.http.request("get", stream), stream, output=self.output)
yield from hlsparse(self.config, self.http.request("get", stream["stream_url"]), stream["stream_url"], output=self.output)