mirror of https://github.com/yt-dlp/yt-dlp.git
[zdf] Fix extraction
This commit is contained in:
Paul Hartmann
Sergey M․
parent
96d315c2be
commit
b6de53ea8a
|
|
@ -6,207 +6,251 @@ import re
|
|||
|
||||
from .common import InfoExtractor
|
||||
from ..utils import (
|
||||
int_or_none,
|
||||
unified_strdate,
|
||||
OnDemandPagedList,
|
||||
xpath_text,
|
||||
determine_ext,
|
||||
qualities,
|
||||
float_or_none,
|
||||
ExtractorError,
|
||||
parse_iso8601,
|
||||
ExtractorError
|
||||
)
|
||||
|
||||
from ..compat import compat_str
|
||||
|
||||
class ZDFIE(InfoExtractor):
|
||||
_VALID_URL = r'(?:zdf:|zdf:video:|https?://www\.zdf\.de/ZDFmediathek(?:#)?/(.*beitrag/(?:video/)?))(?P<id>[0-9]+)(?:/[^/?]+)?(?:\?.*)?'
|
||||
_VALID_URL = r'https?://www\.zdf\.de/.*?/(?P<id>[^/?]*?)\.html'
|
||||
|
||||
_TESTS = [{
|
||||
'url': 'http://www.zdf.de/ZDFmediathek/beitrag/video/2037704/ZDFspezial---Ende-des-Machtpokers--?bc=sts;stt',
|
||||
'url': 'https://www.zdf.de/service-und-hilfe/die-neue-zdf-mediathek/zdfmediathek-trailer-100.html',
|
||||
'info_dict': {
|
||||
'id': '2037704',
|
||||
'ext': 'webm',
|
||||
'title': 'ZDFspezial - Ende des Machtpokers',
|
||||
'description': 'Union und SPD haben sich auf einen Koalitionsvertrag geeinigt. Aber was bedeutet das für die Bürger? Sehen Sie hierzu das ZDFspezial "Ende des Machtpokers - Große Koalition für Deutschland".',
|
||||
'duration': 1022,
|
||||
'uploader': 'spezial',
|
||||
'uploader_id': '225948',
|
||||
'upload_date': '20131127',
|
||||
},
|
||||
'skip': 'Videos on ZDF.de are depublicised in short order',
|
||||
}]
|
||||
|
||||
def _parse_smil_formats(self, smil, smil_url, video_id, namespace=None, f4m_params=None, transform_rtmp_url=None):
|
||||
param_groups = {}
|
||||
for param_group in smil.findall(self._xpath_ns('./head/paramGroup', namespace)):
|
||||
group_id = param_group.attrib.get(self._xpath_ns('id', 'http://www.w3.org/XML/1998/namespace'))
|
||||
params = {}
|
||||
for param in param_group:
|
||||
params[param.get('name')] = param.get('value')
|
||||
param_groups[group_id] = params
|
||||
|
||||
formats = []
|
||||
for video in smil.findall(self._xpath_ns('.//video', namespace)):
|
||||
src = video.get('src')
|
||||
if not src:
|
||||
continue
|
||||
bitrate = float_or_none(video.get('system-bitrate') or video.get('systemBitrate'), 1000)
|
||||
group_id = video.get('paramGroup')
|
||||
param_group = param_groups[group_id]
|
||||
for proto in param_group['protocols'].split(','):
|
||||
formats.append({
|
||||
'url': '%s://%s' % (proto, param_group['host']),
|
||||
'app': param_group['app'],
|
||||
'play_path': src,
|
||||
'ext': 'flv',
|
||||
'format_id': '%s-%d' % (proto, bitrate),
|
||||
'tbr': bitrate,
|
||||
})
|
||||
self._sort_formats(formats)
|
||||
return formats
|
||||
|
||||
def extract_from_xml_url(self, video_id, xml_url):
|
||||
doc = self._download_xml(
|
||||
xml_url, video_id,
|
||||
note='Downloading video info',
|
||||
errnote='Failed to download video info')
|
||||
|
||||
status_code = doc.find('./status/statuscode')
|
||||
if status_code is not None and status_code.text != 'ok':
|
||||
code = status_code.text
|
||||
if code == 'notVisibleAnymore':
|
||||
message = 'Video %s is not available' % video_id
|
||||
else:
|
||||
message = '%s returned error: %s' % (self.IE_NAME, code)
|
||||
raise ExtractorError(message, expected=True)
|
||||
|
||||
title = doc.find('.//information/title').text
|
||||
description = xpath_text(doc, './/information/detail', 'description')
|
||||
duration = int_or_none(xpath_text(doc, './/details/lengthSec', 'duration'))
|
||||
uploader = xpath_text(doc, './/details/originChannelTitle', 'uploader')
|
||||
uploader_id = xpath_text(doc, './/details/originChannelId', 'uploader id')
|
||||
upload_date = unified_strdate(xpath_text(doc, './/details/airtime', 'upload date'))
|
||||
subtitles = {}
|
||||
captions_url = doc.find('.//caption/url')
|
||||
if captions_url is not None:
|
||||
subtitles['de'] = [{
|
||||
'url': captions_url.text,
|
||||
'ext': 'ttml',
|
||||
}]
|
||||
|
||||
def xml_to_thumbnails(fnode):
|
||||
thumbnails = []
|
||||
for node in fnode:
|
||||
thumbnail_url = node.text
|
||||
if not thumbnail_url:
|
||||
continue
|
||||
thumbnail = {
|
||||
'url': thumbnail_url,
|
||||
}
|
||||
if 'key' in node.attrib:
|
||||
m = re.match('^([0-9]+)x([0-9]+)$', node.attrib['key'])
|
||||
if m:
|
||||
thumbnail['width'] = int(m.group(1))
|
||||
thumbnail['height'] = int(m.group(2))
|
||||
thumbnails.append(thumbnail)
|
||||
return thumbnails
|
||||
|
||||
thumbnails = xml_to_thumbnails(doc.findall('.//teaserimages/teaserimage'))
|
||||
|
||||
format_nodes = doc.findall('.//formitaeten/formitaet')
|
||||
quality = qualities(['veryhigh', 'high', 'med', 'low'])
|
||||
|
||||
def get_quality(elem):
|
||||
return quality(xpath_text(elem, 'quality'))
|
||||
format_nodes.sort(key=get_quality)
|
||||
format_ids = []
|
||||
formats = []
|
||||
for fnode in format_nodes:
|
||||
video_url = fnode.find('url').text
|
||||
is_available = 'http://www.metafilegenerator' not in video_url
|
||||
if not is_available:
|
||||
continue
|
||||
format_id = fnode.attrib['basetype']
|
||||
quality = xpath_text(fnode, './quality', 'quality')
|
||||
format_m = re.match(r'''(?x)
|
||||
(?P<vcodec>[^_]+)_(?P<acodec>[^_]+)_(?P<container>[^_]+)_
|
||||
(?P<proto>[^_]+)_(?P<index>[^_]+)_(?P<indexproto>[^_]+)
|
||||
''', format_id)
|
||||
|
||||
ext = determine_ext(video_url, None) or format_m.group('container')
|
||||
if ext not in ('smil', 'f4m', 'm3u8'):
|
||||
format_id = format_id + '-' + quality
|
||||
if format_id in format_ids:
|
||||
continue
|
||||
|
||||
if ext == 'meta':
|
||||
continue
|
||||
elif ext == 'smil':
|
||||
formats.extend(self._extract_smil_formats(
|
||||
video_url, video_id, fatal=False))
|
||||
elif ext == 'm3u8':
|
||||
# the certificates are misconfigured (see
|
||||
# https://github.com/rg3/youtube-dl/issues/8665)
|
||||
if video_url.startswith('https://'):
|
||||
continue
|
||||
formats.extend(self._extract_m3u8_formats(
|
||||
video_url, video_id, 'mp4', m3u8_id=format_id, fatal=False))
|
||||
elif ext == 'f4m':
|
||||
formats.extend(self._extract_f4m_formats(
|
||||
video_url, video_id, f4m_id=format_id, fatal=False))
|
||||
else:
|
||||
proto = format_m.group('proto').lower()
|
||||
|
||||
abr = int_or_none(xpath_text(fnode, './audioBitrate', 'abr'), 1000)
|
||||
vbr = int_or_none(xpath_text(fnode, './videoBitrate', 'vbr'), 1000)
|
||||
|
||||
width = int_or_none(xpath_text(fnode, './width', 'width'))
|
||||
height = int_or_none(xpath_text(fnode, './height', 'height'))
|
||||
|
||||
filesize = int_or_none(xpath_text(fnode, './filesize', 'filesize'))
|
||||
|
||||
format_note = ''
|
||||
if not format_note:
|
||||
format_note = None
|
||||
|
||||
formats.append({
|
||||
'format_id': format_id,
|
||||
'url': video_url,
|
||||
'ext': ext,
|
||||
'acodec': format_m.group('acodec'),
|
||||
'vcodec': format_m.group('vcodec'),
|
||||
'abr': abr,
|
||||
'vbr': vbr,
|
||||
'width': width,
|
||||
'height': height,
|
||||
'filesize': filesize,
|
||||
'format_note': format_note,
|
||||
'protocol': proto,
|
||||
'_available': is_available,
|
||||
})
|
||||
format_ids.append(format_id)
|
||||
|
||||
self._sort_formats(formats)
|
||||
|
||||
return {
|
||||
'id': video_id,
|
||||
'title': title,
|
||||
'description': description,
|
||||
'duration': duration,
|
||||
'thumbnails': thumbnails,
|
||||
'uploader': uploader,
|
||||
'uploader_id': uploader_id,
|
||||
'upload_date': upload_date,
|
||||
'formats': formats,
|
||||
'subtitles': subtitles,
|
||||
'id': 'zdfmediathek-trailer-100',
|
||||
'ext': 'mp4',
|
||||
'title': 'Trailer ZDFmediathek Supermarkt',
|
||||
}
|
||||
}]
|
||||
|
||||
def _real_extract(self, url):
|
||||
video_id = self._match_id(url)
|
||||
xml_url = 'http://www.zdf.de/ZDFmediathek/xmlservice/web/beitragsDetails?ak=web&id=%s' % video_id
|
||||
return self.extract_from_xml_url(video_id, xml_url)
|
||||
try:
|
||||
extr_player = ZDFExtractorPlayer(self, url, video_id)
|
||||
formats = extr_player._real_extract()
|
||||
except (ExtractorError, KeyError) as e:
|
||||
self._downloader.report_warning('%s: %s\nusing fallback method (mobile url)' % (type(e).__name__, compat_str(e)))
|
||||
extr_mobile = ZDFExtractorMobile(self, url, video_id)
|
||||
formats = extr_mobile._real_extract()
|
||||
return formats
|
||||
|
||||
class ZDFExtractor:
|
||||
"""Super class for the 2 extraction methods"""
|
||||
def __init__(self, parent, url, video_id):
|
||||
self.parent = parent
|
||||
self.url = url
|
||||
self.video_id = video_id
|
||||
|
||||
def _real_extract(self):
|
||||
formats = []
|
||||
for entry in self._fetch_entries():
|
||||
video_url = self._get_video_url(entry)
|
||||
if not video_url:
|
||||
continue
|
||||
format_id = self._get_format_id(entry)
|
||||
ext = determine_ext(video_url, None)
|
||||
if ext == 'meta':
|
||||
continue
|
||||
if ext == 'm3u8':
|
||||
formats.extend(self.parent._extract_m3u8_formats(
|
||||
video_url, self.video_id, 'mp4', m3u8_id=format_id, fatal=False))
|
||||
elif ext == 'f4m':
|
||||
formats.extend(self.parent._extract_f4m_formats(
|
||||
video_url, self.video_id, f4m_id=format_id, fatal=False))
|
||||
else:
|
||||
formats.append({
|
||||
'format_id': format_id,
|
||||
'url': video_url,
|
||||
'format_note': self._get_format_note(entry)
|
||||
})
|
||||
self.parent._sort_formats(formats)
|
||||
|
||||
return {
|
||||
'id': self.video_id,
|
||||
'title': self._get_title(),
|
||||
'formats': formats,
|
||||
'subtitles': self._get_subtitles(),
|
||||
'thumbnail': self._get_thumbnail(),
|
||||
'description': self._get_description(),
|
||||
'timestamp': self._get_timestamp()
|
||||
}
|
||||
|
||||
class ZDFExtractorMobile(ZDFExtractor):
|
||||
"""Simple URL extraction method. Disadvantage: fewer formats, no subtitles"""
|
||||
def __init__(self, parent, url, video_id):
|
||||
ZDFExtractor.__init__(self, parent, url, video_id)
|
||||
|
||||
def _fetch_entries(self):
|
||||
meta_data_url = 'https://zdf-cdn.live.cellular.de/mediathekV2/document/' + self.video_id
|
||||
self.meta_data = self.parent._download_json(meta_data_url, self.video_id, note='Downloading meta data')
|
||||
return self.meta_data['document']['formitaeten']
|
||||
|
||||
def _get_title(self):
|
||||
return self.meta_data['document']['titel']
|
||||
|
||||
def _get_video_url(self, entry):
|
||||
return entry['url']
|
||||
|
||||
def _get_format_id(self, entry):
|
||||
format_id = entry['type']
|
||||
if 'quality' in entry:
|
||||
format_id += '-' + entry['quality']
|
||||
return format_id
|
||||
|
||||
def _get_format_note(self, entry):
|
||||
return None
|
||||
|
||||
def _get_subtitles(self):
|
||||
return None
|
||||
|
||||
def _get_description(self):
|
||||
return self.meta_data['document'].get('beschreibung')
|
||||
|
||||
def _get_timestamp(self):
|
||||
meta = self.meta_data['meta']
|
||||
if meta:
|
||||
return parse_iso8601(meta.get('editorialDate'))
|
||||
|
||||
def _get_thumbnail(self):
|
||||
teaser_images = self.meta_data['document'].get('teaserBild')
|
||||
if teaser_images:
|
||||
max_res = max(teaser_images, key=int)
|
||||
return teaser_images[max_res].get('url')
|
||||
|
||||
class ZDFExtractorPlayer(ZDFExtractor):
|
||||
"""Extraction method that requires downloads of several pages.
|
||||
|
||||
Follows the requests of the website."""
|
||||
def __init__(self, parent, url, video_id):
|
||||
ZDFExtractor.__init__(self, parent, url, video_id)
|
||||
|
||||
def _fetch_entries(self):
|
||||
webpage = self.parent._download_webpage(self.url, self.video_id)
|
||||
|
||||
jsb = self.parent._search_regex(r"data-zdfplayer-jsb='([^']*)'", webpage, 'zdfplayer jsb data')
|
||||
jsb_json = self.parent._parse_json(jsb, self.video_id)
|
||||
|
||||
configuration_url = 'https://www.zdf.de' + jsb_json['config']
|
||||
configuration_json = self.parent._download_json(configuration_url, self.video_id, note='Downloading player configuration')
|
||||
api_token = configuration_json['apiToken']
|
||||
|
||||
player_js = self.parent._download_webpage('https://www.zdf.de/ZDFplayer/latest-v2/skins/zdf/zdf-player.js', self.video_id, fatal=False, note='Downloading player script')
|
||||
if player_js:
|
||||
player_id = self.parent._search_regex(r'this\.ptmd_player_id="([^"]*)"', player_js, 'player id', fatal=False)
|
||||
else:
|
||||
player_id = None
|
||||
|
||||
self.content_json = self.parent._download_json(jsb_json['content'], self.video_id, headers={'Api-Auth': 'Bearer %s' % api_token}, note='Downloading content description')
|
||||
|
||||
main_video_content = self.content_json['mainVideoContent']['http://zdf.de/rels/target']
|
||||
meta_data_url = None
|
||||
if not player_id:
|
||||
# could not determine player_id => try alternativ generic URL
|
||||
meta_data_url = main_video_content.get('http://zdf.de/rels/streams/ptmd')
|
||||
if meta_data_url:
|
||||
meta_data_url = 'https://api.zdf.de' + meta_data_url
|
||||
else:
|
||||
# no generic URL found => 2nd fallback: hardcoded player_id
|
||||
player_id = 'ngplayer_2_3'
|
||||
if not meta_data_url:
|
||||
meta_data_url_template = main_video_content['http://zdf.de/rels/streams/ptmd-template']
|
||||
meta_data_url = 'https://api.zdf.de' + meta_data_url_template.replace('{playerId}', player_id)
|
||||
|
||||
self.meta_data = self.parent._download_json(meta_data_url, self.video_id, note='Downloading meta data')
|
||||
|
||||
formats = []
|
||||
for p_list_entry in self.meta_data['priorityList']:
|
||||
for formitaet in p_list_entry['formitaeten']:
|
||||
for entry in formitaet['qualities']:
|
||||
yield (formitaet, entry)
|
||||
|
||||
def _get_title(self):
|
||||
return self.content_json['title']
|
||||
|
||||
def _get_video_url(self, entry_tuple):
|
||||
(formitaet, entry) = entry_tuple
|
||||
tracks = entry['audio'].get('tracks')
|
||||
if not tracks:
|
||||
return
|
||||
if len(tracks) > 1:
|
||||
self._downloader.report_warning('unexpected input: multiple tracks')
|
||||
track = tracks[0]
|
||||
return track['uri']
|
||||
|
||||
def _get_format_id(self, entry_tuple):
|
||||
(formitaet, entry) = entry_tuple
|
||||
facets = self._get_facets(formitaet)
|
||||
add = ''
|
||||
if 'adaptive' in facets:
|
||||
add += 'a'
|
||||
if 'restriction_useragent' in facets:
|
||||
add += 'b'
|
||||
if 'progressive' in facets:
|
||||
add += 'p'
|
||||
type_ = formitaet['type']
|
||||
format_id = type_ + '-'
|
||||
if add:
|
||||
format_id += add + '-'
|
||||
# named qualities are not very useful for sorting the formats:
|
||||
# a 'high' m3u8 entry can be better quality than a 'veryhigh' direct mp4 download
|
||||
format_id += entry['quality']
|
||||
return format_id
|
||||
|
||||
def _get_facets(self, formitaet):
|
||||
facets = formitaet.get('facets') or []
|
||||
if formitaet.get('isAdaptive'):
|
||||
facets.append('adaptive')
|
||||
return facets
|
||||
|
||||
def _get_format_note(self, entry_tuple):
|
||||
(formitaet, entry) = entry_tuple
|
||||
return ', '.join(self._get_facets(formitaet))
|
||||
|
||||
def _get_subtitles(self):
|
||||
subtitles = {}
|
||||
if 'captions' in self.meta_data:
|
||||
for caption in self.meta_data['captions']:
|
||||
lang = caption.get('language')
|
||||
if not lang:
|
||||
continue
|
||||
if lang == 'deu':
|
||||
lang = 'de'
|
||||
subformat = {'url': caption.get('uri')}
|
||||
if caption.get('format') == 'webvtt':
|
||||
subformat['ext'] = 'vtt'
|
||||
elif caption.get('format') == 'ebu-tt-d-basic-de':
|
||||
subformat['ext'] = 'ttml'
|
||||
if not lang in subtitles:
|
||||
subtitles[lang] = []
|
||||
subtitles[lang].append(subformat)
|
||||
return subtitles
|
||||
|
||||
def _get_description(self):
|
||||
return self.content_json.get('teasertext')
|
||||
|
||||
def _get_timestamp(self):
|
||||
return parse_iso8601(self.content_json.get('editorialDate'))
|
||||
|
||||
def _get_thumbnail(self):
|
||||
teaser_images = self.content_json.get('teaserImageRef')
|
||||
if teaser_images:
|
||||
teaser_images_layouts = teaser_images.get('layouts')
|
||||
if teaser_images_layouts:
|
||||
if 'original' in teaser_images_layouts:
|
||||
return teaser_images_layouts['original']
|
||||
teasers = {}
|
||||
for key in teaser_images_layouts:
|
||||
width = self.parent._search_regex(r'(\d+)x\d+', key, 'teaser width', fatal=False)
|
||||
if width:
|
||||
teasers[int(width)] = teaser_images_layouts[key]
|
||||
if teasers:
|
||||
best = max(teasers)
|
||||
return teasers[best]
|
||||
|
||||
class ZDFChannelIE(InfoExtractor):
|
||||
_WORKING = False
|
||||
_VALID_URL = r'(?:zdf:topic:|https?://www\.zdf\.de/ZDFmediathek(?:#)?/.*kanaluebersicht/(?:[^/]+/)?)(?P<id>[0-9]+)'
|
||||
_TESTS = [{
|
||||
'url': 'http://www.zdf.de/ZDFmediathek#/kanaluebersicht/1586442/sendung/Titanic',
|
||||
|
|
|
|||
Loading…
Reference in New Issue