]> jfr.im git - yt-dlp.git/blame - youtube_dl/extractor/youtube.py
[YoutubeDL] Support DASH manifest downloading
[yt-dlp.git] / youtube_dl / extractor / youtube.py
CommitLineData
c5e8d7af 1# coding: utf-8
c5e8d7af 2
78caa52a
PH
3from __future__ import unicode_literals
4
5
0ca96d48 6import itertools
c5e8d7af 7import json
c4417ddb 8import os.path
c5e8d7af 9import re
42939b61 10import time
e0df6211 11import traceback
c5e8d7af 12
b05654f0 13from .common import InfoExtractor, SearchInfoExtractor
2b25cb5d 14from ..jsinterp import JSInterpreter
54256267 15from ..swfinterp import SWFInterpreter
4bb4a188 16from ..compat import (
edf3e38e 17 compat_chr,
c5e8d7af 18 compat_parse_qs,
c5e8d7af
PH
19 compat_urllib_parse,
20 compat_urllib_request,
7c61bd36 21 compat_urlparse,
c5e8d7af 22 compat_str,
4bb4a188
PH
23)
24from ..utils import (
c5e8d7af 25 clean_html,
c5e8d7af 26 ExtractorError,
2d30521a 27 float_or_none,
4bb4a188
PH
28 get_element_by_attribute,
29 get_element_by_id,
dd27fd17 30 int_or_none,
4bb4a188 31 orderedSet,
c5e8d7af
PH
32 unescapeHTML,
33 unified_strdate,
81c2f20b 34 uppercase_escape,
c5e8d7af
PH
35)
36
5f6a1245 37
de7f3446 38class YoutubeBaseInfoExtractor(InfoExtractor):
b2e8bc1b
JMF
39 """Provide base functions for Youtube extractors"""
40 _LOGIN_URL = 'https://accounts.google.com/ServiceLogin'
83317f69 41 _TWOFACTOR_URL = 'https://accounts.google.com/SecondFactor'
b2e8bc1b
JMF
42 _NETRC_MACHINE = 'youtube'
43 # If True it will raise an error if no login info is provided
44 _LOGIN_REQUIRED = False
45
b2e8bc1b 46 def _set_language(self):
810fb84d
PH
47 self._set_cookie(
48 '.youtube.com', 'PREF', 'f1=50000000&hl=en',
42939b61 49 # YouTube sets the expire time to about two months
810fb84d 50 expire_time=time.time() + 2 * 30 * 24 * 3600)
b2e8bc1b 51
25f14e9f
S
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
b2e8bc1b 57 def _login(self):
83317f69 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 """
b2e8bc1b
JMF
65 (username, password) = self._get_login_info()
66 # No authentication to be performed
67 if username is None:
68 if self._LOGIN_REQUIRED:
69ea8ca4 69 raise ExtractorError('No login info available, needed for using %s.' % self.IE_NAME, expected=True)
83317f69 70 return True
b2e8bc1b 71
7cc3570e
PH
72 login_page = self._download_webpage(
73 self._LOGIN_URL, None,
69ea8ca4
PH
74 note='Downloading login page',
75 errnote='unable to fetch login page', fatal=False)
7cc3570e
PH
76 if login_page is False:
77 return
b2e8bc1b 78
795f28f8 79 galx = self._search_regex(r'(?s)<input.+?name="GALX".+?value="(.+?)"',
78caa52a 80 login_page, 'Login GALX parameter')
c5e8d7af 81
b2e8bc1b
JMF
82 # Log in
83 login_form_strs = {
8bcc8756
JW
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',
b2e8bc1b 103 }
83317f69 104
b2e8bc1b
JMF
105 # Convert to UTF-8 *before* urlencode because Python 2.x's urlencode
106 # chokes on unicode
5f6a1245 107 login_form = dict((k.encode('utf-8'), v.encode('utf-8')) for k, v in login_form_strs.items())
b2e8bc1b 108 login_data = compat_urllib_parse.urlencode(login_form).encode('ascii')
7cc3570e
PH
109
110 req = compat_urllib_request.Request(self._LOGIN_URL, login_data)
111 login_results = self._download_webpage(
112 req, None,
69ea8ca4 113 note='Logging in', errnote='unable to log in', fatal=False)
7cc3570e
PH
114 if login_results is False:
115 return False
83317f69 116
117 if re.search(r'id="errormsg_0_Passwd"', login_results) is not None:
69ea8ca4 118 raise ExtractorError('Please use your account password and a two-factor code instead of an application-specific password.', expected=True)
83317f69 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:
69ea8ca4
PH
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.)')
83317f69 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:
69ea8ca4 135 self._downloader.report_warning('Failed to get secTok - did the page structure change?')
83317f69 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:
69ea8ca4 139 self._downloader.report_warning('Failed to get timeStmp - did the page structure change?')
83317f69 140 timeStmp = match.group(1)
141
142 tfa_form_strs = {
78caa52a
PH
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',
83317f69 156 }
5f6a1245 157 tfa_form = dict((k.encode('utf-8'), v.encode('utf-8')) for k, v in tfa_form_strs.items())
83317f69 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,
69ea8ca4 163 note='Submitting TFA code', errnote='unable to submit tfa', fatal=False)
83317f69 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:
69ea8ca4 169 self._downloader.report_warning('Two-factor code expired. Please try again, or use a one-use backup code instead.')
83317f69 170 return False
171 if re.search(r'(?i)<form[^>]* id="gaia_loginform"', tfa_results) is not None:
69ea8ca4 172 self._downloader.report_warning('unable to log in - did the page structure change?')
83317f69 173 return False
174 if re.search(r'smsauth-interstitial-reviewsettings', tfa_results) is not None:
69ea8ca4 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.')
83317f69 176 return False
177
7cc3570e 178 if re.search(r'(?i)<form[^>]* id="gaia_loginform"', login_results) is not None:
69ea8ca4 179 self._downloader.report_warning('unable to log in: bad username or password')
b2e8bc1b
JMF
180 return False
181 return True
182
b2e8bc1b
JMF
183 def _real_initialize(self):
184 if self._downloader is None:
185 return
42939b61 186 self._set_language()
b2e8bc1b
JMF
187 if not self._login():
188 return
c5e8d7af 189
8377574c 190
360e1ca5 191class YoutubeIE(YoutubeBaseInfoExtractor):
78caa52a 192 IE_DESC = 'YouTube.com'
cb7dfeea 193 _VALID_URL = r"""(?x)^
c5e8d7af 194 (
edb53e2d 195 (?:https?://|//) # http(s):// or protocol-independent URL
cb7dfeea 196 (?:(?:(?:(?:\w+\.)?[yY][oO][uU][tT][uU][bB][eE](?:-nocookie)?\.com/|
484aaeb2 197 (?:www\.)?deturl\.com/www\.youtube\.com/|
e70dc1d1 198 (?:www\.)?pwnyoutube\.com/|
f7000f3a 199 (?:www\.)?yourepeat\.com/|
e69ae5b9
JMF
200 tube\.majestyc\.net/|
201 youtube\.googleapis\.com/) # the various hostnames, with wildcard subdomains
c5e8d7af
PH
202 (?:.*?\#/)? # handle anchor (#/) redirect urls
203 (?: # the various things that can precede the ID:
ac7553d0 204 (?:(?:v|embed|e)/(?!videoseries)) # v/ or embed/ or e/
c5e8d7af 205 |(?: # or the v= param in all its forms
f7000f3a 206 (?:(?:watch|movie)(?:_popup)?(?:\.php)?/?)? # preceding watch(_popup|.php) or nothing (like /?v=xxxx)
c5e8d7af
PH
207 (?:\?|\#!?) # the params delimiter ? or # or #!
208 (?:.*?&)? # any other preceding param (like /?s=tuff&v=xxxx)
209 v=
210 )
f4b05232
JMF
211 ))
212 |youtu\.be/ # just youtu.be/xxxx
edb53e2d 213 |(?:www\.)?cleanvideosearch\.com/media/action/yt/watch\?videoId=
f4b05232 214 )
c5e8d7af 215 )? # all until now is optional -> you can pass the naked ID
8963d9c2 216 ([0-9A-Za-z_-]{11}) # here is it! the YouTube video ID
9291475f 217 (?!.*?&list=) # combined list/video URLs are handled by the playlist IE
c5e8d7af
PH
218 (?(1).+)? # if we found the ID, everything can follow
219 $"""
c5e8d7af 220 _NEXT_URL_RE = r'[\?&]next_url=([^&]+)'
2c62dc26
PH
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
1d043b93 238
86fe61c8 239 # 3d videos
43b81eb9
PH
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},
836a086c 247
96fb5605 248 # Apple HTTP Live Streaming
43b81eb9
PH
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},
2c62dc26
PH
256
257 # DASH mp4 video
43b81eb9
PH
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},
e65566a9 263 '138': {'ext': 'mp4', 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40}, # Height can vary (https://github.com/rg3/youtube-dl/issues/4559)
43b81eb9
PH
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},
0d2c1418
PH
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'},
836a086c 269
f6f1fc92 270 # Dash mp4 audio
62cd676c
PH
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'},
836a086c
AZ
274
275 # Dash webm
e75cafe9
A
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},
1cc887cb 282 '278': {'ext': 'webm', 'height': 144, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'container': 'webm', 'vcodec': 'VP9'},
e75cafe9
A
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},
3c80377b 290 '271': {'ext': 'webm', 'height': 1440, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
18061bba 291 '272': {'ext': 'webm', 'height': 2160, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40},
a51d3aa0
PH
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'},
76b3c610 294 '308': {'ext': 'webm', 'height': 1440, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'fps': 60, 'vcodec': 'VP9'},
10a404c3 295 '313': {'ext': 'webm', 'height': 2160, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'vcodec': 'VP9'},
76b3c610 296 '315': {'ext': 'webm', 'height': 2160, 'format_note': 'DASH video', 'acodec': 'none', 'preference': -40, 'fps': 60, 'vcodec': 'VP9'},
2c62dc26
PH
297
298 # Dash webm audio
55db73ef 299 '171': {'ext': 'webm', 'vcodec': 'none', 'format_note': 'DASH audio', 'abr': 128, 'preference': -50},
e75cafe9 300 '172': {'ext': 'webm', 'vcodec': 'none', 'format_note': 'DASH audio', 'abr': 256, 'preference': -50},
ce6b9a2d 301
0857baad
PH
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
ce6b9a2d
PH
307 # RTMP (unnamed)
308 '_rtmp': {'protocol': 'rtmp'},
c5e8d7af 309 }
836a086c 310
78caa52a 311 IE_NAME = 'youtube'
2eb88d95
PH
312 _TESTS = [
313 {
4bc3a23e
PH
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'],
3e7c1224
PH
324 'like_count': int,
325 'dislike_count': int,
2eb88d95 326 }
0e853ca4 327 },
0e853ca4 328 {
4bc3a23e
PH
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',
2eb88d95 339 }
c108eb73
JMF
340 },
341 {
4bc3a23e
PH
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',
c108eb73
JMF
352 }
353 },
fccd3771 354 {
4bc3a23e
PH
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'
fccd3771
PH
365 }
366 },
dd27fd17 367 {
4bc3a23e
PH
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'
4919603f 378 },
4bc3a23e
PH
379 'params': {
380 'youtube_include_dash_manifest': True,
381 'format': '141',
4919603f 382 },
dd27fd17 383 },
3489b7d2
JMF
384 # DASH manifest with encrypted signature
385 {
78caa52a
PH
386 'url': 'https://www.youtube.com/watch?v=IB3lcPjvWLA',
387 'info_dict': {
388 'id': 'IB3lcPjvWLA',
389 'ext': 'm4a',
b766eb27
JMF
390 'title': 'Afrojack, Spree Wilson - The Spark ft. Spree Wilson',
391 'description': 'md5:12e7067fa6735a77bdcbb58cb1187d2d',
78caa52a
PH
392 'uploader': 'AfrojackVEVO',
393 'uploader_id': 'AfrojackVEVO',
394 'upload_date': '20131011',
3489b7d2 395 },
4bc3a23e 396 'params': {
78caa52a
PH
397 'youtube_include_dash_manifest': True,
398 'format': '141',
3489b7d2
JMF
399 },
400 },
aaeb86f6
S
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 },
aa79ac0c
PH
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 }
c522adb1
JMF
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',
9ed99402 438 'description': 're:(?s).{100,}About the Game\n.*?The Witcher 3: Wild Hunt.{100,}',
c522adb1
JMF
439 'uploader': 'The Witcher',
440 'uploader_id': 'WitcherGame',
441 'upload_date': '20140605',
442 },
443 },
fccae2b9
S
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 },
774e208f
PH
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 ]
e52a40ab
PH
472 },
473 # Olympics (https://github.com/rg3/youtube-dl/issues/4431)
474 {
475 'url': 'lqQg6PlCWgI',
476 'info_dict': {
477 'id': 'lqQg6PlCWgI',
478 'ext': 'mp4',
cbe2bd91
PH
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',
e52a40ab 487 }
cbe2bd91 488 },
6271f1ca
PH
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 },
06b491eb
S
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 },
2eb88d95
PH
519 ]
520
e0df6211
PH
521 def __init__(self, *args, **kwargs):
522 super(YoutubeIE, self).__init__(*args, **kwargs)
83799698 523 self._player_cache = {}
e0df6211 524
c5e8d7af
PH
525 def report_video_info_webpage_download(self, video_id):
526 """Report attempt to download video info webpage."""
69ea8ca4 527 self.to_screen('%s: Downloading video info webpage' % video_id)
c5e8d7af 528
c5e8d7af
PH
529 def report_information_extraction(self, video_id):
530 """Report attempt to extract video information."""
69ea8ca4 531 self.to_screen('%s: Extracting video information' % video_id)
c5e8d7af
PH
532
533 def report_unavailable_format(self, video_id, format):
534 """Report extracted video URL."""
69ea8ca4 535 self.to_screen('%s: Format %s not available' % (video_id, format))
c5e8d7af
PH
536
537 def report_rtmp_download(self):
538 """Indicate the download will use the RTMP protocol."""
69ea8ca4 539 self.to_screen('RTMP download detected')
c5e8d7af 540
60064c53
PH
541 def _signature_cache_id(self, example_sig):
542 """ Return a string representation of a signature """
78caa52a 543 return '.'.join(compat_str(len(part)) for part in example_sig.split('.'))
60064c53
PH
544
545 def _extract_signature_function(self, video_id, player_url, example_sig):
cf010131 546 id_m = re.match(
60620368 547 r'.*?-(?P<id>[a-zA-Z0-9_-]+)(?:/watch_as3|/html5player)?\.(?P<ext>[a-z]+)$',
cf010131 548 player_url)
c081b35c
PH
549 if not id_m:
550 raise ExtractorError('Cannot identify player %r' % player_url)
e0df6211
PH
551 player_type = id_m.group('ext')
552 player_id = id_m.group('id')
553
c4417ddb 554 # Read from filesystem cache
60064c53
PH
555 func_id = '%s_%s_%s' % (
556 player_type, player_id, self._signature_cache_id(example_sig))
c4417ddb 557 assert os.path.basename(func_id) == func_id
a0e07d31 558
69ea8ca4 559 cache_spec = self._downloader.cache.load('youtube-sigfuncs', func_id)
a0e07d31 560 if cache_spec is not None:
78caa52a 561 return lambda s: ''.join(s[i] for i in cache_spec)
83799698 562
6d1a55a5
PH
563 download_note = (
564 'Downloading player %s' % player_url
565 if self._downloader.params.get('verbose') else
566 'Downloading %s player %s' % (player_type, player_id)
567 )
e0df6211
PH
568 if player_type == 'js':
569 code = self._download_webpage(
570 player_url, video_id,
6d1a55a5 571 note=download_note,
69ea8ca4 572 errnote='Download of %s failed' % player_url)
83799698 573 res = self._parse_sig_js(code)
c4417ddb 574 elif player_type == 'swf':
e0df6211
PH
575 urlh = self._request_webpage(
576 player_url, video_id,
6d1a55a5 577 note=download_note,
69ea8ca4 578 errnote='Download of %s failed' % player_url)
e0df6211 579 code = urlh.read()
83799698 580 res = self._parse_sig_swf(code)
e0df6211
PH
581 else:
582 assert False, 'Invalid player type %r' % player_type
583
785521bf
PH
584 test_string = ''.join(map(compat_chr, range(len(example_sig))))
585 cache_res = res(test_string)
586 cache_spec = [ord(c) for c in cache_res]
83799698 587
69ea8ca4 588 self._downloader.cache.store('youtube-sigfuncs', func_id, cache_spec)
83799698
PH
589 return res
590
60064c53 591 def _print_sig_code(self, func, example_sig):
edf3e38e
PH
592 def gen_sig_code(idxs):
593 def _genslice(start, end, step):
78caa52a 594 starts = '' if start == 0 else str(start)
8bcc8756 595 ends = (':%d' % (end + step)) if end + step >= 0 else ':'
69ea8ca4 596 steps = '' if step == 1 else (':%d' % step)
78caa52a 597 return 's[%s%s%s]' % (starts, ends, steps)
edf3e38e
PH
598
599 step = None
7af808a5
PH
600 # Quelch pyflakes warnings - start will be set when step is set
601 start = '(Never used)'
edf3e38e
PH
602 for i, prev in zip(idxs[1:], idxs[:-1]):
603 if step is not None:
604 if i - prev == step:
605 continue
606 yield _genslice(start, prev, step)
607 step = None
608 continue
609 if i - prev in [-1, 1]:
610 step = i - prev
611 start = prev
612 continue
613 else:
78caa52a 614 yield 's[%d]' % prev
edf3e38e 615 if step is None:
78caa52a 616 yield 's[%d]' % i
edf3e38e
PH
617 else:
618 yield _genslice(start, i, step)
619
78caa52a 620 test_string = ''.join(map(compat_chr, range(len(example_sig))))
c705320f 621 cache_res = func(test_string)
edf3e38e 622 cache_spec = [ord(c) for c in cache_res]
78caa52a 623 expr_code = ' + '.join(gen_sig_code(cache_spec))
60064c53
PH
624 signature_id_tuple = '(%s)' % (
625 ', '.join(compat_str(len(p)) for p in example_sig.split('.')))
69ea8ca4 626 code = ('if tuple(len(p) for p in s.split(\'.\')) == %s:\n'
78caa52a 627 ' return %s\n') % (signature_id_tuple, expr_code)
69ea8ca4 628 self.to_screen('Extracted signature function:\n' + code)
edf3e38e 629
e0df6211
PH
630 def _parse_sig_js(self, jscode):
631 funcname = self._search_regex(
aaeb86f6 632 r'\.sig\|\|([a-zA-Z0-9$]+)\(', jscode,
8bcc8756 633 'Initial JS player signature function name')
2b25cb5d
PH
634
635 jsi = JSInterpreter(jscode)
636 initial_function = jsi.extract_function(funcname)
e0df6211
PH
637 return lambda s: initial_function([s])
638
639 def _parse_sig_swf(self, file_contents):
54256267 640 swfi = SWFInterpreter(file_contents)
78caa52a 641 TARGET_CLASSNAME = 'SignatureDecipher'
54256267 642 searched_class = swfi.extract_class(TARGET_CLASSNAME)
78caa52a 643 initial_function = swfi.extract_function(searched_class, 'decipher')
e0df6211
PH
644 return lambda s: initial_function([s])
645
83799698 646 def _decrypt_signature(self, s, video_id, player_url, age_gate=False):
257a2501 647 """Turn the encrypted s field into a working signature"""
6b37f0be 648
c8bf86d5 649 if player_url is None:
69ea8ca4 650 raise ExtractorError('Cannot decrypt signature without player_url')
920de7a2 651
69ea8ca4 652 if player_url.startswith('//'):
78caa52a 653 player_url = 'https:' + player_url
c8bf86d5 654 try:
62af3a0e 655 player_id = (player_url, self._signature_cache_id(s))
c8bf86d5
PH
656 if player_id not in self._player_cache:
657 func = self._extract_signature_function(
60064c53 658 video_id, player_url, s
c8bf86d5
PH
659 )
660 self._player_cache[player_id] = func
661 func = self._player_cache[player_id]
662 if self._downloader.params.get('youtube_print_sig_code'):
60064c53 663 self._print_sig_code(func, s)
c8bf86d5
PH
664 return func(s)
665 except Exception as e:
666 tb = traceback.format_exc()
667 raise ExtractorError(
78caa52a 668 'Signature extraction failed: ' + tb, cause=e)
e0df6211 669
360e1ca5 670 def _get_subtitles(self, video_id, webpage):
de7f3446 671 try:
60e47a26 672 subs_doc = self._download_xml(
38c2e5b8 673 'https://video.google.com/timedtext?hl=en&type=list&v=%s' % video_id,
7fad1c63
JMF
674 video_id, note=False)
675 except ExtractorError as err:
69ea8ca4 676 self._downloader.report_warning('unable to download video subtitles: %s' % compat_str(err))
de7f3446 677 return {}
de7f3446
JMF
678
679 sub_lang_list = {}
60e47a26
JMF
680 for track in subs_doc.findall('track'):
681 lang = track.attrib['lang_code']
7e660ac1
LD
682 if lang in sub_lang_list:
683 continue
360e1ca5
JMF
684 sub_formats = []
685 for ext in ['sbv', 'vtt', 'srt']:
686 params = compat_urllib_parse.urlencode({
687 'lang': lang,
688 'v': video_id,
689 'fmt': ext,
690 'name': track.attrib['name'].encode('utf-8'),
691 })
692 sub_formats.append({
693 'url': 'https://www.youtube.com/api/timedtext?' + params,
694 'ext': ext,
695 })
696 sub_lang_list[lang] = sub_formats
de7f3446 697 if not sub_lang_list:
69ea8ca4 698 self._downloader.report_warning('video doesn\'t have subtitles')
de7f3446
JMF
699 return {}
700 return sub_lang_list
701
360e1ca5 702 def _get_automatic_captions(self, video_id, webpage):
de7f3446
JMF
703 """We need the webpage for getting the captions url, pass it as an
704 argument to speed up the process."""
69ea8ca4 705 self.to_screen('%s: Looking for automatic captions' % video_id)
de7f3446 706 mobj = re.search(r';ytplayer.config = ({.*?});', webpage)
78caa52a 707 err_msg = 'Couldn\'t find automatic captions for %s' % video_id
de7f3446
JMF
708 if mobj is None:
709 self._downloader.report_warning(err_msg)
710 return {}
711 player_config = json.loads(mobj.group(1))
712 try:
0792d563
PH
713 args = player_config['args']
714 caption_url = args['ttsurl']
715 timestamp = args['timestamp']
055e6f36
JMF
716 # We get the available subtitles
717 list_params = compat_urllib_parse.urlencode({
718 'type': 'list',
719 'tlangs': 1,
720 'asrs': 1,
de7f3446 721 })
055e6f36 722 list_url = caption_url + '&' + list_params
e26f8712 723 caption_list = self._download_xml(list_url, video_id)
e3dc22ca 724 original_lang_node = caption_list.find('track')
7d900ef1 725 if original_lang_node is None:
69ea8ca4 726 self._downloader.report_warning('Video doesn\'t have automatic captions')
e3dc22ca
JMF
727 return {}
728 original_lang = original_lang_node.attrib['lang_code']
7d900ef1 729 caption_kind = original_lang_node.attrib.get('kind', '')
055e6f36
JMF
730
731 sub_lang_list = {}
732 for lang_node in caption_list.findall('target'):
733 sub_lang = lang_node.attrib['lang_code']
360e1ca5
JMF
734 sub_formats = []
735 for ext in ['sbv', 'vtt', 'srt']:
736 params = compat_urllib_parse.urlencode({
737 'lang': original_lang,
738 'tlang': sub_lang,
739 'fmt': ext,
740 'ts': timestamp,
741 'kind': caption_kind,
742 })
743 sub_formats.append({
744 'url': caption_url + '&' + params,
745 'ext': ext,
746 })
747 sub_lang_list[sub_lang] = sub_formats
055e6f36 748 return sub_lang_list
de7f3446
JMF
749 # An extractor error can be raise by the download process if there are
750 # no automatic captions but there are subtitles
751 except (KeyError, ExtractorError):
752 self._downloader.report_warning(err_msg)
753 return {}
754
97665381
PH
755 @classmethod
756 def extract_id(cls, url):
757 mobj = re.match(cls._VALID_URL, url, re.VERBOSE)
c5e8d7af 758 if mobj is None:
69ea8ca4 759 raise ExtractorError('Invalid URL: %s' % url)
c5e8d7af
PH
760 video_id = mobj.group(2)
761 return video_id
762
1d043b93
JMF
763 def _extract_from_m3u8(self, manifest_url, video_id):
764 url_map = {}
5f6a1245 765
1d043b93
JMF
766 def _get_urls(_manifest):
767 lines = _manifest.split('\n')
768 urls = filter(lambda l: l and not l.startswith('#'),
8bcc8756 769 lines)
1d043b93 770 return urls
78caa52a 771 manifest = self._download_webpage(manifest_url, video_id, 'Downloading formats manifest')
1d043b93
JMF
772 formats_urls = _get_urls(manifest)
773 for format_url in formats_urls:
890f62e8 774 itag = self._search_regex(r'itag/(\d+?)/', format_url, 'itag')
1d043b93
JMF
775 url_map[itag] = format_url
776 return url_map
777
1fb07d10
JG
778 def _extract_annotations(self, video_id):
779 url = 'https://www.youtube.com/annotations_invideo?features=1&legacy=1&video_id=%s' % video_id
69ea8ca4 780 return self._download_webpage(url, video_id, note='Searching for annotations.', errnote='Unable to download video annotations.')
1fb07d10 781
da276600
PH
782 def _parse_dash_manifest(
783 self, video_id, dash_manifest_url, player_url, age_gate):
774e208f
PH
784 def decrypt_sig(mobj):
785 s = mobj.group(1)
786 dec_s = self._decrypt_signature(s, video_id, player_url, age_gate)
787 return '/signature/%s' % dec_s
788 dash_manifest_url = re.sub(r'/s/([\w\.]+)', decrypt_sig, dash_manifest_url)
789 dash_doc = self._download_xml(
790 dash_manifest_url, video_id,
791 note='Downloading DASH manifest',
792 errnote='Could not download DASH manifest')
793
794 formats = []
de5c5456
YCH
795 for a in dash_doc.findall('.//{urn:mpeg:DASH:schema:MPD:2011}AdaptationSet'):
796 mime_type = a.attrib.get('mimeType')
797 for r in a.findall('{urn:mpeg:DASH:schema:MPD:2011}Representation'):
798 url_el = r.find('{urn:mpeg:DASH:schema:MPD:2011}BaseURL')
799 if url_el is None:
800 continue
801 if mime_type == 'text/vtt':
802 # TODO implement WebVTT downloading
803 pass
804 elif mime_type.startswith('audio/') or mime_type.startswith('video/'):
6800d337 805 segment_list = r.find('{urn:mpeg:DASH:schema:MPD:2011}SegmentList')
de5c5456
YCH
806 format_id = r.attrib['id']
807 video_url = url_el.text
808 filesize = int_or_none(url_el.attrib.get('{http://youtube.com/yt/2012/10/10}contentLength'))
809 f = {
810 'format_id': format_id,
811 'url': video_url,
812 'width': int_or_none(r.attrib.get('width')),
813 'height': int_or_none(r.attrib.get('height')),
814 'tbr': int_or_none(r.attrib.get('bandwidth'), 1000),
815 'asr': int_or_none(r.attrib.get('audioSamplingRate')),
816 'filesize': filesize,
817 'fps': int_or_none(r.attrib.get('frameRate')),
818 }
6800d337
YCH
819 if segment_list:
820 f.update({
821 'initialization_url': segment_list.find('{urn:mpeg:DASH:schema:MPD:2011}Initialization').attrib['sourceURL'],
822 'segment_urls': [segment.attrib.get('media') for segment in segment_list.findall('{urn:mpeg:DASH:schema:MPD:2011}SegmentURL')]
823 })
de5c5456
YCH
824 try:
825 existing_format = next(
826 fo for fo in formats
827 if fo['format_id'] == format_id)
828 except StopIteration:
829 full_info = self._formats.get(format_id, {}).copy()
830 full_info.update(f)
831 formats.append(full_info)
832 else:
833 existing_format.update(f)
834 else:
835 self.report_warning('Unknown MIME type %s in DASH manifest' % mime_type)
774e208f
PH
836 return formats
837
c5e8d7af 838 def _real_extract(self, url):
7e8c0af0 839 proto = (
78caa52a
PH
840 'http' if self._downloader.params.get('prefer_insecure', False)
841 else 'https')
7e8c0af0 842
c5e8d7af
PH
843 # Extract original video URL from URL with redirection, like age verification, using next_url parameter
844 mobj = re.search(self._NEXT_URL_RE, url)
845 if mobj:
7e8c0af0 846 url = proto + '://www.youtube.com/' + compat_urllib_parse.unquote(mobj.group(1)).lstrip('/')
97665381 847 video_id = self.extract_id(url)
c5e8d7af
PH
848
849 # Get video webpage
aa79ac0c 850 url = proto + '://www.youtube.com/watch?v=%s&gl=US&hl=en&has_verified=1&bpctr=9999999999' % video_id
a1f934b1 851 video_webpage = self._download_webpage(url, video_id)
c5e8d7af
PH
852
853 # Attempt to extract SWF player URL
e0df6211 854 mobj = re.search(r'swfConfig.*?"(https?:\\/\\/.*?watch.*?-.*?\.swf)"', video_webpage)
c5e8d7af
PH
855 if mobj is not None:
856 player_url = re.sub(r'\\(.)', r'\1', mobj.group(1))
857 else:
858 player_url = None
859
860 # Get video info
6449cd80 861 embed_webpage = None
c108eb73 862 if re.search(r'player-age-gate-content">', video_webpage) is not None:
c108eb73
JMF
863 age_gate = True
864 # We simulate the access to the video from www.youtube.com/v/{video_id}
865 # this can be viewed without login into Youtube
beb95e77
CL
866 url = proto + '://www.youtube.com/embed/%s' % video_id
867 embed_webpage = self._download_webpage(url, video_id, 'Downloading embed webpage')
2c57c7fa
JMF
868 data = compat_urllib_parse.urlencode({
869 'video_id': video_id,
870 'eurl': 'https://youtube.googleapis.com/v/' + video_id,
c084c934 871 'sts': self._search_regex(
beb95e77 872 r'"sts"\s*:\s*(\d+)', embed_webpage, 'sts', default=''),
2c57c7fa 873 })
7e8c0af0 874 video_info_url = proto + '://www.youtube.com/get_video_info?' + data
94bd3613
PH
875 video_info_webpage = self._download_webpage(
876 video_info_url, video_id,
20436c30 877 note='Refetching age-gated info webpage',
94bd3613 878 errnote='unable to download video info webpage')
c5e8d7af 879 video_info = compat_parse_qs(video_info_webpage)
c108eb73
JMF
880 else:
881 age_gate = False
4e62ebe2
JMF
882 try:
883 # Try looking directly into the video webpage
884 mobj = re.search(r';ytplayer\.config\s*=\s*({.*?});', video_webpage)
885 if not mobj:
886 raise ValueError('Could not find ytplayer.config') # caught below
887 json_code = uppercase_escape(mobj.group(1))
888 ytplayer_config = json.loads(json_code)
889 args = ytplayer_config['args']
890 # Convert to the same format returned by compat_parse_qs
891 video_info = dict((k, [v]) for k, v in args.items())
e40bd5f0 892 if not args.get('url_encoded_fmt_stream_map'):
4e62ebe2
JMF
893 raise ValueError('No stream_map present') # caught below
894 except ValueError:
895 # We fallback to the get_video_info pages (used by the embed page)
896 self.report_video_info_webpage_download(video_id)
897 for el_type in ['&el=embedded', '&el=detailpage', '&el=vevo', '']:
810fb84d
PH
898 video_info_url = (
899 '%s://www.youtube.com/get_video_info?&video_id=%s%s&ps=default&eurl=&gl=US&hl=en'
900 % (proto, video_id, el_type))
901 video_info_webpage = self._download_webpage(
902 video_info_url,
4e62ebe2
JMF
903 video_id, note=False,
904 errnote='unable to download video info webpage')
905 video_info = compat_parse_qs(video_info_webpage)
906 if 'token' in video_info:
907 break
c5e8d7af
PH
908 if 'token' not in video_info:
909 if 'reason' in video_info:
d11271dd 910 raise ExtractorError(
78caa52a 911 'YouTube said: %s' % video_info['reason'][0],
d11271dd 912 expected=True, video_id=video_id)
c5e8d7af 913 else:
d11271dd 914 raise ExtractorError(
78caa52a 915 '"token" parameter not in video info for unknown reason',
d11271dd 916 video_id=video_id)
c5e8d7af 917
1d699755
PH
918 if 'view_count' in video_info:
919 view_count = int(video_info['view_count'][0])
920 else:
921 view_count = None
922
c5e8d7af
PH
923 # Check for "rental" videos
924 if 'ypc_video_rental_bar_text' in video_info and 'author' not in video_info:
69ea8ca4 925 raise ExtractorError('"rental" videos not supported')
c5e8d7af
PH
926
927 # Start extracting information
928 self.report_information_extraction(video_id)
929
930 # uploader
931 if 'author' not in video_info:
69ea8ca4 932 raise ExtractorError('Unable to extract uploader name')
c5e8d7af
PH
933 video_uploader = compat_urllib_parse.unquote_plus(video_info['author'][0])
934
935 # uploader_id
936 video_uploader_id = None
937 mobj = re.search(r'<link itemprop="url" href="http://www.youtube.com/(?:user|channel)/([^"]+)">', video_webpage)
938 if mobj is not None:
939 video_uploader_id = mobj.group(1)
940 else:
69ea8ca4 941 self._downloader.report_warning('unable to extract uploader nickname')
c5e8d7af
PH
942
943 # title
a8c6b241 944 if 'title' in video_info:
aa92f063 945 video_title = video_info['title'][0]
a8c6b241 946 else:
69ea8ca4 947 self._downloader.report_warning('Unable to extract video title')
78caa52a 948 video_title = '_'
c5e8d7af
PH
949
950 # thumbnail image
7763b04e
JMF
951 # We try first to get a high quality image:
952 m_thumb = re.search(r'<span itemprop="thumbnail".*?href="(.*?)">',
953 video_webpage, re.DOTALL)
954 if m_thumb is not None:
955 video_thumbnail = m_thumb.group(1)
956 elif 'thumbnail_url' not in video_info:
69ea8ca4 957 self._downloader.report_warning('unable to extract video thumbnail')
f490e77e 958 video_thumbnail = None
c5e8d7af
PH
959 else: # don't panic if we can't find it
960 video_thumbnail = compat_urllib_parse.unquote_plus(video_info['thumbnail_url'][0])
961
962 # upload date
963 upload_date = None
ad3bc6ac 964 mobj = re.search(r'(?s)id="eow-date.*?>(.*?)</span>', video_webpage)
beee53de
PH
965 if mobj is None:
966 mobj = re.search(
263bd4ec 967 r'(?s)id="watch-uploader-info".*?>.*?(?:Published|Uploaded|Streamed live) on (.*?)</strong>',
beee53de 968 video_webpage)
c5e8d7af
PH
969 if mobj is not None:
970 upload_date = ' '.join(re.sub(r'[/,-]', r' ', mobj.group(1)).split())
971 upload_date = unified_strdate(upload_date)
972
55f7bd2d
PH
973 m_cat_container = self._search_regex(
974 r'(?s)<h4[^>]*>\s*Category\s*</h4>\s*<ul[^>]*>(.*?)</ul>',
624dcebf 975 video_webpage, 'categories', default=None)
ec8deefc 976 if m_cat_container:
ad3bc6ac 977 category = self._html_search_regex(
01ed5c9b 978 r'(?s)<a[^<]+>(.*?)</a>', m_cat_container, 'category',
ad3bc6ac
PH
979 default=None)
980 video_categories = None if category is None else [category]
981 else:
982 video_categories = None
ec8deefc 983
c5e8d7af
PH
984 # description
985 video_description = get_element_by_id("eow-description", video_webpage)
986 if video_description:
27dcce19
PH
987 video_description = re.sub(r'''(?x)
988 <a\s+
989 (?:[a-zA-Z-]+="[^"]+"\s+)*?
990 title="([^"]+)"\s+
991 (?:[a-zA-Z-]+="[^"]+"\s+)*?
992 class="yt-uix-redirect-link"\s*>
993 [^<]+
994 </a>
995 ''', r'\1', video_description)
c5e8d7af
PH
996 video_description = clean_html(video_description)
997 else:
998 fd_mobj = re.search(r'<meta name="description" content="([^"]+)"', video_webpage)
999 if fd_mobj:
1000 video_description = unescapeHTML(fd_mobj.group(1))
1001 else:
78caa52a 1002 video_description = ''
c5e8d7af 1003
f30a38be 1004 def _extract_count(count_name):
46374a56 1005 count = self._search_regex(
f30a38be
JMF
1006 r'id="watch-%s"[^>]*>.*?([\d,]+)\s*</span>' % re.escape(count_name),
1007 video_webpage, count_name, default=None)
336c3a69
JMF
1008 if count is not None:
1009 return int(count.replace(',', ''))
1010 return None
69ea8ca4
PH
1011 like_count = _extract_count('like')
1012 dislike_count = _extract_count('dislike')
336c3a69 1013
c5e8d7af 1014 # subtitles
d82134c3 1015 video_subtitles = self.extract_subtitles(video_id, video_webpage)
360e1ca5 1016 automatic_captions = self.extract_automatic_captions(video_id, video_webpage)
c5e8d7af
PH
1017
1018 if 'length_seconds' not in video_info:
69ea8ca4 1019 self._downloader.report_warning('unable to extract video duration')
b466b702 1020 video_duration = None
c5e8d7af 1021 else:
b466b702 1022 video_duration = int(compat_urllib_parse.unquote_plus(video_info['length_seconds'][0]))
c5e8d7af 1023
1fb07d10
JG
1024 # annotations
1025 video_annotations = None
1026 if self._downloader.params.get('writeannotations', False):
5f6a1245 1027 video_annotations = self._extract_annotations(video_id)
1fb07d10 1028
dd27fd17
PH
1029 def _map_to_format_list(urlmap):
1030 formats = []
1031 for itag, video_real_url in urlmap.items():
1032 dct = {
1033 'format_id': itag,
1034 'url': video_real_url,
1035 'player_url': player_url,
1036 }
0b65e5d4
PH
1037 if itag in self._formats:
1038 dct.update(self._formats[itag])
dd27fd17
PH
1039 formats.append(dct)
1040 return formats
1041
c5e8d7af
PH
1042 if 'conn' in video_info and video_info['conn'][0].startswith('rtmp'):
1043 self.report_rtmp_download()
dd27fd17
PH
1044 formats = [{
1045 'format_id': '_rtmp',
1046 'protocol': 'rtmp',
1047 'url': video_info['conn'][0],
1048 'player_url': player_url,
1049 }]
24270b03 1050 elif len(video_info.get('url_encoded_fmt_stream_map', [''])[0]) >= 1 or len(video_info.get('adaptive_fmts', [''])[0]) >= 1:
5f6a1245 1051 encoded_url_map = video_info.get('url_encoded_fmt_stream_map', [''])[0] + ',' + video_info.get('adaptive_fmts', [''])[0]
00fe14fc 1052 if 'rtmpe%3Dyes' in encoded_url_map:
a7055eb9 1053 raise ExtractorError('rtmpe downloads are not supported, see https://github.com/rg3/youtube-dl/issues/343 for more information.', expected=True)
c5e8d7af 1054 url_map = {}
00fe14fc 1055 for url_data_str in encoded_url_map.split(','):
c5e8d7af 1056 url_data = compat_parse_qs(url_data_str)
201e9eaa
PH
1057 if 'itag' not in url_data or 'url' not in url_data:
1058 continue
1059 format_id = url_data['itag'][0]
1060 url = url_data['url'][0]
1061
1062 if 'sig' in url_data:
1063 url += '&signature=' + url_data['sig'][0]
1064 elif 's' in url_data:
1065 encrypted_sig = url_data['s'][0]
6449cd80 1066 ASSETS_RE = r'"assets":.+?"js":\s*("[^"]+")'
201e9eaa 1067
beb95e77 1068 jsplayer_url_json = self._search_regex(
6449cd80
PH
1069 ASSETS_RE,
1070 embed_webpage if age_gate else video_webpage,
1071 'JS player URL (1)', default=None)
1072 if not jsplayer_url_json and not age_gate:
1073 # We need the embed website after all
1074 if embed_webpage is None:
1075 embed_url = proto + '://www.youtube.com/embed/%s' % video_id
1076 embed_webpage = self._download_webpage(
1077 embed_url, video_id, 'Downloading embed webpage')
1078 jsplayer_url_json = self._search_regex(
1079 ASSETS_RE, embed_webpage, 'JS player URL')
1080
beb95e77 1081 player_url = json.loads(jsplayer_url_json)
201e9eaa
PH
1082 if player_url is None:
1083 player_url_json = self._search_regex(
1084 r'ytplayer\.config.*?"url"\s*:\s*("[^"]+")',
78caa52a 1085 video_webpage, 'age gate player URL')
201e9eaa
PH
1086 player_url = json.loads(player_url_json)
1087
1088 if self._downloader.params.get('verbose'):
cf010131 1089 if player_url is None:
201e9eaa
PH
1090 player_version = 'unknown'
1091 player_desc = 'unknown'
1092 else:
1093 if player_url.endswith('swf'):
1094 player_version = self._search_regex(
1095 r'-(.+?)(?:/watch_as3)?\.swf$', player_url,
78caa52a 1096 'flash player', fatal=False)
201e9eaa 1097 player_desc = 'flash player %s' % player_version
cf010131 1098 else:
201e9eaa
PH
1099 player_version = self._search_regex(
1100 r'html5player-([^/]+?)(?:/html5player)?\.js',
1101 player_url,
1102 'html5 player', fatal=False)
78caa52a 1103 player_desc = 'html5 player %s' % player_version
201e9eaa 1104
60064c53 1105 parts_sizes = self._signature_cache_id(encrypted_sig)
69ea8ca4 1106 self.to_screen('{%s} signature length %s, %s' %
9e1a5b84 1107 (format_id, parts_sizes, player_desc))
201e9eaa
PH
1108
1109 signature = self._decrypt_signature(
1110 encrypted_sig, video_id, player_url, age_gate)
1111 url += '&signature=' + signature
1112 if 'ratebypass' not in url:
1113 url += '&ratebypass=yes'
1114 url_map[format_id] = url
dd27fd17 1115 formats = _map_to_format_list(url_map)
1d043b93
JMF
1116 elif video_info.get('hlsvp'):
1117 manifest_url = video_info['hlsvp'][0]
1118 url_map = self._extract_from_m3u8(manifest_url, video_id)
dd27fd17 1119 formats = _map_to_format_list(url_map)
c5e8d7af 1120 else:
69ea8ca4 1121 raise ExtractorError('no conn, hlsvp or url_encoded_fmt_stream_map information found in video info')
c5e8d7af 1122
dd27fd17 1123 # Look for the DASH manifest
203fb43f 1124 if self._downloader.params.get('youtube_include_dash_manifest', True):
774e208f 1125 dash_mpd = video_info.get('dashmpd')
75111274 1126 if dash_mpd:
774e208f
PH
1127 dash_manifest_url = dash_mpd[0]
1128 try:
1129 dash_formats = self._parse_dash_manifest(
da276600 1130 video_id, dash_manifest_url, player_url, age_gate)
774e208f
PH
1131 except (ExtractorError, KeyError) as e:
1132 self.report_warning(
1133 'Skipping DASH manifest: %r' % e, video_id)
1134 else:
04b3b3df
JMF
1135 # Remove the formats we found through non-DASH, they
1136 # contain less info and it can be wrong, because we use
1137 # fixed values (for example the resolution). See
1138 # https://github.com/rg3/youtube-dl/issues/5774 for an
1139 # example.
e65566a9 1140 dash_keys = set(df['format_id'] for df in dash_formats)
04b3b3df 1141 formats = [f for f in formats if f['format_id'] not in dash_keys]
774e208f 1142 formats.extend(dash_formats)
d80044c2 1143
6271f1ca
PH
1144 # Check for malformed aspect ratio
1145 stretched_m = re.search(
1146 r'<meta\s+property="og:video:tag".*?content="yt:stretch=(?P<w>[0-9]+):(?P<h>[0-9]+)">',
1147 video_webpage)
1148 if stretched_m:
1149 ratio = float(stretched_m.group('w')) / float(stretched_m.group('h'))
1150 for f in formats:
1151 if f.get('vcodec') != 'none':
1152 f['stretched_ratio'] = ratio
1153
4bcc7bd1 1154 self._sort_formats(formats)
4ea3be0a 1155
1156 return {
8bcc8756
JW
1157 'id': video_id,
1158 'uploader': video_uploader,
1159 'uploader_id': video_uploader_id,
1160 'upload_date': upload_date,
1161 'title': video_title,
1162 'thumbnail': video_thumbnail,
1163 'description': video_description,
1164 'categories': video_categories,
1165 'subtitles': video_subtitles,
360e1ca5 1166 'automatic_captions': automatic_captions,
8bcc8756
JW
1167 'duration': video_duration,
1168 'age_limit': 18 if age_gate else 0,
1169 'annotations': video_annotations,
7e8c0af0 1170 'webpage_url': proto + '://www.youtube.com/watch?v=%s' % video_id,
8bcc8756 1171 'view_count': view_count,
4ea3be0a 1172 'like_count': like_count,
1173 'dislike_count': dislike_count,
2d30521a 1174 'average_rating': float_or_none(video_info.get('avg_rating', [None])[0]),
8bcc8756 1175 'formats': formats,
4ea3be0a 1176 }
c5e8d7af 1177
5f6a1245 1178
880e1c52 1179class YoutubePlaylistIE(YoutubeBaseInfoExtractor):
78caa52a 1180 IE_DESC = 'YouTube.com playlists'
d67cc9fa 1181 _VALID_URL = r"""(?x)(?:
c5e8d7af
PH
1182 (?:https?://)?
1183 (?:\w+\.)?
1184 youtube\.com/
1185 (?:
ac7553d0 1186 (?:course|view_play_list|my_playlists|artist|playlist|watch|embed/videoseries)
c5e8d7af
PH
1187 \? (?:.*?&)*? (?:p|a|list)=
1188 | p/
1189 )
d67cc9fa 1190 (
99209c29 1191 (?:PL|LL|EC|UU|FL|RD|UL)?[0-9A-Za-z-_]{10,}
5f6a1245 1192 # Top tracks, they can also include dots
d67cc9fa
JMF
1193 |(?:MC)[\w\.]*
1194 )
c5e8d7af
PH
1195 .*
1196 |
99209c29 1197 ((?:PL|LL|EC|UU|FL|RD|UL)[0-9A-Za-z-_]{10,})
c5e8d7af 1198 )"""
dbb94fb0 1199 _TEMPLATE_URL = 'https://www.youtube.com/playlist?list=%s'
dbb94fb0 1200 _VIDEO_RE = r'href="\s*/watch\?v=(?P<id>[0-9A-Za-z_-]{11})&amp;[^"]*?index=(?P<index>\d+)'
78caa52a 1201 IE_NAME = 'youtube:playlist'
81127aa5
PH
1202 _TESTS = [{
1203 'url': 'https://www.youtube.com/playlist?list=PLwiyx1dc3P2JR9N8gQaQN_BCvlSlap7re',
1204 'info_dict': {
1205 'title': 'ytdl test PL',
a1cf99d0 1206 'id': 'PLwiyx1dc3P2JR9N8gQaQN_BCvlSlap7re',
81127aa5
PH
1207 },
1208 'playlist_count': 3,
9291475f
PH
1209 }, {
1210 'url': 'https://www.youtube.com/playlist?list=PLtPgu7CB4gbZDA7i_euNxn75ISqxwZPYx',
1211 'info_dict': {
acf757f4 1212 'id': 'PLtPgu7CB4gbZDA7i_euNxn75ISqxwZPYx',
9291475f
PH
1213 'title': 'YDL_Empty_List',
1214 },
1215 'playlist_count': 0,
1216 }, {
1217 'note': 'Playlist with deleted videos (#651). As a bonus, the video #51 is also twice in this list.',
1218 'url': 'https://www.youtube.com/playlist?list=PLwP_SiAcdui0KVebT0mU9Apz359a4ubsC',
1219 'info_dict': {
1220 'title': '29C3: Not my department',
acf757f4 1221 'id': 'PLwP_SiAcdui0KVebT0mU9Apz359a4ubsC',
9291475f
PH
1222 },
1223 'playlist_count': 95,
1224 }, {
1225 'note': 'issue #673',
1226 'url': 'PLBB231211A4F62143',
1227 'info_dict': {
f46a8702 1228 'title': '[OLD]Team Fortress 2 (Class-based LP)',
acf757f4 1229 'id': 'PLBB231211A4F62143',
9291475f
PH
1230 },
1231 'playlist_mincount': 26,
1232 }, {
1233 'note': 'Large playlist',
1234 'url': 'https://www.youtube.com/playlist?list=UUBABnxM4Ar9ten8Mdjj1j0Q',
1235 'info_dict': {
1236 'title': 'Uploads from Cauchemar',
acf757f4 1237 'id': 'UUBABnxM4Ar9ten8Mdjj1j0Q',
9291475f
PH
1238 },
1239 'playlist_mincount': 799,
1240 }, {
1241 'url': 'PLtPgu7CB4gbY9oDN3drwC3cMbJggS7dKl',
1242 'info_dict': {
1243 'title': 'YDL_safe_search',
acf757f4 1244 'id': 'PLtPgu7CB4gbY9oDN3drwC3cMbJggS7dKl',
9291475f
PH
1245 },
1246 'playlist_count': 2,
ac7553d0
PH
1247 }, {
1248 'note': 'embedded',
1249 'url': 'http://www.youtube.com/embed/videoseries?list=PL6IaIsEjSbf96XFRuNccS_RuEXwNdsoEu',
1250 'playlist_count': 4,
1251 'info_dict': {
1252 'title': 'JODA15',
acf757f4 1253 'id': 'PL6IaIsEjSbf96XFRuNccS_RuEXwNdsoEu',
ac7553d0 1254 }
6b08cdf6
PH
1255 }, {
1256 'note': 'Embedded SWF player',
1257 'url': 'http://www.youtube.com/p/YN5VISEtHet5D4NEvfTd0zcgFk84NqFZ?hl=en_US&fs=1&rel=0',
1258 'playlist_count': 4,
1259 'info_dict': {
1260 'title': 'JODA7',
acf757f4 1261 'id': 'YN5VISEtHet5D4NEvfTd0zcgFk84NqFZ',
6b08cdf6 1262 }
4b7df0d3
JMF
1263 }, {
1264 'note': 'Buggy playlist: the webpage has a "Load more" button but it doesn\'t have more videos',
1265 'url': 'https://www.youtube.com/playlist?list=UUXw-G3eDE9trcvY2sBMM_aA',
1266 'info_dict': {
acf757f4
PH
1267 'title': 'Uploads from Interstellar Movie',
1268 'id': 'UUXw-G3eDE9trcvY2sBMM_aA',
4b7df0d3
JMF
1269 },
1270 'playlist_mincout': 21,
81127aa5 1271 }]
c5e8d7af 1272
880e1c52
JMF
1273 def _real_initialize(self):
1274 self._login()
1275
652cdaa2 1276 def _extract_mix(self, playlist_id):
99209c29 1277 # The mixes are generated from a single video
652cdaa2 1278 # the id of the playlist is just 'RD' + video_id
7d4afc55 1279 url = 'https://youtube.com/watch?v=%s&list=%s' % (playlist_id[-11:], playlist_id)
c9cc0bf5 1280 webpage = self._download_webpage(
78caa52a 1281 url, playlist_id, 'Downloading Youtube mix')
bc2f773b 1282 search_title = lambda class_name: get_element_by_attribute('class', class_name, webpage)
c9cc0bf5
PH
1283 title_span = (
1284 search_title('playlist-title') or
1285 search_title('title long-title') or
1286 search_title('title'))
76d1700b 1287 title = clean_html(title_span)
c9cc0bf5
PH
1288 ids = orderedSet(re.findall(
1289 r'''(?xs)data-video-username=".*?".*?
1290 href="/watch\?v=([0-9A-Za-z_-]{11})&amp;[^"]*?list=%s''' % re.escape(playlist_id),
1291 webpage))
652cdaa2
JMF
1292 url_results = self._ids_to_results(ids)
1293
1294 return self.playlist_result(url_results, playlist_id, title)
1295
448830ce 1296 def _extract_playlist(self, playlist_id):
dbb94fb0
S
1297 url = self._TEMPLATE_URL % playlist_id
1298 page = self._download_webpage(url, playlist_id)
1299 more_widget_html = content_html = page
1300
39b62db1
YCH
1301 for match in re.findall(r'<div class="yt-alert-message">([^<]+)</div>', page):
1302 match = match.strip()
1303 # Check if the playlist exists or is private
1304 if re.match(r'[^<]*(The|This) playlist (does not exist|is private)[^<]*', match):
1305 raise ExtractorError(
1306 'The playlist doesn\'t exist or is private, use --username or '
1307 '--netrc to access it.',
1308 expected=True)
1309 elif re.match(r'[^<]*Invalid parameters[^<]*', match):
1310 raise ExtractorError(
1311 'Invalid parameters. Maybe URL is incorrect.',
1312 expected=True)
1313 elif re.match(r'[^<]*Choose your language[^<]*', match):
1314 continue
1315 else:
1316 self.report_warning('Youtube gives an alert message: ' + match)
10c0e2d8 1317
dcbb4580
JMF
1318 # Extract the video ids from the playlist pages
1319 ids = []
c5e8d7af 1320
755eb032 1321 for page_num in itertools.count(1):
dbb94fb0 1322 matches = re.finditer(self._VIDEO_RE, content_html)
6e47b51e
JMF
1323 # We remove the duplicates and the link with index 0
1324 # (it's not the first video of the playlist)
1325 new_ids = orderedSet(m.group('id') for m in matches if m.group('index') != '0')
dcbb4580 1326 ids.extend(new_ids)
c5e8d7af 1327
dbb94fb0
S
1328 mobj = re.search(r'data-uix-load-more-href="/?(?P<more>[^"]+)"', more_widget_html)
1329 if not mobj:
c5e8d7af
PH
1330 break
1331
dbb94fb0 1332 more = self._download_json(
5912c639
PH
1333 'https://youtube.com/%s' % mobj.group('more'), playlist_id,
1334 'Downloading page #%s' % page_num,
1335 transform_source=uppercase_escape)
dbb94fb0 1336 content_html = more['content_html']
4b7df0d3
JMF
1337 if not content_html.strip():
1338 # Some webpages show a "Load more" button but they don't
1339 # have more videos
1340 break
dbb94fb0
S
1341 more_widget_html = more['load_more_widget_html']
1342
1343 playlist_title = self._html_search_regex(
68eb8e90 1344 r'(?s)<h1 class="pl-header-title[^"]*">\s*(.*?)\s*</h1>',
78caa52a 1345 page, 'title')
c5e8d7af 1346
652cdaa2 1347 url_results = self._ids_to_results(ids)
dcbb4580 1348 return self.playlist_result(url_results, playlist_id, playlist_title)
c5e8d7af 1349
448830ce
S
1350 def _real_extract(self, url):
1351 # Extract playlist id
1352 mobj = re.match(self._VALID_URL, url)
1353 if mobj is None:
1354 raise ExtractorError('Invalid URL: %s' % url)
1355 playlist_id = mobj.group(1) or mobj.group(2)
1356
1357 # Check if it's a video-specific URL
1358 query_dict = compat_urlparse.parse_qs(compat_urlparse.urlparse(url).query)
1359 if 'v' in query_dict:
1360 video_id = query_dict['v'][0]
1361 if self._downloader.params.get('noplaylist'):
1362 self.to_screen('Downloading just video %s because of --no-playlist' % video_id)
1363 return self.url_result(video_id, 'Youtube', video_id=video_id)
1364 else:
1365 self.to_screen('Downloading playlist %s - add --no-playlist to just download video %s' % (playlist_id, video_id))
1366
1367 if playlist_id.startswith('RD') or playlist_id.startswith('UL'):
1368 # Mixes require a custom extraction process
1369 return self._extract_mix(playlist_id)
1370
1371 return self._extract_playlist(playlist_id)
1372
c5e8d7af
PH
1373
1374class YoutubeChannelIE(InfoExtractor):
78caa52a 1375 IE_DESC = 'YouTube.com channels'
9ff67727 1376 _VALID_URL = r'https?://(?:youtu\.be|(?:\w+\.)?youtube(?:-nocookie)?\.com)/channel/(?P<id>[0-9A-Za-z_-]+)'
eb0f3e7e 1377 _TEMPLATE_URL = 'https://www.youtube.com/channel/%s/videos'
78caa52a 1378 IE_NAME = 'youtube:channel'
cdc628a4
PH
1379 _TESTS = [{
1380 'note': 'paginated channel',
1381 'url': 'https://www.youtube.com/channel/UCKfVa3S1e4PHvxWcwyMMg8w',
1382 'playlist_mincount': 91,
acf757f4
PH
1383 'info_dict': {
1384 'id': 'UCKfVa3S1e4PHvxWcwyMMg8w',
1385 }
cdc628a4 1386 }]
c5e8d7af 1387
6de5dbaf
S
1388 @staticmethod
1389 def extract_videos_from_page(page):
c5e8d7af 1390 ids_in_page = []
fb69240c
S
1391 titles_in_page = []
1392 for mobj in re.finditer(r'(?:title="(?P<title>[^"]+)"[^>]+)?href="/watch\?v=(?P<id>[0-9A-Za-z_-]+)&?', page):
1393 video_id = mobj.group('id')
1394 video_title = unescapeHTML(mobj.group('title'))
1395 try:
1396 idx = ids_in_page.index(video_id)
1397 if video_title and not titles_in_page[idx]:
1398 titles_in_page[idx] = video_title
1399 except ValueError:
1400 ids_in_page.append(video_id)
1401 titles_in_page.append(video_title)
1402 return zip(ids_in_page, titles_in_page)
c5e8d7af
PH
1403
1404 def _real_extract(self, url):
9ff67727 1405 channel_id = self._match_id(url)
c5e8d7af 1406
eb0f3e7e 1407 url = self._TEMPLATE_URL % channel_id
386bdfa6
S
1408
1409 # Channel by page listing is restricted to 35 pages of 30 items, i.e. 1050 videos total (see #5778)
1410 # Workaround by extracting as a playlist if managed to obtain channel playlist URL
1411 # otherwise fallback on channel by page extraction
1412 channel_page = self._download_webpage(
1413 url + '?view=57', channel_id,
1414 'Downloading channel page', fatal=False)
1415 channel_playlist_id = self._search_regex(
1416 [r'<meta itemprop="channelId" content="([^"]+)">',
1417 r'data-channel-external-id="([^"]+)"'],
1418 channel_page, 'channel id', default=None)
1419 if channel_playlist_id and channel_playlist_id.startswith('UC'):
1420 playlist_id = 'UU' + channel_playlist_id[2:]
d2a9de78
IK
1421 return self.url_result(
1422 compat_urlparse.urljoin(url, '/playlist?list=%s' % playlist_id), 'YoutubePlaylist')
386bdfa6 1423
60bf45c8 1424 channel_page = self._download_webpage(url, channel_id, 'Downloading page #1')
31812a9e
PH
1425 autogenerated = re.search(r'''(?x)
1426 class="[^"]*?(?:
1427 channel-header-autogenerated-label|
1428 yt-channel-title-autogenerated
1429 )[^"]*"''', channel_page) is not None
c5e8d7af 1430
b9643eed
JMF
1431 if autogenerated:
1432 # The videos are contained in a single page
1433 # the ajax pages can't be used, they are empty
b82f815f 1434 entries = [
fb69240c
S
1435 self.url_result(
1436 video_id, 'Youtube', video_id=video_id,
1437 video_title=video_title)
8f02ad4f 1438 for video_id, video_title in self.extract_videos_from_page(channel_page)]
b82f815f
PH
1439 return self.playlist_result(entries, channel_id)
1440
1441 def _entries():
23d3608c 1442 more_widget_html = content_html = channel_page
b9643eed 1443 for pagenum in itertools.count(1):
81c2f20b 1444
8f02ad4f 1445 for video_id, video_title in self.extract_videos_from_page(content_html):
b82f815f 1446 yield self.url_result(
fb69240c
S
1447 video_id, 'Youtube', video_id=video_id,
1448 video_title=video_title)
5f6a1245 1449
23d3608c
JMF
1450 mobj = re.search(
1451 r'data-uix-load-more-href="/?(?P<more>[^"]+)"',
1452 more_widget_html)
1453 if not mobj:
b9643eed 1454 break
c5e8d7af 1455
23d3608c
JMF
1456 more = self._download_json(
1457 'https://youtube.com/%s' % mobj.group('more'), channel_id,
1458 'Downloading page #%s' % (pagenum + 1),
1459 transform_source=uppercase_escape)
1460 content_html = more['content_html']
1461 more_widget_html = more['load_more_widget_html']
1462
b82f815f 1463 return self.playlist_result(_entries(), channel_id)
c5e8d7af
PH
1464
1465
eb0f3e7e 1466class YoutubeUserIE(YoutubeChannelIE):
78caa52a 1467 IE_DESC = 'YouTube.com user videos (URL or "ytuser" keyword)'
9ff67727 1468 _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_-]+)'
eb0f3e7e 1469 _TEMPLATE_URL = 'https://www.youtube.com/user/%s/videos'
78caa52a 1470 IE_NAME = 'youtube:user'
c5e8d7af 1471
cdc628a4
PH
1472 _TESTS = [{
1473 'url': 'https://www.youtube.com/user/TheLinuxFoundation',
1474 'playlist_mincount': 320,
1475 'info_dict': {
1476 'title': 'TheLinuxFoundation',
1477 }
1478 }, {
1479 'url': 'ytuser:phihag',
1480 'only_matching': True,
1481 }]
1482
e3ea4790 1483 @classmethod
f4b05232 1484 def suitable(cls, url):
e3ea4790
JMF
1485 # Don't return True if the url can be extracted with other youtube
1486 # extractor, the regex would is too permissive and it would match.
1487 other_ies = iter(klass for (name, klass) in globals().items() if name.endswith('IE') and klass is not cls)
5f6a1245
JW
1488 if any(ie.suitable(url) for ie in other_ies):
1489 return False
1490 else:
1491 return super(YoutubeUserIE, cls).suitable(url)
f4b05232 1492
b05654f0 1493
b4c08069 1494class YoutubeSearchIE(SearchInfoExtractor, YoutubePlaylistIE):
78caa52a 1495 IE_DESC = 'YouTube.com searches'
b4c08069
JMF
1496 # there doesn't appear to be a real limit, for example if you search for
1497 # 'python' you get more than 8.000.000 results
1498 _MAX_RESULTS = float('inf')
78caa52a 1499 IE_NAME = 'youtube:search'
b05654f0 1500 _SEARCH_KEY = 'ytsearch'
b4c08069 1501 _EXTRA_QUERY_ARGS = {}
9dd8e46a 1502 _TESTS = []
b05654f0 1503
b05654f0
PH
1504 def _get_n_results(self, query, n):
1505 """Get a specified number of results for a query"""
1506
b4c08069 1507 videos = []
b05654f0
PH
1508 limit = n
1509
b4c08069
JMF
1510 for pagenum in itertools.count(1):
1511 url_query = {
1512 'search_query': query,
1513 'page': pagenum,
1514 'spf': 'navigate',
1515 }
1516 url_query.update(self._EXTRA_QUERY_ARGS)
1517 result_url = 'https://www.youtube.com/results?' + compat_urllib_parse.urlencode(url_query)
1518 data = self._download_json(
69ea8ca4 1519 result_url, video_id='query "%s"' % query,
b4c08069 1520 note='Downloading page %s' % pagenum,
69ea8ca4 1521 errnote='Unable to download API page')
b4c08069 1522 html_content = data[1]['body']['content']
7cc3570e 1523
b4c08069 1524 if 'class="search-message' in html_content:
07ad22b8 1525 raise ExtractorError(
78caa52a 1526 '[youtube] No video results', expected=True)
b05654f0 1527
b4c08069
JMF
1528 new_videos = self._ids_to_results(orderedSet(re.findall(
1529 r'href="/watch\?v=(.{11})', html_content)))
1530 videos += new_videos
1531 if not new_videos or len(videos) > limit:
1532 break
b05654f0 1533
b4c08069
JMF
1534 if len(videos) > n:
1535 videos = videos[:n]
b05654f0 1536 return self.playlist_result(videos, query)
75dff0ee 1537
c9ae7b95 1538
a3dd9248 1539class YoutubeSearchDateIE(YoutubeSearchIE):
cb7fb546 1540 IE_NAME = YoutubeSearchIE.IE_NAME + ':date'
a3dd9248 1541 _SEARCH_KEY = 'ytsearchdate'
78caa52a 1542 IE_DESC = 'YouTube.com searches, newest videos first'
b4c08069 1543 _EXTRA_QUERY_ARGS = {'search_sort': 'video_date_uploaded'}
75dff0ee 1544
c9ae7b95
PH
1545
1546class YoutubeSearchURLIE(InfoExtractor):
78caa52a
PH
1547 IE_DESC = 'YouTube.com search URLs'
1548 IE_NAME = 'youtube:search_url'
c9ae7b95 1549 _VALID_URL = r'https?://(?:www\.)?youtube\.com/results\?(.*?&)?search_query=(?P<query>[^&]+)(?:[&]|$)'
cdc628a4
PH
1550 _TESTS = [{
1551 'url': 'https://www.youtube.com/results?baz=bar&search_query=youtube-dl+test+video&filters=video&lclk=video',
1552 'playlist_mincount': 5,
1553 'info_dict': {
1554 'title': 'youtube-dl test video',
1555 }
1556 }]
c9ae7b95
PH
1557
1558 def _real_extract(self, url):
1559 mobj = re.match(self._VALID_URL, url)
1560 query = compat_urllib_parse.unquote_plus(mobj.group('query'))
1561
1562 webpage = self._download_webpage(url, query)
1563 result_code = self._search_regex(
98998cde 1564 r'(?s)<ol[^>]+class="item-section"(.*?)</ol>', webpage, 'result HTML')
c9ae7b95
PH
1565
1566 part_codes = re.findall(
1567 r'(?s)<h3 class="yt-lockup-title">(.*?)</h3>', result_code)
1568 entries = []
1569 for part_code in part_codes:
1570 part_title = self._html_search_regex(
6feb2d5e 1571 [r'(?s)title="([^"]+)"', r'>([^<]+)</a>'], part_code, 'item title', fatal=False)
c9ae7b95
PH
1572 part_url_snippet = self._html_search_regex(
1573 r'(?s)href="([^"]+)"', part_code, 'item URL')
1574 part_url = compat_urlparse.urljoin(
1575 'https://www.youtube.com/', part_url_snippet)
1576 entries.append({
1577 '_type': 'url',
1578 'url': part_url,
1579 'title': part_title,
1580 })
1581
1582 return {
1583 '_type': 'playlist',
1584 'entries': entries,
1585 'title': query,
1586 }
1587
1588
75dff0ee 1589class YoutubeShowIE(InfoExtractor):
78caa52a 1590 IE_DESC = 'YouTube.com (multi-season) shows'
cdc628a4 1591 _VALID_URL = r'https?://www\.youtube\.com/show/(?P<id>[^?#]*)'
78caa52a 1592 IE_NAME = 'youtube:show'
cdc628a4
PH
1593 _TESTS = [{
1594 'url': 'http://www.youtube.com/show/airdisasters',
1595 'playlist_mincount': 3,
1596 'info_dict': {
1597 'id': 'airdisasters',
1598 'title': 'Air Disasters',
1599 }
1600 }]
75dff0ee
JMF
1601
1602 def _real_extract(self, url):
1603 mobj = re.match(self._VALID_URL, url)
cdc628a4
PH
1604 playlist_id = mobj.group('id')
1605 webpage = self._download_webpage(
1606 url, playlist_id, 'Downloading show webpage')
75dff0ee
JMF
1607 # There's one playlist for each season of the show
1608 m_seasons = list(re.finditer(r'href="(/playlist\?list=.*?)"', webpage))
cdc628a4
PH
1609 self.to_screen('%s: Found %s seasons' % (playlist_id, len(m_seasons)))
1610 entries = [
1611 self.url_result(
1612 'https://www.youtube.com' + season.group(1), 'YoutubePlaylist')
1613 for season in m_seasons
1614 ]
1615 title = self._og_search_title(webpage, fatal=False)
1616
1617 return {
1618 '_type': 'playlist',
1619 'id': playlist_id,
1620 'title': title,
1621 'entries': entries,
1622 }
04cc9617
JMF
1623
1624
b2e8bc1b 1625class YoutubeFeedsInfoExtractor(YoutubeBaseInfoExtractor):
d7ae0639 1626 """
25f14e9f 1627 Base class for feed extractors
d7ae0639
JMF
1628 Subclasses must define the _FEED_NAME and _PLAYLIST_TITLE properties.
1629 """
b2e8bc1b 1630 _LOGIN_REQUIRED = True
d7ae0639
JMF
1631
1632 @property
1633 def IE_NAME(self):
78caa52a 1634 return 'youtube:%s' % self._FEED_NAME
04cc9617 1635
81f0259b 1636 def _real_initialize(self):
b2e8bc1b 1637 self._login()
81f0259b 1638
04cc9617 1639 def _real_extract(self, url):
25f14e9f
S
1640 page = self._download_webpage(
1641 'https://www.youtube.com/feed/%s' % self._FEED_NAME, self._PLAYLIST_TITLE)
2bc43303
JMF
1642
1643 # The extraction process is the same as for playlists, but the regex
1644 # for the video ids doesn't contain an index
1645 ids = []
1646 more_widget_html = content_html = page
2bc43303
JMF
1647 for page_num in itertools.count(1):
1648 matches = re.findall(r'href="\s*/watch\?v=([0-9A-Za-z_-]{11})', content_html)
62c95fd5
S
1649
1650 # 'recommended' feed has infinite 'load more' and each new portion spins
1651 # the same videos in (sometimes) slightly different order, so we'll check
1652 # for unicity and break when portion has no new videos
1653 new_ids = filter(lambda video_id: video_id not in ids, orderedSet(matches))
1654 if not new_ids:
1655 break
1656
2bc43303
JMF
1657 ids.extend(new_ids)
1658
1659 mobj = re.search(r'data-uix-load-more-href="/?(?P<more>[^"]+)"', more_widget_html)
1660 if not mobj:
1661 break
1662
1663 more = self._download_json(
25f14e9f 1664 'https://youtube.com/%s' % mobj.group('more'), self._PLAYLIST_TITLE,
2bc43303
JMF
1665 'Downloading page #%s' % page_num,
1666 transform_source=uppercase_escape)
1667 content_html = more['content_html']
1668 more_widget_html = more['load_more_widget_html']
1669
25f14e9f
S
1670 return self.playlist_result(
1671 self._ids_to_results(ids), playlist_title=self._PLAYLIST_TITLE)
1672
1673
1674class YoutubeWatchLaterIE(YoutubePlaylistIE):
1675 IE_NAME = 'youtube:watchlater'
1676 IE_DESC = 'Youtube watch later list, ":ytwatchlater" for short (requires authentication)'
1677 _VALID_URL = r'https?://www\.youtube\.com/(?:feed/watch_later|playlist\?list=WL)|:ytwatchlater'
1678
1679 _TESTS = [] # override PlaylistIE tests
1680
1681 def _real_extract(self, url):
1682 return self._extract_playlist('WL')
f459d170 1683
5f6a1245 1684
c626a3d9 1685class YoutubeFavouritesIE(YoutubeBaseInfoExtractor):
78caa52a 1686 IE_NAME = 'youtube:favorites'
f3a34072 1687 IE_DESC = 'YouTube.com favourite videos, ":ytfav" for short (requires authentication)'
c7a7750d 1688 _VALID_URL = r'https?://www\.youtube\.com/my_favorites|:ytfav(?:ou?rites)?'
c626a3d9
JMF
1689 _LOGIN_REQUIRED = True
1690
1691 def _real_extract(self, url):
1692 webpage = self._download_webpage('https://www.youtube.com/my_favorites', 'Youtube Favourites videos')
78caa52a 1693 playlist_id = self._search_regex(r'list=(.+?)["&]', webpage, 'favourites playlist id')
c626a3d9 1694 return self.url_result(playlist_id, 'YoutubePlaylist')
15870e90
PH
1695
1696
25f14e9f
S
1697class YoutubeRecommendedIE(YoutubeFeedsInfoExtractor):
1698 IE_DESC = 'YouTube.com recommended videos, ":ytrec" for short (requires authentication)'
1699 _VALID_URL = r'https?://www\.youtube\.com/feed/recommended|:ytrec(?:ommended)?'
1700 _FEED_NAME = 'recommended'
1701 _PLAYLIST_TITLE = 'Youtube Recommended videos'
1ed5b5c9 1702
1ed5b5c9 1703
25f14e9f
S
1704class YoutubeSubscriptionsIE(YoutubeFeedsInfoExtractor):
1705 IE_DESC = 'YouTube.com subscriptions feed, "ytsubs" keyword (requires authentication)'
1706 _VALID_URL = r'https?://www\.youtube\.com/feed/subscriptions|:ytsubs(?:criptions)?'
1707 _FEED_NAME = 'subscriptions'
1708 _PLAYLIST_TITLE = 'Youtube Subscriptions'
1ed5b5c9 1709
1ed5b5c9 1710
25f14e9f
S
1711class YoutubeHistoryIE(YoutubeFeedsInfoExtractor):
1712 IE_DESC = 'Youtube watch history, ":ythistory" for short (requires authentication)'
1713 _VALID_URL = 'https?://www\.youtube\.com/feed/history|:ythistory'
1714 _FEED_NAME = 'history'
1715 _PLAYLIST_TITLE = 'Youtube History'
1ed5b5c9
JMF
1716
1717
15870e90
PH
1718class YoutubeTruncatedURLIE(InfoExtractor):
1719 IE_NAME = 'youtube:truncated_url'
1720 IE_DESC = False # Do not list
975d35db 1721 _VALID_URL = r'''(?x)
b95aab84
PH
1722 (?:https?://)?
1723 (?:\w+\.)?[yY][oO][uU][tT][uU][bB][eE](?:-nocookie)?\.com/
1724 (?:watch\?(?:
c4808c60 1725 feature=[a-z_]+|
b95aab84
PH
1726 annotation_id=annotation_[^&]+|
1727 x-yt-cl=[0-9]+|
c1708b89 1728 hl=[^&]*|
b95aab84
PH
1729 )?
1730 |
1731 attribution_link\?a=[^&]+
1732 )
1733 $
975d35db 1734 '''
15870e90 1735
c4808c60
PH
1736 _TESTS = [{
1737 'url': 'http://www.youtube.com/watch?annotation_id=annotation_3951667041',
1738 'only_matching': True,
dc2fc736
PH
1739 }, {
1740 'url': 'http://www.youtube.com/watch?',
1741 'only_matching': True,
b95aab84
PH
1742 }, {
1743 'url': 'https://www.youtube.com/watch?x-yt-cl=84503534',
1744 'only_matching': True,
1745 }, {
1746 'url': 'https://www.youtube.com/watch?feature=foo',
1747 'only_matching': True,
c1708b89
PH
1748 }, {
1749 'url': 'https://www.youtube.com/watch?hl=en-GB',
1750 'only_matching': True,
c4808c60
PH
1751 }]
1752
15870e90
PH
1753 def _real_extract(self, url):
1754 raise ExtractorError(
78caa52a
PH
1755 'Did you forget to quote the URL? Remember that & is a meta '
1756 'character in most shells, so you want to put the URL in quotes, '
1757 'like youtube-dl '
1758 '"http://www.youtube.com/watch?feature=foo&v=BaW_jenozKc" '
1759 ' or simply youtube-dl BaW_jenozKc .',
15870e90 1760 expected=True)
772fd5cc
PH
1761
1762
1763class YoutubeTruncatedIDIE(InfoExtractor):
1764 IE_NAME = 'youtube:truncated_id'
1765 IE_DESC = False # Do not list
b95aab84 1766 _VALID_URL = r'https?://(?:www\.)?youtube\.com/watch\?v=(?P<id>[0-9A-Za-z_-]{1,10})$'
772fd5cc
PH
1767
1768 _TESTS = [{
1769 'url': 'https://www.youtube.com/watch?v=N_708QY7Ob',
1770 'only_matching': True,
1771 }]
1772
1773 def _real_extract(self, url):
1774 video_id = self._match_id(url)
1775 raise ExtractorError(
1776 'Incomplete YouTube ID %s. URL %s looks truncated.' % (video_id, url),
1777 expected=True)