1
0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-11-30 15:14:14 +01:00
svtplay-dl/lib/svtplay_dl/service/atg.py
2021-05-16 13:55:28 +02:00

41 lines
1.3 KiB
Python

# ex:ts=4:sw=4:sts=4:et
# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
import json
from datetime import datetime
from urllib.parse import urlparse
from svtplay_dl.error import ServiceError
from svtplay_dl.fetcher.hls import hlsparse
from svtplay_dl.service import Service
class Atg(Service):
supported_domains = ["atgplay.se"]
def get(self):
parse = urlparse(self.url)
if not parse.path.startswith("/video"):
yield ServiceError("Can't find video info")
return
wanted_id = parse.path[7:]
current_time = int((datetime.utcnow() - datetime.utcfromtimestamp(0)).total_seconds() * 1000)
api_url = f"https://www.atgplay.se/api/{current_time}/video/{wanted_id}"
video_assets = self.http.request("get", api_url)
try:
janson = json.loads(video_assets.text)
except json.decoder.JSONDecodeError:
yield ServiceError(f"Can't decode api request: {video_assets.text}")
return
if "title" in janson:
self.output["title"] = janson["title"]
if "urls" in janson:
for i in janson["urls"]:
if "m3u" == i:
yield from hlsparse(self.config, self.http.request("get", janson["urls"]["m3u"]), janson["urls"]["m3u"], output=self.output)