[youtube] Add a test for the DASH segment downloader
[youtube-dl] / youtube_dl / extractor / youtube.py
1 # coding: utf-8
2
3 from __future__ import unicode_literals
4
5
6 import itertools
7 import json
8 import os.path
9 import re
10 import time
11 import traceback
12
13 from .common import InfoExtractor, SearchInfoExtractor
14 from ..jsinterp import JSInterpreter
15 from ..swfinterp import SWFInterpreter
16 from ..compat import (
17     compat_chr,
18     compat_parse_qs,
19     compat_urllib_parse,
20     compat_urllib_request,
21     compat_urlparse,
22     compat_str,
23 )
24 from ..utils import (
25     clean_html,
26     ExtractorError,
27     float_or_none,
28     get_element_by_attribute,
29     get_element_by_id,
30     int_or_none,
31     orderedSet,
32     unescapeHTML,
33     unified_strdate,
34     uppercase_escape,
35 )
36
37
38 class YoutubeBaseInfoExtractor(InfoExtractor):
39     """Provide base functions for Youtube extractors"""
40     _LOGIN_URL = 'https://accounts.google.com/ServiceLogin'
41     _TWOFACTOR_URL = 'https://accounts.google.com/SecondFactor'
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         self._set_cookie(
48             '.youtube.com', 'PREF', 'f1=50000000&hl=en',
49             # YouTube sets the expire time to about two months
50             expire_time=time.time() + 2 * 30 * 24 * 3600)
51
52     def _ids_to_results(self, ids):
53         return [
54             self.url_result(vid_id, 'Youtube', video_id=vid_id)
55             for vid_id in ids]
56
57     def _login(self):
58         """
59         Attempt to log in to YouTube.
60         True is returned if successful or skipped.
61         False is returned if login failed.
62
63         If _LOGIN_REQUIRED is set and no authentication was provided, an error is raised.
64         """
65         (username, password) = self._get_login_info()
66         # No authentication to be performed
67         if username is None:
68             if self._LOGIN_REQUIRED:
69                 raise ExtractorError('No login info available, needed for using %s.' % self.IE_NAME, expected=True)
70             return True
71
72         login_page = self._download_webpage(
73             self._LOGIN_URL, None,
74             note='Downloading login page',
75             errnote='unable to fetch login page', fatal=False)
76         if login_page is False:
77             return
78
79         galx = self._search_regex(r'(?s)<input.+?name="GALX".+?value="(.+?)"',
80                                   login_page, 'Login GALX parameter')
81
82         # Log in
83         login_form_strs = {
84             'continue': 'https://www.youtube.com/signin?action_handle_signin=true&feature=sign_in_button&hl=en_US&nomobiletemp=1',
85             'Email': username,
86             'GALX': galx,
87             'Passwd': password,
88
89             'PersistentCookie': 'yes',
90             '_utf8': '霱',
91             'bgresponse': 'js_disabled',
92             'checkConnection': '',
93             'checkedDomains': 'youtube',
94             'dnConn': '',
95             'pstMsg': '0',
96             'rmShown': '1',
97             'secTok': '',
98             'signIn': 'Sign in',
99             'timeStmp': '',
100             'service': 'youtube',
101             'uilel': '3',
102             'hl': 'en_US',
103         }
104
105         # Convert to UTF-8 *before* urlencode because Python 2.x's urlencode
106         # chokes on unicode
107         login_form = dict((k.encode('utf-8'), v.encode('utf-8')) for k, v in login_form_strs.items())
108         login_data = compat_urllib_parse.urlencode(login_form).encode('ascii')
109
110         req = compat_urllib_request.Request(self._LOGIN_URL, login_data)
111         login_results = self._download_webpage(
112             req, None,
113             note='Logging in', errnote='unable to log in', fatal=False)
114         if login_results is False:
115             return False
116
117         if re.search(r'id="errormsg_0_Passwd"', login_results) is not None:
118             raise ExtractorError('Please use your account password and a two-factor code instead of an application-specific password.', expected=True)
119
120         # Two-Factor
121         # TODO add SMS and phone call support - these require making a request and then prompting the user
122
123         if re.search(r'(?i)<form[^>]* id="gaia_secondfactorform"', login_results) is not None:
124             tfa_code = self._get_tfa_info()
125
126             if tfa_code is None:
127                 self._downloader.report_warning('Two-factor authentication required. Provide it with --twofactor <code>')
128                 self._downloader.report_warning('(Note that only TOTP (Google Authenticator App) codes work at this time.)')
129                 return False
130
131             # Unlike the first login form, secTok and timeStmp are both required for the TFA form
132
133             match = re.search(r'id="secTok"\n\s+value=\'(.+)\'/>', login_results, re.M | re.U)
134             if match is None:
135                 self._downloader.report_warning('Failed to get secTok - did the page structure change?')
136             secTok = match.group(1)
137             match = re.search(r'id="timeStmp"\n\s+value=\'(.+)\'/>', login_results, re.M | re.U)
138             if match is None:
139                 self._downloader.report_warning('Failed to get timeStmp - did the page structure change?')
140             timeStmp = match.group(1)
141
142             tfa_form_strs = {
143                 'continue': 'https://www.youtube.com/signin?action_handle_signin=true&feature=sign_in_button&hl=en_US&nomobiletemp=1',
144                 'smsToken': '',
145                 'smsUserPin': tfa_code,
146                 'smsVerifyPin': 'Verify',
147
148                 'PersistentCookie': 'yes',
149                 'checkConnection': '',
150                 'checkedDomains': 'youtube',
151                 'pstMsg': '1',
152                 'secTok': secTok,
153                 'timeStmp': timeStmp,
154                 'service': 'youtube',
155                 'hl': 'en_US',
156             }
157             tfa_form = dict((k.encode('utf-8'), v.encode('utf-8')) for k, v in tfa_form_strs.items())
158             tfa_data = compat_urllib_parse.urlencode(tfa_form).encode('ascii')
159
160             tfa_req = compat_urllib_request.Request(self._TWOFACTOR_URL, tfa_data)
161             tfa_results = self._download_webpage(
162                 tfa_req, None,
163                 note='Submitting TFA code', errnote='unable to submit tfa', fatal=False)
164
165             if tfa_results is False:
166                 return False
167
168             if re.search(r'(?i)<form[^>]* id="gaia_secondfactorform"', tfa_results) is not None:
169                 self._downloader.report_warning('Two-factor code expired. Please try again, or use a one-use backup code instead.')
170                 return False
171             if re.search(r'(?i)<form[^>]* id="gaia_loginform"', tfa_results) is not None:
172                 self._downloader.report_warning('unable to log in - did the page structure change?')
173                 return False
174             if re.search(r'smsauth-interstitial-reviewsettings', tfa_results) is not None:
175                 self._downloader.report_warning('Your Google account has a security notice. Please log in on your web browser, resolve the notice, and try again.')
176                 return False
177
178         if re.search(r'(?i)<form[^>]* id="gaia_loginform"', login_results) is not None:
179             self._downloader.report_warning('unable to log in: bad username or password')
180             return False
181         return True
182
183     def _real_initialize(self):
184         if self._downloader is None:
185             return
186         self._set_language()
187         if not self._login():
188             return
189
190
191 class YoutubeIE(YoutubeBaseInfoExtractor):
192     IE_DESC = 'YouTube.com'
193     _VALID_URL = r"""(?x)^
194                      (
195                          (?:https?://|//)                                    # http(s):// or protocol-independent URL
196                          (?:(?:(?:(?:\w+\.)?[yY][oO][uU][tT][uU][bB][eE](?:-nocookie)?\.com/|
197                             (?:www\.)?deturl\.com/www\.youtube\.com/|
198                             (?:www\.)?pwnyoutube\.com/|
199                             (?:www\.)?yourepeat\.com/|
200                             tube\.majestyc\.net/|
201                             youtube\.googleapis\.com/)                        # the various hostnames, with wildcard subdomains
202                          (?:.*?\#/)?                                          # handle anchor (#/) redirect urls
203                          (?:                                                  # the various things that can precede the ID:
204                              (?:(?:v|embed|e)/(?!videoseries))                # v/ or embed/ or e/
205                              |(?:                                             # or the v= param in all its forms
206                                  (?:(?:watch|movie)(?:_popup)?(?:\.php)?/?)?  # preceding watch(_popup|.php) or nothing (like /?v=xxxx)
207                                  (?:\?|\#!?)                                  # the params delimiter ? or # or #!
208                                  (?:.*?&)?                                    # any other preceding param (like /?s=tuff&v=xxxx)
209                                  v=
210                              )
211                          ))
212                          |youtu\.be/                                          # just youtu.be/xxxx
213                          |(?:www\.)?cleanvideosearch\.com/media/action/yt/watch\?videoId=
214                          )
215                      )?                                                       # all until now is optional -> you can pass the naked ID
216                      ([0-9A-Za-z_-]{11})                                      # here is it! the YouTube video ID
217                      (?!.*?&list=)                                            # combined list/video URLs are handled by the playlist IE
218                      (?(1).+)?                                                # if we found the ID, everything can follow
219                      $"""
220     _NEXT_URL_RE = r'[\?&]next_url=([^&]+)'
221     _formats = {
222         '5': {'ext': 'flv', 'width': 400, 'height': 240},
223         '6': {'ext': 'flv', 'width': 450, 'height': 270},
224         '13': {'ext': '3gp'},
225         '17': {'ext': '3gp', 'width': 176, 'height': 144},
226         '18': {'ext': 'mp4', 'width': 640, 'height': 360},
227         '22': {'ext': 'mp4', 'width': 1280, 'height': 720},
228         '34': {'ext': 'flv', 'width': 640, 'height': 360},
229         '35': {'ext': 'flv', 'width': 854, 'height': 480},
230         '36': {'ext': '3gp', 'width': 320, 'height': 240},
231         '37': {'ext': 'mp4', 'width': 1920, 'height': 1080},
232         '38': {'ext': 'mp4', 'width': 4096, 'height': 3072},
233         '43': {'ext': 'webm', 'width': 640, 'height': 360},
234         '44': {'ext': 'webm', 'width': 854, 'height': 480},
235         '45': {'ext': 'webm', 'width': 1280, 'height': 720},
236         '46': {'ext': 'webm', 'width': 1920, 'height': 1080},
237
238
239         # 3d videos
240         '82': {'ext': 'mp4', 'height': 360, 'format_note': '3D', 'preference': -20},
241         '83': {'ext': 'mp4', 'height': 480, 'format_note': '3D', 'preference': -20},
242         '84': {'ext': 'mp4', 'height': 720, 'format_note': '3D', 'preference': -20},
243         '85': {'ext': 'mp4', 'height': 1080, 'format_note': '3D', 'preference': -20},
244         '100': {'ext': 'webm', 'height': 360, 'format_note': '3D', 'preference': -20},
245         '101': {'ext': 'webm', 'height': 480, 'format_note': '3D', 'preference': -20},
246         '102': {'ext': 'webm', 'height': 720, 'format_note': '3D', 'preference': -20},
247
248         # Apple HTTP Live Streaming
249         '92': {'ext': 'mp4', 'height': 240, 'format_note': 'HLS', 'preference': -10},
250         '93': {'ext': 'mp4', 'height': 360, 'format_note': 'HLS', 'preference': -10},
251         '94': {'ext': 'mp4', 'height': 480, 'format_note': 'HLS', 'preference': -10},
252         '95': {'ext': 'mp4', 'height': 720, 'format_note': 'HLS', 'preference': -10},
253         '96': {'ext': 'mp4', 'height': 1080, 'format_note': 'HLS', 'preference': -10},
254         '132': {'ext': 'mp4', 'height': 240, 'format_note': 'HLS', 'preference': -10},
255         '151': {'ext': 'mp4', 'height': 72, 'format_note': 'HLS', 'preference': -10},
256
257         # DASH mp4 video
258         '133': {'ext': 'mp4', 'height': 240, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
259         '134': {'ext': 'mp4', 'height': 360, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
260         '135': {'ext': 'mp4', 'height': 480, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
261         '136': {'ext': 'mp4', 'height': 720, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
262         '137': {'ext': 'mp4', 'height': 1080, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
263         '138': {'ext': 'mp4', 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},  # Height can vary (https://github.com/rg3/youtube-dl/issues/4559)
264         '160': {'ext': 'mp4', 'height': 144, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
265         '264': {'ext': 'mp4', 'height': 1440, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
266         '298': {'ext': 'mp4', 'height': 720, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'fps': 60, 'vcodec': 'h264'},
267         '299': {'ext': 'mp4', 'height': 1080, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'fps': 60, 'vcodec': 'h264'},
268         '266': {'ext': 'mp4', 'height': 2160, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'vcodec': 'h264'},
269
270         # Dash mp4 audio
271         '139': {'ext': 'm4a', 'format_note': 'DASH audio', 'acodec': 'aac', 'vcodec': 'none', 'abr': 48, 'preference': -50, 'container': 'm4a_dash'},
272         '140': {'ext': 'm4a', 'format_note': 'DASH audio', 'acodec': 'aac', 'vcodec': 'none', 'abr': 128, 'preference': -50, 'container': 'm4a_dash'},
273         '141': {'ext': 'm4a', 'format_note': 'DASH audio', 'acodec': 'aac', 'vcodec': 'none', 'abr': 256, 'preference': -50, 'container': 'm4a_dash'},
274
275         # Dash webm
276         '167': {'ext': 'webm', 'height': 360, 'width': 640, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
277         '168': {'ext': 'webm', 'height': 480, 'width': 854, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
278         '169': {'ext': 'webm', 'height': 720, 'width': 1280, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
279         '170': {'ext': 'webm', 'height': 1080, 'width': 1920, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
280         '218': {'ext': 'webm', 'height': 480, 'width': 854, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
281         '219': {'ext': 'webm', 'height': 480, 'width': 854, 'format_note': 'DASH video', 'acodec': 'none', 'container': 'webm', 'vcodec': 'VP8', 'preference': -40},
282         '278': {'ext': 'webm', 'height': 144, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'container': 'webm', 'vcodec': 'VP9'},
283         '242': {'ext': 'webm', 'height': 240, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
284         '243': {'ext': 'webm', 'height': 360, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
285         '244': {'ext': 'webm', 'height': 480, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
286         '245': {'ext': 'webm', 'height': 480, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
287         '246': {'ext': 'webm', 'height': 480, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
288         '247': {'ext': 'webm', 'height': 720, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
289         '248': {'ext': 'webm', 'height': 1080, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
290         '271': {'ext': 'webm', 'height': 1440, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
291         '272': {'ext': 'webm', 'height': 2160, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
292         '302': {'ext': 'webm', 'height': 720, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'fps': 60, 'vcodec': 'VP9'},
293         '303': {'ext': 'webm', 'height': 1080, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'fps': 60, 'vcodec': 'VP9'},
294         '308': {'ext': 'webm', 'height': 1440, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'fps': 60, 'vcodec': 'VP9'},
295         '313': {'ext': 'webm', 'height': 2160, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'vcodec': 'VP9'},
296         '315': {'ext': 'webm', 'height': 2160, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'fps': 60, 'vcodec': 'VP9'},
297
298         # Dash webm audio
299         '171': {'ext': 'webm', 'vcodec': 'none', 'format_note': 'DASH audio', 'abr': 128, 'preference': -50},
300         '172': {'ext': 'webm', 'vcodec': 'none', 'format_note': 'DASH audio', 'abr': 256, 'preference': -50},
301
302         # Dash webm audio with opus inside
303         '249': {'ext': 'webm', 'vcodec': 'none', 'format_note': 'DASH audio', 'acodec': 'opus', 'abr': 50, 'preference': -50},
304         '250': {'ext': 'webm', 'vcodec': 'none', 'format_note': 'DASH audio', 'acodec': 'opus', 'abr': 70, 'preference': -50},
305         '251': {'ext': 'webm', 'vcodec': 'none', 'format_note': 'DASH audio', 'acodec': 'opus', 'abr': 160, 'preference': -50},
306
307         # RTMP (unnamed)
308         '_rtmp': {'protocol': 'rtmp'},
309     }
310
311     IE_NAME = 'youtube'
312     _TESTS = [
313         {
314             'url': 'http://www.youtube.com/watch?v=BaW_jenozKc',
315             'info_dict': {
316                 'id': 'BaW_jenozKc',
317                 'ext': 'mp4',
318                 'title': 'youtube-dl test video "\'/\\ä↭𝕐',
319                 'uploader': 'Philipp Hagemeister',
320                 'uploader_id': 'phihag',
321                 'upload_date': '20121002',
322                 'description': '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 .',
323                 'categories': ['Science & Technology'],
324                 'like_count': int,
325                 'dislike_count': int,
326             }
327         },
328         {
329             'url': 'http://www.youtube.com/watch?v=UxxajLWwzqY',
330             'note': 'Test generic use_cipher_signature video (#897)',
331             'info_dict': {
332                 'id': 'UxxajLWwzqY',
333                 'ext': 'mp4',
334                 'upload_date': '20120506',
335                 'title': 'Icona Pop - I Love It (feat. Charli XCX) [OFFICIAL VIDEO]',
336                 'description': 'md5:fea86fda2d5a5784273df5c7cc994d9f',
337                 'uploader': 'Icona Pop',
338                 'uploader_id': 'IconaPop',
339             }
340         },
341         {
342             'url': 'https://www.youtube.com/watch?v=07FYdnEawAQ',
343             'note': 'Test VEVO video with age protection (#956)',
344             'info_dict': {
345                 'id': '07FYdnEawAQ',
346                 'ext': 'mp4',
347                 'upload_date': '20130703',
348                 'title': 'Justin Timberlake - Tunnel Vision (Explicit)',
349                 'description': 'md5:64249768eec3bc4276236606ea996373',
350                 'uploader': 'justintimberlakeVEVO',
351                 'uploader_id': 'justintimberlakeVEVO',
352             }
353         },
354         {
355             'url': '//www.YouTube.com/watch?v=yZIXLfi8CZQ',
356             'note': 'Embed-only video (#1746)',
357             'info_dict': {
358                 'id': 'yZIXLfi8CZQ',
359                 'ext': 'mp4',
360                 'upload_date': '20120608',
361                 'title': 'Principal Sexually Assaults A Teacher - Episode 117 - 8th June 2012',
362                 'description': 'md5:09b78bd971f1e3e289601dfba15ca4f7',
363                 'uploader': 'SET India',
364                 'uploader_id': 'setindia'
365             }
366         },
367         {
368             'url': 'http://www.youtube.com/watch?v=a9LDPn-MO4I',
369             'note': '256k DASH audio (format 141) via DASH manifest',
370             'info_dict': {
371                 'id': 'a9LDPn-MO4I',
372                 'ext': 'm4a',
373                 'upload_date': '20121002',
374                 'uploader_id': '8KVIDEO',
375                 'description': '',
376                 'uploader': '8KVIDEO',
377                 'title': 'UHDTV TEST 8K VIDEO.mp4'
378             },
379             'params': {
380                 'youtube_include_dash_manifest': True,
381                 'format': '141',
382             },
383         },
384         # DASH manifest with encrypted signature
385         {
386             'url': 'https://www.youtube.com/watch?v=IB3lcPjvWLA',
387             'info_dict': {
388                 'id': 'IB3lcPjvWLA',
389                 'ext': 'm4a',
390                 'title': 'Afrojack, Spree Wilson - The Spark ft. Spree Wilson',
391                 'description': 'md5:12e7067fa6735a77bdcbb58cb1187d2d',
392                 'uploader': 'AfrojackVEVO',
393                 'uploader_id': 'AfrojackVEVO',
394                 'upload_date': '20131011',
395             },
396             'params': {
397                 'youtube_include_dash_manifest': True,
398                 'format': '141',
399             },
400         },
401         # JS player signature function name containing $
402         {
403             'url': 'https://www.youtube.com/watch?v=nfWlot6h_JM',
404             'info_dict': {
405                 'id': 'nfWlot6h_JM',
406                 'ext': 'm4a',
407                 'title': 'Taylor Swift - Shake It Off',
408                 'description': 'md5:2acfda1b285bdd478ccec22f9918199d',
409                 'uploader': 'TaylorSwiftVEVO',
410                 'uploader_id': 'TaylorSwiftVEVO',
411                 'upload_date': '20140818',
412             },
413             'params': {
414                 'youtube_include_dash_manifest': True,
415                 'format': '141',
416             },
417         },
418         # Controversy video
419         {
420             'url': 'https://www.youtube.com/watch?v=T4XJQO3qol8',
421             'info_dict': {
422                 'id': 'T4XJQO3qol8',
423                 'ext': 'mp4',
424                 'upload_date': '20100909',
425                 'uploader': 'The Amazing Atheist',
426                 'uploader_id': 'TheAmazingAtheist',
427                 'title': 'Burning Everyone\'s Koran',
428                 'description': 'SUBSCRIBE: http://www.youtube.com/saturninefilms\n\nEven Obama has taken a stand against freedom on this issue: http://www.huffingtonpost.com/2010/09/09/obama-gma-interview-quran_n_710282.html',
429             }
430         },
431         # Normal age-gate video (No vevo, embed allowed)
432         {
433             'url': 'http://youtube.com/watch?v=HtVdAasjOgU',
434             'info_dict': {
435                 'id': 'HtVdAasjOgU',
436                 'ext': 'mp4',
437                 'title': 'The Witcher 3: Wild Hunt - The Sword Of Destiny Trailer',
438                 'description': 're:(?s).{100,}About the Game\n.*?The Witcher 3: Wild Hunt.{100,}',
439                 'uploader': 'The Witcher',
440                 'uploader_id': 'WitcherGame',
441                 'upload_date': '20140605',
442             },
443         },
444         # Age-gate video with encrypted signature
445         {
446             'url': 'http://www.youtube.com/watch?v=6kLq3WMV1nU',
447             'info_dict': {
448                 'id': '6kLq3WMV1nU',
449                 'ext': 'mp4',
450                 'title': 'Dedication To My Ex (Miss That) (Lyric Video)',
451                 'description': 'md5:33765bb339e1b47e7e72b5490139bb41',
452                 'uploader': 'LloydVEVO',
453                 'uploader_id': 'LloydVEVO',
454                 'upload_date': '20110629',
455             },
456         },
457         # video_info is None (https://github.com/rg3/youtube-dl/issues/4421)
458         {
459             'url': '__2ABJjxzNo',
460             'info_dict': {
461                 'id': '__2ABJjxzNo',
462                 'ext': 'mp4',
463                 'upload_date': '20100430',
464                 'uploader_id': 'deadmau5',
465                 'description': 'md5:12c56784b8032162bb936a5f76d55360',
466                 'uploader': 'deadmau5',
467                 'title': 'Deadmau5 - Some Chords (HD)',
468             },
469             'expected_warnings': [
470                 'DASH manifest missing',
471             ]
472         },
473         # Olympics (https://github.com/rg3/youtube-dl/issues/4431)
474         {
475             'url': 'lqQg6PlCWgI',
476             'info_dict': {
477                 'id': 'lqQg6PlCWgI',
478                 'ext': 'mp4',
479                 'upload_date': '20120731',
480                 'uploader_id': 'olympic',
481                 'description': 'HO09  - Women -  GER-AUS - Hockey - 31 July 2012 - London 2012 Olympic Games',
482                 'uploader': 'Olympics',
483                 'title': 'Hockey - Women -  GER-AUS - London 2012 Olympic Games',
484             },
485             'params': {
486                 'skip_download': 'requires avconv',
487             }
488         },
489         # Non-square pixels
490         {
491             'url': 'https://www.youtube.com/watch?v=_b-2C3KPAM0',
492             'info_dict': {
493                 'id': '_b-2C3KPAM0',
494                 'ext': 'mp4',
495                 'stretched_ratio': 16 / 9.,
496                 'upload_date': '20110310',
497                 'uploader_id': 'AllenMeow',
498                 'description': 'made by Wacom from Korea | 字幕&加油添醋 by TY\'s Allen | 感謝heylisa00cavey1001同學熱情提供梗及翻譯',
499                 'uploader': '孫艾倫',
500                 'title': '[A-made] 變態妍字幕版 太妍 我就是這樣的人',
501             },
502         },
503         # url_encoded_fmt_stream_map is empty string
504         {
505             'url': 'qEJwOuvDf7I',
506             'info_dict': {
507                 'id': 'qEJwOuvDf7I',
508                 'ext': 'mp4',
509                 'title': 'Обсуждение судебной практики по выборам 14 сентября 2014 года в Санкт-Петербурге',
510                 'description': '',
511                 'upload_date': '20150404',
512                 'uploader_id': 'spbelect',
513                 'uploader': 'Наблюдатели Петербурга',
514             },
515             'params': {
516                 'skip_download': 'requires avconv',
517             }
518         },
519         # DASH manifest with segment_list
520         {
521             'url': 'https://www.youtube.com/embed/CsmdDsKjzN8',
522             'md5': '8ce563a1d667b599d21064e982ab9e31',
523             'info_dict': {
524                 'id': 'CsmdDsKjzN8',
525                 'ext': 'mp4',
526                 'upload_date': '20150510',
527                 'uploader': 'Airtek',
528                 'description': 'Retransmisión en directo de la XVIII media maratón de Zaragoza.',
529                 'uploader_id': 'UCzTzUmjXxxacNnL8I3m4LnQ',
530                 'title': 'Retransmisión XVIII Media maratón Zaragoza 2015',
531             },
532             'params': {
533                 'youtube_include_dash_manifest': True,
534                 'format': '135',  # bestvideo
535             }
536         }
537     ]
538
539     def __init__(self, *args, **kwargs):
540         super(YoutubeIE, self).__init__(*args, **kwargs)
541         self._player_cache = {}
542
543     def report_video_info_webpage_download(self, video_id):
544         """Report attempt to download video info webpage."""
545         self.to_screen('%s: Downloading video info webpage' % video_id)
546
547     def report_information_extraction(self, video_id):
548         """Report attempt to extract video information."""
549         self.to_screen('%s: Extracting video information' % video_id)
550
551     def report_unavailable_format(self, video_id, format):
552         """Report extracted video URL."""
553         self.to_screen('%s: Format %s not available' % (video_id, format))
554
555     def report_rtmp_download(self):
556         """Indicate the download will use the RTMP protocol."""
557         self.to_screen('RTMP download detected')
558
559     def _signature_cache_id(self, example_sig):
560         """ Return a string representation of a signature """
561         return '.'.join(compat_str(len(part)) for part in example_sig.split('.'))
562
563     def _extract_signature_function(self, video_id, player_url, example_sig):
564         id_m = re.match(
565             r'.*?-(?P<id>[a-zA-Z0-9_-]+)(?:/watch_as3|/html5player)?\.(?P<ext>[a-z]+)$',
566             player_url)
567         if not id_m:
568             raise ExtractorError('Cannot identify player %r' % player_url)
569         player_type = id_m.group('ext')
570         player_id = id_m.group('id')
571
572         # Read from filesystem cache
573         func_id = '%s_%s_%s' % (
574             player_type, player_id, self._signature_cache_id(example_sig))
575         assert os.path.basename(func_id) == func_id
576
577         cache_spec = self._downloader.cache.load('youtube-sigfuncs', func_id)
578         if cache_spec is not None:
579             return lambda s: ''.join(s[i] for i in cache_spec)
580
581         download_note = (
582             'Downloading player %s' % player_url
583             if self._downloader.params.get('verbose') else
584             'Downloading %s player %s' % (player_type, player_id)
585         )
586         if player_type == 'js':
587             code = self._download_webpage(
588                 player_url, video_id,
589                 note=download_note,
590                 errnote='Download of %s failed' % player_url)
591             res = self._parse_sig_js(code)
592         elif player_type == 'swf':
593             urlh = self._request_webpage(
594                 player_url, video_id,
595                 note=download_note,
596                 errnote='Download of %s failed' % player_url)
597             code = urlh.read()
598             res = self._parse_sig_swf(code)
599         else:
600             assert False, 'Invalid player type %r' % player_type
601
602         test_string = ''.join(map(compat_chr, range(len(example_sig))))
603         cache_res = res(test_string)
604         cache_spec = [ord(c) for c in cache_res]
605
606         self._downloader.cache.store('youtube-sigfuncs', func_id, cache_spec)
607         return res
608
609     def _print_sig_code(self, func, example_sig):
610         def gen_sig_code(idxs):
611             def _genslice(start, end, step):
612                 starts = '' if start == 0 else str(start)
613                 ends = (':%d' % (end + step)) if end + step >= 0 else ':'
614                 steps = '' if step == 1 else (':%d' % step)
615                 return 's[%s%s%s]' % (starts, ends, steps)
616
617             step = None
618             # Quelch pyflakes warnings - start will be set when step is set
619             start = '(Never used)'
620             for i, prev in zip(idxs[1:], idxs[:-1]):
621                 if step is not None:
622                     if i - prev == step:
623                         continue
624                     yield _genslice(start, prev, step)
625                     step = None
626                     continue
627                 if i - prev in [-1, 1]:
628                     step = i - prev
629                     start = prev
630                     continue
631                 else:
632                     yield 's[%d]' % prev
633             if step is None:
634                 yield 's[%d]' % i
635             else:
636                 yield _genslice(start, i, step)
637
638         test_string = ''.join(map(compat_chr, range(len(example_sig))))
639         cache_res = func(test_string)
640         cache_spec = [ord(c) for c in cache_res]
641         expr_code = ' + '.join(gen_sig_code(cache_spec))
642         signature_id_tuple = '(%s)' % (
643             ', '.join(compat_str(len(p)) for p in example_sig.split('.')))
644         code = ('if tuple(len(p) for p in s.split(\'.\')) == %s:\n'
645                 '    return %s\n') % (signature_id_tuple, expr_code)
646         self.to_screen('Extracted signature function:\n' + code)
647
648     def _parse_sig_js(self, jscode):
649         funcname = self._search_regex(
650             r'\.sig\|\|([a-zA-Z0-9$]+)\(', jscode,
651             'Initial JS player signature function name')
652
653         jsi = JSInterpreter(jscode)
654         initial_function = jsi.extract_function(funcname)
655         return lambda s: initial_function([s])
656
657     def _parse_sig_swf(self, file_contents):
658         swfi = SWFInterpreter(file_contents)
659         TARGET_CLASSNAME = 'SignatureDecipher'
660         searched_class = swfi.extract_class(TARGET_CLASSNAME)
661         initial_function = swfi.extract_function(searched_class, 'decipher')
662         return lambda s: initial_function([s])
663
664     def _decrypt_signature(self, s, video_id, player_url, age_gate=False):
665         """Turn the encrypted s field into a working signature"""
666
667         if player_url is None:
668             raise ExtractorError('Cannot decrypt signature without player_url')
669
670         if player_url.startswith('//'):
671             player_url = 'https:' + player_url
672         try:
673             player_id = (player_url, self._signature_cache_id(s))
674             if player_id not in self._player_cache:
675                 func = self._extract_signature_function(
676                     video_id, player_url, s
677                 )
678                 self._player_cache[player_id] = func
679             func = self._player_cache[player_id]
680             if self._downloader.params.get('youtube_print_sig_code'):
681                 self._print_sig_code(func, s)
682             return func(s)
683         except Exception as e:
684             tb = traceback.format_exc()
685             raise ExtractorError(
686                 'Signature extraction failed: ' + tb, cause=e)
687
688     def _get_subtitles(self, video_id, webpage):
689         try:
690             subs_doc = self._download_xml(
691                 'https://video.google.com/timedtext?hl=en&type=list&v=%s' % video_id,
692                 video_id, note=False)
693         except ExtractorError as err:
694             self._downloader.report_warning('unable to download video subtitles: %s' % compat_str(err))
695             return {}
696
697         sub_lang_list = {}
698         for track in subs_doc.findall('track'):
699             lang = track.attrib['lang_code']
700             if lang in sub_lang_list:
701                 continue
702             sub_formats = []
703             for ext in ['sbv', 'vtt', 'srt']:
704                 params = compat_urllib_parse.urlencode({
705                     'lang': lang,
706                     'v': video_id,
707                     'fmt': ext,
708                     'name': track.attrib['name'].encode('utf-8'),
709                 })
710                 sub_formats.append({
711                     'url': 'https://www.youtube.com/api/timedtext?' + params,
712                     'ext': ext,
713                 })
714             sub_lang_list[lang] = sub_formats
715         if not sub_lang_list:
716             self._downloader.report_warning('video doesn\'t have subtitles')
717             return {}
718         return sub_lang_list
719
720     def _get_automatic_captions(self, video_id, webpage):
721         """We need the webpage for getting the captions url, pass it as an
722            argument to speed up the process."""
723         self.to_screen('%s: Looking for automatic captions' % video_id)
724         mobj = re.search(r';ytplayer.config = ({.*?});', webpage)
725         err_msg = 'Couldn\'t find automatic captions for %s' % video_id
726         if mobj is None:
727             self._downloader.report_warning(err_msg)
728             return {}
729         player_config = json.loads(mobj.group(1))
730         try:
731             args = player_config['args']
732             caption_url = args['ttsurl']
733             timestamp = args['timestamp']
734             # We get the available subtitles
735             list_params = compat_urllib_parse.urlencode({
736                 'type': 'list',
737                 'tlangs': 1,
738                 'asrs': 1,
739             })
740             list_url = caption_url + '&' + list_params
741             caption_list = self._download_xml(list_url, video_id)
742             original_lang_node = caption_list.find('track')
743             if original_lang_node is None:
744                 self._downloader.report_warning('Video doesn\'t have automatic captions')
745                 return {}
746             original_lang = original_lang_node.attrib['lang_code']
747             caption_kind = original_lang_node.attrib.get('kind', '')
748
749             sub_lang_list = {}
750             for lang_node in caption_list.findall('target'):
751                 sub_lang = lang_node.attrib['lang_code']
752                 sub_formats = []
753                 for ext in ['sbv', 'vtt', 'srt']:
754                     params = compat_urllib_parse.urlencode({
755                         'lang': original_lang,
756                         'tlang': sub_lang,
757                         'fmt': ext,
758                         'ts': timestamp,
759                         'kind': caption_kind,
760                     })
761                     sub_formats.append({
762                         'url': caption_url + '&' + params,
763                         'ext': ext,
764                     })
765                 sub_lang_list[sub_lang] = sub_formats
766             return sub_lang_list
767         # An extractor error can be raise by the download process if there are
768         # no automatic captions but there are subtitles
769         except (KeyError, ExtractorError):
770             self._downloader.report_warning(err_msg)
771             return {}
772
773     @classmethod
774     def extract_id(cls, url):
775         mobj = re.match(cls._VALID_URL, url, re.VERBOSE)
776         if mobj is None:
777             raise ExtractorError('Invalid URL: %s' % url)
778         video_id = mobj.group(2)
779         return video_id
780
781     def _extract_from_m3u8(self, manifest_url, video_id):
782         url_map = {}
783
784         def _get_urls(_manifest):
785             lines = _manifest.split('\n')
786             urls = filter(lambda l: l and not l.startswith('#'),
787                           lines)
788             return urls
789         manifest = self._download_webpage(manifest_url, video_id, 'Downloading formats manifest')
790         formats_urls = _get_urls(manifest)
791         for format_url in formats_urls:
792             itag = self._search_regex(r'itag/(\d+?)/', format_url, 'itag')
793             url_map[itag] = format_url
794         return url_map
795
796     def _extract_annotations(self, video_id):
797         url = 'https://www.youtube.com/annotations_invideo?features=1&legacy=1&video_id=%s' % video_id
798         return self._download_webpage(url, video_id, note='Searching for annotations.', errnote='Unable to download video annotations.')
799
800     def _parse_dash_manifest(
801             self, video_id, dash_manifest_url, player_url, age_gate):
802         def decrypt_sig(mobj):
803             s = mobj.group(1)
804             dec_s = self._decrypt_signature(s, video_id, player_url, age_gate)
805             return '/signature/%s' % dec_s
806         dash_manifest_url = re.sub(r'/s/([\w\.]+)', decrypt_sig, dash_manifest_url)
807         dash_doc = self._download_xml(
808             dash_manifest_url, video_id,
809             note='Downloading DASH manifest',
810             errnote='Could not download DASH manifest')
811
812         formats = []
813         for a in dash_doc.findall('.//{urn:mpeg:DASH:schema:MPD:2011}AdaptationSet'):
814             mime_type = a.attrib.get('mimeType')
815             for r in a.findall('{urn:mpeg:DASH:schema:MPD:2011}Representation'):
816                 url_el = r.find('{urn:mpeg:DASH:schema:MPD:2011}BaseURL')
817                 if url_el is None:
818                     continue
819                 if mime_type == 'text/vtt':
820                     # TODO implement WebVTT downloading
821                     pass
822                 elif mime_type.startswith('audio/') or mime_type.startswith('video/'):
823                     segment_list = r.find('{urn:mpeg:DASH:schema:MPD:2011}SegmentList')
824                     format_id = r.attrib['id']
825                     video_url = url_el.text
826                     filesize = int_or_none(url_el.attrib.get('{http://youtube.com/yt/2012/10/10}contentLength'))
827                     f = {
828                         'format_id': format_id,
829                         'url': video_url,
830                         'width': int_or_none(r.attrib.get('width')),
831                         'height': int_or_none(r.attrib.get('height')),
832                         'tbr': int_or_none(r.attrib.get('bandwidth'), 1000),
833                         'asr': int_or_none(r.attrib.get('audioSamplingRate')),
834                         'filesize': filesize,
835                         'fps': int_or_none(r.attrib.get('frameRate')),
836                     }
837                     if segment_list is not None:
838                         f.update({
839                             'initialization_url': segment_list.find('{urn:mpeg:DASH:schema:MPD:2011}Initialization').attrib['sourceURL'],
840                             'segment_urls': [segment.attrib.get('media') for segment in segment_list.findall('{urn:mpeg:DASH:schema:MPD:2011}SegmentURL')],
841                             'protocol': 'http_dash_segments',
842                         })
843                     try:
844                         existing_format = next(
845                             fo for fo in formats
846                             if fo['format_id'] == format_id)
847                     except StopIteration:
848                         full_info = self._formats.get(format_id, {}).copy()
849                         full_info.update(f)
850                         formats.append(full_info)
851                     else:
852                         existing_format.update(f)
853                 else:
854                     self.report_warning('Unknown MIME type %s in DASH manifest' % mime_type)
855         return formats
856
857     def _real_extract(self, url):
858         proto = (
859             'http' if self._downloader.params.get('prefer_insecure', False)
860             else 'https')
861
862         # Extract original video URL from URL with redirection, like age verification, using next_url parameter
863         mobj = re.search(self._NEXT_URL_RE, url)
864         if mobj:
865             url = proto + '://www.youtube.com/' + compat_urllib_parse.unquote(mobj.group(1)).lstrip('/')
866         video_id = self.extract_id(url)
867
868         # Get video webpage
869         url = proto + '://www.youtube.com/watch?v=%s&gl=US&hl=en&has_verified=1&bpctr=9999999999' % video_id
870         video_webpage = self._download_webpage(url, video_id)
871
872         # Attempt to extract SWF player URL
873         mobj = re.search(r'swfConfig.*?"(https?:\\/\\/.*?watch.*?-.*?\.swf)"', video_webpage)
874         if mobj is not None:
875             player_url = re.sub(r'\\(.)', r'\1', mobj.group(1))
876         else:
877             player_url = None
878
879         # Get video info
880         embed_webpage = None
881         if re.search(r'player-age-gate-content">', video_webpage) is not None:
882             age_gate = True
883             # We simulate the access to the video from www.youtube.com/v/{video_id}
884             # this can be viewed without login into Youtube
885             url = proto + '://www.youtube.com/embed/%s' % video_id
886             embed_webpage = self._download_webpage(url, video_id, 'Downloading embed webpage')
887             data = compat_urllib_parse.urlencode({
888                 'video_id': video_id,
889                 'eurl': 'https://youtube.googleapis.com/v/' + video_id,
890                 'sts': self._search_regex(
891                     r'"sts"\s*:\s*(\d+)', embed_webpage, 'sts', default=''),
892             })
893             video_info_url = proto + '://www.youtube.com/get_video_info?' + data
894             video_info_webpage = self._download_webpage(
895                 video_info_url, video_id,
896                 note='Refetching age-gated info webpage',
897                 errnote='unable to download video info webpage')
898             video_info = compat_parse_qs(video_info_webpage)
899         else:
900             age_gate = False
901             try:
902                 # Try looking directly into the video webpage
903                 mobj = re.search(r';ytplayer\.config\s*=\s*({.*?});', video_webpage)
904                 if not mobj:
905                     raise ValueError('Could not find ytplayer.config')  # caught below
906                 json_code = uppercase_escape(mobj.group(1))
907                 ytplayer_config = json.loads(json_code)
908                 args = ytplayer_config['args']
909                 # Convert to the same format returned by compat_parse_qs
910                 video_info = dict((k, [v]) for k, v in args.items())
911                 if not args.get('url_encoded_fmt_stream_map'):
912                     raise ValueError('No stream_map present')  # caught below
913             except ValueError:
914                 # We fallback to the get_video_info pages (used by the embed page)
915                 self.report_video_info_webpage_download(video_id)
916                 for el_type in ['&el=embedded', '&el=detailpage', '&el=vevo', '']:
917                     video_info_url = (
918                         '%s://www.youtube.com/get_video_info?&video_id=%s%s&ps=default&eurl=&gl=US&hl=en'
919                         % (proto, video_id, el_type))
920                     video_info_webpage = self._download_webpage(
921                         video_info_url,
922                         video_id, note=False,
923                         errnote='unable to download video info webpage')
924                     video_info = compat_parse_qs(video_info_webpage)
925                     if 'token' in video_info:
926                         break
927         if 'token' not in video_info:
928             if 'reason' in video_info:
929                 raise ExtractorError(
930                     'YouTube said: %s' % video_info['reason'][0],
931                     expected=True, video_id=video_id)
932             else:
933                 raise ExtractorError(
934                     '"token" parameter not in video info for unknown reason',
935                     video_id=video_id)
936
937         if 'view_count' in video_info:
938             view_count = int(video_info['view_count'][0])
939         else:
940             view_count = None
941
942         # Check for "rental" videos
943         if 'ypc_video_rental_bar_text' in video_info and 'author' not in video_info:
944             raise ExtractorError('"rental" videos not supported')
945
946         # Start extracting information
947         self.report_information_extraction(video_id)
948
949         # uploader
950         if 'author' not in video_info:
951             raise ExtractorError('Unable to extract uploader name')
952         video_uploader = compat_urllib_parse.unquote_plus(video_info['author'][0])
953
954         # uploader_id
955         video_uploader_id = None
956         mobj = re.search(r'<link itemprop="url" href="http://www.youtube.com/(?:user|channel)/([^"]+)">', video_webpage)
957         if mobj is not None:
958             video_uploader_id = mobj.group(1)
959         else:
960             self._downloader.report_warning('unable to extract uploader nickname')
961
962         # title
963         if 'title' in video_info:
964             video_title = video_info['title'][0]
965         else:
966             self._downloader.report_warning('Unable to extract video title')
967             video_title = '_'
968
969         # thumbnail image
970         # We try first to get a high quality image:
971         m_thumb = re.search(r'<span itemprop="thumbnail".*?href="(.*?)">',
972                             video_webpage, re.DOTALL)
973         if m_thumb is not None:
974             video_thumbnail = m_thumb.group(1)
975         elif 'thumbnail_url' not in video_info:
976             self._downloader.report_warning('unable to extract video thumbnail')
977             video_thumbnail = None
978         else:   # don't panic if we can't find it
979             video_thumbnail = compat_urllib_parse.unquote_plus(video_info['thumbnail_url'][0])
980
981         # upload date
982         upload_date = None
983         mobj = re.search(r'(?s)id="eow-date.*?>(.*?)</span>', video_webpage)
984         if mobj is None:
985             mobj = re.search(
986                 r'(?s)id="watch-uploader-info".*?>.*?(?:Published|Uploaded|Streamed live) on (.*?)</strong>',
987                 video_webpage)
988         if mobj is not None:
989             upload_date = ' '.join(re.sub(r'[/,-]', r' ', mobj.group(1)).split())
990             upload_date = unified_strdate(upload_date)
991
992         m_cat_container = self._search_regex(
993             r'(?s)<h4[^>]*>\s*Category\s*</h4>\s*<ul[^>]*>(.*?)</ul>',
994             video_webpage, 'categories', default=None)
995         if m_cat_container:
996             category = self._html_search_regex(
997                 r'(?s)<a[^<]+>(.*?)</a>', m_cat_container, 'category',
998                 default=None)
999             video_categories = None if category is None else [category]
1000         else:
1001             video_categories = None
1002
1003         # description
1004         video_description = get_element_by_id("eow-description", video_webpage)
1005         if video_description:
1006             video_description = re.sub(r'''(?x)
1007                 <a\s+
1008                     (?:[a-zA-Z-]+="[^"]+"\s+)*?
1009                     title="([^"]+)"\s+
1010                     (?:[a-zA-Z-]+="[^"]+"\s+)*?
1011                     class="yt-uix-redirect-link"\s*>
1012                 [^<]+
1013                 </a>
1014             ''', r'\1', video_description)
1015             video_description = clean_html(video_description)
1016         else:
1017             fd_mobj = re.search(r'<meta name="description" content="([^"]+)"', video_webpage)
1018             if fd_mobj:
1019                 video_description = unescapeHTML(fd_mobj.group(1))
1020             else:
1021                 video_description = ''
1022
1023         def _extract_count(count_name):
1024             count = self._search_regex(
1025                 r'id="watch-%s"[^>]*>.*?([\d,]+)\s*</span>' % re.escape(count_name),
1026                 video_webpage, count_name, default=None)
1027             if count is not None:
1028                 return int(count.replace(',', ''))
1029             return None
1030         like_count = _extract_count('like')
1031         dislike_count = _extract_count('dislike')
1032
1033         # subtitles
1034         video_subtitles = self.extract_subtitles(video_id, video_webpage)
1035         automatic_captions = self.extract_automatic_captions(video_id, video_webpage)
1036
1037         if 'length_seconds' not in video_info:
1038             self._downloader.report_warning('unable to extract video duration')
1039             video_duration = None
1040         else:
1041             video_duration = int(compat_urllib_parse.unquote_plus(video_info['length_seconds'][0]))
1042
1043         # annotations
1044         video_annotations = None
1045         if self._downloader.params.get('writeannotations', False):
1046             video_annotations = self._extract_annotations(video_id)
1047
1048         def _map_to_format_list(urlmap):
1049             formats = []
1050             for itag, video_real_url in urlmap.items():
1051                 dct = {
1052                     'format_id': itag,
1053                     'url': video_real_url,
1054                     'player_url': player_url,
1055                 }
1056                 if itag in self._formats:
1057                     dct.update(self._formats[itag])
1058                 formats.append(dct)
1059             return formats
1060
1061         if 'conn' in video_info and video_info['conn'][0].startswith('rtmp'):
1062             self.report_rtmp_download()
1063             formats = [{
1064                 'format_id': '_rtmp',
1065                 'protocol': 'rtmp',
1066                 'url': video_info['conn'][0],
1067                 'player_url': player_url,
1068             }]
1069         elif len(video_info.get('url_encoded_fmt_stream_map', [''])[0]) >= 1 or len(video_info.get('adaptive_fmts', [''])[0]) >= 1:
1070             encoded_url_map = video_info.get('url_encoded_fmt_stream_map', [''])[0] + ',' + video_info.get('adaptive_fmts', [''])[0]
1071             if 'rtmpe%3Dyes' in encoded_url_map:
1072                 raise ExtractorError('rtmpe downloads are not supported, see https://github.com/rg3/youtube-dl/issues/343 for more information.', expected=True)
1073             url_map = {}
1074             for url_data_str in encoded_url_map.split(','):
1075                 url_data = compat_parse_qs(url_data_str)
1076                 if 'itag' not in url_data or 'url' not in url_data:
1077                     continue
1078                 format_id = url_data['itag'][0]
1079                 url = url_data['url'][0]
1080
1081                 if 'sig' in url_data:
1082                     url += '&signature=' + url_data['sig'][0]
1083                 elif 's' in url_data:
1084                     encrypted_sig = url_data['s'][0]
1085                     ASSETS_RE = r'"assets":.+?"js":\s*("[^"]+")'
1086
1087                     jsplayer_url_json = self._search_regex(
1088                         ASSETS_RE,
1089                         embed_webpage if age_gate else video_webpage,
1090                         'JS player URL (1)', default=None)
1091                     if not jsplayer_url_json and not age_gate:
1092                         # We need the embed website after all
1093                         if embed_webpage is None:
1094                             embed_url = proto + '://www.youtube.com/embed/%s' % video_id
1095                             embed_webpage = self._download_webpage(
1096                                 embed_url, video_id, 'Downloading embed webpage')
1097                         jsplayer_url_json = self._search_regex(
1098                             ASSETS_RE, embed_webpage, 'JS player URL')
1099
1100                     player_url = json.loads(jsplayer_url_json)
1101                     if player_url is None:
1102                         player_url_json = self._search_regex(
1103                             r'ytplayer\.config.*?"url"\s*:\s*("[^"]+")',
1104                             video_webpage, 'age gate player URL')
1105                         player_url = json.loads(player_url_json)
1106
1107                     if self._downloader.params.get('verbose'):
1108                         if player_url is None:
1109                             player_version = 'unknown'
1110                             player_desc = 'unknown'
1111                         else:
1112                             if player_url.endswith('swf'):
1113                                 player_version = self._search_regex(
1114                                     r'-(.+?)(?:/watch_as3)?\.swf$', player_url,
1115                                     'flash player', fatal=False)
1116                                 player_desc = 'flash player %s' % player_version
1117                             else:
1118                                 player_version = self._search_regex(
1119                                     r'html5player-([^/]+?)(?:/html5player)?\.js',
1120                                     player_url,
1121                                     'html5 player', fatal=False)
1122                                 player_desc = 'html5 player %s' % player_version
1123
1124                         parts_sizes = self._signature_cache_id(encrypted_sig)
1125                         self.to_screen('{%s} signature length %s, %s' %
1126                                        (format_id, parts_sizes, player_desc))
1127
1128                     signature = self._decrypt_signature(
1129                         encrypted_sig, video_id, player_url, age_gate)
1130                     url += '&signature=' + signature
1131                 if 'ratebypass' not in url:
1132                     url += '&ratebypass=yes'
1133                 url_map[format_id] = url
1134             formats = _map_to_format_list(url_map)
1135         elif video_info.get('hlsvp'):
1136             manifest_url = video_info['hlsvp'][0]
1137             url_map = self._extract_from_m3u8(manifest_url, video_id)
1138             formats = _map_to_format_list(url_map)
1139         else:
1140             raise ExtractorError('no conn, hlsvp or url_encoded_fmt_stream_map information found in video info')
1141
1142         # Look for the DASH manifest
1143         if self._downloader.params.get('youtube_include_dash_manifest', True):
1144             dash_mpd = video_info.get('dashmpd')
1145             if dash_mpd:
1146                 dash_manifest_url = dash_mpd[0]
1147                 try:
1148                     dash_formats = self._parse_dash_manifest(
1149                         video_id, dash_manifest_url, player_url, age_gate)
1150                 except (ExtractorError, KeyError) as e:
1151                     self.report_warning(
1152                         'Skipping DASH manifest: %r' % e, video_id)
1153                 else:
1154                     # Remove the formats we found through non-DASH, they
1155                     # contain less info and it can be wrong, because we use
1156                     # fixed values (for example the resolution). See
1157                     # https://github.com/rg3/youtube-dl/issues/5774 for an
1158                     # example.
1159                     dash_keys = set(df['format_id'] for df in dash_formats)
1160                     formats = [f for f in formats if f['format_id'] not in dash_keys]
1161                     formats.extend(dash_formats)
1162
1163         # Check for malformed aspect ratio
1164         stretched_m = re.search(
1165             r'<meta\s+property="og:video:tag".*?content="yt:stretch=(?P<w>[0-9]+):(?P<h>[0-9]+)">',
1166             video_webpage)
1167         if stretched_m:
1168             ratio = float(stretched_m.group('w')) / float(stretched_m.group('h'))
1169             for f in formats:
1170                 if f.get('vcodec') != 'none':
1171                     f['stretched_ratio'] = ratio
1172
1173         self._sort_formats(formats)
1174
1175         return {
1176             'id': video_id,
1177             'uploader': video_uploader,
1178             'uploader_id': video_uploader_id,
1179             'upload_date': upload_date,
1180             'title': video_title,
1181             'thumbnail': video_thumbnail,
1182             'description': video_description,
1183             'categories': video_categories,
1184             'subtitles': video_subtitles,
1185             'automatic_captions': automatic_captions,
1186             'duration': video_duration,
1187             'age_limit': 18 if age_gate else 0,
1188             'annotations': video_annotations,
1189             'webpage_url': proto + '://www.youtube.com/watch?v=%s' % video_id,
1190             'view_count': view_count,
1191             'like_count': like_count,
1192             'dislike_count': dislike_count,
1193             'average_rating': float_or_none(video_info.get('avg_rating', [None])[0]),
1194             'formats': formats,
1195         }
1196
1197
1198 class YoutubePlaylistIE(YoutubeBaseInfoExtractor):
1199     IE_DESC = 'YouTube.com playlists'
1200     _VALID_URL = r"""(?x)(?:
1201                         (?:https?://)?
1202                         (?:\w+\.)?
1203                         youtube\.com/
1204                         (?:
1205                            (?:course|view_play_list|my_playlists|artist|playlist|watch|embed/videoseries)
1206                            \? (?:.*?&)*? (?:p|a|list)=
1207                         |  p/
1208                         )
1209                         (
1210                             (?:PL|LL|EC|UU|FL|RD|UL)?[0-9A-Za-z-_]{10,}
1211                             # Top tracks, they can also include dots
1212                             |(?:MC)[\w\.]*
1213                         )
1214                         .*
1215                      |
1216                         ((?:PL|LL|EC|UU|FL|RD|UL)[0-9A-Za-z-_]{10,})
1217                      )"""
1218     _TEMPLATE_URL = 'https://www.youtube.com/playlist?list=%s'
1219     _VIDEO_RE = r'href="\s*/watch\?v=(?P<id>[0-9A-Za-z_-]{11})&amp;[^"]*?index=(?P<index>\d+)'
1220     IE_NAME = 'youtube:playlist'
1221     _TESTS = [{
1222         'url': 'https://www.youtube.com/playlist?list=PLwiyx1dc3P2JR9N8gQaQN_BCvlSlap7re',
1223         'info_dict': {
1224             'title': 'ytdl test PL',
1225             'id': 'PLwiyx1dc3P2JR9N8gQaQN_BCvlSlap7re',
1226         },
1227         'playlist_count': 3,
1228     }, {
1229         'url': 'https://www.youtube.com/playlist?list=PLtPgu7CB4gbZDA7i_euNxn75ISqxwZPYx',
1230         'info_dict': {
1231             'id': 'PLtPgu7CB4gbZDA7i_euNxn75ISqxwZPYx',
1232             'title': 'YDL_Empty_List',
1233         },
1234         'playlist_count': 0,
1235     }, {
1236         'note': 'Playlist with deleted videos (#651). As a bonus, the video #51 is also twice in this list.',
1237         'url': 'https://www.youtube.com/playlist?list=PLwP_SiAcdui0KVebT0mU9Apz359a4ubsC',
1238         'info_dict': {
1239             'title': '29C3: Not my department',
1240             'id': 'PLwP_SiAcdui0KVebT0mU9Apz359a4ubsC',
1241         },
1242         'playlist_count': 95,
1243     }, {
1244         'note': 'issue #673',
1245         'url': 'PLBB231211A4F62143',
1246         'info_dict': {
1247             'title': '[OLD]Team Fortress 2 (Class-based LP)',
1248             'id': 'PLBB231211A4F62143',
1249         },
1250         'playlist_mincount': 26,
1251     }, {
1252         'note': 'Large playlist',
1253         'url': 'https://www.youtube.com/playlist?list=UUBABnxM4Ar9ten8Mdjj1j0Q',
1254         'info_dict': {
1255             'title': 'Uploads from Cauchemar',
1256             'id': 'UUBABnxM4Ar9ten8Mdjj1j0Q',
1257         },
1258         'playlist_mincount': 799,
1259     }, {
1260         'url': 'PLtPgu7CB4gbY9oDN3drwC3cMbJggS7dKl',
1261         'info_dict': {
1262             'title': 'YDL_safe_search',
1263             'id': 'PLtPgu7CB4gbY9oDN3drwC3cMbJggS7dKl',
1264         },
1265         'playlist_count': 2,
1266     }, {
1267         'note': 'embedded',
1268         'url': 'http://www.youtube.com/embed/videoseries?list=PL6IaIsEjSbf96XFRuNccS_RuEXwNdsoEu',
1269         'playlist_count': 4,
1270         'info_dict': {
1271             'title': 'JODA15',
1272             'id': 'PL6IaIsEjSbf96XFRuNccS_RuEXwNdsoEu',
1273         }
1274     }, {
1275         'note': 'Embedded SWF player',
1276         'url': 'http://www.youtube.com/p/YN5VISEtHet5D4NEvfTd0zcgFk84NqFZ?hl=en_US&fs=1&rel=0',
1277         'playlist_count': 4,
1278         'info_dict': {
1279             'title': 'JODA7',
1280             'id': 'YN5VISEtHet5D4NEvfTd0zcgFk84NqFZ',
1281         }
1282     }, {
1283         'note': 'Buggy playlist: the webpage has a "Load more" button but it doesn\'t have more videos',
1284         'url': 'https://www.youtube.com/playlist?list=UUXw-G3eDE9trcvY2sBMM_aA',
1285         'info_dict': {
1286             'title': 'Uploads from Interstellar Movie',
1287             'id': 'UUXw-G3eDE9trcvY2sBMM_aA',
1288         },
1289         'playlist_mincout': 21,
1290     }]
1291
1292     def _real_initialize(self):
1293         self._login()
1294
1295     def _extract_mix(self, playlist_id):
1296         # The mixes are generated from a single video
1297         # the id of the playlist is just 'RD' + video_id
1298         url = 'https://youtube.com/watch?v=%s&list=%s' % (playlist_id[-11:], playlist_id)
1299         webpage = self._download_webpage(
1300             url, playlist_id, 'Downloading Youtube mix')
1301         search_title = lambda class_name: get_element_by_attribute('class', class_name, webpage)
1302         title_span = (
1303             search_title('playlist-title') or
1304             search_title('title long-title') or
1305             search_title('title'))
1306         title = clean_html(title_span)
1307         ids = orderedSet(re.findall(
1308             r'''(?xs)data-video-username=".*?".*?
1309                        href="/watch\?v=([0-9A-Za-z_-]{11})&amp;[^"]*?list=%s''' % re.escape(playlist_id),
1310             webpage))
1311         url_results = self._ids_to_results(ids)
1312
1313         return self.playlist_result(url_results, playlist_id, title)
1314
1315     def _extract_playlist(self, playlist_id):
1316         url = self._TEMPLATE_URL % playlist_id
1317         page = self._download_webpage(url, playlist_id)
1318         more_widget_html = content_html = page
1319
1320         for match in re.findall(r'<div class="yt-alert-message">([^<]+)</div>', page):
1321             match = match.strip()
1322             # Check if the playlist exists or is private
1323             if re.match(r'[^<]*(The|This) playlist (does not exist|is private)[^<]*', match):
1324                 raise ExtractorError(
1325                     'The playlist doesn\'t exist or is private, use --username or '
1326                     '--netrc to access it.',
1327                     expected=True)
1328             elif re.match(r'[^<]*Invalid parameters[^<]*', match):
1329                 raise ExtractorError(
1330                     'Invalid parameters. Maybe URL is incorrect.',
1331                     expected=True)
1332             elif re.match(r'[^<]*Choose your language[^<]*', match):
1333                 continue
1334             else:
1335                 self.report_warning('Youtube gives an alert message: ' + match)
1336
1337         # Extract the video ids from the playlist pages
1338         ids = []
1339
1340         for page_num in itertools.count(1):
1341             matches = re.finditer(self._VIDEO_RE, content_html)
1342             # We remove the duplicates and the link with index 0
1343             # (it's not the first video of the playlist)
1344             new_ids = orderedSet(m.group('id') for m in matches if m.group('index') != '0')
1345             ids.extend(new_ids)
1346
1347             mobj = re.search(r'data-uix-load-more-href="/?(?P<more>[^"]+)"', more_widget_html)
1348             if not mobj:
1349                 break
1350
1351             more = self._download_json(
1352                 'https://youtube.com/%s' % mobj.group('more'), playlist_id,
1353                 'Downloading page #%s' % page_num,
1354                 transform_source=uppercase_escape)
1355             content_html = more['content_html']
1356             if not content_html.strip():
1357                 # Some webpages show a "Load more" button but they don't
1358                 # have more videos
1359                 break
1360             more_widget_html = more['load_more_widget_html']
1361
1362         playlist_title = self._html_search_regex(
1363             r'(?s)<h1 class="pl-header-title[^"]*">\s*(.*?)\s*</h1>',
1364             page, 'title')
1365
1366         url_results = self._ids_to_results(ids)
1367         return self.playlist_result(url_results, playlist_id, playlist_title)
1368
1369     def _real_extract(self, url):
1370         # Extract playlist id
1371         mobj = re.match(self._VALID_URL, url)
1372         if mobj is None:
1373             raise ExtractorError('Invalid URL: %s' % url)
1374         playlist_id = mobj.group(1) or mobj.group(2)
1375
1376         # Check if it's a video-specific URL
1377         query_dict = compat_urlparse.parse_qs(compat_urlparse.urlparse(url).query)
1378         if 'v' in query_dict:
1379             video_id = query_dict['v'][0]
1380             if self._downloader.params.get('noplaylist'):
1381                 self.to_screen('Downloading just video %s because of --no-playlist' % video_id)
1382                 return self.url_result(video_id, 'Youtube', video_id=video_id)
1383             else:
1384                 self.to_screen('Downloading playlist %s - add --no-playlist to just download video %s' % (playlist_id, video_id))
1385
1386         if playlist_id.startswith('RD') or playlist_id.startswith('UL'):
1387             # Mixes require a custom extraction process
1388             return self._extract_mix(playlist_id)
1389
1390         return self._extract_playlist(playlist_id)
1391
1392
1393 class YoutubeChannelIE(InfoExtractor):
1394     IE_DESC = 'YouTube.com channels'
1395     _VALID_URL = r'https?://(?:youtu\.be|(?:\w+\.)?youtube(?:-nocookie)?\.com)/channel/(?P<id>[0-9A-Za-z_-]+)'
1396     _TEMPLATE_URL = 'https://www.youtube.com/channel/%s/videos'
1397     IE_NAME = 'youtube:channel'
1398     _TESTS = [{
1399         'note': 'paginated channel',
1400         'url': 'https://www.youtube.com/channel/UCKfVa3S1e4PHvxWcwyMMg8w',
1401         'playlist_mincount': 91,
1402         'info_dict': {
1403             'id': 'UCKfVa3S1e4PHvxWcwyMMg8w',
1404         }
1405     }]
1406
1407     @staticmethod
1408     def extract_videos_from_page(page):
1409         ids_in_page = []
1410         titles_in_page = []
1411         for mobj in re.finditer(r'(?:title="(?P<title>[^"]+)"[^>]+)?href="/watch\?v=(?P<id>[0-9A-Za-z_-]+)&?', page):
1412             video_id = mobj.group('id')
1413             video_title = unescapeHTML(mobj.group('title'))
1414             try:
1415                 idx = ids_in_page.index(video_id)
1416                 if video_title and not titles_in_page[idx]:
1417                     titles_in_page[idx] = video_title
1418             except ValueError:
1419                 ids_in_page.append(video_id)
1420                 titles_in_page.append(video_title)
1421         return zip(ids_in_page, titles_in_page)
1422
1423     def _real_extract(self, url):
1424         channel_id = self._match_id(url)
1425
1426         url = self._TEMPLATE_URL % channel_id
1427
1428         # Channel by page listing is restricted to 35 pages of 30 items, i.e. 1050 videos total (see #5778)
1429         # Workaround by extracting as a playlist if managed to obtain channel playlist URL
1430         # otherwise fallback on channel by page extraction
1431         channel_page = self._download_webpage(
1432             url + '?view=57', channel_id,
1433             'Downloading channel page', fatal=False)
1434         channel_playlist_id = self._search_regex(
1435             [r'<meta itemprop="channelId" content="([^"]+)">',
1436              r'data-channel-external-id="([^"]+)"'],
1437             channel_page, 'channel id', default=None)
1438         if channel_playlist_id and channel_playlist_id.startswith('UC'):
1439             playlist_id = 'UU' + channel_playlist_id[2:]
1440             return self.url_result(
1441                 compat_urlparse.urljoin(url, '/playlist?list=%s' % playlist_id), 'YoutubePlaylist')
1442
1443         channel_page = self._download_webpage(url, channel_id, 'Downloading page #1')
1444         autogenerated = re.search(r'''(?x)
1445                 class="[^"]*?(?:
1446                     channel-header-autogenerated-label|
1447                     yt-channel-title-autogenerated
1448                 )[^"]*"''', channel_page) is not None
1449
1450         if autogenerated:
1451             # The videos are contained in a single page
1452             # the ajax pages can't be used, they are empty
1453             entries = [
1454                 self.url_result(
1455                     video_id, 'Youtube', video_id=video_id,
1456                     video_title=video_title)
1457                 for video_id, video_title in self.extract_videos_from_page(channel_page)]
1458             return self.playlist_result(entries, channel_id)
1459
1460         def _entries():
1461             more_widget_html = content_html = channel_page
1462             for pagenum in itertools.count(1):
1463
1464                 for video_id, video_title in self.extract_videos_from_page(content_html):
1465                     yield self.url_result(
1466                         video_id, 'Youtube', video_id=video_id,
1467                         video_title=video_title)
1468
1469                 mobj = re.search(
1470                     r'data-uix-load-more-href="/?(?P<more>[^"]+)"',
1471                     more_widget_html)
1472                 if not mobj:
1473                     break
1474
1475                 more = self._download_json(
1476                     'https://youtube.com/%s' % mobj.group('more'), channel_id,
1477                     'Downloading page #%s' % (pagenum + 1),
1478                     transform_source=uppercase_escape)
1479                 content_html = more['content_html']
1480                 more_widget_html = more['load_more_widget_html']
1481
1482         return self.playlist_result(_entries(), channel_id)
1483
1484
1485 class YoutubeUserIE(YoutubeChannelIE):
1486     IE_DESC = 'YouTube.com user videos (URL or "ytuser" keyword)'
1487     _VALID_URL = r'(?:(?:(?:https?://)?(?:\w+\.)?youtube\.com/(?:user/)?(?!(?:attribution_link|watch|results)(?:$|[^a-z_A-Z0-9-])))|ytuser:)(?!feed/)(?P<id>[A-Za-z0-9_-]+)'
1488     _TEMPLATE_URL = 'https://www.youtube.com/user/%s/videos'
1489     IE_NAME = 'youtube:user'
1490
1491     _TESTS = [{
1492         'url': 'https://www.youtube.com/user/TheLinuxFoundation',
1493         'playlist_mincount': 320,
1494         'info_dict': {
1495             'title': 'TheLinuxFoundation',
1496         }
1497     }, {
1498         'url': 'ytuser:phihag',
1499         'only_matching': True,
1500     }]
1501
1502     @classmethod
1503     def suitable(cls, url):
1504         # Don't return True if the url can be extracted with other youtube
1505         # extractor, the regex would is too permissive and it would match.
1506         other_ies = iter(klass for (name, klass) in globals().items() if name.endswith('IE') and klass is not cls)
1507         if any(ie.suitable(url) for ie in other_ies):
1508             return False
1509         else:
1510             return super(YoutubeUserIE, cls).suitable(url)
1511
1512
1513 class YoutubeSearchIE(SearchInfoExtractor, YoutubePlaylistIE):
1514     IE_DESC = 'YouTube.com searches'
1515     # there doesn't appear to be a real limit, for example if you search for
1516     # 'python' you get more than 8.000.000 results
1517     _MAX_RESULTS = float('inf')
1518     IE_NAME = 'youtube:search'
1519     _SEARCH_KEY = 'ytsearch'
1520     _EXTRA_QUERY_ARGS = {}
1521     _TESTS = []
1522
1523     def _get_n_results(self, query, n):
1524         """Get a specified number of results for a query"""
1525
1526         videos = []
1527         limit = n
1528
1529         for pagenum in itertools.count(1):
1530             url_query = {
1531                 'search_query': query,
1532                 'page': pagenum,
1533                 'spf': 'navigate',
1534             }
1535             url_query.update(self._EXTRA_QUERY_ARGS)
1536             result_url = 'https://www.youtube.com/results?' + compat_urllib_parse.urlencode(url_query)
1537             data = self._download_json(
1538                 result_url, video_id='query "%s"' % query,
1539                 note='Downloading page %s' % pagenum,
1540                 errnote='Unable to download API page')
1541             html_content = data[1]['body']['content']
1542
1543             if 'class="search-message' in html_content:
1544                 raise ExtractorError(
1545                     '[youtube] No video results', expected=True)
1546
1547             new_videos = self._ids_to_results(orderedSet(re.findall(
1548                 r'href="/watch\?v=(.{11})', html_content)))
1549             videos += new_videos
1550             if not new_videos or len(videos) > limit:
1551                 break
1552
1553         if len(videos) > n:
1554             videos = videos[:n]
1555         return self.playlist_result(videos, query)
1556
1557
1558 class YoutubeSearchDateIE(YoutubeSearchIE):
1559     IE_NAME = YoutubeSearchIE.IE_NAME + ':date'
1560     _SEARCH_KEY = 'ytsearchdate'
1561     IE_DESC = 'YouTube.com searches, newest videos first'
1562     _EXTRA_QUERY_ARGS = {'search_sort': 'video_date_uploaded'}
1563
1564
1565 class YoutubeSearchURLIE(InfoExtractor):
1566     IE_DESC = 'YouTube.com search URLs'
1567     IE_NAME = 'youtube:search_url'
1568     _VALID_URL = r'https?://(?:www\.)?youtube\.com/results\?(.*?&)?search_query=(?P<query>[^&]+)(?:[&]|$)'
1569     _TESTS = [{
1570         'url': 'https://www.youtube.com/results?baz=bar&search_query=youtube-dl+test+video&filters=video&lclk=video',
1571         'playlist_mincount': 5,
1572         'info_dict': {
1573             'title': 'youtube-dl test video',
1574         }
1575     }]
1576
1577     def _real_extract(self, url):
1578         mobj = re.match(self._VALID_URL, url)
1579         query = compat_urllib_parse.unquote_plus(mobj.group('query'))
1580
1581         webpage = self._download_webpage(url, query)
1582         result_code = self._search_regex(
1583             r'(?s)<ol[^>]+class="item-section"(.*?)</ol>', webpage, 'result HTML')
1584
1585         part_codes = re.findall(
1586             r'(?s)<h3 class="yt-lockup-title">(.*?)</h3>', result_code)
1587         entries = []
1588         for part_code in part_codes:
1589             part_title = self._html_search_regex(
1590                 [r'(?s)title="([^"]+)"', r'>([^<]+)</a>'], part_code, 'item title', fatal=False)
1591             part_url_snippet = self._html_search_regex(
1592                 r'(?s)href="([^"]+)"', part_code, 'item URL')
1593             part_url = compat_urlparse.urljoin(
1594                 'https://www.youtube.com/', part_url_snippet)
1595             entries.append({
1596                 '_type': 'url',
1597                 'url': part_url,
1598                 'title': part_title,
1599             })
1600
1601         return {
1602             '_type': 'playlist',
1603             'entries': entries,
1604             'title': query,
1605         }
1606
1607
1608 class YoutubeShowIE(InfoExtractor):
1609     IE_DESC = 'YouTube.com (multi-season) shows'
1610     _VALID_URL = r'https?://www\.youtube\.com/show/(?P<id>[^?#]*)'
1611     IE_NAME = 'youtube:show'
1612     _TESTS = [{
1613         'url': 'http://www.youtube.com/show/airdisasters',
1614         'playlist_mincount': 3,
1615         'info_dict': {
1616             'id': 'airdisasters',
1617             'title': 'Air Disasters',
1618         }
1619     }]
1620
1621     def _real_extract(self, url):
1622         mobj = re.match(self._VALID_URL, url)
1623         playlist_id = mobj.group('id')
1624         webpage = self._download_webpage(
1625             url, playlist_id, 'Downloading show webpage')
1626         # There's one playlist for each season of the show
1627         m_seasons = list(re.finditer(r'href="(/playlist\?list=.*?)"', webpage))
1628         self.to_screen('%s: Found %s seasons' % (playlist_id, len(m_seasons)))
1629         entries = [
1630             self.url_result(
1631                 'https://www.youtube.com' + season.group(1), 'YoutubePlaylist')
1632             for season in m_seasons
1633         ]
1634         title = self._og_search_title(webpage, fatal=False)
1635
1636         return {
1637             '_type': 'playlist',
1638             'id': playlist_id,
1639             'title': title,
1640             'entries': entries,
1641         }
1642
1643
1644 class YoutubeFeedsInfoExtractor(YoutubeBaseInfoExtractor):
1645     """
1646     Base class for feed extractors
1647     Subclasses must define the _FEED_NAME and _PLAYLIST_TITLE properties.
1648     """
1649     _LOGIN_REQUIRED = True
1650
1651     @property
1652     def IE_NAME(self):
1653         return 'youtube:%s' % self._FEED_NAME
1654
1655     def _real_initialize(self):
1656         self._login()
1657
1658     def _real_extract(self, url):
1659         page = self._download_webpage(
1660             'https://www.youtube.com/feed/%s' % self._FEED_NAME, self._PLAYLIST_TITLE)
1661
1662         # The extraction process is the same as for playlists, but the regex
1663         # for the video ids doesn't contain an index
1664         ids = []
1665         more_widget_html = content_html = page
1666         for page_num in itertools.count(1):
1667             matches = re.findall(r'href="\s*/watch\?v=([0-9A-Za-z_-]{11})', content_html)
1668
1669             # 'recommended' feed has infinite 'load more' and each new portion spins
1670             # the same videos in (sometimes) slightly different order, so we'll check
1671             # for unicity and break when portion has no new videos
1672             new_ids = filter(lambda video_id: video_id not in ids, orderedSet(matches))
1673             if not new_ids:
1674                 break
1675
1676             ids.extend(new_ids)
1677
1678             mobj = re.search(r'data-uix-load-more-href="/?(?P<more>[^"]+)"', more_widget_html)
1679             if not mobj:
1680                 break
1681
1682             more = self._download_json(
1683                 'https://youtube.com/%s' % mobj.group('more'), self._PLAYLIST_TITLE,
1684                 'Downloading page #%s' % page_num,
1685                 transform_source=uppercase_escape)
1686             content_html = more['content_html']
1687             more_widget_html = more['load_more_widget_html']
1688
1689         return self.playlist_result(
1690             self._ids_to_results(ids), playlist_title=self._PLAYLIST_TITLE)
1691
1692
1693 class YoutubeWatchLaterIE(YoutubePlaylistIE):
1694     IE_NAME = 'youtube:watchlater'
1695     IE_DESC = 'Youtube watch later list, ":ytwatchlater" for short (requires authentication)'
1696     _VALID_URL = r'https?://www\.youtube\.com/(?:feed/watch_later|playlist\?list=WL)|:ytwatchlater'
1697
1698     _TESTS = []  # override PlaylistIE tests
1699
1700     def _real_extract(self, url):
1701         return self._extract_playlist('WL')
1702
1703
1704 class YoutubeFavouritesIE(YoutubeBaseInfoExtractor):
1705     IE_NAME = 'youtube:favorites'
1706     IE_DESC = 'YouTube.com favourite videos, ":ytfav" for short (requires authentication)'
1707     _VALID_URL = r'https?://www\.youtube\.com/my_favorites|:ytfav(?:ou?rites)?'
1708     _LOGIN_REQUIRED = True
1709
1710     def _real_extract(self, url):
1711         webpage = self._download_webpage('https://www.youtube.com/my_favorites', 'Youtube Favourites videos')
1712         playlist_id = self._search_regex(r'list=(.+?)["&]', webpage, 'favourites playlist id')
1713         return self.url_result(playlist_id, 'YoutubePlaylist')
1714
1715
1716 class YoutubeRecommendedIE(YoutubeFeedsInfoExtractor):
1717     IE_DESC = 'YouTube.com recommended videos, ":ytrec" for short (requires authentication)'
1718     _VALID_URL = r'https?://www\.youtube\.com/feed/recommended|:ytrec(?:ommended)?'
1719     _FEED_NAME = 'recommended'
1720     _PLAYLIST_TITLE = 'Youtube Recommended videos'
1721
1722
1723 class YoutubeSubscriptionsIE(YoutubeFeedsInfoExtractor):
1724     IE_DESC = 'YouTube.com subscriptions feed, "ytsubs" keyword (requires authentication)'
1725     _VALID_URL = r'https?://www\.youtube\.com/feed/subscriptions|:ytsubs(?:criptions)?'
1726     _FEED_NAME = 'subscriptions'
1727     _PLAYLIST_TITLE = 'Youtube Subscriptions'
1728
1729
1730 class YoutubeHistoryIE(YoutubeFeedsInfoExtractor):
1731     IE_DESC = 'Youtube watch history, ":ythistory" for short (requires authentication)'
1732     _VALID_URL = 'https?://www\.youtube\.com/feed/history|:ythistory'
1733     _FEED_NAME = 'history'
1734     _PLAYLIST_TITLE = 'Youtube History'
1735
1736
1737 class YoutubeTruncatedURLIE(InfoExtractor):
1738     IE_NAME = 'youtube:truncated_url'
1739     IE_DESC = False  # Do not list
1740     _VALID_URL = r'''(?x)
1741         (?:https?://)?
1742         (?:\w+\.)?[yY][oO][uU][tT][uU][bB][eE](?:-nocookie)?\.com/
1743         (?:watch\?(?:
1744             feature=[a-z_]+|
1745             annotation_id=annotation_[^&]+|
1746             x-yt-cl=[0-9]+|
1747             hl=[^&]*|
1748         )?
1749         |
1750             attribution_link\?a=[^&]+
1751         )
1752         $
1753     '''
1754
1755     _TESTS = [{
1756         'url': 'http://www.youtube.com/watch?annotation_id=annotation_3951667041',
1757         'only_matching': True,
1758     }, {
1759         'url': 'http://www.youtube.com/watch?',
1760         'only_matching': True,
1761     }, {
1762         'url': 'https://www.youtube.com/watch?x-yt-cl=84503534',
1763         'only_matching': True,
1764     }, {
1765         'url': 'https://www.youtube.com/watch?feature=foo',
1766         'only_matching': True,
1767     }, {
1768         'url': 'https://www.youtube.com/watch?hl=en-GB',
1769         'only_matching': True,
1770     }]
1771
1772     def _real_extract(self, url):
1773         raise ExtractorError(
1774             'Did you forget to quote the URL? Remember that & is a meta '
1775             'character in most shells, so you want to put the URL in quotes, '
1776             'like  youtube-dl '
1777             '"http://www.youtube.com/watch?feature=foo&v=BaW_jenozKc" '
1778             ' or simply  youtube-dl BaW_jenozKc  .',
1779             expected=True)
1780
1781
1782 class YoutubeTruncatedIDIE(InfoExtractor):
1783     IE_NAME = 'youtube:truncated_id'
1784     IE_DESC = False  # Do not list
1785     _VALID_URL = r'https?://(?:www\.)?youtube\.com/watch\?v=(?P<id>[0-9A-Za-z_-]{1,10})$'
1786
1787     _TESTS = [{
1788         'url': 'https://www.youtube.com/watch?v=N_708QY7Ob',
1789         'only_matching': True,
1790     }]
1791
1792     def _real_extract(self, url):
1793         video_id = self._match_id(url)
1794         raise ExtractorError(
1795             'Incomplete YouTube ID %s. URL %s looks truncated.' % (video_id, url),
1796             expected=True)