Raphaël G. Git Repositories - youtubedl/blob - youtube_dl/extractor/vrv.py

   1 # coding: utf-8
   2 from __future__ import unicode_literals
   3
   4 import base64
   5 import json
   6 import hashlib
   7 import hmac
   8 import random
   9 import string
  10 import time
  11
  12 from .common import InfoExtractor
  13 from ..compat import (
  14     compat_urllib_parse_urlencode,
  15     compat_urllib_parse,
  16 )
  17 from ..utils import (
  18     float_or_none,
  19     int_or_none,
  20 )
  21
  22
  23 class VRVBaseIE(InfoExtractor):
  24     _API_DOMAIN = None
  25     _API_PARAMS = {}
  26     _CMS_SIGNING = {}
  27
  28     def _call_api(self, path, video_id, note, data=None):
  29         base_url = self._API_DOMAIN + '/core/' + path
  30         encoded_query = compat_urllib_parse_urlencode({
  31             'oauth_consumer_key': self._API_PARAMS['oAuthKey'],
  32             'oauth_nonce': ''.join([random.choice(string.ascii_letters) for _ in range(32)]),
  33             'oauth_signature_method': 'HMAC-SHA1',
  34             'oauth_timestamp': int(time.time()),
  35             'oauth_version': '1.0',
  36         })
  37         headers = self.geo_verification_headers()
  38         if data:
  39             data = json.dumps(data).encode()
  40             headers['Content-Type'] = 'application/json'
  41         method = 'POST' if data else 'GET'
  42         base_string = '&'.join([method, compat_urllib_parse.quote(base_url, ''), compat_urllib_parse.quote(encoded_query, '')])
  43         oauth_signature = base64.b64encode(hmac.new(
  44             (self._API_PARAMS['oAuthSecret'] + '&').encode('ascii'),
  45             base_string.encode(), hashlib.sha1).digest()).decode()
  46         encoded_query += '&oauth_signature=' + compat_urllib_parse.quote(oauth_signature, '')
  47         return self._download_json(
  48             '?'.join([base_url, encoded_query]), video_id,
  49             note='Downloading %s JSON metadata' % note, headers=headers, data=data)
  50
  51     def _call_cms(self, path, video_id, note):
  52         if not self._CMS_SIGNING:
  53             self._CMS_SIGNING = self._call_api('index', video_id, 'CMS Signing')['cms_signing']
  54         return self._download_json(
  55             self._API_DOMAIN + path, video_id, query=self._CMS_SIGNING,
  56             note='Downloading %s JSON metadata' % note, headers=self.geo_verification_headers())
  57
  58     def _set_api_params(self, webpage, video_id):
  59         if not self._API_PARAMS:
  60             self._API_PARAMS = self._parse_json(self._search_regex(
  61                 r'window\.__APP_CONFIG__\s*=\s*({.+?})</script>',
  62                 webpage, 'api config'), video_id)['cxApiParams']
  63             self._API_DOMAIN = self._API_PARAMS.get('apiDomain', 'https://api.vrv.co')
  64
  65     def _get_cms_resource(self, resource_key, video_id):
  66         return self._call_api(
  67             'cms_resource', video_id, 'resource path', data={
  68                 'resource_key': resource_key,
  69             })['__links__']['cms_resource']['href']
  70
  71
  72 class VRVIE(VRVBaseIE):
  73     IE_NAME = 'vrv'
  74     _VALID_URL = r'https?://(?:www\.)?vrv\.co/watch/(?P<id>[A-Z0-9]+)'
  75     _TEST = {
  76         'url': 'https://vrv.co/watch/GR9PNZ396/Hidden-America-with-Jonah-Ray:BOSTON-WHERE-THE-PAST-IS-THE-PRESENT',
  77         'info_dict': {
  78             'id': 'GR9PNZ396',
  79             'ext': 'mp4',
  80             'title': 'BOSTON: WHERE THE PAST IS THE PRESENT',
  81             'description': 'md5:4ec8844ac262ca2df9e67c0983c6b83f',
  82             'uploader_id': 'seeso',
  83         },
  84         'params': {
  85             # m3u8 download
  86             'skip_download': True,
  87         },
  88     }
  89
  90     def _real_extract(self, url):
  91         video_id = self._match_id(url)
  92         webpage = self._download_webpage(
  93             url, video_id,
  94             headers=self.geo_verification_headers())
  95         media_resource = self._parse_json(self._search_regex(
  96             r'window\.__INITIAL_STATE__\s*=\s*({.+?})</script>',
  97             webpage, 'inital state'), video_id).get('watch', {}).get('mediaResource') or {}
  98
  99         video_data = media_resource.get('json')
 100         if not video_data:
 101             self._set_api_params(webpage, video_id)
 102             episode_path = self._get_cms_resource(
 103                 'cms:/episodes/' + video_id, video_id)
 104             video_data = self._call_cms(episode_path, video_id, 'video')
 105         title = video_data['title']
 106
 107         streams_json = media_resource.get('streams', {}).get('json', {})
 108         if not streams_json:
 109             self._set_api_params(webpage, video_id)
 110             streams_path = video_data['__links__']['streams']['href']
 111             streams_json = self._call_cms(streams_path, video_id, 'streams')
 112
 113         audio_locale = streams_json.get('audio_locale')
 114         formats = []
 115         for stream_type, streams in streams_json.get('streams', {}).items():
 116             if stream_type in ('adaptive_hls', 'adaptive_dash'):
 117                 for stream in streams.values():
 118                     stream_url = stream.get('url')
 119                     if not stream_url:
 120                         continue
 121                     stream_id = stream.get('hardsub_locale') or audio_locale
 122                     format_id = '%s-%s' % (stream_type.split('_')[1], stream_id)
 123                     if stream_type == 'adaptive_hls':
 124                         adaptive_formats = self._extract_m3u8_formats(
 125                             stream_url, video_id, 'mp4', m3u8_id=format_id,
 126                             note='Downloading %s m3u8 information' % stream_id,
 127                             fatal=False)
 128                     else:
 129                         adaptive_formats = self._extract_mpd_formats(
 130                             stream_url, video_id, mpd_id=format_id,
 131                             note='Downloading %s MPD information' % stream_id,
 132                             fatal=False)
 133                     if audio_locale:
 134                         for f in adaptive_formats:
 135                             if f.get('acodec') != 'none':
 136                                 f['language'] = audio_locale
 137                     formats.extend(adaptive_formats)
 138         self._sort_formats(formats)
 139
 140         subtitles = {}
 141         for subtitle in streams_json.get('subtitles', {}).values():
 142             subtitle_url = subtitle.get('url')
 143             if not subtitle_url:
 144                 continue
 145             subtitles.setdefault(subtitle.get('locale', 'en-US'), []).append({
 146                 'url': subtitle_url,
 147                 'ext': subtitle.get('format', 'ass'),
 148             })
 149
 150         thumbnails = []
 151         for thumbnail in video_data.get('images', {}).get('thumbnails', []):
 152             thumbnail_url = thumbnail.get('source')
 153             if not thumbnail_url:
 154                 continue
 155             thumbnails.append({
 156                 'url': thumbnail_url,
 157                 'width': int_or_none(thumbnail.get('width')),
 158                 'height': int_or_none(thumbnail.get('height')),
 159             })
 160
 161         return {
 162             'id': video_id,
 163             'title': title,
 164             'formats': formats,
 165             'subtitles': subtitles,
 166             'thumbnails': thumbnails,
 167             'description': video_data.get('description'),
 168             'duration': float_or_none(video_data.get('duration_ms'), 1000),
 169             'uploader_id': video_data.get('channel_id'),
 170             'series': video_data.get('series_title'),
 171             'season': video_data.get('season_title'),
 172             'season_number': int_or_none(video_data.get('season_number')),
 173             'season_id': video_data.get('season_id'),
 174             'episode': title,
 175             'episode_number': int_or_none(video_data.get('episode_number')),
 176             'episode_id': video_data.get('production_episode_id'),
 177         }
 178
 179
 180 class VRVSeriesIE(VRVBaseIE):
 181     IE_NAME = 'vrv:series'
 182     _VALID_URL = r'https?://(?:www\.)?vrv\.co/series/(?P<id>[A-Z0-9]+)'
 183     _TEST = {
 184         'url': 'https://vrv.co/series/G68VXG3G6/The-Perfect-Insider',
 185         'info_dict': {
 186             'id': 'G68VXG3G6',
 187         },
 188         'playlist_mincount': 11,
 189     }
 190
 191     def _real_extract(self, url):
 192         series_id = self._match_id(url)
 193         webpage = self._download_webpage(
 194             url, series_id,
 195             headers=self.geo_verification_headers())
 196
 197         self._set_api_params(webpage, series_id)
 198         seasons_path = self._get_cms_resource(
 199             'cms:/seasons?series_id=' + series_id, series_id)
 200         seasons_data = self._call_cms(seasons_path, series_id, 'seasons')
 201
 202         entries = []
 203         for season in seasons_data.get('items', []):
 204             episodes_path = season['__links__']['season/episodes']['href']
 205             episodes = self._call_cms(episodes_path, series_id, 'episodes')
 206             for episode in episodes.get('items', []):
 207                 episode_id = episode['id']
 208                 entries.append(self.url_result(
 209                     'https://vrv.co/watch/' + episode_id,
 210                     'VRV', episode_id, episode.get('title')))
 211
 212         return self.playlist_result(entries, series_id)