youtube_dl/extractor/youtube.py

   1 # coding: utf-8
   2
   3 import errno
   4 import io
   5 import itertools
   6 import json
   7 import os.path
   8 import re
   9 import traceback
  10
  11 from .common import InfoExtractor, SearchInfoExtractor
  12 from .subtitles import SubtitlesInfoExtractor
  13 from ..jsinterp import JSInterpreter
  14 from ..swfinterp import SWFInterpreter
  15 from ..utils import (
  16     compat_chr,
  17     compat_parse_qs,
  18     compat_urllib_parse,
  19     compat_urllib_request,
  20     compat_urlparse,
  21     compat_str,
  22
  23     clean_html,
  24     get_cachedir,
  25     get_element_by_id,
  26     get_element_by_attribute,
  27     ExtractorError,
  28     int_or_none,
  29     PagedList,
  30     unescapeHTML,
  31     unified_strdate,
  32     orderedSet,
  33     write_json_file,
  34     uppercase_escape,
  35 )
  36
  37 class YoutubeBaseInfoExtractor(InfoExtractor):
  38     """Provide base functions for Youtube extractors"""
  39     _LOGIN_URL = 'https://accounts.google.com/ServiceLogin'
  40     _LANG_URL = r'https://www.youtube.com/?hl=en&persist_hl=1&gl=US&persist_gl=1&opt_out_ackd=1'
  41     _AGE_URL = 'https://www.youtube.com/verify_age?next_url=/&gl=US&hl=en'
  42     _NETRC_MACHINE = 'youtube'
  43     # If True it will raise an error if no login info is provided
  44     _LOGIN_REQUIRED = False
  45
  46     def _set_language(self):
  47         return bool(self._download_webpage(
  48             self._LANG_URL, None,
  49             note=u'Setting language', errnote='unable to set language',
  50             fatal=False))
  51
  52     def _login(self):
  53         (username, password) = self._get_login_info()
  54         # No authentication to be performed
  55         if username is None:
  56             if self._LOGIN_REQUIRED:
  57                 raise ExtractorError(u'No login info available, needed for using %s.' % self.IE_NAME, expected=True)
  58             return False
  59
  60         login_page = self._download_webpage(
  61             self._LOGIN_URL, None,
  62             note=u'Downloading login page',
  63             errnote=u'unable to fetch login page', fatal=False)
  64         if login_page is False:
  65             return
  66
  67         galx = self._search_regex(r'(?s)<input.+?name="GALX".+?value="(.+?)"',
  68                                   login_page, u'Login GALX parameter')
  69
  70         # Log in
  71         login_form_strs = {
  72                 u'continue': u'https://www.youtube.com/signin?action_handle_signin=true&feature=sign_in_button&hl=en_US&nomobiletemp=1',
  73                 u'Email': username,
  74                 u'GALX': galx,
  75                 u'Passwd': password,
  76                 u'PersistentCookie': u'yes',
  77                 u'_utf8': u'霱',
  78                 u'bgresponse': u'js_disabled',
  79                 u'checkConnection': u'',
  80                 u'checkedDomains': u'youtube',
  81                 u'dnConn': u'',
  82                 u'pstMsg': u'0',
  83                 u'rmShown': u'1',
  84                 u'secTok': u'',
  85                 u'signIn': u'Sign in',
  86                 u'timeStmp': u'',
  87                 u'service': u'youtube',
  88                 u'uilel': u'3',
  89                 u'hl': u'en_US',
  90         }
  91         # Convert to UTF-8 *before* urlencode because Python 2.x's urlencode
  92         # chokes on unicode
  93         login_form = dict((k.encode('utf-8'), v.encode('utf-8')) for k,v in login_form_strs.items())
  94         login_data = compat_urllib_parse.urlencode(login_form).encode('ascii')
  95
  96         req = compat_urllib_request.Request(self._LOGIN_URL, login_data)
  97         login_results = self._download_webpage(
  98             req, None,
  99             note=u'Logging in', errnote=u'unable to log in', fatal=False)
 100         if login_results is False:
 101             return False
 102         if re.search(r'(?i)<form[^>]* id="gaia_loginform"', login_results) is not None:
 103             self._downloader.report_warning(u'unable to log in: bad username or password')
 104             return False
 105         return True
 106
 107     def _confirm_age(self):
 108         age_form = {
 109             'next_url': '/',
 110             'action_confirm': 'Confirm',
 111         }
 112         req = compat_urllib_request.Request(self._AGE_URL,
 113             compat_urllib_parse.urlencode(age_form).encode('ascii'))
 114
 115         self._download_webpage(
 116             req, None,
 117             note=u'Confirming age', errnote=u'Unable to confirm age')
 118         return True
 119
 120     def _real_initialize(self):
 121         if self._downloader is None:
 122             return
 123         if not self._set_language():
 124             return
 125         if not self._login():
 126             return
 127         self._confirm_age()
 128
 129
 130 class YoutubeIE(YoutubeBaseInfoExtractor, SubtitlesInfoExtractor):
 131     IE_DESC = u'YouTube.com'
 132     _VALID_URL = r"""(?x)^
 133                      (
 134                          (?:https?://|//)?                                    # http(s):// or protocol-independent URL (optional)
 135                          (?:(?:(?:(?:\w+\.)?[yY][oO][uU][tT][uU][bB][eE](?:-nocookie)?\.com/|
 136                             (?:www\.)?deturl\.com/www\.youtube\.com/|
 137                             (?:www\.)?pwnyoutube\.com/|
 138                             (?:www\.)?yourepeat\.com/|
 139                             tube\.majestyc\.net/|
 140                             youtube\.googleapis\.com/)                        # the various hostnames, with wildcard subdomains
 141                          (?:.*?\#/)?                                          # handle anchor (#/) redirect urls
 142                          (?:                                                  # the various things that can precede the ID:
 143                              (?:(?:v|embed|e)/)                               # v/ or embed/ or e/
 144                              |(?:                                             # or the v= param in all its forms
 145                                  (?:(?:watch|movie)(?:_popup)?(?:\.php)?/?)?  # preceding watch(_popup|.php) or nothing (like /?v=xxxx)
 146                                  (?:\?|\#!?)                                  # the params delimiter ? or # or #!
 147                                  (?:.*?&)?                                    # any other preceding param (like /?s=tuff&v=xxxx)
 148                                  v=
 149                              )
 150                          ))
 151                          |youtu\.be/                                          # just youtu.be/xxxx
 152                          |https?://(?:www\.)?cleanvideosearch\.com/media/action/yt/watch\?videoId=
 153                          )
 154                      )?                                                       # all until now is optional -> you can pass the naked ID
 155                      ([0-9A-Za-z_-]{11})                                      # here is it! the YouTube video ID
 156                      (?(1).+)?                                                # if we found the ID, everything can follow
 157                      $"""
 158     _NEXT_URL_RE = r'[\?&]next_url=([^&]+)'
 159     _formats = {
 160         '5': {'ext': 'flv', 'width': 400, 'height': 240},
 161         '6': {'ext': 'flv', 'width': 450, 'height': 270},
 162         '13': {'ext': '3gp'},
 163         '17': {'ext': '3gp', 'width': 176, 'height': 144},
 164         '18': {'ext': 'mp4', 'width': 640, 'height': 360},
 165         '22': {'ext': 'mp4', 'width': 1280, 'height': 720},
 166         '34': {'ext': 'flv', 'width': 640, 'height': 360},
 167         '35': {'ext': 'flv', 'width': 854, 'height': 480},
 168         '36': {'ext': '3gp', 'width': 320, 'height': 240},
 169         '37': {'ext': 'mp4', 'width': 1920, 'height': 1080},
 170         '38': {'ext': 'mp4', 'width': 4096, 'height': 3072},
 171         '43': {'ext': 'webm', 'width': 640, 'height': 360},
 172         '44': {'ext': 'webm', 'width': 854, 'height': 480},
 173         '45': {'ext': 'webm', 'width': 1280, 'height': 720},
 174         '46': {'ext': 'webm', 'width': 1920, 'height': 1080},
 175
 176
 177         # 3d videos
 178         '82': {'ext': 'mp4', 'height': 360, 'format_note': '3D', 'preference': -20},
 179         '83': {'ext': 'mp4', 'height': 480, 'format_note': '3D', 'preference': -20},
 180         '84': {'ext': 'mp4', 'height': 720, 'format_note': '3D', 'preference': -20},
 181         '85': {'ext': 'mp4', 'height': 1080, 'format_note': '3D', 'preference': -20},
 182         '100': {'ext': 'webm', 'height': 360, 'format_note': '3D', 'preference': -20},
 183         '101': {'ext': 'webm', 'height': 480, 'format_note': '3D', 'preference': -20},
 184         '102': {'ext': 'webm', 'height': 720, 'format_note': '3D', 'preference': -20},
 185
 186         # Apple HTTP Live Streaming
 187         '92': {'ext': 'mp4', 'height': 240, 'format_note': 'HLS', 'preference': -10},
 188         '93': {'ext': 'mp4', 'height': 360, 'format_note': 'HLS', 'preference': -10},
 189         '94': {'ext': 'mp4', 'height': 480, 'format_note': 'HLS', 'preference': -10},
 190         '95': {'ext': 'mp4', 'height': 720, 'format_note': 'HLS', 'preference': -10},
 191         '96': {'ext': 'mp4', 'height': 1080, 'format_note': 'HLS', 'preference': -10},
 192         '132': {'ext': 'mp4', 'height': 240, 'format_note': 'HLS', 'preference': -10},
 193         '151': {'ext': 'mp4', 'height': 72, 'format_note': 'HLS', 'preference': -10},
 194
 195         # DASH mp4 video
 196         '133': {'ext': 'mp4', 'height': 240, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 197         '134': {'ext': 'mp4', 'height': 360, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 198         '135': {'ext': 'mp4', 'height': 480, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 199         '136': {'ext': 'mp4', 'height': 720, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 200         '137': {'ext': 'mp4', 'height': 1080, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 201         '138': {'ext': 'mp4', 'height': 2160, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 202         '160': {'ext': 'mp4', 'height': 144, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 203         '264': {'ext': 'mp4', 'height': 1440, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 204
 205         # Dash mp4 audio
 206         '139': {'ext': 'm4a', 'format_note': 'DASH audio', 'vcodec': 'none', 'abr': 48, 'preference': -50},
 207         '140': {'ext': 'm4a', 'format_note': 'DASH audio', 'vcodec': 'none', 'abr': 128, 'preference': -50},
 208         '141': {'ext': 'm4a', 'format_note': 'DASH audio', 'vcodec': 'none', 'abr': 256, 'preference': -50},
 209
 210         # Dash webm
 211         '167': {'ext': 'webm', 'height': 360, 'width': 640, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
 212         '168': {'ext': 'webm', 'height': 480, 'width': 854, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
 213         '169': {'ext': 'webm', 'height': 720, 'width': 1280, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
 214         '170': {'ext': 'webm', 'height': 1080, 'width': 1920, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
 215         '218': {'ext': 'webm', 'height': 480, 'width': 854, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
 216         '219': {'ext': 'webm', 'height': 480, 'width': 854, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
 217         '242': {'ext': 'webm', 'height': 240, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 218         '243': {'ext': 'webm', 'height': 360, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 219         '244': {'ext': 'webm', 'height': 480, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 220         '245': {'ext': 'webm', 'height': 480, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 221         '246': {'ext': 'webm', 'height': 480, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 222         '247': {'ext': 'webm', 'height': 720, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 223         '248': {'ext': 'webm', 'height': 1080, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 224         '271': {'ext': 'webm', 'height': 1440, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 225         '272': {'ext': 'webm', 'height': 2160, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
 226
 227         # Dash webm audio
 228         '171': {'ext': 'webm', 'vcodec': 'none', 'format_note': 'DASH audio', 'abr': 128, 'preference': -50},
 229         '172': {'ext': 'webm', 'vcodec': 'none', 'format_note': 'DASH audio', 'abr': 256, 'preference': -50},
 230
 231         # RTMP (unnamed)
 232         '_rtmp': {'protocol': 'rtmp'},
 233     }
 234
 235     IE_NAME = u'youtube'
 236     _TESTS = [
 237         {
 238             u"url":  u"http://www.youtube.com/watch?v=BaW_jenozKc",
 239             u"file":  u"BaW_jenozKc.mp4",
 240             u"info_dict": {
 241                 u"title": u"youtube-dl test video \"'/\\ä↭𝕐",
 242                 u"uploader": u"Philipp Hagemeister",
 243                 u"uploader_id": u"phihag",
 244                 u"upload_date": u"20121002",
 245                 u"description": u"test chars:  \"'/\\ä↭𝕐\ntest URL: https://github.com/rg3/youtube-dl/issues/1892\n\nThis is a test video for youtube-dl.\n\nFor more information, contact phihag@phihag.de .",
 246                 u"categories": [u'Science & Technology'],
 247             }
 248         },
 249         {
 250             u"url":  u"http://www.youtube.com/watch?v=UxxajLWwzqY",
 251             u"file":  u"UxxajLWwzqY.mp4",
 252             u"note": u"Test generic use_cipher_signature video (#897)",
 253             u"info_dict": {
 254                 u"upload_date": u"20120506",
 255                 u"title": u"Icona Pop - I Love It (feat. Charli XCX) [OFFICIAL VIDEO]",
 256                 u"description": u"md5:fea86fda2d5a5784273df5c7cc994d9f",
 257                 u"uploader": u"Icona Pop",
 258                 u"uploader_id": u"IconaPop"
 259             }
 260         },
 261         {
 262             u"url":  u"https://www.youtube.com/watch?v=07FYdnEawAQ",
 263             u"file":  u"07FYdnEawAQ.mp4",
 264             u"note": u"Test VEVO video with age protection (#956)",
 265             u"info_dict": {
 266                 u"upload_date": u"20130703",
 267                 u"title": u"Justin Timberlake - Tunnel Vision (Explicit)",
 268                 u"description": u"md5:64249768eec3bc4276236606ea996373",
 269                 u"uploader": u"justintimberlakeVEVO",
 270                 u"uploader_id": u"justintimberlakeVEVO"
 271             }
 272         },
 273         {
 274             u"url":  u"//www.YouTube.com/watch?v=yZIXLfi8CZQ",
 275             u"file":  u"yZIXLfi8CZQ.mp4",
 276             u"note": u"Embed-only video (#1746)",
 277             u"info_dict": {
 278                 u"upload_date": u"20120608",
 279                 u"title": u"Principal Sexually Assaults A Teacher - Episode 117 - 8th June 2012",
 280                 u"description": u"md5:09b78bd971f1e3e289601dfba15ca4f7",
 281                 u"uploader": u"SET India",
 282                 u"uploader_id": u"setindia"
 283             }
 284         },
 285         {
 286             u"url": u"http://www.youtube.com/watch?v=a9LDPn-MO4I",
 287             u"file": u"a9LDPn-MO4I.m4a",
 288             u"note": u"256k DASH audio (format 141) via DASH manifest",
 289             u"info_dict": {
 290                 u"upload_date": "20121002",
 291                 u"uploader_id": "8KVIDEO",
 292                 u"description": "No description available.",
 293                 u"uploader": "8KVIDEO",
 294                 u"title": "UHDTV TEST 8K VIDEO.mp4"
 295             },
 296             u"params": {
 297                 u"youtube_include_dash_manifest": True,
 298                 u"format": "141",
 299             },
 300         },
 301         # DASH manifest with encrypted signature
 302         {
 303             u'url': u'https://www.youtube.com/watch?v=IB3lcPjvWLA',
 304             u'info_dict': {
 305                 u'id': u'IB3lcPjvWLA',
 306                 u'ext': u'm4a',
 307                 u'title': u'Afrojack - The Spark ft. Spree Wilson',
 308                 u'description': u'md5:9717375db5a9a3992be4668bbf3bc0a8',
 309                 u'uploader': u'AfrojackVEVO',
 310                 u'uploader_id': u'AfrojackVEVO',
 311                 u'upload_date': u'20131011',
 312             },
 313             u"params": {
 314                 u'youtube_include_dash_manifest': True,
 315                 u'format': '141',
 316             },
 317         },
 318     ]
 319
 320
 321     @classmethod
 322     def suitable(cls, url):
 323         """Receives a URL and returns True if suitable for this IE."""
 324         if YoutubePlaylistIE.suitable(url): return False
 325         return re.match(cls._VALID_URL, url) is not None
 326
 327     def __init__(self, *args, **kwargs):
 328         super(YoutubeIE, self).__init__(*args, **kwargs)
 329         self._player_cache = {}
 330
 331     def report_video_info_webpage_download(self, video_id):
 332         """Report attempt to download video info webpage."""
 333         self.to_screen(u'%s: Downloading video info webpage' % video_id)
 334
 335     def report_information_extraction(self, video_id):
 336         """Report attempt to extract video information."""
 337         self.to_screen(u'%s: Extracting video information' % video_id)
 338
 339     def report_unavailable_format(self, video_id, format):
 340         """Report extracted video URL."""
 341         self.to_screen(u'%s: Format %s not available' % (video_id, format))
 342
 343     def report_rtmp_download(self):
 344         """Indicate the download will use the RTMP protocol."""
 345         self.to_screen(u'RTMP download detected')
 346
 347     def _signature_cache_id(self, example_sig):
 348         """ Return a string representation of a signature """
 349         return u'.'.join(compat_str(len(part)) for part in example_sig.split('.'))
 350
 351     def _extract_signature_function(self, video_id, player_url, example_sig):
 352         id_m = re.match(
 353             r'.*-(?P<id>[a-zA-Z0-9_-]+)(?:/watch_as3|/html5player)?\.(?P<ext>[a-z]+)$',
 354             player_url)
 355         if not id_m:
 356             raise ExtractorError('Cannot identify player %r' % player_url)
 357         player_type = id_m.group('ext')
 358         player_id = id_m.group('id')
 359
 360         # Read from filesystem cache
 361         func_id = '%s_%s_%s' % (
 362             player_type, player_id, self._signature_cache_id(example_sig))
 363         assert os.path.basename(func_id) == func_id
 364         cache_dir = get_cachedir(self._downloader.params)
 365
 366         cache_enabled = cache_dir is not None
 367         if cache_enabled:
 368             cache_fn = os.path.join(os.path.expanduser(cache_dir),
 369                                     u'youtube-sigfuncs',
 370                                     func_id + '.json')
 371             try:
 372                 with io.open(cache_fn, 'r', encoding='utf-8') as cachef:
 373                     cache_spec = json.load(cachef)
 374                 return lambda s: u''.join(s[i] for i in cache_spec)
 375             except IOError:
 376                 pass  # No cache available
 377             except ValueError:
 378                 try:
 379                     file_size = os.path.getsize(cache_fn)
 380                 except (OSError, IOError) as oe:
 381                     file_size = str(oe)
 382                 self._downloader.report_warning(
 383                     u'Cache %s failed (%s)' % (cache_fn, file_size))
 384
 385         if player_type == 'js':
 386             code = self._download_webpage(
 387                 player_url, video_id,
 388                 note=u'Downloading %s player %s' % (player_type, player_id),
 389                 errnote=u'Download of %s failed' % player_url)
 390             res = self._parse_sig_js(code)
 391         elif player_type == 'swf':
 392             urlh = self._request_webpage(
 393                 player_url, video_id,
 394                 note=u'Downloading %s player %s' % (player_type, player_id),
 395                 errnote=u'Download of %s failed' % player_url)
 396             code = urlh.read()
 397             res = self._parse_sig_swf(code)
 398         else:
 399             assert False, 'Invalid player type %r' % player_type
 400
 401         if cache_enabled:
 402             try:
 403                 test_string = u''.join(map(compat_chr, range(len(example_sig))))
 404                 cache_res = res(test_string)
 405                 cache_spec = [ord(c) for c in cache_res]
 406                 try:
 407                     os.makedirs(os.path.dirname(cache_fn))
 408                 except OSError as ose:
 409                     if ose.errno != errno.EEXIST:
 410                         raise
 411                 write_json_file(cache_spec, cache_fn)
 412             except Exception:
 413                 tb = traceback.format_exc()
 414                 self._downloader.report_warning(
 415                     u'Writing cache to %r failed: %s' % (cache_fn, tb))
 416
 417         return res
 418
 419     def _print_sig_code(self, func, example_sig):
 420         def gen_sig_code(idxs):
 421             def _genslice(start, end, step):
 422                 starts = u'' if start == 0 else str(start)
 423                 ends = (u':%d' % (end+step)) if end + step >= 0 else u':'
 424                 steps = u'' if step == 1 else (u':%d' % step)
 425                 return u's[%s%s%s]' % (starts, ends, steps)
 426
 427             step = None
 428             start = '(Never used)'  # Quelch pyflakes warnings - start will be
 429                                     # set as soon as step is set
 430             for i, prev in zip(idxs[1:], idxs[:-1]):
 431                 if step is not None:
 432                     if i - prev == step:
 433                         continue
 434                     yield _genslice(start, prev, step)
 435                     step = None
 436                     continue
 437                 if i - prev in [-1, 1]:
 438                     step = i - prev
 439                     start = prev
 440                     continue
 441                 else:
 442                     yield u's[%d]' % prev
 443             if step is None:
 444                 yield u's[%d]' % i
 445             else:
 446                 yield _genslice(start, i, step)
 447
 448         test_string = u''.join(map(compat_chr, range(len(example_sig))))
 449         cache_res = func(test_string)
 450         cache_spec = [ord(c) for c in cache_res]
 451         expr_code = u' + '.join(gen_sig_code(cache_spec))
 452         signature_id_tuple = '(%s)' % (
 453             ', '.join(compat_str(len(p)) for p in example_sig.split('.')))
 454         code = (u'if tuple(len(p) for p in s.split(\'.\')) == %s:\n'
 455                 u'    return %s\n') % (signature_id_tuple, expr_code)
 456         self.to_screen(u'Extracted signature function:\n' + code)
 457
 458     def _parse_sig_js(self, jscode):
 459         funcname = self._search_regex(
 460             r'signature=([$a-zA-Z]+)', jscode,
 461              u'Initial JS player signature function name')
 462
 463         jsi = JSInterpreter(jscode)
 464         initial_function = jsi.extract_function(funcname)
 465         return lambda s: initial_function([s])
 466
 467     def _parse_sig_swf(self, file_contents):
 468         swfi = SWFInterpreter(file_contents)
 469         TARGET_CLASSNAME = u'SignatureDecipher'
 470         searched_class = swfi.extract_class(TARGET_CLASSNAME)
 471         initial_function = swfi.extract_function(searched_class, u'decipher')
 472         return lambda s: initial_function([s])
 473
 474     def _decrypt_signature(self, s, video_id, player_url, age_gate=False):
 475         """Turn the encrypted s field into a working signature"""
 476
 477         if player_url is None:
 478             raise ExtractorError(u'Cannot decrypt signature without player_url')
 479
 480         if player_url.startswith(u'//'):
 481             player_url = u'https:' + player_url
 482         try:
 483             player_id = (player_url, self._signature_cache_id(s))
 484             if player_id not in self._player_cache:
 485                 func = self._extract_signature_function(
 486                     video_id, player_url, s
 487                 )
 488                 self._player_cache[player_id] = func
 489             func = self._player_cache[player_id]
 490             if self._downloader.params.get('youtube_print_sig_code'):
 491                 self._print_sig_code(func, s)
 492             return func(s)
 493         except Exception as e:
 494             tb = traceback.format_exc()
 495             raise ExtractorError(
 496                 u'Signature extraction failed: ' + tb, cause=e)
 497
 498     def _get_available_subtitles(self, video_id, webpage):
 499         try:
 500             sub_list = self._download_webpage(
 501                 'https://video.google.com/timedtext?hl=en&type=list&v=%s' % video_id,
 502                 video_id, note=False)
 503         except ExtractorError as err:
 504             self._downloader.report_warning(u'unable to download video subtitles: %s' % compat_str(err))
 505             return {}
 506         lang_list = re.findall(r'name="([^"]*)"[^>]+lang_code="([\w\-]+)"', sub_list)
 507
 508         sub_lang_list = {}
 509         for l in lang_list:
 510             lang = l[1]
 511             if lang in sub_lang_list:
 512                 continue
 513             params = compat_urllib_parse.urlencode({
 514                 'lang': lang,
 515                 'v': video_id,
 516                 'fmt': self._downloader.params.get('subtitlesformat', 'srt'),
 517                 'name': unescapeHTML(l[0]).encode('utf-8'),
 518             })
 519             url = u'https://www.youtube.com/api/timedtext?' + params
 520             sub_lang_list[lang] = url
 521         if not sub_lang_list:
 522             self._downloader.report_warning(u'video doesn\'t have subtitles')
 523             return {}
 524         return sub_lang_list
 525
 526     def _get_available_automatic_caption(self, video_id, webpage):
 527         """We need the webpage for getting the captions url, pass it as an
 528            argument to speed up the process."""
 529         sub_format = self._downloader.params.get('subtitlesformat', 'srt')
 530         self.to_screen(u'%s: Looking for automatic captions' % video_id)
 531         mobj = re.search(r';ytplayer.config = ({.*?});', webpage)
 532         err_msg = u'Couldn\'t find automatic captions for %s' % video_id
 533         if mobj is None:
 534             self._downloader.report_warning(err_msg)
 535             return {}
 536         player_config = json.loads(mobj.group(1))
 537         try:
 538             args = player_config[u'args']
 539             caption_url = args[u'ttsurl']
 540             timestamp = args[u'timestamp']
 541             # We get the available subtitles
 542             list_params = compat_urllib_parse.urlencode({
 543                 'type': 'list',
 544                 'tlangs': 1,
 545                 'asrs': 1,
 546             })
 547             list_url = caption_url + '&' + list_params
 548             caption_list = self._download_xml(list_url, video_id)
 549             original_lang_node = caption_list.find('track')
 550             if original_lang_node is None or original_lang_node.attrib.get('kind') != 'asr' :
 551                 self._downloader.report_warning(u'Video doesn\'t have automatic captions')
 552                 return {}
 553             original_lang = original_lang_node.attrib['lang_code']
 554
 555             sub_lang_list = {}
 556             for lang_node in caption_list.findall('target'):
 557                 sub_lang = lang_node.attrib['lang_code']
 558                 params = compat_urllib_parse.urlencode({
 559                     'lang': original_lang,
 560                     'tlang': sub_lang,
 561                     'fmt': sub_format,
 562                     'ts': timestamp,
 563                     'kind': 'asr',
 564                 })
 565                 sub_lang_list[sub_lang] = caption_url + '&' + params
 566             return sub_lang_list
 567         # An extractor error can be raise by the download process if there are
 568         # no automatic captions but there are subtitles
 569         except (KeyError, ExtractorError):
 570             self._downloader.report_warning(err_msg)
 571             return {}
 572
 573     @classmethod
 574     def extract_id(cls, url):
 575         mobj = re.match(cls._VALID_URL, url, re.VERBOSE)
 576         if mobj is None:
 577             raise ExtractorError(u'Invalid URL: %s' % url)
 578         video_id = mobj.group(2)
 579         return video_id
 580
 581     def _extract_from_m3u8(self, manifest_url, video_id):
 582         url_map = {}
 583         def _get_urls(_manifest):
 584             lines = _manifest.split('\n')
 585             urls = filter(lambda l: l and not l.startswith('#'),
 586                             lines)
 587             return urls
 588         manifest = self._download_webpage(manifest_url, video_id, u'Downloading formats manifest')
 589         formats_urls = _get_urls(manifest)
 590         for format_url in formats_urls:
 591             itag = self._search_regex(r'itag/(\d+?)/', format_url, 'itag')
 592             url_map[itag] = format_url
 593         return url_map
 594
 595     def _extract_annotations(self, video_id):
 596         url = 'https://www.youtube.com/annotations_invideo?features=1&legacy=1&video_id=%s' % video_id
 597         return self._download_webpage(url, video_id, note=u'Searching for annotations.', errnote=u'Unable to download video annotations.')
 598
 599     def _real_extract(self, url):
 600         proto = (
 601             u'http' if self._downloader.params.get('prefer_insecure', False)
 602             else u'https')
 603
 604         # Extract original video URL from URL with redirection, like age verification, using next_url parameter
 605         mobj = re.search(self._NEXT_URL_RE, url)
 606         if mobj:
 607             url = proto + '://www.youtube.com/' + compat_urllib_parse.unquote(mobj.group(1)).lstrip('/')
 608         video_id = self.extract_id(url)
 609
 610         # Get video webpage
 611         url = proto + '://www.youtube.com/watch?v=%s&gl=US&hl=en&has_verified=1' % video_id
 612         video_webpage = self._download_webpage(url, video_id)
 613
 614         # Attempt to extract SWF player URL
 615         mobj = re.search(r'swfConfig.*?"(https?:\\/\\/.*?watch.*?-.*?\.swf)"', video_webpage)
 616         if mobj is not None:
 617             player_url = re.sub(r'\\(.)', r'\1', mobj.group(1))
 618         else:
 619             player_url = None
 620
 621         # Get video info
 622         self.report_video_info_webpage_download(video_id)
 623         if re.search(r'player-age-gate-content">', video_webpage) is not None:
 624             self.report_age_confirmation()
 625             age_gate = True
 626             # We simulate the access to the video from www.youtube.com/v/{video_id}
 627             # this can be viewed without login into Youtube
 628             data = compat_urllib_parse.urlencode({
 629                 'video_id': video_id,
 630                 'eurl': 'https://youtube.googleapis.com/v/' + video_id,
 631                 'sts': self._search_regex(
 632                     r'"sts"\s*:\s*(\d+)', video_webpage, 'sts'),
 633             })
 634             video_info_url = proto + '://www.youtube.com/get_video_info?' + data
 635             video_info_webpage = self._download_webpage(video_info_url, video_id,
 636                                     note=False,
 637                                     errnote='unable to download video info webpage')
 638             video_info = compat_parse_qs(video_info_webpage)
 639         else:
 640             age_gate = False
 641             for el_type in ['&el=embedded', '&el=detailpage', '&el=vevo', '']:
 642                 video_info_url = (proto + '://www.youtube.com/get_video_info?&video_id=%s%s&ps=default&eurl=&gl=US&hl=en'
 643                         % (video_id, el_type))
 644                 video_info_webpage = self._download_webpage(video_info_url, video_id,
 645                                         note=False,
 646                                         errnote='unable to download video info webpage')
 647                 video_info = compat_parse_qs(video_info_webpage)
 648                 if 'token' in video_info:
 649                     break
 650         if 'token' not in video_info:
 651             if 'reason' in video_info:
 652                 raise ExtractorError(
 653                     u'YouTube said: %s' % video_info['reason'][0],
 654                     expected=True, video_id=video_id)
 655             else:
 656                 raise ExtractorError(
 657                     u'"token" parameter not in video info for unknown reason',
 658                     video_id=video_id)
 659
 660         if 'view_count' in video_info:
 661             view_count = int(video_info['view_count'][0])
 662         else:
 663             view_count = None
 664
 665         # Check for "rental" videos
 666         if 'ypc_video_rental_bar_text' in video_info and 'author' not in video_info:
 667             raise ExtractorError(u'"rental" videos not supported')
 668
 669         # Start extracting information
 670         self.report_information_extraction(video_id)
 671
 672         # uploader
 673         if 'author' not in video_info:
 674             raise ExtractorError(u'Unable to extract uploader name')
 675         video_uploader = compat_urllib_parse.unquote_plus(video_info['author'][0])
 676
 677         # uploader_id
 678         video_uploader_id = None
 679         mobj = re.search(r'<link itemprop="url" href="http://www.youtube.com/(?:user|channel)/([^"]+)">', video_webpage)
 680         if mobj is not None:
 681             video_uploader_id = mobj.group(1)
 682         else:
 683             self._downloader.report_warning(u'unable to extract uploader nickname')
 684
 685         # title
 686         if 'title' in video_info:
 687             video_title = video_info['title'][0]
 688         else:
 689             self._downloader.report_warning(u'Unable to extract video title')
 690             video_title = u'_'
 691
 692         # thumbnail image
 693         # We try first to get a high quality image:
 694         m_thumb = re.search(r'<span itemprop="thumbnail".*?href="(.*?)">',
 695                             video_webpage, re.DOTALL)
 696         if m_thumb is not None:
 697             video_thumbnail = m_thumb.group(1)
 698         elif 'thumbnail_url' not in video_info:
 699             self._downloader.report_warning(u'unable to extract video thumbnail')
 700             video_thumbnail = None
 701         else:   # don't panic if we can't find it
 702             video_thumbnail = compat_urllib_parse.unquote_plus(video_info['thumbnail_url'][0])
 703
 704         # upload date
 705         upload_date = None
 706         mobj = re.search(r'(?s)id="eow-date.*?>(.*?)</span>', video_webpage)
 707         if mobj is None:
 708             mobj = re.search(
 709                 r'(?s)id="watch-uploader-info".*?>.*?(?:Published|Uploaded|Streamed live) on (.*?)</strong>',
 710                 video_webpage)
 711         if mobj is not None:
 712             upload_date = ' '.join(re.sub(r'[/,-]', r' ', mobj.group(1)).split())
 713             upload_date = unified_strdate(upload_date)
 714
 715         m_cat_container = get_element_by_id("eow-category", video_webpage)
 716         if m_cat_container:
 717             category = self._html_search_regex(
 718                 r'(?s)<a[^<]+>(.*?)</a>', m_cat_container, 'category',
 719                 default=None)
 720             video_categories = None if category is None else [category]
 721         else:
 722             video_categories = None
 723
 724         # description
 725         video_description = get_element_by_id("eow-description", video_webpage)
 726         if video_description:
 727             video_description = re.sub(r'''(?x)
 728                 <a\s+
 729                     (?:[a-zA-Z-]+="[^"]+"\s+)*?
 730                     title="([^"]+)"\s+
 731                     (?:[a-zA-Z-]+="[^"]+"\s+)*?
 732                     class="yt-uix-redirect-link"\s*>
 733                 [^<]+
 734                 </a>
 735             ''', r'\1', video_description)
 736             video_description = clean_html(video_description)
 737         else:
 738             fd_mobj = re.search(r'<meta name="description" content="([^"]+)"', video_webpage)
 739             if fd_mobj:
 740                 video_description = unescapeHTML(fd_mobj.group(1))
 741             else:
 742                 video_description = u''
 743
 744         def _extract_count(klass):
 745             count = self._search_regex(
 746                 r'class="%s">([\d,]+)</span>' % re.escape(klass),
 747                 video_webpage, klass, default=None)
 748             if count is not None:
 749                 return int(count.replace(',', ''))
 750             return None
 751         like_count = _extract_count(u'likes-count')
 752         dislike_count = _extract_count(u'dislikes-count')
 753
 754         # subtitles
 755         video_subtitles = self.extract_subtitles(video_id, video_webpage)
 756
 757         if self._downloader.params.get('listsubtitles', False):
 758             self._list_available_subtitles(video_id, video_webpage)
 759             return
 760
 761         if 'length_seconds' not in video_info:
 762             self._downloader.report_warning(u'unable to extract video duration')
 763             video_duration = None
 764         else:
 765             video_duration = int(compat_urllib_parse.unquote_plus(video_info['length_seconds'][0]))
 766
 767         # annotations
 768         video_annotations = None
 769         if self._downloader.params.get('writeannotations', False):
 770                 video_annotations = self._extract_annotations(video_id)
 771
 772         # Decide which formats to download
 773         try:
 774             mobj = re.search(r';ytplayer\.config\s*=\s*({.*?});', video_webpage)
 775             if not mobj:
 776                 raise ValueError('Could not find vevo ID')
 777             json_code = uppercase_escape(mobj.group(1))
 778             ytplayer_config = json.loads(json_code)
 779             args = ytplayer_config['args']
 780             # Easy way to know if the 's' value is in url_encoded_fmt_stream_map
 781             # this signatures are encrypted
 782             if 'url_encoded_fmt_stream_map' not in args:
 783                 raise ValueError(u'No stream_map present')  # caught below
 784             re_signature = re.compile(r'[&,]s=')
 785             m_s = re_signature.search(args['url_encoded_fmt_stream_map'])
 786             if m_s is not None:
 787                 self.to_screen(u'%s: Encrypted signatures detected.' % video_id)
 788                 video_info['url_encoded_fmt_stream_map'] = [args['url_encoded_fmt_stream_map']]
 789             m_s = re_signature.search(args.get('adaptive_fmts', u''))
 790             if m_s is not None:
 791                 if 'adaptive_fmts' in video_info:
 792                     video_info['adaptive_fmts'][0] += ',' + args['adaptive_fmts']
 793                 else:
 794                     video_info['adaptive_fmts'] = [args['adaptive_fmts']]
 795         except ValueError:
 796             pass
 797
 798         def _map_to_format_list(urlmap):
 799             formats = []
 800             for itag, video_real_url in urlmap.items():
 801                 dct = {
 802                     'format_id': itag,
 803                     'url': video_real_url,
 804                     'player_url': player_url,
 805                 }
 806                 if itag in self._formats:
 807                     dct.update(self._formats[itag])
 808                 formats.append(dct)
 809             return formats
 810
 811         if 'conn' in video_info and video_info['conn'][0].startswith('rtmp'):
 812             self.report_rtmp_download()
 813             formats = [{
 814                 'format_id': '_rtmp',
 815                 'protocol': 'rtmp',
 816                 'url': video_info['conn'][0],
 817                 'player_url': player_url,
 818             }]
 819         elif len(video_info.get('url_encoded_fmt_stream_map', [])) >= 1 or len(video_info.get('adaptive_fmts', [])) >= 1:
 820             encoded_url_map = video_info.get('url_encoded_fmt_stream_map', [''])[0] + ',' + video_info.get('adaptive_fmts',[''])[0]
 821             if 'rtmpe%3Dyes' in encoded_url_map:
 822                 raise ExtractorError('rtmpe downloads are not supported, see https://github.com/rg3/youtube-dl/issues/343 for more information.', expected=True)
 823             url_map = {}
 824             for url_data_str in encoded_url_map.split(','):
 825                 url_data = compat_parse_qs(url_data_str)
 826                 if 'itag' not in url_data or 'url' not in url_data:
 827                     continue
 828                 format_id = url_data['itag'][0]
 829                 url = url_data['url'][0]
 830
 831                 if 'sig' in url_data:
 832                     url += '&signature=' + url_data['sig'][0]
 833                 elif 's' in url_data:
 834                     encrypted_sig = url_data['s'][0]
 835
 836                     if not age_gate:
 837                         jsplayer_url_json = self._search_regex(
 838                             r'"assets":.+?"js":\s*("[^"]+")',
 839                             video_webpage, u'JS player URL')
 840                         player_url = json.loads(jsplayer_url_json)
 841                     if player_url is None:
 842                         player_url_json = self._search_regex(
 843                             r'ytplayer\.config.*?"url"\s*:\s*("[^"]+")',
 844                             video_webpage, u'age gate player URL')
 845                         player_url = json.loads(player_url_json)
 846
 847                     if self._downloader.params.get('verbose'):
 848                         if player_url is None:
 849                             player_version = 'unknown'
 850                             player_desc = 'unknown'
 851                         else:
 852                             if player_url.endswith('swf'):
 853                                 player_version = self._search_regex(
 854                                     r'-(.+?)(?:/watch_as3)?\.swf$', player_url,
 855                                     u'flash player', fatal=False)
 856                                 player_desc = 'flash player %s' % player_version
 857                             else:
 858                                 player_version = self._search_regex(
 859                                     r'html5player-([^/]+?)(?:/html5player)?\.js',
 860                                     player_url,
 861                                     'html5 player', fatal=False)
 862                                 player_desc = u'html5 player %s' % player_version
 863
 864                         parts_sizes = self._signature_cache_id(encrypted_sig)
 865                         self.to_screen(u'{%s} signature length %s, %s' %
 866                             (format_id, parts_sizes, player_desc))
 867
 868                     signature = self._decrypt_signature(
 869                         encrypted_sig, video_id, player_url, age_gate)
 870                     url += '&signature=' + signature
 871                 if 'ratebypass' not in url:
 872                     url += '&ratebypass=yes'
 873                 url_map[format_id] = url
 874             formats = _map_to_format_list(url_map)
 875         elif video_info.get('hlsvp'):
 876             manifest_url = video_info['hlsvp'][0]
 877             url_map = self._extract_from_m3u8(manifest_url, video_id)
 878             formats = _map_to_format_list(url_map)
 879         else:
 880             raise ExtractorError(u'no conn, hlsvp or url_encoded_fmt_stream_map information found in video info')
 881
 882         # Look for the DASH manifest
 883         if (self._downloader.params.get('youtube_include_dash_manifest', False)):
 884             try:
 885                 # The DASH manifest used needs to be the one from the original video_webpage.
 886                 # The one found in get_video_info seems to be using different signatures.
 887                 # However, in the case of an age restriction there won't be any embedded dashmpd in the video_webpage.
 888                 # Luckily, it seems, this case uses some kind of default signature (len == 86), so the
 889                 # combination of get_video_info and the _static_decrypt_signature() decryption fallback will work here.
 890                 if age_gate:
 891                     dash_manifest_url = video_info.get('dashmpd')[0]
 892                 else:
 893                     dash_manifest_url = ytplayer_config['args']['dashmpd']
 894                 def decrypt_sig(mobj):
 895                     s = mobj.group(1)
 896                     dec_s = self._decrypt_signature(s, video_id, player_url, age_gate)
 897                     return '/signature/%s' % dec_s
 898                 dash_manifest_url = re.sub(r'/s/([\w\.]+)', decrypt_sig, dash_manifest_url)
 899                 dash_doc = self._download_xml(
 900                     dash_manifest_url, video_id,
 901                     note=u'Downloading DASH manifest',
 902                     errnote=u'Could not download DASH manifest')
 903                 for r in dash_doc.findall(u'.//{urn:mpeg:DASH:schema:MPD:2011}Representation'):
 904                     url_el = r.find('{urn:mpeg:DASH:schema:MPD:2011}BaseURL')
 905                     if url_el is None:
 906                         continue
 907                     format_id = r.attrib['id']
 908                     video_url = url_el.text
 909                     filesize = int_or_none(url_el.attrib.get('{http://youtube.com/yt/2012/10/10}contentLength'))
 910                     f = {
 911                         'format_id': format_id,
 912                         'url': video_url,
 913                         'width': int_or_none(r.attrib.get('width')),
 914                         'tbr': int_or_none(r.attrib.get('bandwidth'), 1000),
 915                         'asr': int_or_none(r.attrib.get('audioSamplingRate')),
 916                         'filesize': filesize,
 917                     }
 918                     try:
 919                         existing_format = next(
 920                             fo for fo in formats
 921                             if fo['format_id'] == format_id)
 922                     except StopIteration:
 923                         f.update(self._formats.get(format_id, {}))
 924                         formats.append(f)
 925                     else:
 926                         existing_format.update(f)
 927
 928             except (ExtractorError, KeyError) as e:
 929                 self.report_warning(u'Skipping DASH manifest: %s' % e, video_id)
 930
 931         self._sort_formats(formats)
 932
 933         return {
 934             'id':           video_id,
 935             'uploader':     video_uploader,
 936             'uploader_id':  video_uploader_id,
 937             'upload_date':  upload_date,
 938             'title':        video_title,
 939             'thumbnail':    video_thumbnail,
 940             'description':  video_description,
 941             'categories':   video_categories,
 942             'subtitles':    video_subtitles,
 943             'duration':     video_duration,
 944             'age_limit':    18 if age_gate else 0,
 945             'annotations':  video_annotations,
 946             'webpage_url': proto + '://www.youtube.com/watch?v=%s' % video_id,
 947             'view_count':   view_count,
 948             'like_count': like_count,
 949             'dislike_count': dislike_count,
 950             'formats':      formats,
 951         }
 952
 953 class YoutubePlaylistIE(YoutubeBaseInfoExtractor):
 954     IE_DESC = u'YouTube.com playlists'
 955     _VALID_URL = r"""(?x)(?:
 956                         (?:https?://)?
 957                         (?:\w+\.)?
 958                         youtube\.com/
 959                         (?:
 960                            (?:course|view_play_list|my_playlists|artist|playlist|watch)
 961                            \? (?:.*?&)*? (?:p|a|list)=
 962                         |  p/
 963                         )
 964                         (
 965                             (?:PL|LL|EC|UU|FL|RD)?[0-9A-Za-z-_]{10,}
 966                             # Top tracks, they can also include dots
 967                             |(?:MC)[\w\.]*
 968                         )
 969                         .*
 970                      |
 971                         ((?:PL|LL|EC|UU|FL|RD)[0-9A-Za-z-_]{10,})
 972                      )"""
 973     _TEMPLATE_URL = 'https://www.youtube.com/playlist?list=%s'
 974     _MORE_PAGES_INDICATOR = r'data-link-type="next"'
 975     _VIDEO_RE = r'href="\s*/watch\?v=(?P<id>[0-9A-Za-z_-]{11})&amp;[^"]*?index=(?P<index>\d+)'
 976     IE_NAME = u'youtube:playlist'
 977
 978     def _real_initialize(self):
 979         self._login()
 980
 981     def _ids_to_results(self, ids):
 982         return [self.url_result(vid_id, 'Youtube', video_id=vid_id)
 983                        for vid_id in ids]
 984
 985     def _extract_mix(self, playlist_id):
 986         # The mixes are generated from a a single video
 987         # the id of the playlist is just 'RD' + video_id
 988         url = 'https://youtube.com/watch?v=%s&list=%s' % (playlist_id[-11:], playlist_id)
 989         webpage = self._download_webpage(url, playlist_id, u'Downloading Youtube mix')
 990         search_title = lambda class_name: get_element_by_attribute('class', class_name, webpage)
 991         title_span = (search_title('playlist-title') or
 992             search_title('title long-title') or search_title('title'))
 993         title = clean_html(title_span)
 994         video_re = r'''(?x)data-video-username=".*?".*?
 995                        href="/watch\?v=([0-9A-Za-z_-]{11})&amp;[^"]*?list=%s''' % re.escape(playlist_id)
 996         ids = orderedSet(re.findall(video_re, webpage, flags=re.DOTALL))
 997         url_results = self._ids_to_results(ids)
 998
 999         return self.playlist_result(url_results, playlist_id, title)
1000
1001     def _real_extract(self, url):
1002         # Extract playlist id
1003         mobj = re.match(self._VALID_URL, url)
1004         if mobj is None:
1005             raise ExtractorError(u'Invalid URL: %s' % url)
1006         playlist_id = mobj.group(1) or mobj.group(2)
1007
1008         # Check if it's a video-specific URL
1009         query_dict = compat_urlparse.parse_qs(compat_urlparse.urlparse(url).query)
1010         if 'v' in query_dict:
1011             video_id = query_dict['v'][0]
1012             if self._downloader.params.get('noplaylist'):
1013                 self.to_screen(u'Downloading just video %s because of --no-playlist' % video_id)
1014                 return self.url_result(video_id, 'Youtube', video_id=video_id)
1015             else:
1016                 self.to_screen(u'Downloading playlist %s - add --no-playlist to just download video %s' % (playlist_id, video_id))
1017
1018         if playlist_id.startswith('RD'):
1019             # Mixes require a custom extraction process
1020             return self._extract_mix(playlist_id)
1021         if playlist_id.startswith('TL'):
1022             raise ExtractorError(u'For downloading YouTube.com top lists, use '
1023                 u'the "yttoplist" keyword, for example "youtube-dl \'yttoplist:music:Top Tracks\'"', expected=True)
1024
1025         url = self._TEMPLATE_URL % playlist_id
1026         page = self._download_webpage(url, playlist_id)
1027         more_widget_html = content_html = page
1028
1029         # Check if the playlist exists or is private
1030         if re.search(r'<div class="yt-alert-message">[^<]*?(The|This) playlist (does not exist|is private)[^<]*?</div>', page) is not None:
1031             raise ExtractorError(
1032                 u'The playlist doesn\'t exist or is private, use --username or '
1033                 '--netrc to access it.',
1034                 expected=True)
1035
1036         # Extract the video ids from the playlist pages
1037         ids = []
1038
1039         for page_num in itertools.count(1):
1040             matches = re.finditer(self._VIDEO_RE, content_html)
1041             # We remove the duplicates and the link with index 0
1042             # (it's not the first video of the playlist)
1043             new_ids = orderedSet(m.group('id') for m in matches if m.group('index') != '0')
1044             ids.extend(new_ids)
1045
1046             mobj = re.search(r'data-uix-load-more-href="/?(?P<more>[^"]+)"', more_widget_html)
1047             if not mobj:
1048                 break
1049
1050             more = self._download_json(
1051                 'https://youtube.com/%s' % mobj.group('more'), playlist_id,
1052                 'Downloading page #%s' % page_num,
1053                 transform_source=uppercase_escape)
1054             content_html = more['content_html']
1055             more_widget_html = more['load_more_widget_html']
1056
1057         playlist_title = self._html_search_regex(
1058             r'(?s)<h1 class="pl-header-title[^"]*">\s*(.*?)\s*</h1>',
1059             page, u'title')
1060
1061         url_results = self._ids_to_results(ids)
1062         return self.playlist_result(url_results, playlist_id, playlist_title)
1063
1064
1065 class YoutubeTopListIE(YoutubePlaylistIE):
1066     IE_NAME = u'youtube:toplist'
1067     IE_DESC = (u'YouTube.com top lists, "yttoplist:{channel}:{list title}"'
1068         u' (Example: "yttoplist:music:Top Tracks")')
1069     _VALID_URL = r'yttoplist:(?P<chann>.*?):(?P<title>.*?)$'
1070
1071     def _real_extract(self, url):
1072         mobj = re.match(self._VALID_URL, url)
1073         channel = mobj.group('chann')
1074         title = mobj.group('title')
1075         query = compat_urllib_parse.urlencode({'title': title})
1076         playlist_re = 'href="([^"]+?%s.*?)"' % re.escape(query)
1077         channel_page = self._download_webpage('https://www.youtube.com/%s' % channel, title)
1078         link = self._html_search_regex(playlist_re, channel_page, u'list')
1079         url = compat_urlparse.urljoin('https://www.youtube.com/', link)
1080
1081         video_re = r'data-index="\d+".*?data-video-id="([0-9A-Za-z_-]{11})"'
1082         ids = []
1083         # sometimes the webpage doesn't contain the videos
1084         # retry until we get them
1085         for i in itertools.count(0):
1086             msg = u'Downloading Youtube mix'
1087             if i > 0:
1088                 msg += ', retry #%d' % i
1089             webpage = self._download_webpage(url, title, msg)
1090             ids = orderedSet(re.findall(video_re, webpage))
1091             if ids:
1092                 break
1093         url_results = self._ids_to_results(ids)
1094         return self.playlist_result(url_results, playlist_title=title)
1095
1096
1097 class YoutubeChannelIE(InfoExtractor):
1098     IE_DESC = u'YouTube.com channels'
1099     _VALID_URL = r"^(?:https?://)?(?:youtu\.be|(?:\w+\.)?youtube(?:-nocookie)?\.com)/channel/([0-9A-Za-z_-]+)"
1100     _MORE_PAGES_INDICATOR = 'yt-uix-load-more'
1101     _MORE_PAGES_URL = 'https://www.youtube.com/c4_browse_ajax?action_load_more_videos=1&flow=list&paging=%s&view=0&sort=da&channel_id=%s'
1102     IE_NAME = u'youtube:channel'
1103
1104     def extract_videos_from_page(self, page):
1105         ids_in_page = []
1106         for mobj in re.finditer(r'href="/watch\?v=([0-9A-Za-z_-]+)&?', page):
1107             if mobj.group(1) not in ids_in_page:
1108                 ids_in_page.append(mobj.group(1))
1109         return ids_in_page
1110
1111     def _real_extract(self, url):
1112         # Extract channel id
1113         mobj = re.match(self._VALID_URL, url)
1114         if mobj is None:
1115             raise ExtractorError(u'Invalid URL: %s' % url)
1116
1117         # Download channel page
1118         channel_id = mobj.group(1)
1119         video_ids = []
1120         url = 'https://www.youtube.com/channel/%s/videos' % channel_id
1121         channel_page = self._download_webpage(url, channel_id)
1122         autogenerated = re.search(r'''(?x)
1123                 class="[^"]*?(?:
1124                     channel-header-autogenerated-label|
1125                     yt-channel-title-autogenerated
1126                 )[^"]*"''', channel_page) is not None
1127
1128         if autogenerated:
1129             # The videos are contained in a single page
1130             # the ajax pages can't be used, they are empty
1131             video_ids = self.extract_videos_from_page(channel_page)
1132         else:
1133             # Download all channel pages using the json-based channel_ajax query
1134             for pagenum in itertools.count(1):
1135                 url = self._MORE_PAGES_URL % (pagenum, channel_id)
1136                 page = self._download_json(
1137                     url, channel_id, note=u'Downloading page #%s' % pagenum,
1138                     transform_source=uppercase_escape)
1139
1140                 ids_in_page = self.extract_videos_from_page(page['content_html'])
1141                 video_ids.extend(ids_in_page)
1142
1143                 if self._MORE_PAGES_INDICATOR not in page['load_more_widget_html']:
1144                     break
1145
1146         self._downloader.to_screen(u'[youtube] Channel %s: Found %i videos' % (channel_id, len(video_ids)))
1147
1148         url_entries = [self.url_result(video_id, 'Youtube', video_id=video_id)
1149                        for video_id in video_ids]
1150         return self.playlist_result(url_entries, channel_id)
1151
1152
1153 class YoutubeUserIE(InfoExtractor):
1154     IE_DESC = u'YouTube.com user videos (URL or "ytuser" keyword)'
1155     _VALID_URL = r'(?:(?:(?:https?://)?(?:\w+\.)?youtube\.com/(?:user/)?(?!(?:attribution_link|watch|results)(?:$|[^a-z_A-Z0-9-])))|ytuser:)(?!feed/)([A-Za-z0-9_-]+)'
1156     _TEMPLATE_URL = 'https://gdata.youtube.com/feeds/api/users/%s'
1157     _GDATA_PAGE_SIZE = 50
1158     _GDATA_URL = 'https://gdata.youtube.com/feeds/api/users/%s/uploads?max-results=%d&start-index=%d&alt=json'
1159     IE_NAME = u'youtube:user'
1160
1161     @classmethod
1162     def suitable(cls, url):
1163         # Don't return True if the url can be extracted with other youtube
1164         # extractor, the regex would is too permissive and it would match.
1165         other_ies = iter(klass for (name, klass) in globals().items() if name.endswith('IE') and klass is not cls)
1166         if any(ie.suitable(url) for ie in other_ies): return False
1167         else: return super(YoutubeUserIE, cls).suitable(url)
1168
1169     def _real_extract(self, url):
1170         # Extract username
1171         mobj = re.match(self._VALID_URL, url)
1172         if mobj is None:
1173             raise ExtractorError(u'Invalid URL: %s' % url)
1174
1175         username = mobj.group(1)
1176
1177         # Download video ids using YouTube Data API. Result size per
1178         # query is limited (currently to 50 videos) so we need to query
1179         # page by page until there are no video ids - it means we got
1180         # all of them.
1181
1182         def download_page(pagenum):
1183             start_index = pagenum * self._GDATA_PAGE_SIZE + 1
1184
1185             gdata_url = self._GDATA_URL % (username, self._GDATA_PAGE_SIZE, start_index)
1186             page = self._download_webpage(
1187                 gdata_url, username,
1188                 u'Downloading video ids from %d to %d' % (
1189                     start_index, start_index + self._GDATA_PAGE_SIZE))
1190
1191             try:
1192                 response = json.loads(page)
1193             except ValueError as err:
1194                 raise ExtractorError(u'Invalid JSON in API response: ' + compat_str(err))
1195             if 'entry' not in response['feed']:
1196                 return
1197
1198             # Extract video identifiers
1199             entries = response['feed']['entry']
1200             for entry in entries:
1201                 title = entry['title']['$t']
1202                 video_id = entry['id']['$t'].split('/')[-1]
1203                 yield {
1204                     '_type': 'url',
1205                     'url': video_id,
1206                     'ie_key': 'Youtube',
1207                     'id': video_id,
1208                     'title': title,
1209                 }
1210         url_results = PagedList(download_page, self._GDATA_PAGE_SIZE)
1211
1212         return self.playlist_result(url_results, playlist_title=username)
1213
1214
1215 class YoutubeSearchIE(SearchInfoExtractor):
1216     IE_DESC = u'YouTube.com searches'
1217     _API_URL = u'https://gdata.youtube.com/feeds/api/videos?q=%s&start-index=%i&max-results=50&v=2&alt=jsonc'
1218     _MAX_RESULTS = 1000
1219     IE_NAME = u'youtube:search'
1220     _SEARCH_KEY = 'ytsearch'
1221
1222     def _get_n_results(self, query, n):
1223         """Get a specified number of results for a query"""
1224
1225         video_ids = []
1226         pagenum = 0
1227         limit = n
1228         PAGE_SIZE = 50
1229
1230         while (PAGE_SIZE * pagenum) < limit:
1231             result_url = self._API_URL % (
1232                 compat_urllib_parse.quote_plus(query.encode('utf-8')),
1233                 (PAGE_SIZE * pagenum) + 1)
1234             data_json = self._download_webpage(
1235                 result_url, video_id=u'query "%s"' % query,
1236                 note=u'Downloading page %s' % (pagenum + 1),
1237                 errnote=u'Unable to download API page')
1238             data = json.loads(data_json)
1239             api_response = data['data']
1240
1241             if 'items' not in api_response:
1242                 raise ExtractorError(
1243                     u'[youtube] No video results', expected=True)
1244
1245             new_ids = list(video['id'] for video in api_response['items'])
1246             video_ids += new_ids
1247
1248             limit = min(n, api_response['totalItems'])
1249             pagenum += 1
1250
1251         if len(video_ids) > n:
1252             video_ids = video_ids[:n]
1253         videos = [self.url_result(video_id, 'Youtube', video_id=video_id)
1254                   for video_id in video_ids]
1255         return self.playlist_result(videos, query)
1256
1257
1258 class YoutubeSearchDateIE(YoutubeSearchIE):
1259     IE_NAME = YoutubeSearchIE.IE_NAME + ':date'
1260     _API_URL = 'https://gdata.youtube.com/feeds/api/videos?q=%s&start-index=%i&max-results=50&v=2&alt=jsonc&orderby=published'
1261     _SEARCH_KEY = 'ytsearchdate'
1262     IE_DESC = u'YouTube.com searches, newest videos first'
1263
1264
1265 class YoutubeSearchURLIE(InfoExtractor):
1266     IE_DESC = u'YouTube.com search URLs'
1267     IE_NAME = u'youtube:search_url'
1268     _VALID_URL = r'https?://(?:www\.)?youtube\.com/results\?(.*?&)?search_query=(?P<query>[^&]+)(?:[&]|$)'
1269
1270     def _real_extract(self, url):
1271         mobj = re.match(self._VALID_URL, url)
1272         query = compat_urllib_parse.unquote_plus(mobj.group('query'))
1273
1274         webpage = self._download_webpage(url, query)
1275         result_code = self._search_regex(
1276             r'(?s)<ol class="item-section"(.*?)</ol>', webpage, u'result HTML')
1277
1278         part_codes = re.findall(
1279             r'(?s)<h3 class="yt-lockup-title">(.*?)</h3>', result_code)
1280         entries = []
1281         for part_code in part_codes:
1282             part_title = self._html_search_regex(
1283                 [r'(?s)title="([^"]+)"', r'>([^<]+)</a>'], part_code, 'item title', fatal=False)
1284             part_url_snippet = self._html_search_regex(
1285                 r'(?s)href="([^"]+)"', part_code, 'item URL')
1286             part_url = compat_urlparse.urljoin(
1287                 'https://www.youtube.com/', part_url_snippet)
1288             entries.append({
1289                 '_type': 'url',
1290                 'url': part_url,
1291                 'title': part_title,
1292             })
1293
1294         return {
1295             '_type': 'playlist',
1296             'entries': entries,
1297             'title': query,
1298         }
1299
1300
1301 class YoutubeShowIE(InfoExtractor):
1302     IE_DESC = u'YouTube.com (multi-season) shows'
1303     _VALID_URL = r'https?://www\.youtube\.com/show/(.*)'
1304     IE_NAME = u'youtube:show'
1305
1306     def _real_extract(self, url):
1307         mobj = re.match(self._VALID_URL, url)
1308         show_name = mobj.group(1)
1309         webpage = self._download_webpage(url, show_name, u'Downloading show webpage')
1310         # There's one playlist for each season of the show
1311         m_seasons = list(re.finditer(r'href="(/playlist\?list=.*?)"', webpage))
1312         self.to_screen(u'%s: Found %s seasons' % (show_name, len(m_seasons)))
1313         return [self.url_result('https://www.youtube.com' + season.group(1), 'YoutubePlaylist') for season in m_seasons]
1314
1315
1316 class YoutubeFeedsInfoExtractor(YoutubeBaseInfoExtractor):
1317     """
1318     Base class for extractors that fetch info from
1319     http://www.youtube.com/feed_ajax
1320     Subclasses must define the _FEED_NAME and _PLAYLIST_TITLE properties.
1321     """
1322     _LOGIN_REQUIRED = True
1323     # use action_load_personal_feed instead of action_load_system_feed
1324     _PERSONAL_FEED = False
1325
1326     @property
1327     def _FEED_TEMPLATE(self):
1328         action = 'action_load_system_feed'
1329         if self._PERSONAL_FEED:
1330             action = 'action_load_personal_feed'
1331         return 'https://www.youtube.com/feed_ajax?%s=1&feed_name=%s&paging=%%s' % (action, self._FEED_NAME)
1332
1333     @property
1334     def IE_NAME(self):
1335         return u'youtube:%s' % self._FEED_NAME
1336
1337     def _real_initialize(self):
1338         self._login()
1339
1340     def _real_extract(self, url):
1341         feed_entries = []
1342         paging = 0
1343         for i in itertools.count(1):
1344             info = self._download_json(self._FEED_TEMPLATE % paging,
1345                                           u'%s feed' % self._FEED_NAME,
1346                                           u'Downloading page %s' % i)
1347             feed_html = info.get('feed_html') or info.get('content_html')
1348             m_ids = re.finditer(r'"/watch\?v=(.*?)["&]', feed_html)
1349             ids = orderedSet(m.group(1) for m in m_ids)
1350             feed_entries.extend(
1351                 self.url_result(video_id, 'Youtube', video_id=video_id)
1352                 for video_id in ids)
1353             mobj = re.search(
1354                 r'data-uix-load-more-href="/?[^"]+paging=(?P<paging>\d+)',
1355                 feed_html)
1356             if mobj is None:
1357                 break
1358             paging = mobj.group('paging')
1359         return self.playlist_result(feed_entries, playlist_title=self._PLAYLIST_TITLE)
1360
1361 class YoutubeSubscriptionsIE(YoutubeFeedsInfoExtractor):
1362     IE_DESC = u'YouTube.com subscriptions feed, "ytsubs" keyword (requires authentication)'
1363     _VALID_URL = r'https?://www\.youtube\.com/feed/subscriptions|:ytsubs(?:criptions)?'
1364     _FEED_NAME = 'subscriptions'
1365     _PLAYLIST_TITLE = u'Youtube Subscriptions'
1366
1367 class YoutubeRecommendedIE(YoutubeFeedsInfoExtractor):
1368     IE_DESC = u'YouTube.com recommended videos, "ytrec" keyword (requires authentication)'
1369     _VALID_URL = r'https?://www\.youtube\.com/feed/recommended|:ytrec(?:ommended)?'
1370     _FEED_NAME = 'recommended'
1371     _PLAYLIST_TITLE = u'Youtube Recommended videos'
1372
1373 class YoutubeWatchLaterIE(YoutubeFeedsInfoExtractor):
1374     IE_DESC = u'Youtube watch later list, "ytwatchlater" keyword (requires authentication)'
1375     _VALID_URL = r'https?://www\.youtube\.com/feed/watch_later|:ytwatchlater'
1376     _FEED_NAME = 'watch_later'
1377     _PLAYLIST_TITLE = u'Youtube Watch Later'
1378     _PERSONAL_FEED = True
1379
1380 class YoutubeHistoryIE(YoutubeFeedsInfoExtractor):
1381     IE_DESC = u'Youtube watch history, "ythistory" keyword (requires authentication)'
1382     _VALID_URL = u'https?://www\.youtube\.com/feed/history|:ythistory'
1383     _FEED_NAME = 'history'
1384     _PERSONAL_FEED = True
1385     _PLAYLIST_TITLE = u'Youtube Watch History'
1386
1387 class YoutubeFavouritesIE(YoutubeBaseInfoExtractor):
1388     IE_NAME = u'youtube:favorites'
1389     IE_DESC = u'YouTube.com favourite videos, "ytfav" keyword (requires authentication)'
1390     _VALID_URL = r'https?://www\.youtube\.com/my_favorites|:ytfav(?:ou?rites)?'
1391     _LOGIN_REQUIRED = True
1392
1393     def _real_extract(self, url):
1394         webpage = self._download_webpage('https://www.youtube.com/my_favorites', 'Youtube Favourites videos')
1395         playlist_id = self._search_regex(r'list=(.+?)["&]', webpage, u'favourites playlist id')
1396         return self.url_result(playlist_id, 'YoutubePlaylist')
1397
1398
1399 class YoutubeTruncatedURLIE(InfoExtractor):
1400     IE_NAME = 'youtube:truncated_url'
1401     IE_DESC = False  # Do not list
1402     _VALID_URL = r'''(?x)
1403         (?:https?://)?[^/]+/watch\?(?:
1404             feature=[a-z_]+|
1405             annotation_id=annotation_[^&]+
1406         )?$|
1407         (?:https?://)?(?:www\.)?youtube\.com/attribution_link\?a=[^&]+$
1408     '''
1409
1410     _TESTS = [{
1411         'url': 'http://www.youtube.com/watch?annotation_id=annotation_3951667041',
1412         'only_matching': True,
1413     }, {
1414         'url': 'http://www.youtube.com/watch?',
1415         'only_matching': True,
1416     }]
1417
1418     def _real_extract(self, url):
1419         raise ExtractorError(
1420             u'Did you forget to quote the URL? Remember that & is a meta '
1421             u'character in most shells, so you want to put the URL in quotes, '
1422             u'like  youtube-dl '
1423             u'"http://www.youtube.com/watch?feature=foo&v=BaW_jenozKc" '
1424             u' or simply  youtube-dl BaW_jenozKc  .',
1425             expected=True)