]> jfr.im git - yt-dlp.git/blob - yt_dlp/extractor/common.py
[cleanup] Misc cleanup
[yt-dlp.git] / yt_dlp / extractor / common.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import base64
5 import datetime
6 import hashlib
7 import json
8 import netrc
9 import os
10 import random
11 import re
12 import sys
13 import time
14 import math
15
16 from ..compat import (
17 compat_cookiejar_Cookie,
18 compat_cookies_SimpleCookie,
19 compat_etree_Element,
20 compat_etree_fromstring,
21 compat_expanduser,
22 compat_getpass,
23 compat_http_client,
24 compat_os_name,
25 compat_str,
26 compat_urllib_error,
27 compat_urllib_parse_unquote,
28 compat_urllib_parse_urlencode,
29 compat_urllib_request,
30 compat_urlparse,
31 compat_xml_parse_error,
32 )
33 from ..downloader import FileDownloader
34 from ..downloader.f4m import (
35 get_base_url,
36 remove_encrypted_media,
37 )
38 from ..utils import (
39 age_restricted,
40 base_url,
41 bug_reports_message,
42 clean_html,
43 compiled_regex_type,
44 determine_ext,
45 determine_protocol,
46 dict_get,
47 error_to_compat_str,
48 extract_attributes,
49 ExtractorError,
50 fix_xml_ampersands,
51 float_or_none,
52 format_field,
53 GeoRestrictedError,
54 GeoUtils,
55 int_or_none,
56 js_to_json,
57 JSON_LD_RE,
58 mimetype2ext,
59 network_exceptions,
60 NO_DEFAULT,
61 orderedSet,
62 parse_bitrate,
63 parse_codecs,
64 parse_duration,
65 parse_iso8601,
66 parse_m3u8_attributes,
67 parse_resolution,
68 RegexNotFoundError,
69 sanitize_filename,
70 sanitized_Request,
71 str_or_none,
72 str_to_int,
73 strip_or_none,
74 traverse_obj,
75 unescapeHTML,
76 unified_strdate,
77 unified_timestamp,
78 update_Request,
79 update_url_query,
80 url_basename,
81 url_or_none,
82 urljoin,
83 variadic,
84 xpath_element,
85 xpath_text,
86 xpath_with_ns,
87 )
88
89
90 class InfoExtractor(object):
91 """Information Extractor class.
92
93 Information extractors are the classes that, given a URL, extract
94 information about the video (or videos) the URL refers to. This
95 information includes the real video URL, the video title, author and
96 others. The information is stored in a dictionary which is then
97 passed to the YoutubeDL. The YoutubeDL processes this
98 information possibly downloading the video to the file system, among
99 other possible outcomes.
100
101 The type field determines the type of the result.
102 By far the most common value (and the default if _type is missing) is
103 "video", which indicates a single video.
104
105 For a video, the dictionaries must include the following fields:
106
107 id: Video identifier.
108 title: Video title, unescaped.
109
110 Additionally, it must contain either a formats entry or a url one:
111
112 formats: A list of dictionaries for each format available, ordered
113 from worst to best quality.
114
115 Potential fields:
116 * url The mandatory URL representing the media:
117 for plain file media - HTTP URL of this file,
118 for RTMP - RTMP URL,
119 for HLS - URL of the M3U8 media playlist,
120 for HDS - URL of the F4M manifest,
121 for DASH
122 - HTTP URL to plain file media (in case of
123 unfragmented media)
124 - URL of the MPD manifest or base URL
125 representing the media if MPD manifest
126 is parsed from a string (in case of
127 fragmented media)
128 for MSS - URL of the ISM manifest.
129 * manifest_url
130 The URL of the manifest file in case of
131 fragmented media:
132 for HLS - URL of the M3U8 master playlist,
133 for HDS - URL of the F4M manifest,
134 for DASH - URL of the MPD manifest,
135 for MSS - URL of the ISM manifest.
136 * ext Will be calculated from URL if missing
137 * format A human-readable description of the format
138 ("mp4 container with h264/opus").
139 Calculated from the format_id, width, height.
140 and format_note fields if missing.
141 * format_id A short description of the format
142 ("mp4_h264_opus" or "19").
143 Technically optional, but strongly recommended.
144 * format_note Additional info about the format
145 ("3D" or "DASH video")
146 * width Width of the video, if known
147 * height Height of the video, if known
148 * resolution Textual description of width and height
149 * tbr Average bitrate of audio and video in KBit/s
150 * abr Average audio bitrate in KBit/s
151 * acodec Name of the audio codec in use
152 * asr Audio sampling rate in Hertz
153 * vbr Average video bitrate in KBit/s
154 * fps Frame rate
155 * vcodec Name of the video codec in use
156 * container Name of the container format
157 * filesize The number of bytes, if known in advance
158 * filesize_approx An estimate for the number of bytes
159 * player_url SWF Player URL (used for rtmpdump).
160 * protocol The protocol that will be used for the actual
161 download, lower-case.
162 "http", "https", "rtsp", "rtmp", "rtmp_ffmpeg", "rtmpe",
163 "m3u8", "m3u8_native" or "http_dash_segments".
164 * fragment_base_url
165 Base URL for fragments. Each fragment's path
166 value (if present) will be relative to
167 this URL.
168 * fragments A list of fragments of a fragmented media.
169 Each fragment entry must contain either an url
170 or a path. If an url is present it should be
171 considered by a client. Otherwise both path and
172 fragment_base_url must be present. Here is
173 the list of all potential fields:
174 * "url" - fragment's URL
175 * "path" - fragment's path relative to
176 fragment_base_url
177 * "duration" (optional, int or float)
178 * "filesize" (optional, int)
179 * preference Order number of this format. If this field is
180 present and not None, the formats get sorted
181 by this field, regardless of all other values.
182 -1 for default (order by other properties),
183 -2 or smaller for less than default.
184 < -1000 to hide the format (if there is
185 another one which is strictly better)
186 * language Language code, e.g. "de" or "en-US".
187 * language_preference Is this in the language mentioned in
188 the URL?
189 10 if it's what the URL is about,
190 -1 for default (don't know),
191 -10 otherwise, other values reserved for now.
192 * quality Order number of the video quality of this
193 format, irrespective of the file format.
194 -1 for default (order by other properties),
195 -2 or smaller for less than default.
196 * source_preference Order number for this video source
197 (quality takes higher priority)
198 -1 for default (order by other properties),
199 -2 or smaller for less than default.
200 * http_headers A dictionary of additional HTTP headers
201 to add to the request.
202 * stretched_ratio If given and not 1, indicates that the
203 video's pixels are not square.
204 width : height ratio as float.
205 * no_resume The server does not support resuming the
206 (HTTP or RTMP) download. Boolean.
207 * has_drm The format has DRM and cannot be downloaded. Boolean
208 * downloader_options A dictionary of downloader options as
209 described in FileDownloader
210 RTMP formats can also have the additional fields: page_url,
211 app, play_path, tc_url, flash_version, rtmp_live, rtmp_conn,
212 rtmp_protocol, rtmp_real_time
213
214 url: Final video URL.
215 ext: Video filename extension.
216 format: The video format, defaults to ext (used for --get-format)
217 player_url: SWF Player URL (used for rtmpdump).
218
219 The following fields are optional:
220
221 alt_title: A secondary title of the video.
222 display_id An alternative identifier for the video, not necessarily
223 unique, but available before title. Typically, id is
224 something like "4234987", title "Dancing naked mole rats",
225 and display_id "dancing-naked-mole-rats"
226 thumbnails: A list of dictionaries, with the following entries:
227 * "id" (optional, string) - Thumbnail format ID
228 * "url"
229 * "preference" (optional, int) - quality of the image
230 * "width" (optional, int)
231 * "height" (optional, int)
232 * "resolution" (optional, string "{width}x{height}",
233 deprecated)
234 * "filesize" (optional, int)
235 * "_test_url" (optional, bool) - If true, test the URL
236 thumbnail: Full URL to a video thumbnail image.
237 description: Full video description.
238 uploader: Full name of the video uploader.
239 license: License name the video is licensed under.
240 creator: The creator of the video.
241 release_timestamp: UNIX timestamp of the moment the video was released.
242 release_date: The date (YYYYMMDD) when the video was released.
243 timestamp: UNIX timestamp of the moment the video was uploaded
244 upload_date: Video upload date (YYYYMMDD).
245 If not explicitly set, calculated from timestamp.
246 uploader_id: Nickname or id of the video uploader.
247 uploader_url: Full URL to a personal webpage of the video uploader.
248 channel: Full name of the channel the video is uploaded on.
249 Note that channel fields may or may not repeat uploader
250 fields. This depends on a particular extractor.
251 channel_id: Id of the channel.
252 channel_url: Full URL to a channel webpage.
253 location: Physical location where the video was filmed.
254 subtitles: The available subtitles as a dictionary in the format
255 {tag: subformats}. "tag" is usually a language code, and
256 "subformats" is a list sorted from lower to higher
257 preference, each element is a dictionary with the "ext"
258 entry and one of:
259 * "data": The subtitles file contents
260 * "url": A URL pointing to the subtitles file
261 It can optionally also have:
262 * "name": Name or description of the subtitles
263 "ext" will be calculated from URL if missing
264 automatic_captions: Like 'subtitles'; contains automatically generated
265 captions instead of normal subtitles
266 duration: Length of the video in seconds, as an integer or float.
267 view_count: How many users have watched the video on the platform.
268 like_count: Number of positive ratings of the video
269 dislike_count: Number of negative ratings of the video
270 repost_count: Number of reposts of the video
271 average_rating: Average rating give by users, the scale used depends on the webpage
272 comment_count: Number of comments on the video
273 comments: A list of comments, each with one or more of the following
274 properties (all but one of text or html optional):
275 * "author" - human-readable name of the comment author
276 * "author_id" - user ID of the comment author
277 * "author_thumbnail" - The thumbnail of the comment author
278 * "id" - Comment ID
279 * "html" - Comment as HTML
280 * "text" - Plain text of the comment
281 * "timestamp" - UNIX timestamp of comment
282 * "parent" - ID of the comment this one is replying to.
283 Set to "root" to indicate that this is a
284 comment to the original video.
285 * "like_count" - Number of positive ratings of the comment
286 * "dislike_count" - Number of negative ratings of the comment
287 * "is_favorited" - Whether the comment is marked as
288 favorite by the video uploader
289 * "author_is_uploader" - Whether the comment is made by
290 the video uploader
291 age_limit: Age restriction for the video, as an integer (years)
292 webpage_url: The URL to the video webpage, if given to yt-dlp it
293 should allow to get the same result again. (It will be set
294 by YoutubeDL if it's missing)
295 categories: A list of categories that the video falls in, for example
296 ["Sports", "Berlin"]
297 tags: A list of tags assigned to the video, e.g. ["sweden", "pop music"]
298 cast: A list of the video cast
299 is_live: True, False, or None (=unknown). Whether this video is a
300 live stream that goes on instead of a fixed-length video.
301 was_live: True, False, or None (=unknown). Whether this video was
302 originally a live stream.
303 live_status: 'is_live', 'is_upcoming', 'was_live', 'not_live' or None (=unknown)
304 If absent, automatically set from is_live, was_live
305 start_time: Time in seconds where the reproduction should start, as
306 specified in the URL.
307 end_time: Time in seconds where the reproduction should end, as
308 specified in the URL.
309 chapters: A list of dictionaries, with the following entries:
310 * "start_time" - The start time of the chapter in seconds
311 * "end_time" - The end time of the chapter in seconds
312 * "title" (optional, string)
313 playable_in_embed: Whether this video is allowed to play in embedded
314 players on other sites. Can be True (=always allowed),
315 False (=never allowed), None (=unknown), or a string
316 specifying the criteria for embedability (Eg: 'whitelist')
317 availability: Under what condition the video is available. One of
318 'private', 'premium_only', 'subscriber_only', 'needs_auth',
319 'unlisted' or 'public'. Use 'InfoExtractor._availability'
320 to set it
321 __post_extractor: A function to be called just before the metadata is
322 written to either disk, logger or console. The function
323 must return a dict which will be added to the info_dict.
324 This is usefull for additional information that is
325 time-consuming to extract. Note that the fields thus
326 extracted will not be available to output template and
327 match_filter. So, only "comments" and "comment_count" are
328 currently allowed to be extracted via this method.
329
330 The following fields should only be used when the video belongs to some logical
331 chapter or section:
332
333 chapter: Name or title of the chapter the video belongs to.
334 chapter_number: Number of the chapter the video belongs to, as an integer.
335 chapter_id: Id of the chapter the video belongs to, as a unicode string.
336
337 The following fields should only be used when the video is an episode of some
338 series, programme or podcast:
339
340 series: Title of the series or programme the video episode belongs to.
341 season: Title of the season the video episode belongs to.
342 season_number: Number of the season the video episode belongs to, as an integer.
343 season_id: Id of the season the video episode belongs to, as a unicode string.
344 episode: Title of the video episode. Unlike mandatory video title field,
345 this field should denote the exact title of the video episode
346 without any kind of decoration.
347 episode_number: Number of the video episode within a season, as an integer.
348 episode_id: Id of the video episode, as a unicode string.
349
350 The following fields should only be used when the media is a track or a part of
351 a music album:
352
353 track: Title of the track.
354 track_number: Number of the track within an album or a disc, as an integer.
355 track_id: Id of the track (useful in case of custom indexing, e.g. 6.iii),
356 as a unicode string.
357 artist: Artist(s) of the track.
358 genre: Genre(s) of the track.
359 album: Title of the album the track belongs to.
360 album_type: Type of the album (e.g. "Demo", "Full-length", "Split", "Compilation", etc).
361 album_artist: List of all artists appeared on the album (e.g.
362 "Ash Borer / Fell Voices" or "Various Artists", useful for splits
363 and compilations).
364 disc_number: Number of the disc or other physical medium the track belongs to,
365 as an integer.
366 release_year: Year (YYYY) when the album was released.
367
368 Unless mentioned otherwise, the fields should be Unicode strings.
369
370 Unless mentioned otherwise, None is equivalent to absence of information.
371
372
373 _type "playlist" indicates multiple videos.
374 There must be a key "entries", which is a list, an iterable, or a PagedList
375 object, each element of which is a valid dictionary by this specification.
376
377 Additionally, playlists can have "id", "title", and any other relevent
378 attributes with the same semantics as videos (see above).
379
380
381 _type "multi_video" indicates that there are multiple videos that
382 form a single show, for examples multiple acts of an opera or TV episode.
383 It must have an entries key like a playlist and contain all the keys
384 required for a video at the same time.
385
386
387 _type "url" indicates that the video must be extracted from another
388 location, possibly by a different extractor. Its only required key is:
389 "url" - the next URL to extract.
390 The key "ie_key" can be set to the class name (minus the trailing "IE",
391 e.g. "Youtube") if the extractor class is known in advance.
392 Additionally, the dictionary may have any properties of the resolved entity
393 known in advance, for example "title" if the title of the referred video is
394 known ahead of time.
395
396
397 _type "url_transparent" entities have the same specification as "url", but
398 indicate that the given additional information is more precise than the one
399 associated with the resolved URL.
400 This is useful when a site employs a video service that hosts the video and
401 its technical metadata, but that video service does not embed a useful
402 title, description etc.
403
404
405 Subclasses of this one should re-define the _real_initialize() and
406 _real_extract() methods and define a _VALID_URL regexp.
407 Probably, they should also be added to the list of extractors.
408
409 Subclasses may also override suitable() if necessary, but ensure the function
410 signature is preserved and that this function imports everything it needs
411 (except other extractors), so that lazy_extractors works correctly
412
413 _GEO_BYPASS attribute may be set to False in order to disable
414 geo restriction bypass mechanisms for a particular extractor.
415 Though it won't disable explicit geo restriction bypass based on
416 country code provided with geo_bypass_country.
417
418 _GEO_COUNTRIES attribute may contain a list of presumably geo unrestricted
419 countries for this extractor. One of these countries will be used by
420 geo restriction bypass mechanism right away in order to bypass
421 geo restriction, of course, if the mechanism is not disabled.
422
423 _GEO_IP_BLOCKS attribute may contain a list of presumably geo unrestricted
424 IP blocks in CIDR notation for this extractor. One of these IP blocks
425 will be used by geo restriction bypass mechanism similarly
426 to _GEO_COUNTRIES.
427
428 The _WORKING attribute should be set to False for broken IEs
429 in order to warn the users and skip the tests.
430 """
431
432 _ready = False
433 _downloader = None
434 _x_forwarded_for_ip = None
435 _GEO_BYPASS = True
436 _GEO_COUNTRIES = None
437 _GEO_IP_BLOCKS = None
438 _WORKING = True
439
440 _LOGIN_HINTS = {
441 'any': 'Use --cookies, --username and --password or --netrc to provide account credentials',
442 'cookies': (
443 'Use --cookies for the authentication. '
444 'See https://github.com/ytdl-org/youtube-dl#how-do-i-pass-cookies-to-youtube-dl for how to pass cookies'),
445 'password': 'Use --username and --password or --netrc to provide account credentials',
446 }
447
448 def __init__(self, downloader=None):
449 """Constructor. Receives an optional downloader."""
450 self._ready = False
451 self._x_forwarded_for_ip = None
452 self._printed_messages = set()
453 self.set_downloader(downloader)
454
455 @classmethod
456 def _match_valid_url(cls, url):
457 # This does not use has/getattr intentionally - we want to know whether
458 # we have cached the regexp for *this* class, whereas getattr would also
459 # match the superclass
460 if '_VALID_URL_RE' not in cls.__dict__:
461 cls._VALID_URL_RE = re.compile(cls._VALID_URL)
462 return cls._VALID_URL_RE.match(url)
463
464 @classmethod
465 def suitable(cls, url):
466 """Receives a URL and returns True if suitable for this IE."""
467 # This function must import everything it needs (except other extractors),
468 # so that lazy_extractors works correctly
469 return cls._match_valid_url(url) is not None
470
471 @classmethod
472 def _match_id(cls, url):
473 return cls._match_valid_url(url).group('id')
474
475 @classmethod
476 def get_temp_id(cls, url):
477 try:
478 return cls._match_id(url)
479 except (IndexError, AttributeError):
480 return None
481
482 @classmethod
483 def working(cls):
484 """Getter method for _WORKING."""
485 return cls._WORKING
486
487 def initialize(self):
488 """Initializes an instance (authentication, etc)."""
489 self._printed_messages = set()
490 self._initialize_geo_bypass({
491 'countries': self._GEO_COUNTRIES,
492 'ip_blocks': self._GEO_IP_BLOCKS,
493 })
494 if not self._ready:
495 self._real_initialize()
496 self._ready = True
497
498 def _initialize_geo_bypass(self, geo_bypass_context):
499 """
500 Initialize geo restriction bypass mechanism.
501
502 This method is used to initialize geo bypass mechanism based on faking
503 X-Forwarded-For HTTP header. A random country from provided country list
504 is selected and a random IP belonging to this country is generated. This
505 IP will be passed as X-Forwarded-For HTTP header in all subsequent
506 HTTP requests.
507
508 This method will be used for initial geo bypass mechanism initialization
509 during the instance initialization with _GEO_COUNTRIES and
510 _GEO_IP_BLOCKS.
511
512 You may also manually call it from extractor's code if geo bypass
513 information is not available beforehand (e.g. obtained during
514 extraction) or due to some other reason. In this case you should pass
515 this information in geo bypass context passed as first argument. It may
516 contain following fields:
517
518 countries: List of geo unrestricted countries (similar
519 to _GEO_COUNTRIES)
520 ip_blocks: List of geo unrestricted IP blocks in CIDR notation
521 (similar to _GEO_IP_BLOCKS)
522
523 """
524 if not self._x_forwarded_for_ip:
525
526 # Geo bypass mechanism is explicitly disabled by user
527 if not self.get_param('geo_bypass', True):
528 return
529
530 if not geo_bypass_context:
531 geo_bypass_context = {}
532
533 # Backward compatibility: previously _initialize_geo_bypass
534 # expected a list of countries, some 3rd party code may still use
535 # it this way
536 if isinstance(geo_bypass_context, (list, tuple)):
537 geo_bypass_context = {
538 'countries': geo_bypass_context,
539 }
540
541 # The whole point of geo bypass mechanism is to fake IP
542 # as X-Forwarded-For HTTP header based on some IP block or
543 # country code.
544
545 # Path 1: bypassing based on IP block in CIDR notation
546
547 # Explicit IP block specified by user, use it right away
548 # regardless of whether extractor is geo bypassable or not
549 ip_block = self.get_param('geo_bypass_ip_block', None)
550
551 # Otherwise use random IP block from geo bypass context but only
552 # if extractor is known as geo bypassable
553 if not ip_block:
554 ip_blocks = geo_bypass_context.get('ip_blocks')
555 if self._GEO_BYPASS and ip_blocks:
556 ip_block = random.choice(ip_blocks)
557
558 if ip_block:
559 self._x_forwarded_for_ip = GeoUtils.random_ipv4(ip_block)
560 self._downloader.write_debug(
561 '[debug] Using fake IP %s as X-Forwarded-For' % self._x_forwarded_for_ip)
562 return
563
564 # Path 2: bypassing based on country code
565
566 # Explicit country code specified by user, use it right away
567 # regardless of whether extractor is geo bypassable or not
568 country = self.get_param('geo_bypass_country', None)
569
570 # Otherwise use random country code from geo bypass context but
571 # only if extractor is known as geo bypassable
572 if not country:
573 countries = geo_bypass_context.get('countries')
574 if self._GEO_BYPASS and countries:
575 country = random.choice(countries)
576
577 if country:
578 self._x_forwarded_for_ip = GeoUtils.random_ipv4(country)
579 self._downloader.write_debug(
580 'Using fake IP %s (%s) as X-Forwarded-For' % (self._x_forwarded_for_ip, country.upper()))
581
582 def extract(self, url):
583 """Extracts URL information and returns it in list of dicts."""
584 try:
585 for _ in range(2):
586 try:
587 self.initialize()
588 self.write_debug('Extracting URL: %s' % url)
589 ie_result = self._real_extract(url)
590 if ie_result is None:
591 return None
592 if self._x_forwarded_for_ip:
593 ie_result['__x_forwarded_for_ip'] = self._x_forwarded_for_ip
594 subtitles = ie_result.get('subtitles')
595 if (subtitles and 'live_chat' in subtitles
596 and 'no-live-chat' in self.get_param('compat_opts', [])):
597 del subtitles['live_chat']
598 return ie_result
599 except GeoRestrictedError as e:
600 if self.__maybe_fake_ip_and_retry(e.countries):
601 continue
602 raise
603 except ExtractorError as e:
604 video_id = e.video_id or self.get_temp_id(url)
605 raise ExtractorError(
606 e.msg, video_id=video_id, ie=self.IE_NAME, tb=e.traceback, expected=e.expected, cause=e.cause)
607 except compat_http_client.IncompleteRead as e:
608 raise ExtractorError('A network error has occurred.', cause=e, expected=True, video_id=self.get_temp_id(url))
609 except (KeyError, StopIteration) as e:
610 raise ExtractorError('An extractor error has occurred.', cause=e, video_id=self.get_temp_id(url))
611
612 def __maybe_fake_ip_and_retry(self, countries):
613 if (not self.get_param('geo_bypass_country', None)
614 and self._GEO_BYPASS
615 and self.get_param('geo_bypass', True)
616 and not self._x_forwarded_for_ip
617 and countries):
618 country_code = random.choice(countries)
619 self._x_forwarded_for_ip = GeoUtils.random_ipv4(country_code)
620 if self._x_forwarded_for_ip:
621 self.report_warning(
622 'Video is geo restricted. Retrying extraction with fake IP %s (%s) as X-Forwarded-For.'
623 % (self._x_forwarded_for_ip, country_code.upper()))
624 return True
625 return False
626
627 def set_downloader(self, downloader):
628 """Sets the downloader for this IE."""
629 self._downloader = downloader
630
631 def _real_initialize(self):
632 """Real initialization process. Redefine in subclasses."""
633 pass
634
635 def _real_extract(self, url):
636 """Real extraction process. Redefine in subclasses."""
637 pass
638
639 @classmethod
640 def ie_key(cls):
641 """A string for getting the InfoExtractor with get_info_extractor"""
642 return cls.__name__[:-2]
643
644 @property
645 def IE_NAME(self):
646 return compat_str(type(self).__name__[:-2])
647
648 @staticmethod
649 def __can_accept_status_code(err, expected_status):
650 assert isinstance(err, compat_urllib_error.HTTPError)
651 if expected_status is None:
652 return False
653 elif callable(expected_status):
654 return expected_status(err.code) is True
655 else:
656 return err.code in variadic(expected_status)
657
658 def _request_webpage(self, url_or_request, video_id, note=None, errnote=None, fatal=True, data=None, headers={}, query={}, expected_status=None):
659 """
660 Return the response handle.
661
662 See _download_webpage docstring for arguments specification.
663 """
664 if not self._downloader._first_webpage_request:
665 sleep_interval = float_or_none(self.get_param('sleep_interval_requests')) or 0
666 if sleep_interval > 0:
667 self.to_screen('Sleeping %s seconds ...' % sleep_interval)
668 time.sleep(sleep_interval)
669 else:
670 self._downloader._first_webpage_request = False
671
672 if note is None:
673 self.report_download_webpage(video_id)
674 elif note is not False:
675 if video_id is None:
676 self.to_screen('%s' % (note,))
677 else:
678 self.to_screen('%s: %s' % (video_id, note))
679
680 # Some sites check X-Forwarded-For HTTP header in order to figure out
681 # the origin of the client behind proxy. This allows bypassing geo
682 # restriction by faking this header's value to IP that belongs to some
683 # geo unrestricted country. We will do so once we encounter any
684 # geo restriction error.
685 if self._x_forwarded_for_ip:
686 if 'X-Forwarded-For' not in headers:
687 headers['X-Forwarded-For'] = self._x_forwarded_for_ip
688
689 if isinstance(url_or_request, compat_urllib_request.Request):
690 url_or_request = update_Request(
691 url_or_request, data=data, headers=headers, query=query)
692 else:
693 if query:
694 url_or_request = update_url_query(url_or_request, query)
695 if data is not None or headers:
696 url_or_request = sanitized_Request(url_or_request, data, headers)
697 try:
698 return self._downloader.urlopen(url_or_request)
699 except network_exceptions as err:
700 if isinstance(err, compat_urllib_error.HTTPError):
701 if self.__can_accept_status_code(err, expected_status):
702 # Retain reference to error to prevent file object from
703 # being closed before it can be read. Works around the
704 # effects of <https://bugs.python.org/issue15002>
705 # introduced in Python 3.4.1.
706 err.fp._error = err
707 return err.fp
708
709 if errnote is False:
710 return False
711 if errnote is None:
712 errnote = 'Unable to download webpage'
713
714 errmsg = '%s: %s' % (errnote, error_to_compat_str(err))
715 if fatal:
716 raise ExtractorError(errmsg, sys.exc_info()[2], cause=err)
717 else:
718 self.report_warning(errmsg)
719 return False
720
721 def _download_webpage_handle(self, url_or_request, video_id, note=None, errnote=None, fatal=True, encoding=None, data=None, headers={}, query={}, expected_status=None):
722 """
723 Return a tuple (page content as string, URL handle).
724
725 See _download_webpage docstring for arguments specification.
726 """
727 # Strip hashes from the URL (#1038)
728 if isinstance(url_or_request, (compat_str, str)):
729 url_or_request = url_or_request.partition('#')[0]
730
731 urlh = self._request_webpage(url_or_request, video_id, note, errnote, fatal, data=data, headers=headers, query=query, expected_status=expected_status)
732 if urlh is False:
733 assert not fatal
734 return False
735 content = self._webpage_read_content(urlh, url_or_request, video_id, note, errnote, fatal, encoding=encoding)
736 return (content, urlh)
737
738 @staticmethod
739 def _guess_encoding_from_content(content_type, webpage_bytes):
740 m = re.match(r'[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+\s*;\s*charset=(.+)', content_type)
741 if m:
742 encoding = m.group(1)
743 else:
744 m = re.search(br'<meta[^>]+charset=[\'"]?([^\'")]+)[ /\'">]',
745 webpage_bytes[:1024])
746 if m:
747 encoding = m.group(1).decode('ascii')
748 elif webpage_bytes.startswith(b'\xff\xfe'):
749 encoding = 'utf-16'
750 else:
751 encoding = 'utf-8'
752
753 return encoding
754
755 def __check_blocked(self, content):
756 first_block = content[:512]
757 if ('<title>Access to this site is blocked</title>' in content
758 and 'Websense' in first_block):
759 msg = 'Access to this webpage has been blocked by Websense filtering software in your network.'
760 blocked_iframe = self._html_search_regex(
761 r'<iframe src="([^"]+)"', content,
762 'Websense information URL', default=None)
763 if blocked_iframe:
764 msg += ' Visit %s for more details' % blocked_iframe
765 raise ExtractorError(msg, expected=True)
766 if '<title>The URL you requested has been blocked</title>' in first_block:
767 msg = (
768 'Access to this webpage has been blocked by Indian censorship. '
769 'Use a VPN or proxy server (with --proxy) to route around it.')
770 block_msg = self._html_search_regex(
771 r'</h1><p>(.*?)</p>',
772 content, 'block message', default=None)
773 if block_msg:
774 msg += ' (Message: "%s")' % block_msg.replace('\n', ' ')
775 raise ExtractorError(msg, expected=True)
776 if ('<title>TTK :: Доступ к ресурсу ограничен</title>' in content
777 and 'blocklist.rkn.gov.ru' in content):
778 raise ExtractorError(
779 'Access to this webpage has been blocked by decision of the Russian government. '
780 'Visit http://blocklist.rkn.gov.ru/ for a block reason.',
781 expected=True)
782
783 def _webpage_read_content(self, urlh, url_or_request, video_id, note=None, errnote=None, fatal=True, prefix=None, encoding=None):
784 content_type = urlh.headers.get('Content-Type', '')
785 webpage_bytes = urlh.read()
786 if prefix is not None:
787 webpage_bytes = prefix + webpage_bytes
788 if not encoding:
789 encoding = self._guess_encoding_from_content(content_type, webpage_bytes)
790 if self.get_param('dump_intermediate_pages', False):
791 self.to_screen('Dumping request to ' + urlh.geturl())
792 dump = base64.b64encode(webpage_bytes).decode('ascii')
793 self._downloader.to_screen(dump)
794 if self.get_param('write_pages', False):
795 basen = '%s_%s' % (video_id, urlh.geturl())
796 trim_length = self.get_param('trim_file_name') or 240
797 if len(basen) > trim_length:
798 h = '___' + hashlib.md5(basen.encode('utf-8')).hexdigest()
799 basen = basen[:trim_length - len(h)] + h
800 raw_filename = basen + '.dump'
801 filename = sanitize_filename(raw_filename, restricted=True)
802 self.to_screen('Saving request to ' + filename)
803 # Working around MAX_PATH limitation on Windows (see
804 # http://msdn.microsoft.com/en-us/library/windows/desktop/aa365247(v=vs.85).aspx)
805 if compat_os_name == 'nt':
806 absfilepath = os.path.abspath(filename)
807 if len(absfilepath) > 259:
808 filename = '\\\\?\\' + absfilepath
809 with open(filename, 'wb') as outf:
810 outf.write(webpage_bytes)
811
812 try:
813 content = webpage_bytes.decode(encoding, 'replace')
814 except LookupError:
815 content = webpage_bytes.decode('utf-8', 'replace')
816
817 self.__check_blocked(content)
818
819 return content
820
821 def _download_webpage(
822 self, url_or_request, video_id, note=None, errnote=None,
823 fatal=True, tries=1, timeout=5, encoding=None, data=None,
824 headers={}, query={}, expected_status=None):
825 """
826 Return the data of the page as a string.
827
828 Arguments:
829 url_or_request -- plain text URL as a string or
830 a compat_urllib_request.Requestobject
831 video_id -- Video/playlist/item identifier (string)
832
833 Keyword arguments:
834 note -- note printed before downloading (string)
835 errnote -- note printed in case of an error (string)
836 fatal -- flag denoting whether error should be considered fatal,
837 i.e. whether it should cause ExtractionError to be raised,
838 otherwise a warning will be reported and extraction continued
839 tries -- number of tries
840 timeout -- sleep interval between tries
841 encoding -- encoding for a page content decoding, guessed automatically
842 when not explicitly specified
843 data -- POST data (bytes)
844 headers -- HTTP headers (dict)
845 query -- URL query (dict)
846 expected_status -- allows to accept failed HTTP requests (non 2xx
847 status code) by explicitly specifying a set of accepted status
848 codes. Can be any of the following entities:
849 - an integer type specifying an exact failed status code to
850 accept
851 - a list or a tuple of integer types specifying a list of
852 failed status codes to accept
853 - a callable accepting an actual failed status code and
854 returning True if it should be accepted
855 Note that this argument does not affect success status codes (2xx)
856 which are always accepted.
857 """
858
859 success = False
860 try_count = 0
861 while success is False:
862 try:
863 res = self._download_webpage_handle(
864 url_or_request, video_id, note, errnote, fatal,
865 encoding=encoding, data=data, headers=headers, query=query,
866 expected_status=expected_status)
867 success = True
868 except compat_http_client.IncompleteRead as e:
869 try_count += 1
870 if try_count >= tries:
871 raise e
872 self._sleep(timeout, video_id)
873 if res is False:
874 return res
875 else:
876 content, _ = res
877 return content
878
879 def _download_xml_handle(
880 self, url_or_request, video_id, note='Downloading XML',
881 errnote='Unable to download XML', transform_source=None,
882 fatal=True, encoding=None, data=None, headers={}, query={},
883 expected_status=None):
884 """
885 Return a tuple (xml as an compat_etree_Element, URL handle).
886
887 See _download_webpage docstring for arguments specification.
888 """
889 res = self._download_webpage_handle(
890 url_or_request, video_id, note, errnote, fatal=fatal,
891 encoding=encoding, data=data, headers=headers, query=query,
892 expected_status=expected_status)
893 if res is False:
894 return res
895 xml_string, urlh = res
896 return self._parse_xml(
897 xml_string, video_id, transform_source=transform_source,
898 fatal=fatal), urlh
899
900 def _download_xml(
901 self, url_or_request, video_id,
902 note='Downloading XML', errnote='Unable to download XML',
903 transform_source=None, fatal=True, encoding=None,
904 data=None, headers={}, query={}, expected_status=None):
905 """
906 Return the xml as an compat_etree_Element.
907
908 See _download_webpage docstring for arguments specification.
909 """
910 res = self._download_xml_handle(
911 url_or_request, video_id, note=note, errnote=errnote,
912 transform_source=transform_source, fatal=fatal, encoding=encoding,
913 data=data, headers=headers, query=query,
914 expected_status=expected_status)
915 return res if res is False else res[0]
916
917 def _parse_xml(self, xml_string, video_id, transform_source=None, fatal=True):
918 if transform_source:
919 xml_string = transform_source(xml_string)
920 try:
921 return compat_etree_fromstring(xml_string.encode('utf-8'))
922 except compat_xml_parse_error as ve:
923 errmsg = '%s: Failed to parse XML ' % video_id
924 if fatal:
925 raise ExtractorError(errmsg, cause=ve)
926 else:
927 self.report_warning(errmsg + str(ve))
928
929 def _download_json_handle(
930 self, url_or_request, video_id, note='Downloading JSON metadata',
931 errnote='Unable to download JSON metadata', transform_source=None,
932 fatal=True, encoding=None, data=None, headers={}, query={},
933 expected_status=None):
934 """
935 Return a tuple (JSON object, URL handle).
936
937 See _download_webpage docstring for arguments specification.
938 """
939 res = self._download_webpage_handle(
940 url_or_request, video_id, note, errnote, fatal=fatal,
941 encoding=encoding, data=data, headers=headers, query=query,
942 expected_status=expected_status)
943 if res is False:
944 return res
945 json_string, urlh = res
946 return self._parse_json(
947 json_string, video_id, transform_source=transform_source,
948 fatal=fatal), urlh
949
950 def _download_json(
951 self, url_or_request, video_id, note='Downloading JSON metadata',
952 errnote='Unable to download JSON metadata', transform_source=None,
953 fatal=True, encoding=None, data=None, headers={}, query={},
954 expected_status=None):
955 """
956 Return the JSON object as a dict.
957
958 See _download_webpage docstring for arguments specification.
959 """
960 res = self._download_json_handle(
961 url_or_request, video_id, note=note, errnote=errnote,
962 transform_source=transform_source, fatal=fatal, encoding=encoding,
963 data=data, headers=headers, query=query,
964 expected_status=expected_status)
965 return res if res is False else res[0]
966
967 def _parse_json(self, json_string, video_id, transform_source=None, fatal=True):
968 if transform_source:
969 json_string = transform_source(json_string)
970 try:
971 return json.loads(json_string)
972 except ValueError as ve:
973 errmsg = '%s: Failed to parse JSON ' % video_id
974 if fatal:
975 raise ExtractorError(errmsg, cause=ve)
976 else:
977 self.report_warning(errmsg + str(ve))
978
979 def _parse_socket_response_as_json(self, data, video_id, transform_source=None, fatal=True):
980 return self._parse_json(
981 data[data.find('{'):data.rfind('}') + 1],
982 video_id, transform_source, fatal)
983
984 def _download_socket_json_handle(
985 self, url_or_request, video_id, note='Polling socket',
986 errnote='Unable to poll socket', transform_source=None,
987 fatal=True, encoding=None, data=None, headers={}, query={},
988 expected_status=None):
989 """
990 Return a tuple (JSON object, URL handle).
991
992 See _download_webpage docstring for arguments specification.
993 """
994 res = self._download_webpage_handle(
995 url_or_request, video_id, note, errnote, fatal=fatal,
996 encoding=encoding, data=data, headers=headers, query=query,
997 expected_status=expected_status)
998 if res is False:
999 return res
1000 webpage, urlh = res
1001 return self._parse_socket_response_as_json(
1002 webpage, video_id, transform_source=transform_source,
1003 fatal=fatal), urlh
1004
1005 def _download_socket_json(
1006 self, url_or_request, video_id, note='Polling socket',
1007 errnote='Unable to poll socket', transform_source=None,
1008 fatal=True, encoding=None, data=None, headers={}, query={},
1009 expected_status=None):
1010 """
1011 Return the JSON object as a dict.
1012
1013 See _download_webpage docstring for arguments specification.
1014 """
1015 res = self._download_socket_json_handle(
1016 url_or_request, video_id, note=note, errnote=errnote,
1017 transform_source=transform_source, fatal=fatal, encoding=encoding,
1018 data=data, headers=headers, query=query,
1019 expected_status=expected_status)
1020 return res if res is False else res[0]
1021
1022 def report_warning(self, msg, video_id=None, *args, only_once=False, **kwargs):
1023 idstr = format_field(video_id, template='%s: ')
1024 msg = f'[{self.IE_NAME}] {idstr}{msg}'
1025 if only_once:
1026 if f'WARNING: {msg}' in self._printed_messages:
1027 return
1028 self._printed_messages.add(f'WARNING: {msg}')
1029 self._downloader.report_warning(msg, *args, **kwargs)
1030
1031 def to_screen(self, msg, *args, **kwargs):
1032 """Print msg to screen, prefixing it with '[ie_name]'"""
1033 self._downloader.to_screen('[%s] %s' % (self.IE_NAME, msg), *args, **kwargs)
1034
1035 def write_debug(self, msg, *args, **kwargs):
1036 self._downloader.write_debug('[%s] %s' % (self.IE_NAME, msg), *args, **kwargs)
1037
1038 def get_param(self, name, default=None, *args, **kwargs):
1039 if self._downloader:
1040 return self._downloader.params.get(name, default, *args, **kwargs)
1041 return default
1042
1043 def report_drm(self, video_id, partial=False):
1044 self.raise_no_formats('This video is DRM protected', expected=True, video_id=video_id)
1045
1046 def report_extraction(self, id_or_name):
1047 """Report information extraction."""
1048 self.to_screen('%s: Extracting information' % id_or_name)
1049
1050 def report_download_webpage(self, video_id):
1051 """Report webpage download."""
1052 self.to_screen('%s: Downloading webpage' % video_id)
1053
1054 def report_age_confirmation(self):
1055 """Report attempt to confirm age."""
1056 self.to_screen('Confirming age')
1057
1058 def report_login(self):
1059 """Report attempt to log in."""
1060 self.to_screen('Logging in')
1061
1062 def raise_login_required(
1063 self, msg='This video is only available for registered users',
1064 metadata_available=False, method='any'):
1065 if metadata_available and self.get_param('ignore_no_formats_error'):
1066 self.report_warning(msg)
1067 if method is not None:
1068 msg = '%s. %s' % (msg, self._LOGIN_HINTS[method])
1069 raise ExtractorError(msg, expected=True)
1070
1071 def raise_geo_restricted(
1072 self, msg='This video is not available from your location due to geo restriction',
1073 countries=None, metadata_available=False):
1074 if metadata_available and self.get_param('ignore_no_formats_error'):
1075 self.report_warning(msg)
1076 else:
1077 raise GeoRestrictedError(msg, countries=countries)
1078
1079 def raise_no_formats(self, msg, expected=False, video_id=None):
1080 if expected and self.get_param('ignore_no_formats_error'):
1081 self.report_warning(msg, video_id)
1082 elif isinstance(msg, ExtractorError):
1083 raise msg
1084 else:
1085 raise ExtractorError(msg, expected=expected, video_id=video_id)
1086
1087 # Methods for following #608
1088 @staticmethod
1089 def url_result(url, ie=None, video_id=None, video_title=None):
1090 """Returns a URL that points to a page that should be processed"""
1091 # TODO: ie should be the class used for getting the info
1092 video_info = {'_type': 'url',
1093 'url': url,
1094 'ie_key': ie}
1095 if video_id is not None:
1096 video_info['id'] = video_id
1097 if video_title is not None:
1098 video_info['title'] = video_title
1099 return video_info
1100
1101 def playlist_from_matches(self, matches, playlist_id=None, playlist_title=None, getter=None, ie=None):
1102 urls = orderedSet(
1103 self.url_result(self._proto_relative_url(getter(m) if getter else m), ie)
1104 for m in matches)
1105 return self.playlist_result(
1106 urls, playlist_id=playlist_id, playlist_title=playlist_title)
1107
1108 @staticmethod
1109 def playlist_result(entries, playlist_id=None, playlist_title=None, playlist_description=None, **kwargs):
1110 """Returns a playlist"""
1111 video_info = {'_type': 'playlist',
1112 'entries': entries}
1113 video_info.update(kwargs)
1114 if playlist_id:
1115 video_info['id'] = playlist_id
1116 if playlist_title:
1117 video_info['title'] = playlist_title
1118 if playlist_description is not None:
1119 video_info['description'] = playlist_description
1120 return video_info
1121
1122 def _search_regex(self, pattern, string, name, default=NO_DEFAULT, fatal=True, flags=0, group=None):
1123 """
1124 Perform a regex search on the given string, using a single or a list of
1125 patterns returning the first matching group.
1126 In case of failure return a default value or raise a WARNING or a
1127 RegexNotFoundError, depending on fatal, specifying the field name.
1128 """
1129 if isinstance(pattern, (str, compat_str, compiled_regex_type)):
1130 mobj = re.search(pattern, string, flags)
1131 else:
1132 for p in pattern:
1133 mobj = re.search(p, string, flags)
1134 if mobj:
1135 break
1136
1137 _name = self._downloader._color_text(name, 'blue')
1138
1139 if mobj:
1140 if group is None:
1141 # return the first matching group
1142 return next(g for g in mobj.groups() if g is not None)
1143 elif isinstance(group, (list, tuple)):
1144 return tuple(mobj.group(g) for g in group)
1145 else:
1146 return mobj.group(group)
1147 elif default is not NO_DEFAULT:
1148 return default
1149 elif fatal:
1150 raise RegexNotFoundError('Unable to extract %s' % _name)
1151 else:
1152 self.report_warning('unable to extract %s' % _name + bug_reports_message())
1153 return None
1154
1155 def _html_search_regex(self, pattern, string, name, default=NO_DEFAULT, fatal=True, flags=0, group=None):
1156 """
1157 Like _search_regex, but strips HTML tags and unescapes entities.
1158 """
1159 res = self._search_regex(pattern, string, name, default, fatal, flags, group)
1160 if res:
1161 return clean_html(res).strip()
1162 else:
1163 return res
1164
1165 def _get_netrc_login_info(self, netrc_machine=None):
1166 username = None
1167 password = None
1168 netrc_machine = netrc_machine or self._NETRC_MACHINE
1169
1170 if self.get_param('usenetrc', False):
1171 try:
1172 netrc_file = compat_expanduser(self.get_param('netrc_location') or '~')
1173 if os.path.isdir(netrc_file):
1174 netrc_file = os.path.join(netrc_file, '.netrc')
1175 info = netrc.netrc(file=netrc_file).authenticators(netrc_machine)
1176 if info is not None:
1177 username = info[0]
1178 password = info[2]
1179 else:
1180 raise netrc.NetrcParseError(
1181 'No authenticators for %s' % netrc_machine)
1182 except (IOError, netrc.NetrcParseError) as err:
1183 self.report_warning(
1184 'parsing .netrc: %s' % error_to_compat_str(err))
1185
1186 return username, password
1187
1188 def _get_login_info(self, username_option='username', password_option='password', netrc_machine=None):
1189 """
1190 Get the login info as (username, password)
1191 First look for the manually specified credentials using username_option
1192 and password_option as keys in params dictionary. If no such credentials
1193 available look in the netrc file using the netrc_machine or _NETRC_MACHINE
1194 value.
1195 If there's no info available, return (None, None)
1196 """
1197
1198 # Attempt to use provided username and password or .netrc data
1199 username = self.get_param(username_option)
1200 if username is not None:
1201 password = self.get_param(password_option)
1202 else:
1203 username, password = self._get_netrc_login_info(netrc_machine)
1204
1205 return username, password
1206
1207 def _get_tfa_info(self, note='two-factor verification code'):
1208 """
1209 Get the two-factor authentication info
1210 TODO - asking the user will be required for sms/phone verify
1211 currently just uses the command line option
1212 If there's no info available, return None
1213 """
1214
1215 tfa = self.get_param('twofactor')
1216 if tfa is not None:
1217 return tfa
1218
1219 return compat_getpass('Type %s and press [Return]: ' % note)
1220
1221 # Helper functions for extracting OpenGraph info
1222 @staticmethod
1223 def _og_regexes(prop):
1224 content_re = r'content=(?:"([^"]+?)"|\'([^\']+?)\'|\s*([^\s"\'=<>`]+?))'
1225 property_re = (r'(?:name|property)=(?:\'og[:-]%(prop)s\'|"og[:-]%(prop)s"|\s*og[:-]%(prop)s\b)'
1226 % {'prop': re.escape(prop)})
1227 template = r'<meta[^>]+?%s[^>]+?%s'
1228 return [
1229 template % (property_re, content_re),
1230 template % (content_re, property_re),
1231 ]
1232
1233 @staticmethod
1234 def _meta_regex(prop):
1235 return r'''(?isx)<meta
1236 (?=[^>]+(?:itemprop|name|property|id|http-equiv)=(["\']?)%s\1)
1237 [^>]+?content=(["\'])(?P<content>.*?)\2''' % re.escape(prop)
1238
1239 def _og_search_property(self, prop, html, name=None, **kargs):
1240 prop = variadic(prop)
1241 if name is None:
1242 name = 'OpenGraph %s' % prop[0]
1243 og_regexes = []
1244 for p in prop:
1245 og_regexes.extend(self._og_regexes(p))
1246 escaped = self._search_regex(og_regexes, html, name, flags=re.DOTALL, **kargs)
1247 if escaped is None:
1248 return None
1249 return unescapeHTML(escaped)
1250
1251 def _og_search_thumbnail(self, html, **kargs):
1252 return self._og_search_property('image', html, 'thumbnail URL', fatal=False, **kargs)
1253
1254 def _og_search_description(self, html, **kargs):
1255 return self._og_search_property('description', html, fatal=False, **kargs)
1256
1257 def _og_search_title(self, html, **kargs):
1258 return self._og_search_property('title', html, **kargs)
1259
1260 def _og_search_video_url(self, html, name='video url', secure=True, **kargs):
1261 regexes = self._og_regexes('video') + self._og_regexes('video:url')
1262 if secure:
1263 regexes = self._og_regexes('video:secure_url') + regexes
1264 return self._html_search_regex(regexes, html, name, **kargs)
1265
1266 def _og_search_url(self, html, **kargs):
1267 return self._og_search_property('url', html, **kargs)
1268
1269 def _html_search_meta(self, name, html, display_name=None, fatal=False, **kwargs):
1270 name = variadic(name)
1271 if display_name is None:
1272 display_name = name[0]
1273 return self._html_search_regex(
1274 [self._meta_regex(n) for n in name],
1275 html, display_name, fatal=fatal, group='content', **kwargs)
1276
1277 def _dc_search_uploader(self, html):
1278 return self._html_search_meta('dc.creator', html, 'uploader')
1279
1280 def _rta_search(self, html):
1281 # See http://www.rtalabel.org/index.php?content=howtofaq#single
1282 if re.search(r'(?ix)<meta\s+name="rating"\s+'
1283 r' content="RTA-5042-1996-1400-1577-RTA"',
1284 html):
1285 return 18
1286 return 0
1287
1288 def _media_rating_search(self, html):
1289 # See http://www.tjg-designs.com/WP/metadata-code-examples-adding-metadata-to-your-web-pages/
1290 rating = self._html_search_meta('rating', html)
1291
1292 if not rating:
1293 return None
1294
1295 RATING_TABLE = {
1296 'safe for kids': 0,
1297 'general': 8,
1298 '14 years': 14,
1299 'mature': 17,
1300 'restricted': 19,
1301 }
1302 return RATING_TABLE.get(rating.lower())
1303
1304 def _family_friendly_search(self, html):
1305 # See http://schema.org/VideoObject
1306 family_friendly = self._html_search_meta(
1307 'isFamilyFriendly', html, default=None)
1308
1309 if not family_friendly:
1310 return None
1311
1312 RATING_TABLE = {
1313 '1': 0,
1314 'true': 0,
1315 '0': 18,
1316 'false': 18,
1317 }
1318 return RATING_TABLE.get(family_friendly.lower())
1319
1320 def _twitter_search_player(self, html):
1321 return self._html_search_meta('twitter:player', html,
1322 'twitter card player')
1323
1324 def _search_json_ld(self, html, video_id, expected_type=None, **kwargs):
1325 json_ld_list = list(re.finditer(JSON_LD_RE, html))
1326 default = kwargs.get('default', NO_DEFAULT)
1327 # JSON-LD may be malformed and thus `fatal` should be respected.
1328 # At the same time `default` may be passed that assumes `fatal=False`
1329 # for _search_regex. Let's simulate the same behavior here as well.
1330 fatal = kwargs.get('fatal', True) if default is NO_DEFAULT else False
1331 json_ld = []
1332 for mobj in json_ld_list:
1333 json_ld_item = self._parse_json(
1334 mobj.group('json_ld'), video_id, fatal=fatal)
1335 if not json_ld_item:
1336 continue
1337 if isinstance(json_ld_item, dict):
1338 json_ld.append(json_ld_item)
1339 elif isinstance(json_ld_item, (list, tuple)):
1340 json_ld.extend(json_ld_item)
1341 if json_ld:
1342 json_ld = self._json_ld(json_ld, video_id, fatal=fatal, expected_type=expected_type)
1343 if json_ld:
1344 return json_ld
1345 if default is not NO_DEFAULT:
1346 return default
1347 elif fatal:
1348 raise RegexNotFoundError('Unable to extract JSON-LD')
1349 else:
1350 self.report_warning('unable to extract JSON-LD %s' % bug_reports_message())
1351 return {}
1352
1353 def _json_ld(self, json_ld, video_id, fatal=True, expected_type=None):
1354 if isinstance(json_ld, compat_str):
1355 json_ld = self._parse_json(json_ld, video_id, fatal=fatal)
1356 if not json_ld:
1357 return {}
1358 info = {}
1359 if not isinstance(json_ld, (list, tuple, dict)):
1360 return info
1361 if isinstance(json_ld, dict):
1362 json_ld = [json_ld]
1363
1364 INTERACTION_TYPE_MAP = {
1365 'CommentAction': 'comment',
1366 'AgreeAction': 'like',
1367 'DisagreeAction': 'dislike',
1368 'LikeAction': 'like',
1369 'DislikeAction': 'dislike',
1370 'ListenAction': 'view',
1371 'WatchAction': 'view',
1372 'ViewAction': 'view',
1373 }
1374
1375 def extract_interaction_type(e):
1376 interaction_type = e.get('interactionType')
1377 if isinstance(interaction_type, dict):
1378 interaction_type = interaction_type.get('@type')
1379 return str_or_none(interaction_type)
1380
1381 def extract_interaction_statistic(e):
1382 interaction_statistic = e.get('interactionStatistic')
1383 if isinstance(interaction_statistic, dict):
1384 interaction_statistic = [interaction_statistic]
1385 if not isinstance(interaction_statistic, list):
1386 return
1387 for is_e in interaction_statistic:
1388 if not isinstance(is_e, dict):
1389 continue
1390 if is_e.get('@type') != 'InteractionCounter':
1391 continue
1392 interaction_type = extract_interaction_type(is_e)
1393 if not interaction_type:
1394 continue
1395 # For interaction count some sites provide string instead of
1396 # an integer (as per spec) with non digit characters (e.g. ",")
1397 # so extracting count with more relaxed str_to_int
1398 interaction_count = str_to_int(is_e.get('userInteractionCount'))
1399 if interaction_count is None:
1400 continue
1401 count_kind = INTERACTION_TYPE_MAP.get(interaction_type.split('/')[-1])
1402 if not count_kind:
1403 continue
1404 count_key = '%s_count' % count_kind
1405 if info.get(count_key) is not None:
1406 continue
1407 info[count_key] = interaction_count
1408
1409 def extract_video_object(e):
1410 assert e['@type'] == 'VideoObject'
1411 author = e.get('author')
1412 info.update({
1413 'url': url_or_none(e.get('contentUrl')),
1414 'title': unescapeHTML(e.get('name')),
1415 'description': unescapeHTML(e.get('description')),
1416 'thumbnail': url_or_none(e.get('thumbnailUrl') or e.get('thumbnailURL')),
1417 'duration': parse_duration(e.get('duration')),
1418 'timestamp': unified_timestamp(e.get('uploadDate')),
1419 # author can be an instance of 'Organization' or 'Person' types.
1420 # both types can have 'name' property(inherited from 'Thing' type). [1]
1421 # however some websites are using 'Text' type instead.
1422 # 1. https://schema.org/VideoObject
1423 'uploader': author.get('name') if isinstance(author, dict) else author if isinstance(author, compat_str) else None,
1424 'filesize': float_or_none(e.get('contentSize')),
1425 'tbr': int_or_none(e.get('bitrate')),
1426 'width': int_or_none(e.get('width')),
1427 'height': int_or_none(e.get('height')),
1428 'view_count': int_or_none(e.get('interactionCount')),
1429 })
1430 extract_interaction_statistic(e)
1431
1432 for e in json_ld:
1433 if '@context' in e:
1434 item_type = e.get('@type')
1435 if expected_type is not None and expected_type != item_type:
1436 continue
1437 if item_type in ('TVEpisode', 'Episode'):
1438 episode_name = unescapeHTML(e.get('name'))
1439 info.update({
1440 'episode': episode_name,
1441 'episode_number': int_or_none(e.get('episodeNumber')),
1442 'description': unescapeHTML(e.get('description')),
1443 })
1444 if not info.get('title') and episode_name:
1445 info['title'] = episode_name
1446 part_of_season = e.get('partOfSeason')
1447 if isinstance(part_of_season, dict) and part_of_season.get('@type') in ('TVSeason', 'Season', 'CreativeWorkSeason'):
1448 info.update({
1449 'season': unescapeHTML(part_of_season.get('name')),
1450 'season_number': int_or_none(part_of_season.get('seasonNumber')),
1451 })
1452 part_of_series = e.get('partOfSeries') or e.get('partOfTVSeries')
1453 if isinstance(part_of_series, dict) and part_of_series.get('@type') in ('TVSeries', 'Series', 'CreativeWorkSeries'):
1454 info['series'] = unescapeHTML(part_of_series.get('name'))
1455 elif item_type == 'Movie':
1456 info.update({
1457 'title': unescapeHTML(e.get('name')),
1458 'description': unescapeHTML(e.get('description')),
1459 'duration': parse_duration(e.get('duration')),
1460 'timestamp': unified_timestamp(e.get('dateCreated')),
1461 })
1462 elif item_type in ('Article', 'NewsArticle'):
1463 info.update({
1464 'timestamp': parse_iso8601(e.get('datePublished')),
1465 'title': unescapeHTML(e.get('headline')),
1466 'description': unescapeHTML(e.get('articleBody')),
1467 })
1468 elif item_type == 'VideoObject':
1469 extract_video_object(e)
1470 if expected_type is None:
1471 continue
1472 else:
1473 break
1474 video = e.get('video')
1475 if isinstance(video, dict) and video.get('@type') == 'VideoObject':
1476 extract_video_object(video)
1477 if expected_type is None:
1478 continue
1479 else:
1480 break
1481 return dict((k, v) for k, v in info.items() if v is not None)
1482
1483 @staticmethod
1484 def _hidden_inputs(html):
1485 html = re.sub(r'<!--(?:(?!<!--).)*-->', '', html)
1486 hidden_inputs = {}
1487 for input in re.findall(r'(?i)(<input[^>]+>)', html):
1488 attrs = extract_attributes(input)
1489 if not input:
1490 continue
1491 if attrs.get('type') not in ('hidden', 'submit'):
1492 continue
1493 name = attrs.get('name') or attrs.get('id')
1494 value = attrs.get('value')
1495 if name and value is not None:
1496 hidden_inputs[name] = value
1497 return hidden_inputs
1498
1499 def _form_hidden_inputs(self, form_id, html):
1500 form = self._search_regex(
1501 r'(?is)<form[^>]+?id=(["\'])%s\1[^>]*>(?P<form>.+?)</form>' % form_id,
1502 html, '%s form' % form_id, group='form')
1503 return self._hidden_inputs(form)
1504
1505 class FormatSort:
1506 regex = r' *((?P<reverse>\+)?(?P<field>[a-zA-Z0-9_]+)((?P<separator>[~:])(?P<limit>.*?))?)? *$'
1507
1508 default = ('hidden', 'aud_or_vid', 'hasvid', 'ie_pref', 'lang', 'quality',
1509 'res', 'fps', 'codec:vp9.2', 'size', 'br', 'asr',
1510 'proto', 'ext', 'hasaud', 'source', 'format_id') # These must not be aliases
1511 ytdl_default = ('hasaud', 'lang', 'quality', 'tbr', 'filesize', 'vbr',
1512 'height', 'width', 'proto', 'vext', 'abr', 'aext',
1513 'fps', 'fs_approx', 'source', 'format_id')
1514
1515 settings = {
1516 'vcodec': {'type': 'ordered', 'regex': True,
1517 'order': ['av0?1', 'vp0?9.2', 'vp0?9', '[hx]265|he?vc?', '[hx]264|avc', 'vp0?8', 'mp4v|h263', 'theora', '', None, 'none']},
1518 'acodec': {'type': 'ordered', 'regex': True,
1519 'order': ['opus', 'vorbis', 'aac', 'mp?4a?', 'mp3', 'e?a?c-?3', 'dts', '', None, 'none']},
1520 'proto': {'type': 'ordered', 'regex': True, 'field': 'protocol',
1521 'order': ['(ht|f)tps', '(ht|f)tp$', 'm3u8.+', '.*dash', 'ws|websocket', '', 'mms|rtsp', 'none', 'f4']},
1522 'vext': {'type': 'ordered', 'field': 'video_ext',
1523 'order': ('mp4', 'webm', 'flv', '', 'none'),
1524 'order_free': ('webm', 'mp4', 'flv', '', 'none')},
1525 'aext': {'type': 'ordered', 'field': 'audio_ext',
1526 'order': ('m4a', 'aac', 'mp3', 'ogg', 'opus', 'webm', '', 'none'),
1527 'order_free': ('opus', 'ogg', 'webm', 'm4a', 'mp3', 'aac', '', 'none')},
1528 'hidden': {'visible': False, 'forced': True, 'type': 'extractor', 'max': -1000},
1529 'aud_or_vid': {'visible': False, 'forced': True, 'type': 'multiple',
1530 'field': ('vcodec', 'acodec'),
1531 'function': lambda it: int(any(v != 'none' for v in it))},
1532 'ie_pref': {'priority': True, 'type': 'extractor'},
1533 'hasvid': {'priority': True, 'field': 'vcodec', 'type': 'boolean', 'not_in_list': ('none',)},
1534 'hasaud': {'field': 'acodec', 'type': 'boolean', 'not_in_list': ('none',)},
1535 'lang': {'convert': 'ignore', 'field': 'language_preference'},
1536 'quality': {'convert': 'float_none', 'default': -1},
1537 'filesize': {'convert': 'bytes'},
1538 'fs_approx': {'convert': 'bytes', 'field': 'filesize_approx'},
1539 'id': {'convert': 'string', 'field': 'format_id'},
1540 'height': {'convert': 'float_none'},
1541 'width': {'convert': 'float_none'},
1542 'fps': {'convert': 'float_none'},
1543 'tbr': {'convert': 'float_none'},
1544 'vbr': {'convert': 'float_none'},
1545 'abr': {'convert': 'float_none'},
1546 'asr': {'convert': 'float_none'},
1547 'source': {'convert': 'ignore', 'field': 'source_preference'},
1548
1549 'codec': {'type': 'combined', 'field': ('vcodec', 'acodec')},
1550 'br': {'type': 'combined', 'field': ('tbr', 'vbr', 'abr'), 'same_limit': True},
1551 'size': {'type': 'combined', 'same_limit': True, 'field': ('filesize', 'fs_approx')},
1552 'ext': {'type': 'combined', 'field': ('vext', 'aext')},
1553 'res': {'type': 'multiple', 'field': ('height', 'width'),
1554 'function': lambda it: (lambda l: min(l) if l else 0)(tuple(filter(None, it)))},
1555
1556 # Most of these exist only for compatibility reasons
1557 'dimension': {'type': 'alias', 'field': 'res'},
1558 'resolution': {'type': 'alias', 'field': 'res'},
1559 'extension': {'type': 'alias', 'field': 'ext'},
1560 'bitrate': {'type': 'alias', 'field': 'br'},
1561 'total_bitrate': {'type': 'alias', 'field': 'tbr'},
1562 'video_bitrate': {'type': 'alias', 'field': 'vbr'},
1563 'audio_bitrate': {'type': 'alias', 'field': 'abr'},
1564 'framerate': {'type': 'alias', 'field': 'fps'},
1565 'language_preference': {'type': 'alias', 'field': 'lang'}, # not named as 'language' because such a field exists
1566 'protocol': {'type': 'alias', 'field': 'proto'},
1567 'source_preference': {'type': 'alias', 'field': 'source'},
1568 'filesize_approx': {'type': 'alias', 'field': 'fs_approx'},
1569 'filesize_estimate': {'type': 'alias', 'field': 'size'},
1570 'samplerate': {'type': 'alias', 'field': 'asr'},
1571 'video_ext': {'type': 'alias', 'field': 'vext'},
1572 'audio_ext': {'type': 'alias', 'field': 'aext'},
1573 'video_codec': {'type': 'alias', 'field': 'vcodec'},
1574 'audio_codec': {'type': 'alias', 'field': 'acodec'},
1575 'video': {'type': 'alias', 'field': 'hasvid'},
1576 'has_video': {'type': 'alias', 'field': 'hasvid'},
1577 'audio': {'type': 'alias', 'field': 'hasaud'},
1578 'has_audio': {'type': 'alias', 'field': 'hasaud'},
1579 'extractor': {'type': 'alias', 'field': 'ie_pref'},
1580 'preference': {'type': 'alias', 'field': 'ie_pref'},
1581 'extractor_preference': {'type': 'alias', 'field': 'ie_pref'},
1582 'format_id': {'type': 'alias', 'field': 'id'},
1583 }
1584
1585 _order = []
1586
1587 def _get_field_setting(self, field, key):
1588 if field not in self.settings:
1589 self.settings[field] = {}
1590 propObj = self.settings[field]
1591 if key not in propObj:
1592 type = propObj.get('type')
1593 if key == 'field':
1594 default = 'preference' if type == 'extractor' else (field,) if type in ('combined', 'multiple') else field
1595 elif key == 'convert':
1596 default = 'order' if type == 'ordered' else 'float_string' if field else 'ignore'
1597 else:
1598 default = {'type': 'field', 'visible': True, 'order': [], 'not_in_list': (None,)}.get(key, None)
1599 propObj[key] = default
1600 return propObj[key]
1601
1602 def _resolve_field_value(self, field, value, convertNone=False):
1603 if value is None:
1604 if not convertNone:
1605 return None
1606 else:
1607 value = value.lower()
1608 conversion = self._get_field_setting(field, 'convert')
1609 if conversion == 'ignore':
1610 return None
1611 if conversion == 'string':
1612 return value
1613 elif conversion == 'float_none':
1614 return float_or_none(value)
1615 elif conversion == 'bytes':
1616 return FileDownloader.parse_bytes(value)
1617 elif conversion == 'order':
1618 order_list = (self._use_free_order and self._get_field_setting(field, 'order_free')) or self._get_field_setting(field, 'order')
1619 use_regex = self._get_field_setting(field, 'regex')
1620 list_length = len(order_list)
1621 empty_pos = order_list.index('') if '' in order_list else list_length + 1
1622 if use_regex and value is not None:
1623 for i, regex in enumerate(order_list):
1624 if regex and re.match(regex, value):
1625 return list_length - i
1626 return list_length - empty_pos # not in list
1627 else: # not regex or value = None
1628 return list_length - (order_list.index(value) if value in order_list else empty_pos)
1629 else:
1630 if value.isnumeric():
1631 return float(value)
1632 else:
1633 self.settings[field]['convert'] = 'string'
1634 return value
1635
1636 def evaluate_params(self, params, sort_extractor):
1637 self._use_free_order = params.get('prefer_free_formats', False)
1638 self._sort_user = params.get('format_sort', [])
1639 self._sort_extractor = sort_extractor
1640
1641 def add_item(field, reverse, closest, limit_text):
1642 field = field.lower()
1643 if field in self._order:
1644 return
1645 self._order.append(field)
1646 limit = self._resolve_field_value(field, limit_text)
1647 data = {
1648 'reverse': reverse,
1649 'closest': False if limit is None else closest,
1650 'limit_text': limit_text,
1651 'limit': limit}
1652 if field in self.settings:
1653 self.settings[field].update(data)
1654 else:
1655 self.settings[field] = data
1656
1657 sort_list = (
1658 tuple(field for field in self.default if self._get_field_setting(field, 'forced'))
1659 + (tuple() if params.get('format_sort_force', False)
1660 else tuple(field for field in self.default if self._get_field_setting(field, 'priority')))
1661 + tuple(self._sort_user) + tuple(sort_extractor) + self.default)
1662
1663 for item in sort_list:
1664 match = re.match(self.regex, item)
1665 if match is None:
1666 raise ExtractorError('Invalid format sort string "%s" given by extractor' % item)
1667 field = match.group('field')
1668 if field is None:
1669 continue
1670 if self._get_field_setting(field, 'type') == 'alias':
1671 field = self._get_field_setting(field, 'field')
1672 reverse = match.group('reverse') is not None
1673 closest = match.group('separator') == '~'
1674 limit_text = match.group('limit')
1675
1676 has_limit = limit_text is not None
1677 has_multiple_fields = self._get_field_setting(field, 'type') == 'combined'
1678 has_multiple_limits = has_limit and has_multiple_fields and not self._get_field_setting(field, 'same_limit')
1679
1680 fields = self._get_field_setting(field, 'field') if has_multiple_fields else (field,)
1681 limits = limit_text.split(':') if has_multiple_limits else (limit_text,) if has_limit else tuple()
1682 limit_count = len(limits)
1683 for (i, f) in enumerate(fields):
1684 add_item(f, reverse, closest,
1685 limits[i] if i < limit_count
1686 else limits[0] if has_limit and not has_multiple_limits
1687 else None)
1688
1689 def print_verbose_info(self, write_debug):
1690 if self._sort_user:
1691 write_debug('Sort order given by user: %s' % ', '.join(self._sort_user))
1692 if self._sort_extractor:
1693 write_debug('Sort order given by extractor: %s' % ', '.join(self._sort_extractor))
1694 write_debug('Formats sorted by: %s' % ', '.join(['%s%s%s' % (
1695 '+' if self._get_field_setting(field, 'reverse') else '', field,
1696 '%s%s(%s)' % ('~' if self._get_field_setting(field, 'closest') else ':',
1697 self._get_field_setting(field, 'limit_text'),
1698 self._get_field_setting(field, 'limit'))
1699 if self._get_field_setting(field, 'limit_text') is not None else '')
1700 for field in self._order if self._get_field_setting(field, 'visible')]))
1701
1702 def _calculate_field_preference_from_value(self, format, field, type, value):
1703 reverse = self._get_field_setting(field, 'reverse')
1704 closest = self._get_field_setting(field, 'closest')
1705 limit = self._get_field_setting(field, 'limit')
1706
1707 if type == 'extractor':
1708 maximum = self._get_field_setting(field, 'max')
1709 if value is None or (maximum is not None and value >= maximum):
1710 value = -1
1711 elif type == 'boolean':
1712 in_list = self._get_field_setting(field, 'in_list')
1713 not_in_list = self._get_field_setting(field, 'not_in_list')
1714 value = 0 if ((in_list is None or value in in_list) and (not_in_list is None or value not in not_in_list)) else -1
1715 elif type == 'ordered':
1716 value = self._resolve_field_value(field, value, True)
1717
1718 # try to convert to number
1719 val_num = float_or_none(value, default=self._get_field_setting(field, 'default'))
1720 is_num = self._get_field_setting(field, 'convert') != 'string' and val_num is not None
1721 if is_num:
1722 value = val_num
1723
1724 return ((-10, 0) if value is None
1725 else (1, value, 0) if not is_num # if a field has mixed strings and numbers, strings are sorted higher
1726 else (0, -abs(value - limit), value - limit if reverse else limit - value) if closest
1727 else (0, value, 0) if not reverse and (limit is None or value <= limit)
1728 else (0, -value, 0) if limit is None or (reverse and value == limit) or value > limit
1729 else (-1, value, 0))
1730
1731 def _calculate_field_preference(self, format, field):
1732 type = self._get_field_setting(field, 'type') # extractor, boolean, ordered, field, multiple
1733 get_value = lambda f: format.get(self._get_field_setting(f, 'field'))
1734 if type == 'multiple':
1735 type = 'field' # Only 'field' is allowed in multiple for now
1736 actual_fields = self._get_field_setting(field, 'field')
1737
1738 value = self._get_field_setting(field, 'function')(get_value(f) for f in actual_fields)
1739 else:
1740 value = get_value(field)
1741 return self._calculate_field_preference_from_value(format, field, type, value)
1742
1743 def calculate_preference(self, format):
1744 # Determine missing protocol
1745 if not format.get('protocol'):
1746 format['protocol'] = determine_protocol(format)
1747
1748 # Determine missing ext
1749 if not format.get('ext') and 'url' in format:
1750 format['ext'] = determine_ext(format['url'])
1751 if format.get('vcodec') == 'none':
1752 format['audio_ext'] = format['ext'] if format.get('acodec') != 'none' else 'none'
1753 format['video_ext'] = 'none'
1754 else:
1755 format['video_ext'] = format['ext']
1756 format['audio_ext'] = 'none'
1757 # if format.get('preference') is None and format.get('ext') in ('f4f', 'f4m'): # Not supported?
1758 # format['preference'] = -1000
1759
1760 # Determine missing bitrates
1761 if format.get('tbr') is None:
1762 if format.get('vbr') is not None and format.get('abr') is not None:
1763 format['tbr'] = format.get('vbr', 0) + format.get('abr', 0)
1764 else:
1765 if format.get('vcodec') != 'none' and format.get('vbr') is None:
1766 format['vbr'] = format.get('tbr') - format.get('abr', 0)
1767 if format.get('acodec') != 'none' and format.get('abr') is None:
1768 format['abr'] = format.get('tbr') - format.get('vbr', 0)
1769
1770 return tuple(self._calculate_field_preference(format, field) for field in self._order)
1771
1772 def _sort_formats(self, formats, field_preference=[]):
1773 if not formats:
1774 return
1775 format_sort = self.FormatSort() # params and to_screen are taken from the downloader
1776 format_sort.evaluate_params(self._downloader.params, field_preference)
1777 if self.get_param('verbose', False):
1778 format_sort.print_verbose_info(self._downloader.write_debug)
1779 formats.sort(key=lambda f: format_sort.calculate_preference(f))
1780
1781 def _check_formats(self, formats, video_id):
1782 if formats:
1783 formats[:] = filter(
1784 lambda f: self._is_valid_url(
1785 f['url'], video_id,
1786 item='%s video format' % f.get('format_id') if f.get('format_id') else 'video'),
1787 formats)
1788
1789 @staticmethod
1790 def _remove_duplicate_formats(formats):
1791 format_urls = set()
1792 unique_formats = []
1793 for f in formats:
1794 if f['url'] not in format_urls:
1795 format_urls.add(f['url'])
1796 unique_formats.append(f)
1797 formats[:] = unique_formats
1798
1799 def _is_valid_url(self, url, video_id, item='video', headers={}):
1800 url = self._proto_relative_url(url, scheme='http:')
1801 # For now assume non HTTP(S) URLs always valid
1802 if not (url.startswith('http://') or url.startswith('https://')):
1803 return True
1804 try:
1805 self._request_webpage(url, video_id, 'Checking %s URL' % item, headers=headers)
1806 return True
1807 except ExtractorError as e:
1808 self.to_screen(
1809 '%s: %s URL is invalid, skipping: %s'
1810 % (video_id, item, error_to_compat_str(e.cause)))
1811 return False
1812
1813 def http_scheme(self):
1814 """ Either "http:" or "https:", depending on the user's preferences """
1815 return (
1816 'http:'
1817 if self.get_param('prefer_insecure', False)
1818 else 'https:')
1819
1820 def _proto_relative_url(self, url, scheme=None):
1821 if url is None:
1822 return url
1823 if url.startswith('//'):
1824 if scheme is None:
1825 scheme = self.http_scheme()
1826 return scheme + url
1827 else:
1828 return url
1829
1830 def _sleep(self, timeout, video_id, msg_template=None):
1831 if msg_template is None:
1832 msg_template = '%(video_id)s: Waiting for %(timeout)s seconds'
1833 msg = msg_template % {'video_id': video_id, 'timeout': timeout}
1834 self.to_screen(msg)
1835 time.sleep(timeout)
1836
1837 def _extract_f4m_formats(self, manifest_url, video_id, preference=None, quality=None, f4m_id=None,
1838 transform_source=lambda s: fix_xml_ampersands(s).strip(),
1839 fatal=True, m3u8_id=None, data=None, headers={}, query={}):
1840 manifest = self._download_xml(
1841 manifest_url, video_id, 'Downloading f4m manifest',
1842 'Unable to download f4m manifest',
1843 # Some manifests may be malformed, e.g. prosiebensat1 generated manifests
1844 # (see https://github.com/ytdl-org/youtube-dl/issues/6215#issuecomment-121704244)
1845 transform_source=transform_source,
1846 fatal=fatal, data=data, headers=headers, query=query)
1847
1848 if manifest is False:
1849 return []
1850
1851 return self._parse_f4m_formats(
1852 manifest, manifest_url, video_id, preference=preference, quality=quality, f4m_id=f4m_id,
1853 transform_source=transform_source, fatal=fatal, m3u8_id=m3u8_id)
1854
1855 def _parse_f4m_formats(self, manifest, manifest_url, video_id, preference=None, quality=None, f4m_id=None,
1856 transform_source=lambda s: fix_xml_ampersands(s).strip(),
1857 fatal=True, m3u8_id=None):
1858 if not isinstance(manifest, compat_etree_Element) and not fatal:
1859 return []
1860
1861 # currently yt-dlp cannot decode the playerVerificationChallenge as Akamai uses Adobe Alchemy
1862 akamai_pv = manifest.find('{http://ns.adobe.com/f4m/1.0}pv-2.0')
1863 if akamai_pv is not None and ';' in akamai_pv.text:
1864 playerVerificationChallenge = akamai_pv.text.split(';')[0]
1865 if playerVerificationChallenge.strip() != '':
1866 return []
1867
1868 formats = []
1869 manifest_version = '1.0'
1870 media_nodes = manifest.findall('{http://ns.adobe.com/f4m/1.0}media')
1871 if not media_nodes:
1872 manifest_version = '2.0'
1873 media_nodes = manifest.findall('{http://ns.adobe.com/f4m/2.0}media')
1874 # Remove unsupported DRM protected media from final formats
1875 # rendition (see https://github.com/ytdl-org/youtube-dl/issues/8573).
1876 media_nodes = remove_encrypted_media(media_nodes)
1877 if not media_nodes:
1878 return formats
1879
1880 manifest_base_url = get_base_url(manifest)
1881
1882 bootstrap_info = xpath_element(
1883 manifest, ['{http://ns.adobe.com/f4m/1.0}bootstrapInfo', '{http://ns.adobe.com/f4m/2.0}bootstrapInfo'],
1884 'bootstrap info', default=None)
1885
1886 vcodec = None
1887 mime_type = xpath_text(
1888 manifest, ['{http://ns.adobe.com/f4m/1.0}mimeType', '{http://ns.adobe.com/f4m/2.0}mimeType'],
1889 'base URL', default=None)
1890 if mime_type and mime_type.startswith('audio/'):
1891 vcodec = 'none'
1892
1893 for i, media_el in enumerate(media_nodes):
1894 tbr = int_or_none(media_el.attrib.get('bitrate'))
1895 width = int_or_none(media_el.attrib.get('width'))
1896 height = int_or_none(media_el.attrib.get('height'))
1897 format_id = '-'.join(filter(None, [f4m_id, compat_str(i if tbr is None else tbr)]))
1898 # If <bootstrapInfo> is present, the specified f4m is a
1899 # stream-level manifest, and only set-level manifests may refer to
1900 # external resources. See section 11.4 and section 4 of F4M spec
1901 if bootstrap_info is None:
1902 media_url = None
1903 # @href is introduced in 2.0, see section 11.6 of F4M spec
1904 if manifest_version == '2.0':
1905 media_url = media_el.attrib.get('href')
1906 if media_url is None:
1907 media_url = media_el.attrib.get('url')
1908 if not media_url:
1909 continue
1910 manifest_url = (
1911 media_url if media_url.startswith('http://') or media_url.startswith('https://')
1912 else ((manifest_base_url or '/'.join(manifest_url.split('/')[:-1])) + '/' + media_url))
1913 # If media_url is itself a f4m manifest do the recursive extraction
1914 # since bitrates in parent manifest (this one) and media_url manifest
1915 # may differ leading to inability to resolve the format by requested
1916 # bitrate in f4m downloader
1917 ext = determine_ext(manifest_url)
1918 if ext == 'f4m':
1919 f4m_formats = self._extract_f4m_formats(
1920 manifest_url, video_id, preference=preference, quality=quality, f4m_id=f4m_id,
1921 transform_source=transform_source, fatal=fatal)
1922 # Sometimes stream-level manifest contains single media entry that
1923 # does not contain any quality metadata (e.g. http://matchtv.ru/#live-player).
1924 # At the same time parent's media entry in set-level manifest may
1925 # contain it. We will copy it from parent in such cases.
1926 if len(f4m_formats) == 1:
1927 f = f4m_formats[0]
1928 f.update({
1929 'tbr': f.get('tbr') or tbr,
1930 'width': f.get('width') or width,
1931 'height': f.get('height') or height,
1932 'format_id': f.get('format_id') if not tbr else format_id,
1933 'vcodec': vcodec,
1934 })
1935 formats.extend(f4m_formats)
1936 continue
1937 elif ext == 'm3u8':
1938 formats.extend(self._extract_m3u8_formats(
1939 manifest_url, video_id, 'mp4', preference=preference,
1940 quality=quality, m3u8_id=m3u8_id, fatal=fatal))
1941 continue
1942 formats.append({
1943 'format_id': format_id,
1944 'url': manifest_url,
1945 'manifest_url': manifest_url,
1946 'ext': 'flv' if bootstrap_info is not None else None,
1947 'protocol': 'f4m',
1948 'tbr': tbr,
1949 'width': width,
1950 'height': height,
1951 'vcodec': vcodec,
1952 'preference': preference,
1953 'quality': quality,
1954 })
1955 return formats
1956
1957 def _m3u8_meta_format(self, m3u8_url, ext=None, preference=None, quality=None, m3u8_id=None):
1958 return {
1959 'format_id': '-'.join(filter(None, [m3u8_id, 'meta'])),
1960 'url': m3u8_url,
1961 'ext': ext,
1962 'protocol': 'm3u8',
1963 'preference': preference - 100 if preference else -100,
1964 'quality': quality,
1965 'resolution': 'multiple',
1966 'format_note': 'Quality selection URL',
1967 }
1968
1969 def _report_ignoring_subs(self, name):
1970 self.report_warning(bug_reports_message(
1971 f'Ignoring subtitle tracks found in the {name} manifest; '
1972 'if any subtitle tracks are missing,'
1973 ), only_once=True)
1974
1975 def _extract_m3u8_formats(self, *args, **kwargs):
1976 fmts, subs = self._extract_m3u8_formats_and_subtitles(*args, **kwargs)
1977 if subs:
1978 self._report_ignoring_subs('HLS')
1979 return fmts
1980
1981 def _extract_m3u8_formats_and_subtitles(
1982 self, m3u8_url, video_id, ext=None, entry_protocol='m3u8_native',
1983 preference=None, quality=None, m3u8_id=None, note=None,
1984 errnote=None, fatal=True, live=False, data=None, headers={},
1985 query={}):
1986
1987 res = self._download_webpage_handle(
1988 m3u8_url, video_id,
1989 note='Downloading m3u8 information' if note is None else note,
1990 errnote='Failed to download m3u8 information' if errnote is None else errnote,
1991 fatal=fatal, data=data, headers=headers, query=query)
1992
1993 if res is False:
1994 return [], {}
1995
1996 m3u8_doc, urlh = res
1997 m3u8_url = urlh.geturl()
1998
1999 return self._parse_m3u8_formats_and_subtitles(
2000 m3u8_doc, m3u8_url, ext=ext, entry_protocol=entry_protocol,
2001 preference=preference, quality=quality, m3u8_id=m3u8_id,
2002 note=note, errnote=errnote, fatal=fatal, live=live, data=data,
2003 headers=headers, query=query, video_id=video_id)
2004
2005 def _parse_m3u8_formats_and_subtitles(
2006 self, m3u8_doc, m3u8_url, ext=None, entry_protocol='m3u8_native',
2007 preference=None, quality=None, m3u8_id=None, live=False, note=None,
2008 errnote=None, fatal=True, data=None, headers={}, query={},
2009 video_id=None):
2010 formats, subtitles = [], {}
2011
2012 if '#EXT-X-FAXS-CM:' in m3u8_doc: # Adobe Flash Access
2013 return formats, subtitles
2014
2015 has_drm = re.search(r'#EXT-X-SESSION-KEY:.*?URI="skd://', m3u8_doc)
2016
2017 def format_url(url):
2018 return url if re.match(r'^https?://', url) else compat_urlparse.urljoin(m3u8_url, url)
2019
2020 if self.get_param('hls_split_discontinuity', False):
2021 def _extract_m3u8_playlist_indices(manifest_url=None, m3u8_doc=None):
2022 if not m3u8_doc:
2023 if not manifest_url:
2024 return []
2025 m3u8_doc = self._download_webpage(
2026 manifest_url, video_id, fatal=fatal, data=data, headers=headers,
2027 note=False, errnote='Failed to download m3u8 playlist information')
2028 if m3u8_doc is False:
2029 return []
2030 return range(1 + sum(line.startswith('#EXT-X-DISCONTINUITY') for line in m3u8_doc.splitlines()))
2031
2032 else:
2033 def _extract_m3u8_playlist_indices(*args, **kwargs):
2034 return [None]
2035
2036 # References:
2037 # 1. https://tools.ietf.org/html/draft-pantos-http-live-streaming-21
2038 # 2. https://github.com/ytdl-org/youtube-dl/issues/12211
2039 # 3. https://github.com/ytdl-org/youtube-dl/issues/18923
2040
2041 # We should try extracting formats only from master playlists [1, 4.3.4],
2042 # i.e. playlists that describe available qualities. On the other hand
2043 # media playlists [1, 4.3.3] should be returned as is since they contain
2044 # just the media without qualities renditions.
2045 # Fortunately, master playlist can be easily distinguished from media
2046 # playlist based on particular tags availability. As of [1, 4.3.3, 4.3.4]
2047 # master playlist tags MUST NOT appear in a media playlist and vice versa.
2048 # As of [1, 4.3.3.1] #EXT-X-TARGETDURATION tag is REQUIRED for every
2049 # media playlist and MUST NOT appear in master playlist thus we can
2050 # clearly detect media playlist with this criterion.
2051
2052 if '#EXT-X-TARGETDURATION' in m3u8_doc: # media playlist, return as is
2053 formats = [{
2054 'format_id': '-'.join(map(str, filter(None, [m3u8_id, idx]))),
2055 'format_index': idx,
2056 'url': m3u8_url,
2057 'ext': ext,
2058 'protocol': entry_protocol,
2059 'preference': preference,
2060 'quality': quality,
2061 'has_drm': has_drm,
2062 } for idx in _extract_m3u8_playlist_indices(m3u8_doc=m3u8_doc)]
2063
2064 return formats, subtitles
2065
2066 groups = {}
2067 last_stream_inf = {}
2068
2069 def extract_media(x_media_line):
2070 media = parse_m3u8_attributes(x_media_line)
2071 # As per [1, 4.3.4.1] TYPE, GROUP-ID and NAME are REQUIRED
2072 media_type, group_id, name = media.get('TYPE'), media.get('GROUP-ID'), media.get('NAME')
2073 if not (media_type and group_id and name):
2074 return
2075 groups.setdefault(group_id, []).append(media)
2076 # <https://tools.ietf.org/html/rfc8216#section-4.3.4.1>
2077 if media_type == 'SUBTITLES':
2078 # According to RFC 8216 §4.3.4.2.1, URI is REQUIRED in the
2079 # EXT-X-MEDIA tag if the media type is SUBTITLES.
2080 # However, lack of URI has been spotted in the wild.
2081 # e.g. NebulaIE; see https://github.com/yt-dlp/yt-dlp/issues/339
2082 if not media.get('URI'):
2083 return
2084 url = format_url(media['URI'])
2085 sub_info = {
2086 'url': url,
2087 'ext': determine_ext(url),
2088 }
2089 if sub_info['ext'] == 'm3u8':
2090 # Per RFC 8216 §3.1, the only possible subtitle format m3u8
2091 # files may contain is WebVTT:
2092 # <https://tools.ietf.org/html/rfc8216#section-3.1>
2093 sub_info['ext'] = 'vtt'
2094 sub_info['protocol'] = 'm3u8_native'
2095 lang = media.get('LANGUAGE') or 'und'
2096 subtitles.setdefault(lang, []).append(sub_info)
2097 if media_type not in ('VIDEO', 'AUDIO'):
2098 return
2099 media_url = media.get('URI')
2100 if media_url:
2101 manifest_url = format_url(media_url)
2102 formats.extend({
2103 'format_id': '-'.join(map(str, filter(None, (m3u8_id, group_id, name, idx)))),
2104 'format_note': name,
2105 'format_index': idx,
2106 'url': manifest_url,
2107 'manifest_url': m3u8_url,
2108 'language': media.get('LANGUAGE'),
2109 'ext': ext,
2110 'protocol': entry_protocol,
2111 'preference': preference,
2112 'quality': quality,
2113 'vcodec': 'none' if media_type == 'AUDIO' else None,
2114 } for idx in _extract_m3u8_playlist_indices(manifest_url))
2115
2116 def build_stream_name():
2117 # Despite specification does not mention NAME attribute for
2118 # EXT-X-STREAM-INF tag it still sometimes may be present (see [1]
2119 # or vidio test in TestInfoExtractor.test_parse_m3u8_formats)
2120 # 1. http://www.vidio.com/watch/165683-dj_ambred-booyah-live-2015
2121 stream_name = last_stream_inf.get('NAME')
2122 if stream_name:
2123 return stream_name
2124 # If there is no NAME in EXT-X-STREAM-INF it will be obtained
2125 # from corresponding rendition group
2126 stream_group_id = last_stream_inf.get('VIDEO')
2127 if not stream_group_id:
2128 return
2129 stream_group = groups.get(stream_group_id)
2130 if not stream_group:
2131 return stream_group_id
2132 rendition = stream_group[0]
2133 return rendition.get('NAME') or stream_group_id
2134
2135 # parse EXT-X-MEDIA tags before EXT-X-STREAM-INF in order to have the
2136 # chance to detect video only formats when EXT-X-STREAM-INF tags
2137 # precede EXT-X-MEDIA tags in HLS manifest such as [3].
2138 for line in m3u8_doc.splitlines():
2139 if line.startswith('#EXT-X-MEDIA:'):
2140 extract_media(line)
2141
2142 for line in m3u8_doc.splitlines():
2143 if line.startswith('#EXT-X-STREAM-INF:'):
2144 last_stream_inf = parse_m3u8_attributes(line)
2145 elif line.startswith('#') or not line.strip():
2146 continue
2147 else:
2148 tbr = float_or_none(
2149 last_stream_inf.get('AVERAGE-BANDWIDTH')
2150 or last_stream_inf.get('BANDWIDTH'), scale=1000)
2151 manifest_url = format_url(line.strip())
2152
2153 for idx in _extract_m3u8_playlist_indices(manifest_url):
2154 format_id = [m3u8_id, None, idx]
2155 # Bandwidth of live streams may differ over time thus making
2156 # format_id unpredictable. So it's better to keep provided
2157 # format_id intact.
2158 if not live:
2159 stream_name = build_stream_name()
2160 format_id[1] = stream_name if stream_name else '%d' % (tbr if tbr else len(formats))
2161 f = {
2162 'format_id': '-'.join(map(str, filter(None, format_id))),
2163 'format_index': idx,
2164 'url': manifest_url,
2165 'manifest_url': m3u8_url,
2166 'tbr': tbr,
2167 'ext': ext,
2168 'fps': float_or_none(last_stream_inf.get('FRAME-RATE')),
2169 'protocol': entry_protocol,
2170 'preference': preference,
2171 'quality': quality,
2172 }
2173 resolution = last_stream_inf.get('RESOLUTION')
2174 if resolution:
2175 mobj = re.search(r'(?P<width>\d+)[xX](?P<height>\d+)', resolution)
2176 if mobj:
2177 f['width'] = int(mobj.group('width'))
2178 f['height'] = int(mobj.group('height'))
2179 # Unified Streaming Platform
2180 mobj = re.search(
2181 r'audio.*?(?:%3D|=)(\d+)(?:-video.*?(?:%3D|=)(\d+))?', f['url'])
2182 if mobj:
2183 abr, vbr = mobj.groups()
2184 abr, vbr = float_or_none(abr, 1000), float_or_none(vbr, 1000)
2185 f.update({
2186 'vbr': vbr,
2187 'abr': abr,
2188 })
2189 codecs = parse_codecs(last_stream_inf.get('CODECS'))
2190 f.update(codecs)
2191 audio_group_id = last_stream_inf.get('AUDIO')
2192 # As per [1, 4.3.4.1.1] any EXT-X-STREAM-INF tag which
2193 # references a rendition group MUST have a CODECS attribute.
2194 # However, this is not always respected, for example, [2]
2195 # contains EXT-X-STREAM-INF tag which references AUDIO
2196 # rendition group but does not have CODECS and despite
2197 # referencing an audio group it represents a complete
2198 # (with audio and video) format. So, for such cases we will
2199 # ignore references to rendition groups and treat them
2200 # as complete formats.
2201 if audio_group_id and codecs and f.get('vcodec') != 'none':
2202 audio_group = groups.get(audio_group_id)
2203 if audio_group and audio_group[0].get('URI'):
2204 # TODO: update acodec for audio only formats with
2205 # the same GROUP-ID
2206 f['acodec'] = 'none'
2207 if not f.get('ext'):
2208 f['ext'] = 'm4a' if f.get('vcodec') == 'none' else 'mp4'
2209 formats.append(f)
2210
2211 # for DailyMotion
2212 progressive_uri = last_stream_inf.get('PROGRESSIVE-URI')
2213 if progressive_uri:
2214 http_f = f.copy()
2215 del http_f['manifest_url']
2216 http_f.update({
2217 'format_id': f['format_id'].replace('hls-', 'http-'),
2218 'protocol': 'http',
2219 'url': progressive_uri,
2220 })
2221 formats.append(http_f)
2222
2223 last_stream_inf = {}
2224 return formats, subtitles
2225
2226 def _extract_m3u8_vod_duration(
2227 self, m3u8_vod_url, video_id, note=None, errnote=None, data=None, headers={}, query={}):
2228
2229 m3u8_vod = self._download_webpage(
2230 m3u8_vod_url, video_id,
2231 note='Downloading m3u8 VOD manifest' if note is None else note,
2232 errnote='Failed to download VOD manifest' if errnote is None else errnote,
2233 fatal=False, data=data, headers=headers, query=query)
2234
2235 return self._parse_m3u8_vod_duration(m3u8_vod or '', video_id)
2236
2237 def _parse_m3u8_vod_duration(self, m3u8_vod, video_id):
2238 if '#EXT-X-PLAYLIST-TYPE:VOD' not in m3u8_vod:
2239 return None
2240
2241 return int(sum(
2242 float(line[len('#EXTINF:'):].split(',')[0])
2243 for line in m3u8_vod.splitlines() if line.startswith('#EXTINF:'))) or None
2244
2245 @staticmethod
2246 def _xpath_ns(path, namespace=None):
2247 if not namespace:
2248 return path
2249 out = []
2250 for c in path.split('/'):
2251 if not c or c == '.':
2252 out.append(c)
2253 else:
2254 out.append('{%s}%s' % (namespace, c))
2255 return '/'.join(out)
2256
2257 def _extract_smil_formats_and_subtitles(self, smil_url, video_id, fatal=True, f4m_params=None, transform_source=None):
2258 smil = self._download_smil(smil_url, video_id, fatal=fatal, transform_source=transform_source)
2259
2260 if smil is False:
2261 assert not fatal
2262 return []
2263
2264 namespace = self._parse_smil_namespace(smil)
2265
2266 fmts = self._parse_smil_formats(
2267 smil, smil_url, video_id, namespace=namespace, f4m_params=f4m_params)
2268 subs = self._parse_smil_subtitles(
2269 smil, namespace=namespace)
2270
2271 return fmts, subs
2272
2273 def _extract_smil_formats(self, *args, **kwargs):
2274 fmts, subs = self._extract_smil_formats_and_subtitles(*args, **kwargs)
2275 if subs:
2276 self._report_ignoring_subs('SMIL')
2277 return fmts
2278
2279 def _extract_smil_info(self, smil_url, video_id, fatal=True, f4m_params=None):
2280 smil = self._download_smil(smil_url, video_id, fatal=fatal)
2281 if smil is False:
2282 return {}
2283 return self._parse_smil(smil, smil_url, video_id, f4m_params=f4m_params)
2284
2285 def _download_smil(self, smil_url, video_id, fatal=True, transform_source=None):
2286 return self._download_xml(
2287 smil_url, video_id, 'Downloading SMIL file',
2288 'Unable to download SMIL file', fatal=fatal, transform_source=transform_source)
2289
2290 def _parse_smil(self, smil, smil_url, video_id, f4m_params=None):
2291 namespace = self._parse_smil_namespace(smil)
2292
2293 formats = self._parse_smil_formats(
2294 smil, smil_url, video_id, namespace=namespace, f4m_params=f4m_params)
2295 subtitles = self._parse_smil_subtitles(smil, namespace=namespace)
2296
2297 video_id = os.path.splitext(url_basename(smil_url))[0]
2298 title = None
2299 description = None
2300 upload_date = None
2301 for meta in smil.findall(self._xpath_ns('./head/meta', namespace)):
2302 name = meta.attrib.get('name')
2303 content = meta.attrib.get('content')
2304 if not name or not content:
2305 continue
2306 if not title and name == 'title':
2307 title = content
2308 elif not description and name in ('description', 'abstract'):
2309 description = content
2310 elif not upload_date and name == 'date':
2311 upload_date = unified_strdate(content)
2312
2313 thumbnails = [{
2314 'id': image.get('type'),
2315 'url': image.get('src'),
2316 'width': int_or_none(image.get('width')),
2317 'height': int_or_none(image.get('height')),
2318 } for image in smil.findall(self._xpath_ns('.//image', namespace)) if image.get('src')]
2319
2320 return {
2321 'id': video_id,
2322 'title': title or video_id,
2323 'description': description,
2324 'upload_date': upload_date,
2325 'thumbnails': thumbnails,
2326 'formats': formats,
2327 'subtitles': subtitles,
2328 }
2329
2330 def _parse_smil_namespace(self, smil):
2331 return self._search_regex(
2332 r'(?i)^{([^}]+)?}smil$', smil.tag, 'namespace', default=None)
2333
2334 def _parse_smil_formats(self, smil, smil_url, video_id, namespace=None, f4m_params=None, transform_rtmp_url=None):
2335 base = smil_url
2336 for meta in smil.findall(self._xpath_ns('./head/meta', namespace)):
2337 b = meta.get('base') or meta.get('httpBase')
2338 if b:
2339 base = b
2340 break
2341
2342 formats = []
2343 rtmp_count = 0
2344 http_count = 0
2345 m3u8_count = 0
2346 imgs_count = 0
2347
2348 srcs = set()
2349 media = smil.findall(self._xpath_ns('.//video', namespace)) + smil.findall(self._xpath_ns('.//audio', namespace))
2350 for medium in media:
2351 src = medium.get('src')
2352 if not src or src in srcs:
2353 continue
2354 srcs.add(src)
2355
2356 bitrate = float_or_none(medium.get('system-bitrate') or medium.get('systemBitrate'), 1000)
2357 filesize = int_or_none(medium.get('size') or medium.get('fileSize'))
2358 width = int_or_none(medium.get('width'))
2359 height = int_or_none(medium.get('height'))
2360 proto = medium.get('proto')
2361 ext = medium.get('ext')
2362 src_ext = determine_ext(src)
2363 streamer = medium.get('streamer') or base
2364
2365 if proto == 'rtmp' or streamer.startswith('rtmp'):
2366 rtmp_count += 1
2367 formats.append({
2368 'url': streamer,
2369 'play_path': src,
2370 'ext': 'flv',
2371 'format_id': 'rtmp-%d' % (rtmp_count if bitrate is None else bitrate),
2372 'tbr': bitrate,
2373 'filesize': filesize,
2374 'width': width,
2375 'height': height,
2376 })
2377 if transform_rtmp_url:
2378 streamer, src = transform_rtmp_url(streamer, src)
2379 formats[-1].update({
2380 'url': streamer,
2381 'play_path': src,
2382 })
2383 continue
2384
2385 src_url = src if src.startswith('http') else compat_urlparse.urljoin(base, src)
2386 src_url = src_url.strip()
2387
2388 if proto == 'm3u8' or src_ext == 'm3u8':
2389 m3u8_formats = self._extract_m3u8_formats(
2390 src_url, video_id, ext or 'mp4', m3u8_id='hls', fatal=False)
2391 if len(m3u8_formats) == 1:
2392 m3u8_count += 1
2393 m3u8_formats[0].update({
2394 'format_id': 'hls-%d' % (m3u8_count if bitrate is None else bitrate),
2395 'tbr': bitrate,
2396 'width': width,
2397 'height': height,
2398 })
2399 formats.extend(m3u8_formats)
2400 elif src_ext == 'f4m':
2401 f4m_url = src_url
2402 if not f4m_params:
2403 f4m_params = {
2404 'hdcore': '3.2.0',
2405 'plugin': 'flowplayer-3.2.0.1',
2406 }
2407 f4m_url += '&' if '?' in f4m_url else '?'
2408 f4m_url += compat_urllib_parse_urlencode(f4m_params)
2409 formats.extend(self._extract_f4m_formats(f4m_url, video_id, f4m_id='hds', fatal=False))
2410 elif src_ext == 'mpd':
2411 formats.extend(self._extract_mpd_formats(
2412 src_url, video_id, mpd_id='dash', fatal=False))
2413 elif re.search(r'\.ism/[Mm]anifest', src_url):
2414 formats.extend(self._extract_ism_formats(
2415 src_url, video_id, ism_id='mss', fatal=False))
2416 elif src_url.startswith('http') and self._is_valid_url(src, video_id):
2417 http_count += 1
2418 formats.append({
2419 'url': src_url,
2420 'ext': ext or src_ext or 'flv',
2421 'format_id': 'http-%d' % (bitrate or http_count),
2422 'tbr': bitrate,
2423 'filesize': filesize,
2424 'width': width,
2425 'height': height,
2426 })
2427
2428 for medium in smil.findall(self._xpath_ns('.//imagestream', namespace)):
2429 src = medium.get('src')
2430 if not src or src in srcs:
2431 continue
2432 srcs.add(src)
2433
2434 imgs_count += 1
2435 formats.append({
2436 'format_id': 'imagestream-%d' % (imgs_count),
2437 'url': src,
2438 'ext': mimetype2ext(medium.get('type')),
2439 'acodec': 'none',
2440 'vcodec': 'none',
2441 'width': int_or_none(medium.get('width')),
2442 'height': int_or_none(medium.get('height')),
2443 'format_note': 'SMIL storyboards',
2444 })
2445
2446 return formats
2447
2448 def _parse_smil_subtitles(self, smil, namespace=None, subtitles_lang='en'):
2449 urls = []
2450 subtitles = {}
2451 for num, textstream in enumerate(smil.findall(self._xpath_ns('.//textstream', namespace))):
2452 src = textstream.get('src')
2453 if not src or src in urls:
2454 continue
2455 urls.append(src)
2456 ext = textstream.get('ext') or mimetype2ext(textstream.get('type')) or determine_ext(src)
2457 lang = textstream.get('systemLanguage') or textstream.get('systemLanguageName') or textstream.get('lang') or subtitles_lang
2458 subtitles.setdefault(lang, []).append({
2459 'url': src,
2460 'ext': ext,
2461 })
2462 return subtitles
2463
2464 def _extract_xspf_playlist(self, xspf_url, playlist_id, fatal=True):
2465 xspf = self._download_xml(
2466 xspf_url, playlist_id, 'Downloading xpsf playlist',
2467 'Unable to download xspf manifest', fatal=fatal)
2468 if xspf is False:
2469 return []
2470 return self._parse_xspf(
2471 xspf, playlist_id, xspf_url=xspf_url,
2472 xspf_base_url=base_url(xspf_url))
2473
2474 def _parse_xspf(self, xspf_doc, playlist_id, xspf_url=None, xspf_base_url=None):
2475 NS_MAP = {
2476 'xspf': 'http://xspf.org/ns/0/',
2477 's1': 'http://static.streamone.nl/player/ns/0',
2478 }
2479
2480 entries = []
2481 for track in xspf_doc.findall(xpath_with_ns('./xspf:trackList/xspf:track', NS_MAP)):
2482 title = xpath_text(
2483 track, xpath_with_ns('./xspf:title', NS_MAP), 'title', default=playlist_id)
2484 description = xpath_text(
2485 track, xpath_with_ns('./xspf:annotation', NS_MAP), 'description')
2486 thumbnail = xpath_text(
2487 track, xpath_with_ns('./xspf:image', NS_MAP), 'thumbnail')
2488 duration = float_or_none(
2489 xpath_text(track, xpath_with_ns('./xspf:duration', NS_MAP), 'duration'), 1000)
2490
2491 formats = []
2492 for location in track.findall(xpath_with_ns('./xspf:location', NS_MAP)):
2493 format_url = urljoin(xspf_base_url, location.text)
2494 if not format_url:
2495 continue
2496 formats.append({
2497 'url': format_url,
2498 'manifest_url': xspf_url,
2499 'format_id': location.get(xpath_with_ns('s1:label', NS_MAP)),
2500 'width': int_or_none(location.get(xpath_with_ns('s1:width', NS_MAP))),
2501 'height': int_or_none(location.get(xpath_with_ns('s1:height', NS_MAP))),
2502 })
2503 self._sort_formats(formats)
2504
2505 entries.append({
2506 'id': playlist_id,
2507 'title': title,
2508 'description': description,
2509 'thumbnail': thumbnail,
2510 'duration': duration,
2511 'formats': formats,
2512 })
2513 return entries
2514
2515 def _extract_mpd_formats(self, *args, **kwargs):
2516 fmts, subs = self._extract_mpd_formats_and_subtitles(*args, **kwargs)
2517 if subs:
2518 self._report_ignoring_subs('DASH')
2519 return fmts
2520
2521 def _extract_mpd_formats_and_subtitles(
2522 self, mpd_url, video_id, mpd_id=None, note=None, errnote=None,
2523 fatal=True, data=None, headers={}, query={}):
2524 res = self._download_xml_handle(
2525 mpd_url, video_id,
2526 note='Downloading MPD manifest' if note is None else note,
2527 errnote='Failed to download MPD manifest' if errnote is None else errnote,
2528 fatal=fatal, data=data, headers=headers, query=query)
2529 if res is False:
2530 return [], {}
2531 mpd_doc, urlh = res
2532 if mpd_doc is None:
2533 return [], {}
2534 mpd_base_url = base_url(urlh.geturl())
2535
2536 return self._parse_mpd_formats_and_subtitles(
2537 mpd_doc, mpd_id, mpd_base_url, mpd_url)
2538
2539 def _parse_mpd_formats(self, *args, **kwargs):
2540 fmts, subs = self._parse_mpd_formats_and_subtitles(*args, **kwargs)
2541 if subs:
2542 self._report_ignoring_subs('DASH')
2543 return fmts
2544
2545 def _parse_mpd_formats_and_subtitles(
2546 self, mpd_doc, mpd_id=None, mpd_base_url='', mpd_url=None):
2547 """
2548 Parse formats from MPD manifest.
2549 References:
2550 1. MPEG-DASH Standard, ISO/IEC 23009-1:2014(E),
2551 http://standards.iso.org/ittf/PubliclyAvailableStandards/c065274_ISO_IEC_23009-1_2014.zip
2552 2. https://en.wikipedia.org/wiki/Dynamic_Adaptive_Streaming_over_HTTP
2553 """
2554 if not self.get_param('dynamic_mpd', True):
2555 if mpd_doc.get('type') == 'dynamic':
2556 return [], {}
2557
2558 namespace = self._search_regex(r'(?i)^{([^}]+)?}MPD$', mpd_doc.tag, 'namespace', default=None)
2559
2560 def _add_ns(path):
2561 return self._xpath_ns(path, namespace)
2562
2563 def is_drm_protected(element):
2564 return element.find(_add_ns('ContentProtection')) is not None
2565
2566 def extract_multisegment_info(element, ms_parent_info):
2567 ms_info = ms_parent_info.copy()
2568
2569 # As per [1, 5.3.9.2.2] SegmentList and SegmentTemplate share some
2570 # common attributes and elements. We will only extract relevant
2571 # for us.
2572 def extract_common(source):
2573 segment_timeline = source.find(_add_ns('SegmentTimeline'))
2574 if segment_timeline is not None:
2575 s_e = segment_timeline.findall(_add_ns('S'))
2576 if s_e:
2577 ms_info['total_number'] = 0
2578 ms_info['s'] = []
2579 for s in s_e:
2580 r = int(s.get('r', 0))
2581 ms_info['total_number'] += 1 + r
2582 ms_info['s'].append({
2583 't': int(s.get('t', 0)),
2584 # @d is mandatory (see [1, 5.3.9.6.2, Table 17, page 60])
2585 'd': int(s.attrib['d']),
2586 'r': r,
2587 })
2588 start_number = source.get('startNumber')
2589 if start_number:
2590 ms_info['start_number'] = int(start_number)
2591 timescale = source.get('timescale')
2592 if timescale:
2593 ms_info['timescale'] = int(timescale)
2594 segment_duration = source.get('duration')
2595 if segment_duration:
2596 ms_info['segment_duration'] = float(segment_duration)
2597
2598 def extract_Initialization(source):
2599 initialization = source.find(_add_ns('Initialization'))
2600 if initialization is not None:
2601 ms_info['initialization_url'] = initialization.attrib['sourceURL']
2602
2603 segment_list = element.find(_add_ns('SegmentList'))
2604 if segment_list is not None:
2605 extract_common(segment_list)
2606 extract_Initialization(segment_list)
2607 segment_urls_e = segment_list.findall(_add_ns('SegmentURL'))
2608 if segment_urls_e:
2609 ms_info['segment_urls'] = [segment.attrib['media'] for segment in segment_urls_e]
2610 else:
2611 segment_template = element.find(_add_ns('SegmentTemplate'))
2612 if segment_template is not None:
2613 extract_common(segment_template)
2614 media = segment_template.get('media')
2615 if media:
2616 ms_info['media'] = media
2617 initialization = segment_template.get('initialization')
2618 if initialization:
2619 ms_info['initialization'] = initialization
2620 else:
2621 extract_Initialization(segment_template)
2622 return ms_info
2623
2624 mpd_duration = parse_duration(mpd_doc.get('mediaPresentationDuration'))
2625 formats, subtitles = [], {}
2626 stream_numbers = {'audio': 0, 'video': 0}
2627 for period in mpd_doc.findall(_add_ns('Period')):
2628 period_duration = parse_duration(period.get('duration')) or mpd_duration
2629 period_ms_info = extract_multisegment_info(period, {
2630 'start_number': 1,
2631 'timescale': 1,
2632 })
2633 for adaptation_set in period.findall(_add_ns('AdaptationSet')):
2634 adaption_set_ms_info = extract_multisegment_info(adaptation_set, period_ms_info)
2635 for representation in adaptation_set.findall(_add_ns('Representation')):
2636 representation_attrib = adaptation_set.attrib.copy()
2637 representation_attrib.update(representation.attrib)
2638 # According to [1, 5.3.7.2, Table 9, page 41], @mimeType is mandatory
2639 mime_type = representation_attrib['mimeType']
2640 content_type = representation_attrib.get('contentType', mime_type.split('/')[0])
2641
2642 codecs = representation_attrib.get('codecs', '')
2643 if content_type not in ('video', 'audio', 'text'):
2644 if mime_type == 'image/jpeg':
2645 content_type = mime_type
2646 elif codecs.split('.')[0] == 'stpp':
2647 content_type = 'text'
2648 else:
2649 self.report_warning('Unknown MIME type %s in DASH manifest' % mime_type)
2650 continue
2651
2652 base_url = ''
2653 for element in (representation, adaptation_set, period, mpd_doc):
2654 base_url_e = element.find(_add_ns('BaseURL'))
2655 if base_url_e is not None:
2656 base_url = base_url_e.text + base_url
2657 if re.match(r'^https?://', base_url):
2658 break
2659 if mpd_base_url and base_url.startswith('/'):
2660 base_url = compat_urlparse.urljoin(mpd_base_url, base_url)
2661 elif mpd_base_url and not re.match(r'^https?://', base_url):
2662 if not mpd_base_url.endswith('/'):
2663 mpd_base_url += '/'
2664 base_url = mpd_base_url + base_url
2665 representation_id = representation_attrib.get('id')
2666 lang = representation_attrib.get('lang')
2667 url_el = representation.find(_add_ns('BaseURL'))
2668 filesize = int_or_none(url_el.attrib.get('{http://youtube.com/yt/2012/10/10}contentLength') if url_el is not None else None)
2669 bandwidth = int_or_none(representation_attrib.get('bandwidth'))
2670 if representation_id is not None:
2671 format_id = representation_id
2672 else:
2673 format_id = content_type
2674 if mpd_id:
2675 format_id = mpd_id + '-' + format_id
2676 if content_type in ('video', 'audio'):
2677 f = {
2678 'format_id': format_id,
2679 'manifest_url': mpd_url,
2680 'ext': mimetype2ext(mime_type),
2681 'width': int_or_none(representation_attrib.get('width')),
2682 'height': int_or_none(representation_attrib.get('height')),
2683 'tbr': float_or_none(bandwidth, 1000),
2684 'asr': int_or_none(representation_attrib.get('audioSamplingRate')),
2685 'fps': int_or_none(representation_attrib.get('frameRate')),
2686 'language': lang if lang not in ('mul', 'und', 'zxx', 'mis') else None,
2687 'format_note': 'DASH %s' % content_type,
2688 'filesize': filesize,
2689 'container': mimetype2ext(mime_type) + '_dash',
2690 'manifest_stream_number': stream_numbers[content_type]
2691 }
2692 f.update(parse_codecs(codecs))
2693 stream_numbers[content_type] += 1
2694 elif content_type == 'text':
2695 f = {
2696 'ext': mimetype2ext(mime_type),
2697 'manifest_url': mpd_url,
2698 'filesize': filesize,
2699 }
2700 elif content_type == 'image/jpeg':
2701 # See test case in VikiIE
2702 # https://www.viki.com/videos/1175236v-choosing-spouse-by-lottery-episode-1
2703 f = {
2704 'format_id': format_id,
2705 'ext': 'mhtml',
2706 'manifest_url': mpd_url,
2707 'format_note': 'DASH storyboards (jpeg)',
2708 'acodec': 'none',
2709 'vcodec': 'none',
2710 }
2711 if is_drm_protected(adaptation_set) or is_drm_protected(representation):
2712 f['has_drm'] = True
2713 representation_ms_info = extract_multisegment_info(representation, adaption_set_ms_info)
2714
2715 def prepare_template(template_name, identifiers):
2716 tmpl = representation_ms_info[template_name]
2717 # First of, % characters outside $...$ templates
2718 # must be escaped by doubling for proper processing
2719 # by % operator string formatting used further (see
2720 # https://github.com/ytdl-org/youtube-dl/issues/16867).
2721 t = ''
2722 in_template = False
2723 for c in tmpl:
2724 t += c
2725 if c == '$':
2726 in_template = not in_template
2727 elif c == '%' and not in_template:
2728 t += c
2729 # Next, $...$ templates are translated to their
2730 # %(...) counterparts to be used with % operator
2731 if representation_id is not None:
2732 t = t.replace('$RepresentationID$', representation_id)
2733 t = re.sub(r'\$(%s)\$' % '|'.join(identifiers), r'%(\1)d', t)
2734 t = re.sub(r'\$(%s)%%([^$]+)\$' % '|'.join(identifiers), r'%(\1)\2', t)
2735 t.replace('$$', '$')
2736 return t
2737
2738 # @initialization is a regular template like @media one
2739 # so it should be handled just the same way (see
2740 # https://github.com/ytdl-org/youtube-dl/issues/11605)
2741 if 'initialization' in representation_ms_info:
2742 initialization_template = prepare_template(
2743 'initialization',
2744 # As per [1, 5.3.9.4.2, Table 15, page 54] $Number$ and
2745 # $Time$ shall not be included for @initialization thus
2746 # only $Bandwidth$ remains
2747 ('Bandwidth', ))
2748 representation_ms_info['initialization_url'] = initialization_template % {
2749 'Bandwidth': bandwidth,
2750 }
2751
2752 def location_key(location):
2753 return 'url' if re.match(r'^https?://', location) else 'path'
2754
2755 if 'segment_urls' not in representation_ms_info and 'media' in representation_ms_info:
2756
2757 media_template = prepare_template('media', ('Number', 'Bandwidth', 'Time'))
2758 media_location_key = location_key(media_template)
2759
2760 # As per [1, 5.3.9.4.4, Table 16, page 55] $Number$ and $Time$
2761 # can't be used at the same time
2762 if '%(Number' in media_template and 's' not in representation_ms_info:
2763 segment_duration = None
2764 if 'total_number' not in representation_ms_info and 'segment_duration' in representation_ms_info:
2765 segment_duration = float_or_none(representation_ms_info['segment_duration'], representation_ms_info['timescale'])
2766 representation_ms_info['total_number'] = int(math.ceil(float(period_duration) / segment_duration))
2767 representation_ms_info['fragments'] = [{
2768 media_location_key: media_template % {
2769 'Number': segment_number,
2770 'Bandwidth': bandwidth,
2771 },
2772 'duration': segment_duration,
2773 } for segment_number in range(
2774 representation_ms_info['start_number'],
2775 representation_ms_info['total_number'] + representation_ms_info['start_number'])]
2776 else:
2777 # $Number*$ or $Time$ in media template with S list available
2778 # Example $Number*$: http://www.svtplay.se/klipp/9023742/stopptid-om-bjorn-borg
2779 # Example $Time$: https://play.arkena.com/embed/avp/v2/player/media/b41dda37-d8e7-4d3f-b1b5-9a9db578bdfe/1/129411
2780 representation_ms_info['fragments'] = []
2781 segment_time = 0
2782 segment_d = None
2783 segment_number = representation_ms_info['start_number']
2784
2785 def add_segment_url():
2786 segment_url = media_template % {
2787 'Time': segment_time,
2788 'Bandwidth': bandwidth,
2789 'Number': segment_number,
2790 }
2791 representation_ms_info['fragments'].append({
2792 media_location_key: segment_url,
2793 'duration': float_or_none(segment_d, representation_ms_info['timescale']),
2794 })
2795
2796 for num, s in enumerate(representation_ms_info['s']):
2797 segment_time = s.get('t') or segment_time
2798 segment_d = s['d']
2799 add_segment_url()
2800 segment_number += 1
2801 for r in range(s.get('r', 0)):
2802 segment_time += segment_d
2803 add_segment_url()
2804 segment_number += 1
2805 segment_time += segment_d
2806 elif 'segment_urls' in representation_ms_info and 's' in representation_ms_info:
2807 # No media template
2808 # Example: https://www.youtube.com/watch?v=iXZV5uAYMJI
2809 # or any YouTube dashsegments video
2810 fragments = []
2811 segment_index = 0
2812 timescale = representation_ms_info['timescale']
2813 for s in representation_ms_info['s']:
2814 duration = float_or_none(s['d'], timescale)
2815 for r in range(s.get('r', 0) + 1):
2816 segment_uri = representation_ms_info['segment_urls'][segment_index]
2817 fragments.append({
2818 location_key(segment_uri): segment_uri,
2819 'duration': duration,
2820 })
2821 segment_index += 1
2822 representation_ms_info['fragments'] = fragments
2823 elif 'segment_urls' in representation_ms_info:
2824 # Segment URLs with no SegmentTimeline
2825 # Example: https://www.seznam.cz/zpravy/clanek/cesko-zasahne-vitr-o-sile-vichrice-muze-byt-i-zivotu-nebezpecny-39091
2826 # https://github.com/ytdl-org/youtube-dl/pull/14844
2827 fragments = []
2828 segment_duration = float_or_none(
2829 representation_ms_info['segment_duration'],
2830 representation_ms_info['timescale']) if 'segment_duration' in representation_ms_info else None
2831 for segment_url in representation_ms_info['segment_urls']:
2832 fragment = {
2833 location_key(segment_url): segment_url,
2834 }
2835 if segment_duration:
2836 fragment['duration'] = segment_duration
2837 fragments.append(fragment)
2838 representation_ms_info['fragments'] = fragments
2839 # If there is a fragments key available then we correctly recognized fragmented media.
2840 # Otherwise we will assume unfragmented media with direct access. Technically, such
2841 # assumption is not necessarily correct since we may simply have no support for
2842 # some forms of fragmented media renditions yet, but for now we'll use this fallback.
2843 if 'fragments' in representation_ms_info:
2844 f.update({
2845 # NB: mpd_url may be empty when MPD manifest is parsed from a string
2846 'url': mpd_url or base_url,
2847 'fragment_base_url': base_url,
2848 'fragments': [],
2849 'protocol': 'http_dash_segments' if mime_type != 'image/jpeg' else 'mhtml',
2850 })
2851 if 'initialization_url' in representation_ms_info:
2852 initialization_url = representation_ms_info['initialization_url']
2853 if not f.get('url'):
2854 f['url'] = initialization_url
2855 f['fragments'].append({location_key(initialization_url): initialization_url})
2856 f['fragments'].extend(representation_ms_info['fragments'])
2857 else:
2858 # Assuming direct URL to unfragmented media.
2859 f['url'] = base_url
2860 if content_type in ('video', 'audio') or mime_type == 'image/jpeg':
2861 formats.append(f)
2862 elif content_type == 'text':
2863 subtitles.setdefault(lang or 'und', []).append(f)
2864
2865 return formats, subtitles
2866
2867 def _extract_ism_formats(self, *args, **kwargs):
2868 fmts, subs = self._extract_ism_formats_and_subtitles(*args, **kwargs)
2869 if subs:
2870 self._report_ignoring_subs('ISM')
2871 return fmts
2872
2873 def _extract_ism_formats_and_subtitles(self, ism_url, video_id, ism_id=None, note=None, errnote=None, fatal=True, data=None, headers={}, query={}):
2874 res = self._download_xml_handle(
2875 ism_url, video_id,
2876 note='Downloading ISM manifest' if note is None else note,
2877 errnote='Failed to download ISM manifest' if errnote is None else errnote,
2878 fatal=fatal, data=data, headers=headers, query=query)
2879 if res is False:
2880 return [], {}
2881 ism_doc, urlh = res
2882 if ism_doc is None:
2883 return [], {}
2884
2885 return self._parse_ism_formats_and_subtitles(ism_doc, urlh.geturl(), ism_id)
2886
2887 def _parse_ism_formats_and_subtitles(self, ism_doc, ism_url, ism_id=None):
2888 """
2889 Parse formats from ISM manifest.
2890 References:
2891 1. [MS-SSTR]: Smooth Streaming Protocol,
2892 https://msdn.microsoft.com/en-us/library/ff469518.aspx
2893 """
2894 if ism_doc.get('IsLive') == 'TRUE':
2895 return [], {}
2896
2897 duration = int(ism_doc.attrib['Duration'])
2898 timescale = int_or_none(ism_doc.get('TimeScale')) or 10000000
2899
2900 formats = []
2901 subtitles = {}
2902 for stream in ism_doc.findall('StreamIndex'):
2903 stream_type = stream.get('Type')
2904 if stream_type not in ('video', 'audio', 'text'):
2905 continue
2906 url_pattern = stream.attrib['Url']
2907 stream_timescale = int_or_none(stream.get('TimeScale')) or timescale
2908 stream_name = stream.get('Name')
2909 stream_language = stream.get('Language', 'und')
2910 for track in stream.findall('QualityLevel'):
2911 fourcc = track.get('FourCC') or ('AACL' if track.get('AudioTag') == '255' else None)
2912 # TODO: add support for WVC1 and WMAP
2913 if fourcc not in ('H264', 'AVC1', 'AACL', 'TTML'):
2914 self.report_warning('%s is not a supported codec' % fourcc)
2915 continue
2916 tbr = int(track.attrib['Bitrate']) // 1000
2917 # [1] does not mention Width and Height attributes. However,
2918 # they're often present while MaxWidth and MaxHeight are
2919 # missing, so should be used as fallbacks
2920 width = int_or_none(track.get('MaxWidth') or track.get('Width'))
2921 height = int_or_none(track.get('MaxHeight') or track.get('Height'))
2922 sampling_rate = int_or_none(track.get('SamplingRate'))
2923
2924 track_url_pattern = re.sub(r'{[Bb]itrate}', track.attrib['Bitrate'], url_pattern)
2925 track_url_pattern = compat_urlparse.urljoin(ism_url, track_url_pattern)
2926
2927 fragments = []
2928 fragment_ctx = {
2929 'time': 0,
2930 }
2931 stream_fragments = stream.findall('c')
2932 for stream_fragment_index, stream_fragment in enumerate(stream_fragments):
2933 fragment_ctx['time'] = int_or_none(stream_fragment.get('t')) or fragment_ctx['time']
2934 fragment_repeat = int_or_none(stream_fragment.get('r')) or 1
2935 fragment_ctx['duration'] = int_or_none(stream_fragment.get('d'))
2936 if not fragment_ctx['duration']:
2937 try:
2938 next_fragment_time = int(stream_fragment[stream_fragment_index + 1].attrib['t'])
2939 except IndexError:
2940 next_fragment_time = duration
2941 fragment_ctx['duration'] = (next_fragment_time - fragment_ctx['time']) / fragment_repeat
2942 for _ in range(fragment_repeat):
2943 fragments.append({
2944 'url': re.sub(r'{start[ _]time}', compat_str(fragment_ctx['time']), track_url_pattern),
2945 'duration': fragment_ctx['duration'] / stream_timescale,
2946 })
2947 fragment_ctx['time'] += fragment_ctx['duration']
2948
2949 format_id = []
2950 if ism_id:
2951 format_id.append(ism_id)
2952 if stream_name:
2953 format_id.append(stream_name)
2954 format_id.append(compat_str(tbr))
2955
2956 if stream_type == 'text':
2957 subtitles.setdefault(stream_language, []).append({
2958 'ext': 'ismt',
2959 'protocol': 'ism',
2960 'url': ism_url,
2961 'manifest_url': ism_url,
2962 'fragments': fragments,
2963 '_download_params': {
2964 'stream_type': stream_type,
2965 'duration': duration,
2966 'timescale': stream_timescale,
2967 'fourcc': fourcc,
2968 'language': stream_language,
2969 'codec_private_data': track.get('CodecPrivateData'),
2970 }
2971 })
2972 elif stream_type in ('video', 'audio'):
2973 formats.append({
2974 'format_id': '-'.join(format_id),
2975 'url': ism_url,
2976 'manifest_url': ism_url,
2977 'ext': 'ismv' if stream_type == 'video' else 'isma',
2978 'width': width,
2979 'height': height,
2980 'tbr': tbr,
2981 'asr': sampling_rate,
2982 'vcodec': 'none' if stream_type == 'audio' else fourcc,
2983 'acodec': 'none' if stream_type == 'video' else fourcc,
2984 'protocol': 'ism',
2985 'fragments': fragments,
2986 'has_drm': ism_doc.find('Protection') is not None,
2987 '_download_params': {
2988 'stream_type': stream_type,
2989 'duration': duration,
2990 'timescale': stream_timescale,
2991 'width': width or 0,
2992 'height': height or 0,
2993 'fourcc': fourcc,
2994 'language': stream_language,
2995 'codec_private_data': track.get('CodecPrivateData'),
2996 'sampling_rate': sampling_rate,
2997 'channels': int_or_none(track.get('Channels', 2)),
2998 'bits_per_sample': int_or_none(track.get('BitsPerSample', 16)),
2999 'nal_unit_length_field': int_or_none(track.get('NALUnitLengthField', 4)),
3000 },
3001 })
3002 return formats, subtitles
3003
3004 def _parse_html5_media_entries(self, base_url, webpage, video_id, m3u8_id=None, m3u8_entry_protocol='m3u8', mpd_id=None, preference=None, quality=None):
3005 def absolute_url(item_url):
3006 return urljoin(base_url, item_url)
3007
3008 def parse_content_type(content_type):
3009 if not content_type:
3010 return {}
3011 ctr = re.search(r'(?P<mimetype>[^/]+/[^;]+)(?:;\s*codecs="?(?P<codecs>[^"]+))?', content_type)
3012 if ctr:
3013 mimetype, codecs = ctr.groups()
3014 f = parse_codecs(codecs)
3015 f['ext'] = mimetype2ext(mimetype)
3016 return f
3017 return {}
3018
3019 def _media_formats(src, cur_media_type, type_info={}):
3020 full_url = absolute_url(src)
3021 ext = type_info.get('ext') or determine_ext(full_url)
3022 if ext == 'm3u8':
3023 is_plain_url = False
3024 formats = self._extract_m3u8_formats(
3025 full_url, video_id, ext='mp4',
3026 entry_protocol=m3u8_entry_protocol, m3u8_id=m3u8_id,
3027 preference=preference, quality=quality, fatal=False)
3028 elif ext == 'mpd':
3029 is_plain_url = False
3030 formats = self._extract_mpd_formats(
3031 full_url, video_id, mpd_id=mpd_id, fatal=False)
3032 else:
3033 is_plain_url = True
3034 formats = [{
3035 'url': full_url,
3036 'vcodec': 'none' if cur_media_type == 'audio' else None,
3037 }]
3038 return is_plain_url, formats
3039
3040 entries = []
3041 # amp-video and amp-audio are very similar to their HTML5 counterparts
3042 # so we wll include them right here (see
3043 # https://www.ampproject.org/docs/reference/components/amp-video)
3044 # For dl8-* tags see https://delight-vr.com/documentation/dl8-video/
3045 _MEDIA_TAG_NAME_RE = r'(?:(?:amp|dl8(?:-live)?)-)?(video|audio)'
3046 media_tags = [(media_tag, media_tag_name, media_type, '')
3047 for media_tag, media_tag_name, media_type
3048 in re.findall(r'(?s)(<(%s)[^>]*/>)' % _MEDIA_TAG_NAME_RE, webpage)]
3049 media_tags.extend(re.findall(
3050 # We only allow video|audio followed by a whitespace or '>'.
3051 # Allowing more characters may end up in significant slow down (see
3052 # https://github.com/ytdl-org/youtube-dl/issues/11979, example URL:
3053 # http://www.porntrex.com/maps/videositemap.xml).
3054 r'(?s)(<(?P<tag>%s)(?:\s+[^>]*)?>)(.*?)</(?P=tag)>' % _MEDIA_TAG_NAME_RE, webpage))
3055 for media_tag, _, media_type, media_content in media_tags:
3056 media_info = {
3057 'formats': [],
3058 'subtitles': {},
3059 }
3060 media_attributes = extract_attributes(media_tag)
3061 src = strip_or_none(media_attributes.get('src'))
3062 if src:
3063 _, formats = _media_formats(src, media_type)
3064 media_info['formats'].extend(formats)
3065 media_info['thumbnail'] = absolute_url(media_attributes.get('poster'))
3066 if media_content:
3067 for source_tag in re.findall(r'<source[^>]+>', media_content):
3068 s_attr = extract_attributes(source_tag)
3069 # data-video-src and data-src are non standard but seen
3070 # several times in the wild
3071 src = strip_or_none(dict_get(s_attr, ('src', 'data-video-src', 'data-src')))
3072 if not src:
3073 continue
3074 f = parse_content_type(s_attr.get('type'))
3075 is_plain_url, formats = _media_formats(src, media_type, f)
3076 if is_plain_url:
3077 # width, height, res, label and title attributes are
3078 # all not standard but seen several times in the wild
3079 labels = [
3080 s_attr.get(lbl)
3081 for lbl in ('label', 'title')
3082 if str_or_none(s_attr.get(lbl))
3083 ]
3084 width = int_or_none(s_attr.get('width'))
3085 height = (int_or_none(s_attr.get('height'))
3086 or int_or_none(s_attr.get('res')))
3087 if not width or not height:
3088 for lbl in labels:
3089 resolution = parse_resolution(lbl)
3090 if not resolution:
3091 continue
3092 width = width or resolution.get('width')
3093 height = height or resolution.get('height')
3094 for lbl in labels:
3095 tbr = parse_bitrate(lbl)
3096 if tbr:
3097 break
3098 else:
3099 tbr = None
3100 f.update({
3101 'width': width,
3102 'height': height,
3103 'tbr': tbr,
3104 'format_id': s_attr.get('label') or s_attr.get('title'),
3105 })
3106 f.update(formats[0])
3107 media_info['formats'].append(f)
3108 else:
3109 media_info['formats'].extend(formats)
3110 for track_tag in re.findall(r'<track[^>]+>', media_content):
3111 track_attributes = extract_attributes(track_tag)
3112 kind = track_attributes.get('kind')
3113 if not kind or kind in ('subtitles', 'captions'):
3114 src = strip_or_none(track_attributes.get('src'))
3115 if not src:
3116 continue
3117 lang = track_attributes.get('srclang') or track_attributes.get('lang') or track_attributes.get('label')
3118 media_info['subtitles'].setdefault(lang, []).append({
3119 'url': absolute_url(src),
3120 })
3121 for f in media_info['formats']:
3122 f.setdefault('http_headers', {})['Referer'] = base_url
3123 if media_info['formats'] or media_info['subtitles']:
3124 entries.append(media_info)
3125 return entries
3126
3127 def _extract_akamai_formats(self, *args, **kwargs):
3128 fmts, subs = self._extract_akamai_formats_and_subtitles(*args, **kwargs)
3129 if subs:
3130 self._report_ignoring_subs('akamai')
3131 return fmts
3132
3133 def _extract_akamai_formats_and_subtitles(self, manifest_url, video_id, hosts={}):
3134 signed = 'hdnea=' in manifest_url
3135 if not signed:
3136 # https://learn.akamai.com/en-us/webhelp/media-services-on-demand/stream-packaging-user-guide/GUID-BE6C0F73-1E06-483B-B0EA-57984B91B7F9.html
3137 manifest_url = re.sub(
3138 r'(?:b=[\d,-]+|(?:__a__|attributes)=off|__b__=\d+)&?',
3139 '', manifest_url).strip('?')
3140
3141 formats = []
3142 subtitles = {}
3143
3144 hdcore_sign = 'hdcore=3.7.0'
3145 f4m_url = re.sub(r'(https?://[^/]+)/i/', r'\1/z/', manifest_url).replace('/master.m3u8', '/manifest.f4m')
3146 hds_host = hosts.get('hds')
3147 if hds_host:
3148 f4m_url = re.sub(r'(https?://)[^/]+', r'\1' + hds_host, f4m_url)
3149 if 'hdcore=' not in f4m_url:
3150 f4m_url += ('&' if '?' in f4m_url else '?') + hdcore_sign
3151 f4m_formats = self._extract_f4m_formats(
3152 f4m_url, video_id, f4m_id='hds', fatal=False)
3153 for entry in f4m_formats:
3154 entry.update({'extra_param_to_segment_url': hdcore_sign})
3155 formats.extend(f4m_formats)
3156
3157 m3u8_url = re.sub(r'(https?://[^/]+)/z/', r'\1/i/', manifest_url).replace('/manifest.f4m', '/master.m3u8')
3158 hls_host = hosts.get('hls')
3159 if hls_host:
3160 m3u8_url = re.sub(r'(https?://)[^/]+', r'\1' + hls_host, m3u8_url)
3161 m3u8_formats, m3u8_subtitles = self._extract_m3u8_formats_and_subtitles(
3162 m3u8_url, video_id, 'mp4', 'm3u8_native',
3163 m3u8_id='hls', fatal=False)
3164 formats.extend(m3u8_formats)
3165 subtitles = self._merge_subtitles(subtitles, m3u8_subtitles)
3166
3167 http_host = hosts.get('http')
3168 if http_host and m3u8_formats and not signed:
3169 REPL_REGEX = r'https?://[^/]+/i/([^,]+),([^/]+),([^/]+)\.csmil/.+'
3170 qualities = re.match(REPL_REGEX, m3u8_url).group(2).split(',')
3171 qualities_length = len(qualities)
3172 if len(m3u8_formats) in (qualities_length, qualities_length + 1):
3173 i = 0
3174 for f in m3u8_formats:
3175 if f['vcodec'] != 'none':
3176 for protocol in ('http', 'https'):
3177 http_f = f.copy()
3178 del http_f['manifest_url']
3179 http_url = re.sub(
3180 REPL_REGEX, protocol + r'://%s/\g<1>%s\3' % (http_host, qualities[i]), f['url'])
3181 http_f.update({
3182 'format_id': http_f['format_id'].replace('hls-', protocol + '-'),
3183 'url': http_url,
3184 'protocol': protocol,
3185 })
3186 formats.append(http_f)
3187 i += 1
3188
3189 return formats, subtitles
3190
3191 def _extract_wowza_formats(self, url, video_id, m3u8_entry_protocol='m3u8_native', skip_protocols=[]):
3192 query = compat_urlparse.urlparse(url).query
3193 url = re.sub(r'/(?:manifest|playlist|jwplayer)\.(?:m3u8|f4m|mpd|smil)', '', url)
3194 mobj = re.search(
3195 r'(?:(?:http|rtmp|rtsp)(?P<s>s)?:)?(?P<url>//[^?]+)', url)
3196 url_base = mobj.group('url')
3197 http_base_url = '%s%s:%s' % ('http', mobj.group('s') or '', url_base)
3198 formats = []
3199
3200 def manifest_url(manifest):
3201 m_url = '%s/%s' % (http_base_url, manifest)
3202 if query:
3203 m_url += '?%s' % query
3204 return m_url
3205
3206 if 'm3u8' not in skip_protocols:
3207 formats.extend(self._extract_m3u8_formats(
3208 manifest_url('playlist.m3u8'), video_id, 'mp4',
3209 m3u8_entry_protocol, m3u8_id='hls', fatal=False))
3210 if 'f4m' not in skip_protocols:
3211 formats.extend(self._extract_f4m_formats(
3212 manifest_url('manifest.f4m'),
3213 video_id, f4m_id='hds', fatal=False))
3214 if 'dash' not in skip_protocols:
3215 formats.extend(self._extract_mpd_formats(
3216 manifest_url('manifest.mpd'),
3217 video_id, mpd_id='dash', fatal=False))
3218 if re.search(r'(?:/smil:|\.smil)', url_base):
3219 if 'smil' not in skip_protocols:
3220 rtmp_formats = self._extract_smil_formats(
3221 manifest_url('jwplayer.smil'),
3222 video_id, fatal=False)
3223 for rtmp_format in rtmp_formats:
3224 rtsp_format = rtmp_format.copy()
3225 rtsp_format['url'] = '%s/%s' % (rtmp_format['url'], rtmp_format['play_path'])
3226 del rtsp_format['play_path']
3227 del rtsp_format['ext']
3228 rtsp_format.update({
3229 'url': rtsp_format['url'].replace('rtmp://', 'rtsp://'),
3230 'format_id': rtmp_format['format_id'].replace('rtmp', 'rtsp'),
3231 'protocol': 'rtsp',
3232 })
3233 formats.extend([rtmp_format, rtsp_format])
3234 else:
3235 for protocol in ('rtmp', 'rtsp'):
3236 if protocol not in skip_protocols:
3237 formats.append({
3238 'url': '%s:%s' % (protocol, url_base),
3239 'format_id': protocol,
3240 'protocol': protocol,
3241 })
3242 return formats
3243
3244 def _find_jwplayer_data(self, webpage, video_id=None, transform_source=js_to_json):
3245 mobj = re.search(
3246 r'(?s)jwplayer\((?P<quote>[\'"])[^\'" ]+(?P=quote)\)(?!</script>).*?\.setup\s*\((?P<options>[^)]+)\)',
3247 webpage)
3248 if mobj:
3249 try:
3250 jwplayer_data = self._parse_json(mobj.group('options'),
3251 video_id=video_id,
3252 transform_source=transform_source)
3253 except ExtractorError:
3254 pass
3255 else:
3256 if isinstance(jwplayer_data, dict):
3257 return jwplayer_data
3258
3259 def _extract_jwplayer_data(self, webpage, video_id, *args, **kwargs):
3260 jwplayer_data = self._find_jwplayer_data(
3261 webpage, video_id, transform_source=js_to_json)
3262 return self._parse_jwplayer_data(
3263 jwplayer_data, video_id, *args, **kwargs)
3264
3265 def _parse_jwplayer_data(self, jwplayer_data, video_id=None, require_title=True,
3266 m3u8_id=None, mpd_id=None, rtmp_params=None, base_url=None):
3267 # JWPlayer backward compatibility: flattened playlists
3268 # https://github.com/jwplayer/jwplayer/blob/v7.4.3/src/js/api/config.js#L81-L96
3269 if 'playlist' not in jwplayer_data:
3270 jwplayer_data = {'playlist': [jwplayer_data]}
3271
3272 entries = []
3273
3274 # JWPlayer backward compatibility: single playlist item
3275 # https://github.com/jwplayer/jwplayer/blob/v7.7.0/src/js/playlist/playlist.js#L10
3276 if not isinstance(jwplayer_data['playlist'], list):
3277 jwplayer_data['playlist'] = [jwplayer_data['playlist']]
3278
3279 for video_data in jwplayer_data['playlist']:
3280 # JWPlayer backward compatibility: flattened sources
3281 # https://github.com/jwplayer/jwplayer/blob/v7.4.3/src/js/playlist/item.js#L29-L35
3282 if 'sources' not in video_data:
3283 video_data['sources'] = [video_data]
3284
3285 this_video_id = video_id or video_data['mediaid']
3286
3287 formats = self._parse_jwplayer_formats(
3288 video_data['sources'], video_id=this_video_id, m3u8_id=m3u8_id,
3289 mpd_id=mpd_id, rtmp_params=rtmp_params, base_url=base_url)
3290
3291 subtitles = {}
3292 tracks = video_data.get('tracks')
3293 if tracks and isinstance(tracks, list):
3294 for track in tracks:
3295 if not isinstance(track, dict):
3296 continue
3297 track_kind = track.get('kind')
3298 if not track_kind or not isinstance(track_kind, compat_str):
3299 continue
3300 if track_kind.lower() not in ('captions', 'subtitles'):
3301 continue
3302 track_url = urljoin(base_url, track.get('file'))
3303 if not track_url:
3304 continue
3305 subtitles.setdefault(track.get('label') or 'en', []).append({
3306 'url': self._proto_relative_url(track_url)
3307 })
3308
3309 entry = {
3310 'id': this_video_id,
3311 'title': unescapeHTML(video_data['title'] if require_title else video_data.get('title')),
3312 'description': clean_html(video_data.get('description')),
3313 'thumbnail': urljoin(base_url, self._proto_relative_url(video_data.get('image'))),
3314 'timestamp': int_or_none(video_data.get('pubdate')),
3315 'duration': float_or_none(jwplayer_data.get('duration') or video_data.get('duration')),
3316 'subtitles': subtitles,
3317 }
3318 # https://github.com/jwplayer/jwplayer/blob/master/src/js/utils/validator.js#L32
3319 if len(formats) == 1 and re.search(r'^(?:http|//).*(?:youtube\.com|youtu\.be)/.+', formats[0]['url']):
3320 entry.update({
3321 '_type': 'url_transparent',
3322 'url': formats[0]['url'],
3323 })
3324 else:
3325 self._sort_formats(formats)
3326 entry['formats'] = formats
3327 entries.append(entry)
3328 if len(entries) == 1:
3329 return entries[0]
3330 else:
3331 return self.playlist_result(entries)
3332
3333 def _parse_jwplayer_formats(self, jwplayer_sources_data, video_id=None,
3334 m3u8_id=None, mpd_id=None, rtmp_params=None, base_url=None):
3335 urls = []
3336 formats = []
3337 for source in jwplayer_sources_data:
3338 if not isinstance(source, dict):
3339 continue
3340 source_url = urljoin(
3341 base_url, self._proto_relative_url(source.get('file')))
3342 if not source_url or source_url in urls:
3343 continue
3344 urls.append(source_url)
3345 source_type = source.get('type') or ''
3346 ext = mimetype2ext(source_type) or determine_ext(source_url)
3347 if source_type == 'hls' or ext == 'm3u8':
3348 formats.extend(self._extract_m3u8_formats(
3349 source_url, video_id, 'mp4', entry_protocol='m3u8_native',
3350 m3u8_id=m3u8_id, fatal=False))
3351 elif source_type == 'dash' or ext == 'mpd':
3352 formats.extend(self._extract_mpd_formats(
3353 source_url, video_id, mpd_id=mpd_id, fatal=False))
3354 elif ext == 'smil':
3355 formats.extend(self._extract_smil_formats(
3356 source_url, video_id, fatal=False))
3357 # https://github.com/jwplayer/jwplayer/blob/master/src/js/providers/default.js#L67
3358 elif source_type.startswith('audio') or ext in (
3359 'oga', 'aac', 'mp3', 'mpeg', 'vorbis'):
3360 formats.append({
3361 'url': source_url,
3362 'vcodec': 'none',
3363 'ext': ext,
3364 })
3365 else:
3366 height = int_or_none(source.get('height'))
3367 if height is None:
3368 # Often no height is provided but there is a label in
3369 # format like "1080p", "720p SD", or 1080.
3370 height = int_or_none(self._search_regex(
3371 r'^(\d{3,4})[pP]?(?:\b|$)', compat_str(source.get('label') or ''),
3372 'height', default=None))
3373 a_format = {
3374 'url': source_url,
3375 'width': int_or_none(source.get('width')),
3376 'height': height,
3377 'tbr': int_or_none(source.get('bitrate')),
3378 'ext': ext,
3379 }
3380 if source_url.startswith('rtmp'):
3381 a_format['ext'] = 'flv'
3382 # See com/longtailvideo/jwplayer/media/RTMPMediaProvider.as
3383 # of jwplayer.flash.swf
3384 rtmp_url_parts = re.split(
3385 r'((?:mp4|mp3|flv):)', source_url, 1)
3386 if len(rtmp_url_parts) == 3:
3387 rtmp_url, prefix, play_path = rtmp_url_parts
3388 a_format.update({
3389 'url': rtmp_url,
3390 'play_path': prefix + play_path,
3391 })
3392 if rtmp_params:
3393 a_format.update(rtmp_params)
3394 formats.append(a_format)
3395 return formats
3396
3397 def _live_title(self, name):
3398 """ Generate the title for a live video """
3399 now = datetime.datetime.now()
3400 now_str = now.strftime('%Y-%m-%d %H:%M')
3401 return name + ' ' + now_str
3402
3403 def _int(self, v, name, fatal=False, **kwargs):
3404 res = int_or_none(v, **kwargs)
3405 if 'get_attr' in kwargs:
3406 print(getattr(v, kwargs['get_attr']))
3407 if res is None:
3408 msg = 'Failed to extract %s: Could not parse value %r' % (name, v)
3409 if fatal:
3410 raise ExtractorError(msg)
3411 else:
3412 self.report_warning(msg)
3413 return res
3414
3415 def _float(self, v, name, fatal=False, **kwargs):
3416 res = float_or_none(v, **kwargs)
3417 if res is None:
3418 msg = 'Failed to extract %s: Could not parse value %r' % (name, v)
3419 if fatal:
3420 raise ExtractorError(msg)
3421 else:
3422 self.report_warning(msg)
3423 return res
3424
3425 def _set_cookie(self, domain, name, value, expire_time=None, port=None,
3426 path='/', secure=False, discard=False, rest={}, **kwargs):
3427 cookie = compat_cookiejar_Cookie(
3428 0, name, value, port, port is not None, domain, True,
3429 domain.startswith('.'), path, True, secure, expire_time,
3430 discard, None, None, rest)
3431 self._downloader.cookiejar.set_cookie(cookie)
3432
3433 def _get_cookies(self, url):
3434 """ Return a compat_cookies_SimpleCookie with the cookies for the url """
3435 req = sanitized_Request(url)
3436 self._downloader.cookiejar.add_cookie_header(req)
3437 return compat_cookies_SimpleCookie(req.get_header('Cookie'))
3438
3439 def _apply_first_set_cookie_header(self, url_handle, cookie):
3440 """
3441 Apply first Set-Cookie header instead of the last. Experimental.
3442
3443 Some sites (e.g. [1-3]) may serve two cookies under the same name
3444 in Set-Cookie header and expect the first (old) one to be set rather
3445 than second (new). However, as of RFC6265 the newer one cookie
3446 should be set into cookie store what actually happens.
3447 We will workaround this issue by resetting the cookie to
3448 the first one manually.
3449 1. https://new.vk.com/
3450 2. https://github.com/ytdl-org/youtube-dl/issues/9841#issuecomment-227871201
3451 3. https://learning.oreilly.com/
3452 """
3453 for header, cookies in url_handle.headers.items():
3454 if header.lower() != 'set-cookie':
3455 continue
3456 if sys.version_info[0] >= 3:
3457 cookies = cookies.encode('iso-8859-1')
3458 cookies = cookies.decode('utf-8')
3459 cookie_value = re.search(
3460 r'%s=(.+?);.*?\b[Dd]omain=(.+?)(?:[,;]|$)' % cookie, cookies)
3461 if cookie_value:
3462 value, domain = cookie_value.groups()
3463 self._set_cookie(domain, cookie, value)
3464 break
3465
3466 def get_testcases(self, include_onlymatching=False):
3467 t = getattr(self, '_TEST', None)
3468 if t:
3469 assert not hasattr(self, '_TESTS'), \
3470 '%s has _TEST and _TESTS' % type(self).__name__
3471 tests = [t]
3472 else:
3473 tests = getattr(self, '_TESTS', [])
3474 for t in tests:
3475 if not include_onlymatching and t.get('only_matching', False):
3476 continue
3477 t['name'] = type(self).__name__[:-len('IE')]
3478 yield t
3479
3480 def is_suitable(self, age_limit):
3481 """ Test whether the extractor is generally suitable for the given
3482 age limit (i.e. pornographic sites are not, all others usually are) """
3483
3484 any_restricted = False
3485 for tc in self.get_testcases(include_onlymatching=False):
3486 if tc.get('playlist', []):
3487 tc = tc['playlist'][0]
3488 is_restricted = age_restricted(
3489 tc.get('info_dict', {}).get('age_limit'), age_limit)
3490 if not is_restricted:
3491 return True
3492 any_restricted = any_restricted or is_restricted
3493 return not any_restricted
3494
3495 def extract_subtitles(self, *args, **kwargs):
3496 if (self.get_param('writesubtitles', False)
3497 or self.get_param('listsubtitles')):
3498 return self._get_subtitles(*args, **kwargs)
3499 return {}
3500
3501 def _get_subtitles(self, *args, **kwargs):
3502 raise NotImplementedError('This method must be implemented by subclasses')
3503
3504 @staticmethod
3505 def _merge_subtitle_items(subtitle_list1, subtitle_list2):
3506 """ Merge subtitle items for one language. Items with duplicated URLs
3507 will be dropped. """
3508 list1_urls = set([item['url'] for item in subtitle_list1])
3509 ret = list(subtitle_list1)
3510 ret.extend([item for item in subtitle_list2 if item['url'] not in list1_urls])
3511 return ret
3512
3513 @classmethod
3514 def _merge_subtitles(cls, *dicts, target=None):
3515 """ Merge subtitle dictionaries, language by language. """
3516 if target is None:
3517 target = {}
3518 for d in dicts:
3519 for lang, subs in d.items():
3520 target[lang] = cls._merge_subtitle_items(target.get(lang, []), subs)
3521 return target
3522
3523 def extract_automatic_captions(self, *args, **kwargs):
3524 if (self.get_param('writeautomaticsub', False)
3525 or self.get_param('listsubtitles')):
3526 return self._get_automatic_captions(*args, **kwargs)
3527 return {}
3528
3529 def _get_automatic_captions(self, *args, **kwargs):
3530 raise NotImplementedError('This method must be implemented by subclasses')
3531
3532 def mark_watched(self, *args, **kwargs):
3533 if not self.get_param('mark_watched', False):
3534 return
3535 if (self._get_login_info()[0] is not None
3536 or self.get_param('cookiefile')
3537 or self.get_param('cookiesfrombrowser')):
3538 self._mark_watched(*args, **kwargs)
3539
3540 def _mark_watched(self, *args, **kwargs):
3541 raise NotImplementedError('This method must be implemented by subclasses')
3542
3543 def geo_verification_headers(self):
3544 headers = {}
3545 geo_verification_proxy = self.get_param('geo_verification_proxy')
3546 if geo_verification_proxy:
3547 headers['Ytdl-request-proxy'] = geo_verification_proxy
3548 return headers
3549
3550 def _generic_id(self, url):
3551 return compat_urllib_parse_unquote(os.path.splitext(url.rstrip('/').split('/')[-1])[0])
3552
3553 def _generic_title(self, url):
3554 return compat_urllib_parse_unquote(os.path.splitext(url_basename(url))[0])
3555
3556 @staticmethod
3557 def _availability(is_private=None, needs_premium=None, needs_subscription=None, needs_auth=None, is_unlisted=None):
3558 all_known = all(map(
3559 lambda x: x is not None,
3560 (is_private, needs_premium, needs_subscription, needs_auth, is_unlisted)))
3561 return (
3562 'private' if is_private
3563 else 'premium_only' if needs_premium
3564 else 'subscriber_only' if needs_subscription
3565 else 'needs_auth' if needs_auth
3566 else 'unlisted' if is_unlisted
3567 else 'public' if all_known
3568 else None)
3569
3570 def _configuration_arg(self, key, default=NO_DEFAULT, casesense=False):
3571 '''
3572 @returns A list of values for the extractor argument given by "key"
3573 or "default" if no such key is present
3574 @param default The default value to return when the key is not present (default: [])
3575 @param casesense When false, the values are converted to lower case
3576 '''
3577 val = traverse_obj(
3578 self._downloader.params, ('extractor_args', self.ie_key().lower(), key))
3579 if val is None:
3580 return [] if default is NO_DEFAULT else default
3581 return list(val) if casesense else [x.lower() for x in val]
3582
3583
3584 class SearchInfoExtractor(InfoExtractor):
3585 """
3586 Base class for paged search queries extractors.
3587 They accept URLs in the format _SEARCH_KEY(|all|[0-9]):{query}
3588 Instances should define _SEARCH_KEY and _MAX_RESULTS.
3589 """
3590
3591 @classmethod
3592 def _make_valid_url(cls):
3593 return r'%s(?P<prefix>|[1-9][0-9]*|all):(?P<query>[\s\S]+)' % cls._SEARCH_KEY
3594
3595 @classmethod
3596 def suitable(cls, url):
3597 return re.match(cls._make_valid_url(), url) is not None
3598
3599 def _real_extract(self, query):
3600 mobj = re.match(self._make_valid_url(), query)
3601 if mobj is None:
3602 raise ExtractorError('Invalid search query "%s"' % query)
3603
3604 prefix = mobj.group('prefix')
3605 query = mobj.group('query')
3606 if prefix == '':
3607 return self._get_n_results(query, 1)
3608 elif prefix == 'all':
3609 return self._get_n_results(query, self._MAX_RESULTS)
3610 else:
3611 n = int(prefix)
3612 if n <= 0:
3613 raise ExtractorError('invalid download number %s for query "%s"' % (n, query))
3614 elif n > self._MAX_RESULTS:
3615 self.report_warning('%s returns max %i results (you requested %i)' % (self._SEARCH_KEY, self._MAX_RESULTS, n))
3616 n = self._MAX_RESULTS
3617 return self._get_n_results(query, n)
3618
3619 def _get_n_results(self, query, n):
3620 """Get a specified number of results for a query"""
3621 raise NotImplementedError('This method must be implemented by subclasses')
3622
3623 @property
3624 def SEARCH_KEY(self):
3625 return self._SEARCH_KEY