From 2975fe1a7be39b66d14fc7ef68ea7db948f0e40f Mon Sep 17 00:00:00 2001 From: remitamine Date: Mon, 25 Jan 2016 22:35:06 +0100 Subject: [PATCH] [vevo] extract all formats and bypass geo restriction --- youtube_dl/extractor/vevo.py | 236 +++++++++++++++-------------------- 1 file changed, 103 insertions(+), 133 deletions(-) diff --git a/youtube_dl/extractor/vevo.py b/youtube_dl/extractor/vevo.py index 02dfd36f4..ff854e74b 100644 --- a/youtube_dl/extractor/vevo.py +++ b/youtube_dl/extractor/vevo.py @@ -3,22 +3,18 @@ from __future__ import unicode_literals import re from .common import InfoExtractor -from ..compat import ( - compat_etree_fromstring, - compat_urlparse, -) +from ..compat import compat_etree_fromstring from ..utils import ( ExtractorError, int_or_none, - sanitized_Request, ) class VevoIE(InfoExtractor): - """ + ''' Accepts urls from vevo.com or in the format 'vevo:{id}' (currently used by MTVIE and MySpaceIE) - """ + ''' _VALID_URL = r'''(?x) (?:https?://www\.vevo\.com/watch/(?:[^/]+/(?:[^/]+/)?)?| https?://cache\.vevo\.com/m/html/embed\.html\?video=| @@ -28,48 +24,51 @@ class VevoIE(InfoExtractor): _TESTS = [{ 'url': 'http://www.vevo.com/watch/hurts/somebody-to-die-for/GB1101300280', - "md5": "95ee28ee45e70130e3ab02b0f579ae23", + 'md5': '2dbc7e9fd4f1c60436c9aa73a5406193', 'info_dict': { - 'id': 'GB1101300280', + 'id': 'Pt1kc_FniKM', 'ext': 'mp4', - "upload_date": "20130624", - "uploader": "Hurts", - "title": "Somebody to Die For", - "duration": 230.12, - "width": 1920, - "height": 1080, - # timestamp and upload_date are often incorrect; seem to change randomly - 'timestamp': int, - } + 'title': 'Hurts - Somebody to Die For', + 'description': 'md5:13e925b89af6b01c7e417332bd23c4bf', + 'uploader_id': 'HurtsVEVO', + 'uploader': 'HurtsVEVO', + 'upload_date': '20130624', + 'duration': 230, + }, + 'add_ie': ['Youtube'], }, { 'note': 'v3 SMIL format', 'url': 'http://www.vevo.com/watch/cassadee-pope/i-wish-i-could-break-your-heart/USUV71302923', - 'md5': 'f6ab09b034f8c22969020b042e5ac7fc', + 'md5': '13d5204f520af905eeffa675040b8e76', 'info_dict': { - 'id': 'USUV71302923', + 'id': 'ByGmQn1uxJw', 'ext': 'mp4', + 'title': 'Cassadee Pope - I Wish I Could Break Your Heart', + 'description': 'md5:5e9721c92ef117a6f69d00e9b42ceba7', + 'uploader_id': 'CassadeeVEVO', + 'uploader': 'CassadeeVEVO', 'upload_date': '20140219', - 'uploader': 'Cassadee Pope', - 'title': 'I Wish I Could Break Your Heart', - 'duration': 226.101, + 'duration': 226, 'age_limit': 0, - 'timestamp': int, - } + }, + 'add_ie': ['Youtube'], }, { 'note': 'Age-limited video', 'url': 'https://www.vevo.com/watch/justin-timberlake/tunnel-vision-explicit/USRV81300282', 'info_dict': { - 'id': 'USRV81300282', + 'id': '07FYdnEawAQ', 'ext': 'mp4', 'age_limit': 18, - 'title': 'Tunnel Vision (Explicit)', - 'uploader': 'Justin Timberlake', - 'upload_date': 're:2013070[34]', - 'timestamp': int, + 'title': 'Justin Timberlake - Tunnel Vision (Explicit)', + 'description': 'md5:64249768eec3bc4276236606ea996373', + 'uploader_id': 'justintimberlakeVEVO', + 'uploader': 'justintimberlakeVEVO', + 'upload_date': '20130703', }, 'params': { 'skip_download': 'true', - } + }, + 'add_ie': ['Youtube'], }, { 'note': 'No video_info', 'url': 'http://www.vevo.com/watch/k-camp-1/Till-I-Die/USUV71503000', @@ -80,66 +79,40 @@ class VevoIE(InfoExtractor): 'title': 'Till I Die - K Camp ft. T.I.', 'duration': 193, }, - 'expected_warnings': ['Unable to download SMIL file'], }] - _SMIL_BASE_URL = 'http://smil.lvl3.vevo.com/' - - def _real_initialize(self): - req = sanitized_Request( - 'http://www.vevo.com/auth', data=b'') - webpage = self._download_webpage( - req, None, - note='Retrieving oauth token', - errnote='Unable to retrieve oauth token', - fatal=False) - if webpage is False: - self._oauth_token = None - else: - if 'THIS PAGE IS CURRENTLY UNAVAILABLE IN YOUR REGION' in webpage: - raise ExtractorError('%s said: This page is currently unavailable in your region.' % self.IE_NAME, expected=True) - - self._oauth_token = self._search_regex( - r'access_token":\s*"([^"]+)"', - webpage, 'access token', fatal=False) - - def _formats_from_json(self, video_info): - if not video_info: - return [] - - last_version = {'version': -1} - for version in video_info['videoVersions']: - # These are the HTTP downloads, other types are for different manifests - if version['sourceType'] == 2: - if version['version'] > last_version['version']: - last_version = version - if last_version['version'] == -1: - raise ExtractorError('Unable to extract last version of the video') - - renditions = compat_etree_fromstring(last_version['data']) - formats = [] - # Already sorted from worst to best quality - for rend in renditions.findall('rendition'): - attr = rend.attrib - format_note = '%(videoCodec)s@%(videoBitrate)4sk, %(audioCodec)s@%(audioBitrate)3sk' % attr - formats.append({ - 'url': attr['url'], - 'format_id': attr['name'], - 'format_note': format_note, - 'height': int(attr['frameheight']), - 'width': int(attr['frameWidth']), - }) - return formats - - def _formats_from_smil(self, smil_doc): + _SMIL_BASE_URL = 'http://smil.lvl3.vevo.com' + _SOURCE_TYPES = { + 0: 'youtube', + 1: 'brightcove', + 2: 'http', + 3: 'hls_ios', + 4: 'hls', + 5: 'smil', # http + 7: 'f4m_cc', + 8: 'f4m_ak', + 9: 'f4m_l3', + 10: 'ism', + 13: 'smil', # rtmp + 18: 'dash', + } + _VERSIONS = { + 0: 'youtube', + 1: 'level3', + 2: 'akamai', + 3: 'level3', + 4: 'amazon', + } + + def _parse_smil_formats(self, smil, smil_url, video_id, namespace=None, f4m_params=None, transform_rtmp_url=None): formats = [] - els = smil_doc.findall('.//{http://www.w3.org/2001/SMIL20/Language}video') + els = smil.findall('.//{http://www.w3.org/2001/SMIL20/Language}video') for el in els: src = el.attrib['src'] m = re.match(r'''(?xi) (?P[a-z0-9]+): (?P [/a-z0-9]+ # The directory and main part of the URL - _(?P[0-9]+)k + _(?P[0-9]+)k _(?P[0-9]+)x(?P[0-9]+) _(?P[a-z0-9]+) _(?P[0-9]+) @@ -153,9 +126,10 @@ class VevoIE(InfoExtractor): format_url = self._SMIL_BASE_URL + m.group('path') formats.append({ 'url': format_url, - 'format_id': 'SMIL_' + m.group('cbr'), + 'format_id': 'smil_' + m.group('tbr'), 'vcodec': m.group('vcodec'), 'acodec': m.group('acodec'), + 'tbr': int(m.group('tbr')), 'vbr': int(m.group('vbr')), 'abr': int(m.group('abr')), 'ext': m.group('ext'), @@ -164,26 +138,6 @@ class VevoIE(InfoExtractor): }) return formats - def _download_api_formats(self, video_id, video_url): - if not self._oauth_token: - self._downloader.report_warning( - 'No oauth token available, skipping API HLS download') - return [] - - api_url = compat_urlparse.urljoin(video_url, '//apiv2.vevo.com/video/%s/streams/hls?token=%s' % ( - video_id, self._oauth_token)) - api_data = self._download_json( - api_url, video_id, - note='Downloading HLS formats', - errnote='Failed to download HLS format list', fatal=False) - if api_data is None: - return [] - - m3u8_url = api_data[0]['url'] - return self._extract_m3u8_formats( - m3u8_url, video_id, entry_protocol='m3u8_native', ext='mp4', - preference=0) - def _real_extract(self, url): video_id = self._match_id(url) @@ -193,19 +147,57 @@ class VevoIE(InfoExtractor): response = self._download_json(json_url, video_id) video_info = response['video'] or {} - if not video_info and response.get('statusCode') != 909: - if 'statusMessage' in response: - raise ExtractorError('%s said: %s' % (self.IE_NAME, response['statusMessage']), expected=True) - raise ExtractorError('Unable to extract videos') - if not video_info: + ytid = response.get('errorInfo', {}).get('ytid') + if ytid: + return self.url_result(ytid, 'Youtube', ytid) + + if response.get('statusCode') != 909: + if 'statusMessage' in response: + raise ExtractorError('%s said: %s' % ( + self.IE_NAME, response['statusMessage']), expected=True) + raise ExtractorError('Unable to extract videos') + if url.startswith('vevo:'): - raise ExtractorError('Please specify full Vevo URL for downloading', expected=True) + raise ExtractorError( + 'Please specify full Vevo URL for downloading', expected=True) webpage = self._download_webpage(url, video_id) title = video_info.get('title') or self._og_search_title(webpage) - formats = self._formats_from_json(video_info) + smil_parsed = False + formats = [] + for video_version in video_info['videoVersions']: + version = self._VERSIONS.get(video_version['version']) + if version == 'youtube': + return self.url_result( + video_version['id'], 'Youtube', video_version['id']) + else: + source_type = self._SOURCE_TYPES.get(video_version['sourceType']) + renditions = compat_etree_fromstring(video_version['data']) + if source_type == 'http': + for rend in renditions.findall('rendition'): + attr = rend.attrib + formats.append({ + 'url': attr['url'], + 'format_id': '%s-%s' % (version, attr['name']), + 'height': int_or_none(attr.get('frameheight')), + 'width': int_or_none(attr.get('frameWidth')), + 'tbr': int_or_none(attr.get('totalBitrate')), + 'vbr': int_or_none(attr.get('videoBitrate')), + 'abr': int_or_none(attr.get('audioBitrate')), + 'vcodec': attr.get('videoCodec'), + 'acodec': attr.get('audioCodec'), + }) + elif source_type == 'hls': + formats.extend(self._extract_m3u8_formats( + renditions.find('rendition').attrib['url'], video_id, + 'mp4', 'm3u8_native', m3u8_id='hls-%s' % version, fatal=False)) + elif source_type == 'smil' and not smil_parsed: + formats.extend(self._extract_smil_formats( + renditions.find('rendition').attrib['url'], video_id, False)) + smil_parsed = True + self._sort_formats(formats) is_explicit = video_info.get('isExplicit') if is_explicit is True: @@ -215,28 +207,6 @@ class VevoIE(InfoExtractor): else: age_limit = None - # Download via HLS API - formats.extend(self._download_api_formats(video_id, url)) - - # Download SMIL - smil_blocks = sorted(( - f for f in video_info.get('videoVersions', []) - if f['sourceType'] == 13), - key=lambda f: f['version']) - smil_url = '%s/Video/V2/VFILE/%s/%sr.smil' % ( - self._SMIL_BASE_URL, video_id, video_id.lower()) - if smil_blocks: - smil_url_m = self._search_regex( - r'url="([^"]+)"', smil_blocks[-1]['data'], 'SMIL URL', - default=None) - if smil_url_m is not None: - smil_url = smil_url_m - if smil_url: - smil_doc = self._download_smil(smil_url, video_id, fatal=False) - if smil_doc: - formats.extend(self._formats_from_smil(smil_doc)) - - self._sort_formats(formats) timestamp = int_or_none(self._search_regex( r'/Date\((\d+)\)/', video_info['launchDate'], 'launch date', fatal=False),