1
0
mirror of https://github.com/spaam/svtplay-dl.git synced 2024-12-03 00:24:21 +01:00
svtplay-dl/lib/svtplay_dl/service/dr.py
Anders Waldenborg baa8d76551 Add get_urldata() method to service
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.
2014-02-18 19:00:20 +01:00

74 lines
2.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 re
import json
import sys
from svtplay_dl.service import Service, OpenGraphThumbMixin
from svtplay_dl.utils import get_http_data, select_quality
from svtplay_dl.fetcher.rtmp import download_rtmp
from svtplay_dl.fetcher.hls import download_hls
from svtplay_dl.log import log
class Dr(Service, OpenGraphThumbMixin):
supported_domains = ['dr.dk']
def get(self, options):
data = self.get_urldata()
match = re.search(r'resource:[ ]*"([^"]*)",', data)
if match:
resource_url = match.group(1)
resource_data = get_http_data(resource_url)
resource = json.loads(resource_data)
tempresource = resource['Data'][0]['Assets']
# To find the VideoResource, they have Images as well
for resources in tempresource:
if resources['Kind'] == 'VideoResource':
links = resources['Links']
break
streams = {}
for i in links:
if options.hls:
if i["Target"] == "Ios":
stream = {}
stream["uri"] = i["Uri"]
streams[int(i["Bitrate"])] = stream
else:
if i["Target"] == "Streaming":
stream = {}
stream["uri"] = i["Uri"]
streams[int(i["Bitrate"])] = stream
if len(streams) == 1:
test = streams[list(streams.keys())[0]]
else:
test = select_quality(options, streams)
if options.hls:
download_hls(options, test["uri"])
else:
options.other = "-y '%s'" % test["uri"].replace("rtmp://vod.dr.dk/cms/", "")
rtmp = "rtmp://vod.dr.dk/cms/"
download_rtmp(options, rtmp)
else:
match = re.search(r'resource="([^"]*)"', data)
if not match:
log.error("Cant find resource info for this video")
sys.exit(2)
resource_url = "http://www.dr.dk%s" % match.group(1)
resource_data = get_http_data(resource_url)
resource = json.loads(resource_data)
streams = {}
for stream in resource['links']:
streams[stream['bitrateKbps']] = stream['uri']
if len(streams) == 1:
uri = streams[list(streams.keys())[0]]
else:
uri = select_quality(options, streams)
options.other = "-v -y '" + uri.replace("rtmp://vod.dr.dk/cms/", "") + "'"
rtmp = "rtmp://vod.dr.dk/cms/"
download_rtmp(options, rtmp)