[youtube] Prefer info from YouTube than _formats (#8293)

This commit is contained in:
Yen Chi Hsuan 2016-01-25 01:02:19 +08:00
parent a0d8d704df
commit 94278f7202

View File

@ -32,6 +32,7 @@ from ..utils import (
get_element_by_attribute, get_element_by_attribute,
get_element_by_id, get_element_by_id,
int_or_none, int_or_none,
mimetype2ext,
orderedSet, orderedSet,
parse_duration, parse_duration,
remove_quotes, remove_quotes,
@ -1090,9 +1091,9 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
full_info.update(f) full_info.update(f)
codecs = r.attrib.get('codecs') codecs = r.attrib.get('codecs')
if codecs: if codecs:
if full_info.get('acodec') == 'none' and 'vcodec' not in full_info: if full_info.get('acodec') == 'none':
full_info['vcodec'] = codecs full_info['vcodec'] = codecs
elif full_info.get('vcodec') == 'none' and 'acodec' not in full_info: elif full_info.get('vcodec') == 'none':
full_info['acodec'] = codecs full_info['acodec'] = codecs
formats.append(full_info) formats.append(full_info)
else: else:
@ -1461,15 +1462,21 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
if 'ratebypass' not in url: if 'ratebypass' not in url:
url += '&ratebypass=yes' url += '&ratebypass=yes'
dct = {
'format_id': format_id,
'url': url,
'player_url': player_url,
}
if format_id in self._formats:
dct.update(self._formats[format_id])
# Some itags are not included in DASH manifest thus corresponding formats will # Some itags are not included in DASH manifest thus corresponding formats will
# lack metadata (see https://github.com/rg3/youtube-dl/pull/5993). # lack metadata (see https://github.com/rg3/youtube-dl/pull/5993).
# Trying to extract metadata from url_encoded_fmt_stream_map entry. # Trying to extract metadata from url_encoded_fmt_stream_map entry.
mobj = re.search(r'^(?P<width>\d+)[xX](?P<height>\d+)$', url_data.get('size', [''])[0]) mobj = re.search(r'^(?P<width>\d+)[xX](?P<height>\d+)$', url_data.get('size', [''])[0])
width, height = (int(mobj.group('width')), int(mobj.group('height'))) if mobj else (None, None) width, height = (int(mobj.group('width')), int(mobj.group('height'))) if mobj else (None, None)
dct = {
'format_id': format_id, more_fields = {
'url': url,
'player_url': player_url,
'filesize': int_or_none(url_data.get('clen', [None])[0]), 'filesize': int_or_none(url_data.get('clen', [None])[0]),
'tbr': float_or_none(url_data.get('bitrate', [None])[0], 1000), 'tbr': float_or_none(url_data.get('bitrate', [None])[0], 1000),
'width': width, 'width': width,
@ -1477,13 +1484,16 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
'fps': int_or_none(url_data.get('fps', [None])[0]), 'fps': int_or_none(url_data.get('fps', [None])[0]),
'format_note': url_data.get('quality_label', [None])[0] or url_data.get('quality', [None])[0], 'format_note': url_data.get('quality_label', [None])[0] or url_data.get('quality', [None])[0],
} }
for key, value in more_fields.items():
if value:
dct[key] = value
type_ = url_data.get('type', [None])[0] type_ = url_data.get('type', [None])[0]
if type_: if type_:
type_split = type_.split(';') type_split = type_.split(';')
kind_ext = type_split[0].split('/') kind_ext = type_split[0].split('/')
if len(kind_ext) == 2: if len(kind_ext) == 2:
kind, ext = kind_ext kind, _ = kind_ext
dct['ext'] = ext dct['ext'] = mimetype2ext(type_split[0])
if kind in ('audio', 'video'): if kind in ('audio', 'video'):
codecs = None codecs = None
for mobj in re.finditer( for mobj in re.finditer(
@ -1501,8 +1511,6 @@ class YoutubeIE(YoutubeBaseInfoExtractor):
'acodec': acodec, 'acodec': acodec,
'vcodec': vcodec, 'vcodec': vcodec,
}) })
if format_id in self._formats:
dct.update(self._formats[format_id])
formats.append(dct) formats.append(dct)
elif video_info.get('hlsvp'): elif video_info.get('hlsvp'):
manifest_url = video_info['hlsvp'][0] manifest_url = video_info['hlsvp'][0]