mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-12-01 07:34:16 +01:00
56 lines
1.8 KiB
Python
56 lines
1.8 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 json
|
|
import os
|
|
from datetime import datetime
|
|
from urllib.parse import urlparse
|
|
|
|
from svtplay_dl.service import Service
|
|
from svtplay_dl.error import ServiceError
|
|
from svtplay_dl.fetcher.hls import hlsparse
|
|
from svtplay_dl.utils import filenamify
|
|
|
|
|
|
class Atg(Service):
|
|
supported_domains = ["atgplay.se"]
|
|
|
|
def get(self):
|
|
|
|
if self.exclude():
|
|
yield ServiceError("Excluding video")
|
|
return
|
|
|
|
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 = "https://www.atgplay.se/api/{0}/video/{1}".format(current_time, wanted_id)
|
|
video_assets = self.http.request("get", api_url)
|
|
|
|
try:
|
|
janson = json.loads(video_assets.text)
|
|
except json.decoder.JSONDecodeError:
|
|
yield ServiceError("Can't decode api request: {0}".format(video_assets.text))
|
|
return
|
|
|
|
if "title" in janson:
|
|
directory = os.path.dirname(self.options.output)
|
|
title = filenamify(janson["title"])
|
|
self.options.output = os.path.join(directory, title)
|
|
|
|
if "urls" in janson:
|
|
for i in janson["urls"]:
|
|
stream = None
|
|
if "m3u" == i:
|
|
stream = hlsparse(self.options, self.http.request("get", janson["urls"]["m3u"]), janson["urls"]["m3u"])
|
|
|
|
if stream:
|
|
for key in list(stream.keys()):
|
|
yield stream[key]
|