mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-25 04:35:45 +01:00
baa8d76551
self.get_urldata() is eqivalent to get_http_data(self.url), but also caches the data, so no additional requests are made if it is called multiple times (e.g when grabbing title or downloading thumbnail). Generic().get(url) still causes it to be fetched an extra time.
80 lines
3.1 KiB
Python
80 lines
3.1 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 re
|
|
import json
|
|
import sys
|
|
|
|
from svtplay_dl.service import Service, OpenGraphThumbMixin
|
|
from svtplay_dl.utils import get_http_data, subtitle_tt
|
|
from svtplay_dl.fetcher.rtmp import download_rtmp
|
|
from svtplay_dl.fetcher.hls import download_hls
|
|
from svtplay_dl.log import log
|
|
|
|
class Urplay(Service, OpenGraphThumbMixin):
|
|
supported_domains = ['urplay.se', 'ur.se']
|
|
|
|
def __init__(self, url):
|
|
Service.__init__(self, url)
|
|
self.subtitle = None
|
|
|
|
def get(self, options):
|
|
match = re.search(r"urPlayer.init\((.*)\);", self.get_urldata())
|
|
if not match:
|
|
log.error("Can't find json info")
|
|
sys.exit(2)
|
|
data = match.group(1)
|
|
jsondata = json.loads(data)
|
|
self.subtitle = jsondata["subtitles"].split(",")[0]
|
|
basedomain = jsondata["streaming_config"]["streamer"]["redirect"]
|
|
http = "http://%s/%s" % (basedomain, jsondata["file_html5"])
|
|
hd = None
|
|
if len(jsondata["file_html5_hd"]) > 0:
|
|
http_hd = "http://%s/%s" % (basedomain, jsondata["file_html5_hd"])
|
|
hls_hd = "%s%s" % (http_hd, jsondata["streaming_config"]["http_streaming"]["hls_file"])
|
|
tmp = jsondata["file_html5_hd"]
|
|
match = re.search(".*(mp[34]:.*$)", tmp)
|
|
path_hd = match.group(1)
|
|
hd = True
|
|
#hds = "%s%s" % (http, jsondata["streaming_config"]["http_streaming"]["hds_file"])
|
|
hls = "%s%s" % (http, jsondata["streaming_config"]["http_streaming"]["hls_file"])
|
|
rtmp = "rtmp://%s/%s" % (basedomain, jsondata["streaming_config"]["rtmp"]["application"])
|
|
path = "mp%s:%s" % (jsondata["file_flash"][-1], jsondata["file_flash"])
|
|
available = {"sd":{"hls":{"http":http, "playlist":hls}, "rtmp":{"server":rtmp, "path":path}}}
|
|
if hd:
|
|
available.update({"hd":{"hls":{"http":http_hd, "playlist":hls_hd}, "rtmp":{"server":rtmp, "path":path_hd}}})
|
|
|
|
if options.quality:
|
|
try:
|
|
selected = available[options.quality]
|
|
except KeyError:
|
|
log.error("Can't find that quality. (Try one of: %s)",
|
|
", ".join([str(elm) for elm in available]))
|
|
sys.exit(4)
|
|
else:
|
|
try:
|
|
selected = self.select_highest_quality(available)
|
|
except KeyError:
|
|
log.error("Can't find any streams.")
|
|
sys.exit(4)
|
|
|
|
options.other = "-v -a %s -y %s" % (jsondata["streaming_config"]["rtmp"]["application"], selected["rtmp"]["path"])
|
|
if options.hls:
|
|
download_hls(options, selected["hls"]["playlist"])
|
|
else:
|
|
download_rtmp(options, selected["rtmp"]["server"])
|
|
|
|
def select_highest_quality(self, available):
|
|
if 'hd' in available:
|
|
return available["hd"]
|
|
elif 'sd' in available:
|
|
return available["sd"]
|
|
else:
|
|
raise KeyError()
|
|
|
|
|
|
def get_subtitle(self, options):
|
|
if self.subtitle:
|
|
data = get_http_data(self.subtitle)
|
|
subtitle_tt(options, data)
|