]> jfr.im git - yt-dlp.git/blame - youtube_dl/extractor/youtube.py
Merge pull request #1297 from iemejia/master
[yt-dlp.git] / youtube_dl / extractor / youtube.py
CommitLineData
c5e8d7af 1# coding: utf-8
c5e8d7af
PH
2
3import json
4import netrc
5import re
6import socket
04cc9617 7import itertools
c5e8d7af 8
b05654f0 9from .common import InfoExtractor, SearchInfoExtractor
c5e8d7af
PH
10from ..utils import (
11 compat_http_client,
12 compat_parse_qs,
13 compat_urllib_error,
14 compat_urllib_parse,
15 compat_urllib_request,
16 compat_str,
17
18 clean_html,
19 get_element_by_id,
20 ExtractorError,
21 unescapeHTML,
22 unified_strdate,
04cc9617 23 orderedSet,
c5e8d7af
PH
24)
25
b2e8bc1b
JMF
26class YoutubeBaseInfoExtractor(InfoExtractor):
27 """Provide base functions for Youtube extractors"""
28 _LOGIN_URL = 'https://accounts.google.com/ServiceLogin'
29 _LANG_URL = r'https://www.youtube.com/?hl=en&persist_hl=1&gl=US&persist_gl=1&opt_out_ackd=1'
30 _AGE_URL = 'http://www.youtube.com/verify_age?next_url=/&gl=US&hl=en'
31 _NETRC_MACHINE = 'youtube'
32 # If True it will raise an error if no login info is provided
33 _LOGIN_REQUIRED = False
34
35 def report_lang(self):
36 """Report attempt to set language."""
37 self.to_screen(u'Setting language')
38
39 def _set_language(self):
40 request = compat_urllib_request.Request(self._LANG_URL)
41 try:
42 self.report_lang()
43 compat_urllib_request.urlopen(request).read()
44 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
45 self._downloader.report_warning(u'unable to set language: %s' % compat_str(err))
46 return False
47 return True
48
49 def _login(self):
50 (username, password) = self._get_login_info()
51 # No authentication to be performed
52 if username is None:
53 if self._LOGIN_REQUIRED:
54 raise ExtractorError(u'No login info available, needed for using %s.' % self.IE_NAME, expected=True)
55 return False
56
57 request = compat_urllib_request.Request(self._LOGIN_URL)
58 try:
59 login_page = compat_urllib_request.urlopen(request).read().decode('utf-8')
60 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
61 self._downloader.report_warning(u'unable to fetch login page: %s' % compat_str(err))
62 return False
63
64 galx = None
65 dsh = None
66 match = re.search(re.compile(r'<input.+?name="GALX".+?value="(.+?)"', re.DOTALL), login_page)
67 if match:
68 galx = match.group(1)
69 match = re.search(re.compile(r'<input.+?name="dsh".+?value="(.+?)"', re.DOTALL), login_page)
70 if match:
71 dsh = match.group(1)
c5e8d7af 72
b2e8bc1b
JMF
73 # Log in
74 login_form_strs = {
75 u'continue': u'https://www.youtube.com/signin?action_handle_signin=true&feature=sign_in_button&hl=en_US&nomobiletemp=1',
76 u'Email': username,
77 u'GALX': galx,
78 u'Passwd': password,
79 u'PersistentCookie': u'yes',
80 u'_utf8': u'霱',
81 u'bgresponse': u'js_disabled',
82 u'checkConnection': u'',
83 u'checkedDomains': u'youtube',
84 u'dnConn': u'',
85 u'dsh': dsh,
86 u'pstMsg': u'0',
87 u'rmShown': u'1',
88 u'secTok': u'',
89 u'signIn': u'Sign in',
90 u'timeStmp': u'',
91 u'service': u'youtube',
92 u'uilel': u'3',
93 u'hl': u'en_US',
94 }
95 # Convert to UTF-8 *before* urlencode because Python 2.x's urlencode
96 # chokes on unicode
97 login_form = dict((k.encode('utf-8'), v.encode('utf-8')) for k,v in login_form_strs.items())
98 login_data = compat_urllib_parse.urlencode(login_form).encode('ascii')
99 request = compat_urllib_request.Request(self._LOGIN_URL, login_data)
100 try:
101 self.report_login()
102 login_results = compat_urllib_request.urlopen(request).read().decode('utf-8')
103 if re.search(r'(?i)<form[^>]* id="gaia_loginform"', login_results) is not None:
104 self._downloader.report_warning(u'unable to log in: bad username or password')
105 return False
106 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
107 self._downloader.report_warning(u'unable to log in: %s' % compat_str(err))
108 return False
109 return True
110
111 def _confirm_age(self):
112 age_form = {
113 'next_url': '/',
114 'action_confirm': 'Confirm',
115 }
116 request = compat_urllib_request.Request(self._AGE_URL, compat_urllib_parse.urlencode(age_form))
117 try:
118 self.report_age_confirmation()
119 compat_urllib_request.urlopen(request).read().decode('utf-8')
120 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
121 raise ExtractorError(u'Unable to confirm age: %s' % compat_str(err))
122 return True
123
124 def _real_initialize(self):
125 if self._downloader is None:
126 return
127 if not self._set_language():
128 return
129 if not self._login():
130 return
131 self._confirm_age()
132
133class YoutubeIE(YoutubeBaseInfoExtractor):
0f818663 134 IE_DESC = u'YouTube.com'
c5e8d7af
PH
135 _VALID_URL = r"""^
136 (
137 (?:https?://)? # http(s):// (optional)
138 (?:youtu\.be/|(?:\w+\.)?youtube(?:-nocookie)?\.com/|
139 tube\.majestyc\.net/) # the various hostnames, with wildcard subdomains
140 (?:.*?\#/)? # handle anchor (#/) redirect urls
141 (?: # the various things that can precede the ID:
142 (?:(?:v|embed|e)/) # v/ or embed/ or e/
143 |(?: # or the v= param in all its forms
d741e55a 144 (?:(?:watch|movie)(?:_popup)?(?:\.php)?)? # preceding watch(_popup|.php) or nothing (like /?v=xxxx)
c5e8d7af
PH
145 (?:\?|\#!?) # the params delimiter ? or # or #!
146 (?:.*?&)? # any other preceding param (like /?s=tuff&v=xxxx)
147 v=
148 )
149 )? # optional -> youtube.com/xxxx is OK
150 )? # all until now is optional -> you can pass the naked ID
151 ([0-9A-Za-z_-]+) # here is it! the YouTube video ID
152 (?(1).+)? # if we found the ID, everything can follow
153 $"""
c5e8d7af 154 _NEXT_URL_RE = r'[\?&]next_url=([^&]+)'
c5e8d7af 155 # Listed in order of quality
1d043b93
JMF
156 _available_formats = ['38', '37', '46', '22', '45', '35', '44', '34', '18', '43', '6', '5', '17', '13',
157 '95', '94', '93', '92', '132', '151',
939fbd26
JMF
158 # 3D
159 '85', '84', '102', '83', '101', '82', '100',
160 # Dash video
161 '138', '137', '248', '136', '247', '135', '246',
162 '245', '244', '134', '243', '133', '242', '160',
163 # Dash audio
164 '141', '172', '140', '171', '139',
1d043b93
JMF
165 ]
166 _available_formats_prefer_free = ['38', '46', '37', '45', '22', '44', '35', '43', '34', '18', '6', '5', '17', '13',
167 '95', '94', '93', '92', '132', '151',
86fe61c8 168 '85', '102', '84', '101', '83', '100', '82',
939fbd26
JMF
169 # Dash video
170 '138', '248', '137', '247', '136', '246', '245',
171 '244', '135', '243', '134', '242', '133', '160',
172 # Dash audio
173 '172', '141', '171', '140', '139',
1d043b93 174 ]
c5e8d7af
PH
175 _video_extensions = {
176 '13': '3gp',
177 '17': 'mp4',
178 '18': 'mp4',
179 '22': 'mp4',
180 '37': 'mp4',
d69cf69a 181 '38': 'mp4',
c5e8d7af
PH
182 '43': 'webm',
183 '44': 'webm',
184 '45': 'webm',
185 '46': 'webm',
1d043b93 186
86fe61c8
AZ
187 # 3d videos
188 '82': 'mp4',
189 '83': 'mp4',
190 '84': 'mp4',
191 '85': 'mp4',
192 '100': 'webm',
193 '101': 'webm',
194 '102': 'webm',
836a086c 195
1d043b93
JMF
196 # videos that use m3u8
197 '92': 'mp4',
198 '93': 'mp4',
199 '94': 'mp4',
200 '95': 'mp4',
201 '96': 'mp4',
202 '132': 'mp4',
203 '151': 'mp4',
836a086c
AZ
204
205 # Dash mp4
206 '133': 'mp4',
207 '134': 'mp4',
208 '135': 'mp4',
209 '136': 'mp4',
210 '137': 'mp4',
211 '138': 'mp4',
212 '139': 'mp4',
213 '140': 'mp4',
214 '141': 'mp4',
215 '160': 'mp4',
216
217 # Dash webm
218 '171': 'webm',
219 '172': 'webm',
220 '242': 'webm',
221 '243': 'webm',
222 '244': 'webm',
223 '245': 'webm',
224 '246': 'webm',
225 '247': 'webm',
226 '248': 'webm',
c5e8d7af
PH
227 }
228 _video_dimensions = {
229 '5': '240x400',
230 '6': '???',
231 '13': '???',
232 '17': '144x176',
233 '18': '360x640',
234 '22': '720x1280',
235 '34': '360x640',
236 '35': '480x854',
237 '37': '1080x1920',
238 '38': '3072x4096',
239 '43': '360x640',
240 '44': '480x854',
241 '45': '720x1280',
242 '46': '1080x1920',
86fe61c8
AZ
243 '82': '360p',
244 '83': '480p',
245 '84': '720p',
246 '85': '1080p',
1d043b93
JMF
247 '92': '240p',
248 '93': '360p',
249 '94': '480p',
250 '95': '720p',
251 '96': '1080p',
86fe61c8
AZ
252 '100': '360p',
253 '101': '480p',
836a086c 254 '102': '720p',
1d043b93
JMF
255 '132': '240p',
256 '151': '72p',
836a086c
AZ
257 '133': '240p',
258 '134': '360p',
259 '135': '480p',
260 '136': '720p',
261 '137': '1080p',
262 '138': '>1080p',
263 '139': '48k',
264 '140': '128k',
265 '141': '256k',
266 '160': '192p',
267 '171': '128k',
268 '172': '256k',
269 '242': '240p',
270 '243': '360p',
271 '244': '480p',
272 '245': '480p',
273 '246': '480p',
274 '247': '720p',
275 '248': '1080p',
c5e8d7af 276 }
836a086c
AZ
277 _special_itags = {
278 '82': '3D',
279 '83': '3D',
280 '84': '3D',
281 '85': '3D',
282 '100': '3D',
283 '101': '3D',
284 '102': '3D',
285 '133': 'DASH Video',
286 '134': 'DASH Video',
287 '135': 'DASH Video',
288 '136': 'DASH Video',
289 '137': 'DASH Video',
290 '138': 'DASH Video',
291 '139': 'DASH Audio',
292 '140': 'DASH Audio',
293 '141': 'DASH Audio',
294 '160': 'DASH Video',
295 '171': 'DASH Audio',
296 '172': 'DASH Audio',
297 '242': 'DASH Video',
298 '243': 'DASH Video',
299 '244': 'DASH Video',
300 '245': 'DASH Video',
301 '246': 'DASH Video',
302 '247': 'DASH Video',
303 '248': 'DASH Video',
c5e8d7af 304 }
836a086c 305
c5e8d7af 306 IE_NAME = u'youtube'
2eb88d95
PH
307 _TESTS = [
308 {
0e853ca4
PH
309 u"url": u"http://www.youtube.com/watch?v=BaW_jenozKc",
310 u"file": u"BaW_jenozKc.mp4",
311 u"info_dict": {
312 u"title": u"youtube-dl test video \"'/\\ä↭𝕐",
313 u"uploader": u"Philipp Hagemeister",
314 u"uploader_id": u"phihag",
315 u"upload_date": u"20121002",
316 u"description": u"test chars: \"'/\\ä↭𝕐\n\nThis is a test video for youtube-dl.\n\nFor more information, contact phihag@phihag.de ."
2eb88d95 317 }
0e853ca4
PH
318 },
319 {
320 u"url": u"http://www.youtube.com/watch?v=1ltcDfZMA3U",
321 u"file": u"1ltcDfZMA3U.flv",
322 u"note": u"Test VEVO video (#897)",
323 u"info_dict": {
324 u"upload_date": u"20070518",
325 u"title": u"Maps - It Will Find You",
326 u"description": u"Music video by Maps performing It Will Find You.",
327 u"uploader": u"MuteUSA",
328 u"uploader_id": u"MuteUSA"
2eb88d95 329 }
0e853ca4
PH
330 },
331 {
332 u"url": u"http://www.youtube.com/watch?v=UxxajLWwzqY",
333 u"file": u"UxxajLWwzqY.mp4",
334 u"note": u"Test generic use_cipher_signature video (#897)",
335 u"info_dict": {
336 u"upload_date": u"20120506",
337 u"title": u"Icona Pop - I Love It (feat. Charli XCX) [OFFICIAL VIDEO]",
338 u"description": u"md5:b085c9804f5ab69f4adea963a2dceb3c",
45ed795c 339 u"uploader": u"Icona Pop",
0e853ca4 340 u"uploader_id": u"IconaPop"
2eb88d95 341 }
c108eb73
JMF
342 },
343 {
344 u"url": u"https://www.youtube.com/watch?v=07FYdnEawAQ",
345 u"file": u"07FYdnEawAQ.mp4",
346 u"note": u"Test VEVO video with age protection (#956)",
347 u"info_dict": {
348 u"upload_date": u"20130703",
349 u"title": u"Justin Timberlake - Tunnel Vision (Explicit)",
350 u"description": u"md5:64249768eec3bc4276236606ea996373",
351 u"uploader": u"justintimberlakeVEVO",
352 u"uploader_id": u"justintimberlakeVEVO"
353 }
354 },
1d043b93
JMF
355 {
356 u'url': u'https://www.youtube.com/watch?v=TGi3HqYrWHE',
357 u'file': u'TGi3HqYrWHE.mp4',
358 u'note': u'm3u8 video',
359 u'info_dict': {
360 u'title': u'Triathlon - Men - London 2012 Olympic Games',
361 u'description': u'- Men - TR02 - Triathlon - 07 August 2012 - London 2012 Olympic Games',
362 u'uploader': u'olympic',
363 u'upload_date': u'20120807',
364 u'uploader_id': u'olympic',
365 },
366 u'params': {
367 u'skip_download': True,
368 },
369 },
2eb88d95
PH
370 ]
371
c5e8d7af
PH
372
373 @classmethod
374 def suitable(cls, url):
375 """Receives a URL and returns True if suitable for this IE."""
04cc9617 376 if YoutubePlaylistIE.suitable(url) or YoutubeSubscriptionsIE.suitable(url): return False
c5e8d7af
PH
377 return re.match(cls._VALID_URL, url, re.VERBOSE) is not None
378
c5e8d7af
PH
379 def report_video_webpage_download(self, video_id):
380 """Report attempt to download video webpage."""
381 self.to_screen(u'%s: Downloading video webpage' % video_id)
382
383 def report_video_info_webpage_download(self, video_id):
384 """Report attempt to download video info webpage."""
385 self.to_screen(u'%s: Downloading video info webpage' % video_id)
386
387 def report_video_subtitles_download(self, video_id):
388 """Report attempt to download video info webpage."""
389 self.to_screen(u'%s: Checking available subtitles' % video_id)
390
391 def report_video_subtitles_request(self, video_id, sub_lang, format):
392 """Report attempt to download video info webpage."""
393 self.to_screen(u'%s: Downloading video subtitles for %s.%s' % (video_id, sub_lang, format))
394
395 def report_video_subtitles_available(self, video_id, sub_lang_list):
396 """Report available subtitles."""
397 sub_lang = ",".join(list(sub_lang_list.keys()))
398 self.to_screen(u'%s: Available subtitles for video: %s' % (video_id, sub_lang))
399
400 def report_information_extraction(self, video_id):
401 """Report attempt to extract video information."""
402 self.to_screen(u'%s: Extracting video information' % video_id)
403
404 def report_unavailable_format(self, video_id, format):
405 """Report extracted video URL."""
406 self.to_screen(u'%s: Format %s not available' % (video_id, format))
407
408 def report_rtmp_download(self):
409 """Indicate the download will use the RTMP protocol."""
410 self.to_screen(u'RTMP download detected')
411
98bcd283 412 def _decrypt_signature(self, s):
257a2501 413 """Turn the encrypted s field into a working signature"""
6b37f0be 414
444b1165
JMF
415 if len(s) == 92:
416 return s[25] + s[3:25] + s[0] + s[26:42] + s[79] + s[43:79] + s[91] + s[80:83]
417 elif len(s) == 90:
418 return s[25] + s[3:25] + s[2] + s[26:40] + s[77] + s[41:77] + s[89] + s[78:81]
8a9d86a2 419 elif len(s) == 89:
420 return s[84:78:-1] + s[87] + s[77:60:-1] + s[0] + s[59:3:-1]
444b1165 421 elif len(s) == 88:
be547e1d
PH
422 return s[48] + s[81:67:-1] + s[82] + s[66:62:-1] + s[85] + s[61:48:-1] + s[67] + s[47:12:-1] + s[3] + s[11:3:-1] + s[2] + s[12]
423 elif len(s) == 87:
3a725669 424 return s[6:27] + s[4] + s[28:39] + s[27] + s[40:59] + s[2] + s[60:]
be547e1d 425 elif len(s) == 86:
02cf62e2 426 return s[5:20] + s[2] + s[21:]
be547e1d 427 elif len(s) == 85:
f9c3c90c 428 return s[83:34:-1] + s[0] + s[33:27:-1] + s[3] + s[26:19:-1] + s[34] + s[18:3:-1] + s[27]
be547e1d 429 elif len(s) == 84:
7a4c6cc9 430 return s[83:27:-1] + s[0] + s[26:5:-1] + s[2:0:-1] + s[27]
be547e1d 431 elif len(s) == 83:
e1842025 432 return s[81:64:-1] + s[82] + s[63:52:-1] + s[45] + s[51:45:-1] + s[1] + s[44:1:-1] + s[0]
be547e1d 433 elif len(s) == 82:
3669cdba 434 return s[1:19] + s[0] + s[20:68] + s[19] + s[69:82]
be547e1d 435 elif len(s) == 81:
aedd6bb9 436 return s[56] + s[79:56:-1] + s[41] + s[55:41:-1] + s[80] + s[40:34:-1] + s[0] + s[33:29:-1] + s[34] + s[28:9:-1] + s[29] + s[8:0:-1] + s[9]
5c468ca8
JMF
437 elif len(s) == 79:
438 return s[54] + s[77:54:-1] + s[39] + s[53:39:-1] + s[78] + s[38:34:-1] + s[0] + s[33:29:-1] + s[34] + s[28:9:-1] + s[29] + s[8:0:-1] + s[9]
be547e1d
PH
439
440 else:
441 raise ExtractorError(u'Unable to decrypt signature, key length %d not supported; retrying might work' % (len(s)))
c5e8d7af 442
75952c6e
JMF
443 def _decrypt_signature_age_gate(self, s):
444 # The videos with age protection use another player, so the algorithms
445 # can be different.
446 if len(s) == 86:
447 return s[2:63] + s[82] + s[64:82] + s[63]
448 else:
449 # Fallback to the other algortihms
b072a9de 450 return self._decrypt_signature(s)
75952c6e
JMF
451
452
c5e8d7af
PH
453 def _get_available_subtitles(self, video_id):
454 self.report_video_subtitles_download(video_id)
455 request = compat_urllib_request.Request('http://video.google.com/timedtext?hl=en&type=list&v=%s' % video_id)
456 try:
457 sub_list = compat_urllib_request.urlopen(request).read().decode('utf-8')
458 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
459 return (u'unable to download video subtitles: %s' % compat_str(err), None)
460 sub_lang_list = re.findall(r'name="([^"]*)"[^>]+lang_code="([\w\-]+)"', sub_list)
461 sub_lang_list = dict((l[1], l[0]) for l in sub_lang_list)
462 if not sub_lang_list:
463 return (u'video doesn\'t have subtitles', None)
464 return sub_lang_list
465
466 def _list_available_subtitles(self, video_id):
467 sub_lang_list = self._get_available_subtitles(video_id)
468 self.report_video_subtitles_available(video_id, sub_lang_list)
469
470 def _request_subtitle(self, sub_lang, sub_name, video_id, format):
471 """
472 Return tuple:
473 (error_message, sub_lang, sub)
474 """
475 self.report_video_subtitles_request(video_id, sub_lang, format)
476 params = compat_urllib_parse.urlencode({
477 'lang': sub_lang,
478 'name': sub_name,
479 'v': video_id,
480 'fmt': format,
481 })
482 url = 'http://www.youtube.com/api/timedtext?' + params
483 try:
484 sub = compat_urllib_request.urlopen(url).read().decode('utf-8')
485 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
486 return (u'unable to download video subtitles: %s' % compat_str(err), None, None)
487 if not sub:
488 return (u'Did not fetch video subtitles', None, None)
489 return (None, sub_lang, sub)
490
491 def _request_automatic_caption(self, video_id, webpage):
492 """We need the webpage for getting the captions url, pass it as an
493 argument to speed up the process."""
494 sub_lang = self._downloader.params.get('subtitleslang') or 'en'
495 sub_format = self._downloader.params.get('subtitlesformat')
496 self.to_screen(u'%s: Looking for automatic captions' % video_id)
497 mobj = re.search(r';ytplayer.config = ({.*?});', webpage)
498 err_msg = u'Couldn\'t find automatic captions for "%s"' % sub_lang
499 if mobj is None:
500 return [(err_msg, None, None)]
501 player_config = json.loads(mobj.group(1))
502 try:
503 args = player_config[u'args']
504 caption_url = args[u'ttsurl']
505 timestamp = args[u'timestamp']
506 params = compat_urllib_parse.urlencode({
507 'lang': 'en',
508 'tlang': sub_lang,
509 'fmt': sub_format,
510 'ts': timestamp,
511 'kind': 'asr',
512 })
513 subtitles_url = caption_url + '&' + params
514 sub = self._download_webpage(subtitles_url, video_id, u'Downloading automatic captions')
515 return [(None, sub_lang, sub)]
516 except KeyError:
517 return [(err_msg, None, None)]
518
519 def _extract_subtitle(self, video_id):
520 """
521 Return a list with a tuple:
522 [(error_message, sub_lang, sub)]
523 """
524 sub_lang_list = self._get_available_subtitles(video_id)
525 sub_format = self._downloader.params.get('subtitlesformat')
526 if isinstance(sub_lang_list,tuple): #There was some error, it didn't get the available subtitles
527 return [(sub_lang_list[0], None, None)]
528 if self._downloader.params.get('subtitleslang', False):
529 sub_lang = self._downloader.params.get('subtitleslang')
530 elif 'en' in sub_lang_list:
531 sub_lang = 'en'
532 else:
533 sub_lang = list(sub_lang_list.keys())[0]
534 if not sub_lang in sub_lang_list:
535 return [(u'no closed captions found in the specified language "%s"' % sub_lang, None, None)]
536
537 subtitle = self._request_subtitle(sub_lang, sub_lang_list[sub_lang].encode('utf-8'), video_id, sub_format)
538 return [subtitle]
539
540 def _extract_all_subtitles(self, video_id):
541 sub_lang_list = self._get_available_subtitles(video_id)
542 sub_format = self._downloader.params.get('subtitlesformat')
543 if isinstance(sub_lang_list,tuple): #There was some error, it didn't get the available subtitles
544 return [(sub_lang_list[0], None, None)]
545 subtitles = []
546 for sub_lang in sub_lang_list:
547 subtitle = self._request_subtitle(sub_lang, sub_lang_list[sub_lang].encode('utf-8'), video_id, sub_format)
548 subtitles.append(subtitle)
549 return subtitles
550
551 def _print_formats(self, formats):
552 print('Available formats:')
553 for x in formats:
03cc7c20
JMF
554 print('%s\t:\t%s\t[%s]%s' %(x, self._video_extensions.get(x, 'flv'),
555 self._video_dimensions.get(x, '???'),
836a086c 556 ' ('+self._special_itags[x]+')' if x in self._special_itags else ''))
c5e8d7af 557
c5e8d7af
PH
558 def _extract_id(self, url):
559 mobj = re.match(self._VALID_URL, url, re.VERBOSE)
560 if mobj is None:
561 raise ExtractorError(u'Invalid URL: %s' % url)
562 video_id = mobj.group(2)
563 return video_id
564
1d043b93
JMF
565 def _get_video_url_list(self, url_map):
566 """
567 Transform a dictionary in the format {itag:url} to a list of (itag, url)
568 with the requested formats.
569 """
570 req_format = self._downloader.params.get('format', None)
571 format_limit = self._downloader.params.get('format_limit', None)
572 available_formats = self._available_formats_prefer_free if self._downloader.params.get('prefer_free_formats', False) else self._available_formats
573 if format_limit is not None and format_limit in available_formats:
574 format_list = available_formats[available_formats.index(format_limit):]
575 else:
576 format_list = available_formats
577 existing_formats = [x for x in format_list if x in url_map]
578 if len(existing_formats) == 0:
579 raise ExtractorError(u'no known formats available for video')
580 if self._downloader.params.get('listformats', None):
581 self._print_formats(existing_formats)
582 return
583 if req_format is None or req_format == 'best':
584 video_url_list = [(existing_formats[0], url_map[existing_formats[0]])] # Best quality
585 elif req_format == 'worst':
586 video_url_list = [(existing_formats[-1], url_map[existing_formats[-1]])] # worst quality
587 elif req_format in ('-1', 'all'):
588 video_url_list = [(f, url_map[f]) for f in existing_formats] # All formats
589 else:
590 # Specific formats. We pick the first in a slash-delimeted sequence.
591 # For example, if '1/2/3/4' is requested and '2' and '4' are available, we pick '2'.
592 req_formats = req_format.split('/')
593 video_url_list = None
594 for rf in req_formats:
595 if rf in url_map:
596 video_url_list = [(rf, url_map[rf])]
597 break
598 if video_url_list is None:
599 raise ExtractorError(u'requested format not available')
600 return video_url_list
601
602 def _extract_from_m3u8(self, manifest_url, video_id):
603 url_map = {}
604 def _get_urls(_manifest):
605 lines = _manifest.split('\n')
606 urls = filter(lambda l: l and not l.startswith('#'),
607 lines)
608 return urls
609 manifest = self._download_webpage(manifest_url, video_id, u'Downloading formats manifest')
610 formats_urls = _get_urls(manifest)
611 for format_url in formats_urls:
612 itag = self._search_regex(r'itag/(\d+?)/', format_url, 'itag')
613 url_map[itag] = format_url
614 return url_map
615
c5e8d7af 616 def _real_extract(self, url):
d7f44b5b
PH
617 if re.match(r'(?:https?://)?[^/]+/watch\?feature=[a-z_]+$', url):
618 self._downloader.report_warning(u'Did you forget to quote the URL? Remember that & is a meta-character in most shells, so you want to put the URL in quotes, like youtube-dl \'http://www.youtube.com/watch?feature=foo&v=BaW_jenozKc\' (or simply youtube-dl BaW_jenozKc ).')
619
c5e8d7af
PH
620 # Extract original video URL from URL with redirection, like age verification, using next_url parameter
621 mobj = re.search(self._NEXT_URL_RE, url)
622 if mobj:
623 url = 'https://www.youtube.com/' + compat_urllib_parse.unquote(mobj.group(1)).lstrip('/')
624 video_id = self._extract_id(url)
625
626 # Get video webpage
627 self.report_video_webpage_download(video_id)
628 url = 'https://www.youtube.com/watch?v=%s&gl=US&hl=en&has_verified=1' % video_id
629 request = compat_urllib_request.Request(url)
630 try:
631 video_webpage_bytes = compat_urllib_request.urlopen(request).read()
632 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
633 raise ExtractorError(u'Unable to download video webpage: %s' % compat_str(err))
634
635 video_webpage = video_webpage_bytes.decode('utf-8', 'ignore')
636
637 # Attempt to extract SWF player URL
638 mobj = re.search(r'swfConfig.*?"(http:\\/\\/.*?watch.*?-.*?\.swf)"', video_webpage)
639 if mobj is not None:
640 player_url = re.sub(r'\\(.)', r'\1', mobj.group(1))
641 else:
642 player_url = None
643
644 # Get video info
645 self.report_video_info_webpage_download(video_id)
c108eb73
JMF
646 if re.search(r'player-age-gate-content">', video_webpage) is not None:
647 self.report_age_confirmation()
648 age_gate = True
649 # We simulate the access to the video from www.youtube.com/v/{video_id}
650 # this can be viewed without login into Youtube
651 data = compat_urllib_parse.urlencode({'video_id': video_id,
652 'el': 'embedded',
653 'gl': 'US',
654 'hl': 'en',
655 'eurl': 'https://youtube.googleapis.com/v/' + video_id,
656 'asv': 3,
657 'sts':'1588',
658 })
659 video_info_url = 'https://www.youtube.com/get_video_info?' + data
c5e8d7af
PH
660 video_info_webpage = self._download_webpage(video_info_url, video_id,
661 note=False,
662 errnote='unable to download video info webpage')
663 video_info = compat_parse_qs(video_info_webpage)
c108eb73
JMF
664 else:
665 age_gate = False
666 for el_type in ['&el=embedded', '&el=detailpage', '&el=vevo', '']:
667 video_info_url = ('https://www.youtube.com/get_video_info?&video_id=%s%s&ps=default&eurl=&gl=US&hl=en'
668 % (video_id, el_type))
669 video_info_webpage = self._download_webpage(video_info_url, video_id,
670 note=False,
671 errnote='unable to download video info webpage')
672 video_info = compat_parse_qs(video_info_webpage)
673 if 'token' in video_info:
674 break
c5e8d7af
PH
675 if 'token' not in video_info:
676 if 'reason' in video_info:
9a82b238 677 raise ExtractorError(u'YouTube said: %s' % video_info['reason'][0], expected=True)
c5e8d7af
PH
678 else:
679 raise ExtractorError(u'"token" parameter not in video info for unknown reason')
680
681 # Check for "rental" videos
682 if 'ypc_video_rental_bar_text' in video_info and 'author' not in video_info:
683 raise ExtractorError(u'"rental" videos not supported')
684
685 # Start extracting information
686 self.report_information_extraction(video_id)
687
688 # uploader
689 if 'author' not in video_info:
690 raise ExtractorError(u'Unable to extract uploader name')
691 video_uploader = compat_urllib_parse.unquote_plus(video_info['author'][0])
692
693 # uploader_id
694 video_uploader_id = None
695 mobj = re.search(r'<link itemprop="url" href="http://www.youtube.com/(?:user|channel)/([^"]+)">', video_webpage)
696 if mobj is not None:
697 video_uploader_id = mobj.group(1)
698 else:
699 self._downloader.report_warning(u'unable to extract uploader nickname')
700
701 # title
702 if 'title' not in video_info:
703 raise ExtractorError(u'Unable to extract video title')
704 video_title = compat_urllib_parse.unquote_plus(video_info['title'][0])
705
706 # thumbnail image
7763b04e
JMF
707 # We try first to get a high quality image:
708 m_thumb = re.search(r'<span itemprop="thumbnail".*?href="(.*?)">',
709 video_webpage, re.DOTALL)
710 if m_thumb is not None:
711 video_thumbnail = m_thumb.group(1)
712 elif 'thumbnail_url' not in video_info:
c5e8d7af
PH
713 self._downloader.report_warning(u'unable to extract video thumbnail')
714 video_thumbnail = ''
715 else: # don't panic if we can't find it
716 video_thumbnail = compat_urllib_parse.unquote_plus(video_info['thumbnail_url'][0])
717
718 # upload date
719 upload_date = None
720 mobj = re.search(r'id="eow-date.*?>(.*?)</span>', video_webpage, re.DOTALL)
721 if mobj is not None:
722 upload_date = ' '.join(re.sub(r'[/,-]', r' ', mobj.group(1)).split())
723 upload_date = unified_strdate(upload_date)
724
725 # description
726 video_description = get_element_by_id("eow-description", video_webpage)
727 if video_description:
728 video_description = clean_html(video_description)
729 else:
730 fd_mobj = re.search(r'<meta name="description" content="([^"]+)"', video_webpage)
731 if fd_mobj:
732 video_description = unescapeHTML(fd_mobj.group(1))
733 else:
734 video_description = u''
735
736 # subtitles
737 video_subtitles = None
738
739 if self._downloader.params.get('writesubtitles', False):
740 video_subtitles = self._extract_subtitle(video_id)
741 if video_subtitles:
742 (sub_error, sub_lang, sub) = video_subtitles[0]
743 if sub_error:
b004821f
JMF
744 self._downloader.report_warning(sub_error)
745
746 if self._downloader.params.get('writeautomaticsub', False):
747 video_subtitles = self._request_automatic_caption(video_id, video_webpage)
748 (sub_error, sub_lang, sub) = video_subtitles[0]
749 if sub_error:
750 self._downloader.report_warning(sub_error)
c5e8d7af
PH
751
752 if self._downloader.params.get('allsubtitles', False):
753 video_subtitles = self._extract_all_subtitles(video_id)
754 for video_subtitle in video_subtitles:
755 (sub_error, sub_lang, sub) = video_subtitle
756 if sub_error:
757 self._downloader.report_warning(sub_error)
758
759 if self._downloader.params.get('listsubtitles', False):
93d3a642 760 self._list_available_subtitles(video_id)
c5e8d7af
PH
761 return
762
763 if 'length_seconds' not in video_info:
764 self._downloader.report_warning(u'unable to extract video duration')
765 video_duration = ''
766 else:
767 video_duration = compat_urllib_parse.unquote_plus(video_info['length_seconds'][0])
768
c5e8d7af 769 # Decide which formats to download
c5e8d7af
PH
770
771 try:
772 mobj = re.search(r';ytplayer.config = ({.*?});', video_webpage)
50be92c1
PH
773 if not mobj:
774 raise ValueError('Could not find vevo ID')
c5e8d7af
PH
775 info = json.loads(mobj.group(1))
776 args = info['args']
7ce7e394
JMF
777 # Easy way to know if the 's' value is in url_encoded_fmt_stream_map
778 # this signatures are encrypted
779 m_s = re.search(r'[&,]s=', args['url_encoded_fmt_stream_map'])
780 if m_s is not None:
781 self.to_screen(u'%s: Encrypted signatures detected.' % video_id)
c5e8d7af 782 video_info['url_encoded_fmt_stream_map'] = [args['url_encoded_fmt_stream_map']]
cde846b3 783 m_s = re.search(r'[&,]s=', args.get('adaptive_fmts', u''))
b7a68384 784 if m_s is not None:
37b6d5f6
AZ
785 if 'url_encoded_fmt_stream_map' in video_info:
786 video_info['url_encoded_fmt_stream_map'][0] += ',' + args['adaptive_fmts']
787 else:
788 video_info['url_encoded_fmt_stream_map'] = [args['adaptive_fmts']]
211fbc13 789 elif 'adaptive_fmts' in video_info:
37b6d5f6
AZ
790 if 'url_encoded_fmt_stream_map' in video_info:
791 video_info['url_encoded_fmt_stream_map'][0] += ',' + video_info['adaptive_fmts'][0]
792 else:
793 video_info['url_encoded_fmt_stream_map'] = video_info['adaptive_fmts']
c5e8d7af
PH
794 except ValueError:
795 pass
796
797 if 'conn' in video_info and video_info['conn'][0].startswith('rtmp'):
798 self.report_rtmp_download()
799 video_url_list = [(None, video_info['conn'][0])]
800 elif 'url_encoded_fmt_stream_map' in video_info and len(video_info['url_encoded_fmt_stream_map']) >= 1:
a7055eb9
JMF
801 if 'rtmpe%3Dyes' in video_info['url_encoded_fmt_stream_map'][0]:
802 raise ExtractorError('rtmpe downloads are not supported, see https://github.com/rg3/youtube-dl/issues/343 for more information.', expected=True)
c5e8d7af
PH
803 url_map = {}
804 for url_data_str in video_info['url_encoded_fmt_stream_map'][0].split(','):
805 url_data = compat_parse_qs(url_data_str)
806 if 'itag' in url_data and 'url' in url_data:
807 url = url_data['url'][0]
808 if 'sig' in url_data:
809 url += '&signature=' + url_data['sig'][0]
810 elif 's' in url_data:
769fda3c
FV
811 if self._downloader.params.get('verbose'):
812 s = url_data['s'][0]
c108eb73
JMF
813 if age_gate:
814 player_version = self._search_regex(r'ad3-(.+?)\.swf',
ed27d356
JMF
815 video_info['ad3_module'][0] if 'ad3_module' in video_info else 'NOT FOUND',
816 'flash player', fatal=False)
c108eb73
JMF
817 player = 'flash player %s' % player_version
818 else:
819 player = u'html5 player %s' % self._search_regex(r'html5player-(.+?)\.js', video_webpage,
820 'html5 player', fatal=False)
5a76c651
JMF
821 parts_sizes = u'.'.join(compat_str(len(part)) for part in s.split('.'))
822 self.to_screen(u'encrypted signature length %d (%s), itag %s, %s' %
823 (len(s), parts_sizes, url_data['itag'][0], player))
75952c6e
JMF
824 encrypted_sig = url_data['s'][0]
825 if age_gate:
826 signature = self._decrypt_signature_age_gate(encrypted_sig)
827 else:
828 signature = self._decrypt_signature(encrypted_sig)
c5e8d7af
PH
829 url += '&signature=' + signature
830 if 'ratebypass' not in url:
831 url += '&ratebypass=yes'
832 url_map[url_data['itag'][0]] = url
1d043b93
JMF
833 video_url_list = self._get_video_url_list(url_map)
834 if not video_url_list:
c5e8d7af 835 return
1d043b93
JMF
836 elif video_info.get('hlsvp'):
837 manifest_url = video_info['hlsvp'][0]
838 url_map = self._extract_from_m3u8(manifest_url, video_id)
839 video_url_list = self._get_video_url_list(url_map)
840 if not video_url_list:
841 return
842
c5e8d7af
PH
843 else:
844 raise ExtractorError(u'no conn or url_encoded_fmt_stream_map information found in video info')
845
846 results = []
847 for format_param, video_real_url in video_url_list:
848 # Extension
849 video_extension = self._video_extensions.get(format_param, 'flv')
850
03cc7c20
JMF
851 video_format = '{0} - {1}{2}'.format(format_param if format_param else video_extension,
852 self._video_dimensions.get(format_param, '???'),
836a086c 853 ' ('+self._special_itags[format_param]+')' if format_param in self._special_itags else '')
c5e8d7af
PH
854
855 results.append({
856 'id': video_id,
857 'url': video_real_url,
858 'uploader': video_uploader,
859 'uploader_id': video_uploader_id,
860 'upload_date': upload_date,
861 'title': video_title,
862 'ext': video_extension,
863 'format': video_format,
864 'thumbnail': video_thumbnail,
865 'description': video_description,
866 'player_url': player_url,
867 'subtitles': video_subtitles,
868 'duration': video_duration
869 })
870 return results
871
872class YoutubePlaylistIE(InfoExtractor):
0f818663 873 IE_DESC = u'YouTube.com playlists'
c5e8d7af
PH
874 _VALID_URL = r"""(?:
875 (?:https?://)?
876 (?:\w+\.)?
877 youtube\.com/
878 (?:
879 (?:course|view_play_list|my_playlists|artist|playlist|watch)
880 \? (?:.*?&)*? (?:p|a|list)=
881 | p/
882 )
c626a3d9 883 ((?:PL|EC|UU|FL)?[0-9A-Za-z-_]{10,})
c5e8d7af
PH
884 .*
885 |
c626a3d9 886 ((?:PL|EC|UU|FL)[0-9A-Za-z-_]{10,})
c5e8d7af
PH
887 )"""
888 _TEMPLATE_URL = 'https://gdata.youtube.com/feeds/api/playlists/%s?max-results=%i&start-index=%i&v=2&alt=json&safeSearch=none'
889 _MAX_RESULTS = 50
890 IE_NAME = u'youtube:playlist'
891
892 @classmethod
893 def suitable(cls, url):
894 """Receives a URL and returns True if suitable for this IE."""
895 return re.match(cls._VALID_URL, url, re.VERBOSE) is not None
896
897 def _real_extract(self, url):
898 # Extract playlist id
899 mobj = re.match(self._VALID_URL, url, re.VERBOSE)
900 if mobj is None:
901 raise ExtractorError(u'Invalid URL: %s' % url)
902
903 # Download playlist videos from API
904 playlist_id = mobj.group(1) or mobj.group(2)
c5e8d7af
PH
905 videos = []
906
755eb032 907 for page_num in itertools.count(1):
771822eb
JMF
908 start_index = self._MAX_RESULTS * (page_num - 1) + 1
909 if start_index >= 1000:
910 self._downloader.report_warning(u'Max number of results reached')
911 break
912 url = self._TEMPLATE_URL % (playlist_id, self._MAX_RESULTS, start_index)
c5e8d7af
PH
913 page = self._download_webpage(url, playlist_id, u'Downloading page #%s' % page_num)
914
915 try:
916 response = json.loads(page)
917 except ValueError as err:
918 raise ExtractorError(u'Invalid JSON in API response: ' + compat_str(err))
919
920 if 'feed' not in response:
921 raise ExtractorError(u'Got a malformed response from YouTube API')
922 playlist_title = response['feed']['title']['$t']
923 if 'entry' not in response['feed']:
924 # Number of videos is a multiple of self._MAX_RESULTS
925 break
926
927 for entry in response['feed']['entry']:
928 index = entry['yt$position']['$t']
929 if 'media$group' in entry and 'media$player' in entry['media$group']:
930 videos.append((index, entry['media$group']['media$player']['url']))
c5e8d7af
PH
931
932 videos = [v[1] for v in sorted(videos)]
933
20c3893f 934 url_results = [self.url_result(vurl, 'Youtube') for vurl in videos]
c5e8d7af
PH
935 return [self.playlist_result(url_results, playlist_id, playlist_title)]
936
937
938class YoutubeChannelIE(InfoExtractor):
0f818663 939 IE_DESC = u'YouTube.com channels'
c5e8d7af
PH
940 _VALID_URL = r"^(?:https?://)?(?:youtu\.be|(?:\w+\.)?youtube(?:-nocookie)?\.com)/channel/([0-9A-Za-z_-]+)"
941 _TEMPLATE_URL = 'http://www.youtube.com/channel/%s/videos?sort=da&flow=list&view=0&page=%s&gl=US&hl=en'
942 _MORE_PAGES_INDICATOR = 'yt-uix-load-more'
252580c5 943 _MORE_PAGES_URL = 'http://www.youtube.com/c4_browse_ajax?action_load_more_videos=1&flow=list&paging=%s&view=0&sort=da&channel_id=%s'
c5e8d7af
PH
944 IE_NAME = u'youtube:channel'
945
946 def extract_videos_from_page(self, page):
947 ids_in_page = []
948 for mobj in re.finditer(r'href="/watch\?v=([0-9A-Za-z_-]+)&?', page):
949 if mobj.group(1) not in ids_in_page:
950 ids_in_page.append(mobj.group(1))
951 return ids_in_page
952
953 def _real_extract(self, url):
954 # Extract channel id
955 mobj = re.match(self._VALID_URL, url)
956 if mobj is None:
957 raise ExtractorError(u'Invalid URL: %s' % url)
958
959 # Download channel page
960 channel_id = mobj.group(1)
961 video_ids = []
962 pagenum = 1
963
964 url = self._TEMPLATE_URL % (channel_id, pagenum)
965 page = self._download_webpage(url, channel_id,
966 u'Downloading page #%s' % pagenum)
967
968 # Extract video identifiers
969 ids_in_page = self.extract_videos_from_page(page)
970 video_ids.extend(ids_in_page)
971
972 # Download any subsequent channel pages using the json-based channel_ajax query
973 if self._MORE_PAGES_INDICATOR in page:
755eb032 974 for pagenum in itertools.count(1):
c5e8d7af
PH
975 url = self._MORE_PAGES_URL % (pagenum, channel_id)
976 page = self._download_webpage(url, channel_id,
977 u'Downloading page #%s' % pagenum)
978
979 page = json.loads(page)
980
981 ids_in_page = self.extract_videos_from_page(page['content_html'])
982 video_ids.extend(ids_in_page)
983
984 if self._MORE_PAGES_INDICATOR not in page['load_more_widget_html']:
985 break
986
987 self._downloader.to_screen(u'[youtube] Channel %s: Found %i videos' % (channel_id, len(video_ids)))
988
989 urls = ['http://www.youtube.com/watch?v=%s' % id for id in video_ids]
20c3893f 990 url_entries = [self.url_result(eurl, 'Youtube') for eurl in urls]
c5e8d7af
PH
991 return [self.playlist_result(url_entries, channel_id)]
992
993
994class YoutubeUserIE(InfoExtractor):
0f818663 995 IE_DESC = u'YouTube.com user videos (URL or "ytuser" keyword)'
c5e8d7af
PH
996 _VALID_URL = r'(?:(?:(?:https?://)?(?:\w+\.)?youtube\.com/user/)|ytuser:)([A-Za-z0-9_-]+)'
997 _TEMPLATE_URL = 'http://gdata.youtube.com/feeds/api/users/%s'
998 _GDATA_PAGE_SIZE = 50
999 _GDATA_URL = 'http://gdata.youtube.com/feeds/api/users/%s/uploads?max-results=%d&start-index=%d'
1000 _VIDEO_INDICATOR = r'/watch\?v=(.+?)[\<&]'
1001 IE_NAME = u'youtube:user'
1002
1003 def _real_extract(self, url):
1004 # Extract username
1005 mobj = re.match(self._VALID_URL, url)
1006 if mobj is None:
1007 raise ExtractorError(u'Invalid URL: %s' % url)
1008
1009 username = mobj.group(1)
1010
1011 # Download video ids using YouTube Data API. Result size per
1012 # query is limited (currently to 50 videos) so we need to query
1013 # page by page until there are no video ids - it means we got
1014 # all of them.
1015
1016 video_ids = []
c5e8d7af 1017
755eb032 1018 for pagenum in itertools.count(0):
c5e8d7af
PH
1019 start_index = pagenum * self._GDATA_PAGE_SIZE + 1
1020
1021 gdata_url = self._GDATA_URL % (username, self._GDATA_PAGE_SIZE, start_index)
1022 page = self._download_webpage(gdata_url, username,
1023 u'Downloading video ids from %d to %d' % (start_index, start_index + self._GDATA_PAGE_SIZE))
1024
1025 # Extract video identifiers
1026 ids_in_page = []
1027
1028 for mobj in re.finditer(self._VIDEO_INDICATOR, page):
1029 if mobj.group(1) not in ids_in_page:
1030 ids_in_page.append(mobj.group(1))
1031
1032 video_ids.extend(ids_in_page)
1033
1034 # A little optimization - if current page is not
1035 # "full", ie. does not contain PAGE_SIZE video ids then
1036 # we can assume that this page is the last one - there
1037 # are no more ids on further pages - no need to query
1038 # again.
1039
1040 if len(ids_in_page) < self._GDATA_PAGE_SIZE:
1041 break
1042
c5e8d7af 1043 urls = ['http://www.youtube.com/watch?v=%s' % video_id for video_id in video_ids]
20c3893f 1044 url_results = [self.url_result(rurl, 'Youtube') for rurl in urls]
c5e8d7af 1045 return [self.playlist_result(url_results, playlist_title = username)]
b05654f0
PH
1046
1047class YoutubeSearchIE(SearchInfoExtractor):
0f818663 1048 IE_DESC = u'YouTube.com searches'
b05654f0
PH
1049 _API_URL = 'https://gdata.youtube.com/feeds/api/videos?q=%s&start-index=%i&max-results=50&v=2&alt=jsonc'
1050 _MAX_RESULTS = 1000
1051 IE_NAME = u'youtube:search'
1052 _SEARCH_KEY = 'ytsearch'
1053
1054 def report_download_page(self, query, pagenum):
1055 """Report attempt to download search page with given number."""
1056 self._downloader.to_screen(u'[youtube] query "%s": Downloading page %s' % (query, pagenum))
1057
1058 def _get_n_results(self, query, n):
1059 """Get a specified number of results for a query"""
1060
1061 video_ids = []
1062 pagenum = 0
1063 limit = n
1064
1065 while (50 * pagenum) < limit:
1066 self.report_download_page(query, pagenum+1)
1067 result_url = self._API_URL % (compat_urllib_parse.quote_plus(query), (50*pagenum)+1)
1068 request = compat_urllib_request.Request(result_url)
1069 try:
1070 data = compat_urllib_request.urlopen(request).read().decode('utf-8')
1071 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
1072 raise ExtractorError(u'Unable to download API page: %s' % compat_str(err))
1073 api_response = json.loads(data)['data']
1074
1075 if not 'items' in api_response:
1076 raise ExtractorError(u'[youtube] No video results')
1077
1078 new_ids = list(video['id'] for video in api_response['items'])
1079 video_ids += new_ids
1080
1081 limit = min(n, api_response['totalItems'])
1082 pagenum += 1
1083
1084 if len(video_ids) > n:
1085 video_ids = video_ids[:n]
1086 videos = [self.url_result('http://www.youtube.com/watch?v=%s' % id, 'Youtube') for id in video_ids]
1087 return self.playlist_result(videos, query)
75dff0ee
JMF
1088
1089
1090class YoutubeShowIE(InfoExtractor):
0f818663 1091 IE_DESC = u'YouTube.com (multi-season) shows'
75dff0ee
JMF
1092 _VALID_URL = r'https?://www\.youtube\.com/show/(.*)'
1093 IE_NAME = u'youtube:show'
1094
1095 def _real_extract(self, url):
1096 mobj = re.match(self._VALID_URL, url)
1097 show_name = mobj.group(1)
1098 webpage = self._download_webpage(url, show_name, u'Downloading show webpage')
1099 # There's one playlist for each season of the show
1100 m_seasons = list(re.finditer(r'href="(/playlist\?list=.*?)"', webpage))
1101 self.to_screen(u'%s: Found %s seasons' % (show_name, len(m_seasons)))
1102 return [self.url_result('https://www.youtube.com' + season.group(1), 'YoutubePlaylist') for season in m_seasons]
04cc9617
JMF
1103
1104
b2e8bc1b 1105class YoutubeFeedsInfoExtractor(YoutubeBaseInfoExtractor):
d7ae0639
JMF
1106 """
1107 Base class for extractors that fetch info from
1108 http://www.youtube.com/feed_ajax
1109 Subclasses must define the _FEED_NAME and _PLAYLIST_TITLE properties.
1110 """
b2e8bc1b 1111 _LOGIN_REQUIRED = True
04cc9617 1112 _PAGING_STEP = 30
43ba5456
JMF
1113 # use action_load_personal_feed instead of action_load_system_feed
1114 _PERSONAL_FEED = False
04cc9617 1115
d7ae0639
JMF
1116 @property
1117 def _FEED_TEMPLATE(self):
43ba5456
JMF
1118 action = 'action_load_system_feed'
1119 if self._PERSONAL_FEED:
1120 action = 'action_load_personal_feed'
1121 return 'http://www.youtube.com/feed_ajax?%s=1&feed_name=%s&paging=%%s' % (action, self._FEED_NAME)
d7ae0639
JMF
1122
1123 @property
1124 def IE_NAME(self):
1125 return u'youtube:%s' % self._FEED_NAME
1126
81f0259b 1127 def _real_initialize(self):
b2e8bc1b 1128 self._login()
81f0259b 1129
04cc9617
JMF
1130 def _real_extract(self, url):
1131 feed_entries = []
1132 # The step argument is available only in 2.7 or higher
1133 for i in itertools.count(0):
1134 paging = i*self._PAGING_STEP
d7ae0639
JMF
1135 info = self._download_webpage(self._FEED_TEMPLATE % paging,
1136 u'%s feed' % self._FEED_NAME,
04cc9617
JMF
1137 u'Downloading page %s' % i)
1138 info = json.loads(info)
1139 feed_html = info['feed_html']
43ba5456 1140 m_ids = re.finditer(r'"/watch\?v=(.*?)["&]', feed_html)
04cc9617
JMF
1141 ids = orderedSet(m.group(1) for m in m_ids)
1142 feed_entries.extend(self.url_result(id, 'Youtube') for id in ids)
1143 if info['paging'] is None:
1144 break
d7ae0639
JMF
1145 return self.playlist_result(feed_entries, playlist_title=self._PLAYLIST_TITLE)
1146
1147class YoutubeSubscriptionsIE(YoutubeFeedsInfoExtractor):
1148 IE_DESC = u'YouTube.com subscriptions feed, "ytsubs" keyword(requires authentication)'
1149 _VALID_URL = r'https?://www\.youtube\.com/feed/subscriptions|:ytsubs(?:criptions)?'
1150 _FEED_NAME = 'subscriptions'
1151 _PLAYLIST_TITLE = u'Youtube Subscriptions'
1152
1153class YoutubeRecommendedIE(YoutubeFeedsInfoExtractor):
1154 IE_DESC = u'YouTube.com recommended videos, "ytrec" keyword (requires authentication)'
1155 _VALID_URL = r'https?://www\.youtube\.com/feed/recommended|:ytrec(?:ommended)?'
1156 _FEED_NAME = 'recommended'
1157 _PLAYLIST_TITLE = u'Youtube Recommended videos'
c626a3d9 1158
43ba5456
JMF
1159class YoutubeWatchLaterIE(YoutubeFeedsInfoExtractor):
1160 IE_DESC = u'Youtube watch later list, "ytwatchlater" keyword (requires authentication)'
1161 _VALID_URL = r'https?://www\.youtube\.com/feed/watch_later|:ytwatchlater'
1162 _FEED_NAME = 'watch_later'
1163 _PLAYLIST_TITLE = u'Youtube Watch Later'
1164 _PAGING_STEP = 100
1165 _PERSONAL_FEED = True
c626a3d9
JMF
1166
1167class YoutubeFavouritesIE(YoutubeBaseInfoExtractor):
1168 IE_NAME = u'youtube:favorites'
1169 IE_DESC = u'YouTube.com favourite videos, "ytfav" keyword (requires authentication)'
1170 _VALID_URL = r'https?://www\.youtube\.com/my_favorites|:ytfav(?:o?rites)?'
1171 _LOGIN_REQUIRED = True
1172
1173 def _real_extract(self, url):
1174 webpage = self._download_webpage('https://www.youtube.com/my_favorites', 'Youtube Favourites videos')
1175 playlist_id = self._search_regex(r'list=(.+?)["&]', webpage, u'favourites playlist id')
1176 return self.url_result(playlist_id, 'YoutubePlaylist')