]> jfr.im git - yt-dlp.git/blob - yt_dlp/extractor/common.py
[docs,cleanup] Some minor refactoring and improve docs
[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 if not self.get_param('no_color') and compat_os_name != 'nt' and sys.stderr.isatty():
1138 _name = '\033[0;34m%s\033[0m' % name
1139 else:
1140 _name = name
1141
1142 if mobj:
1143 if group is None:
1144 # return the first matching group
1145 return next(g for g in mobj.groups() if g is not None)
1146 elif isinstance(group, (list, tuple)):
1147 return tuple(mobj.group(g) for g in group)
1148 else:
1149 return mobj.group(group)
1150 elif default is not NO_DEFAULT:
1151 return default
1152 elif fatal:
1153 raise RegexNotFoundError('Unable to extract %s' % _name)
1154 else:
1155 self.report_warning('unable to extract %s' % _name + bug_reports_message())
1156 return None
1157
1158 def _html_search_regex(self, pattern, string, name, default=NO_DEFAULT, fatal=True, flags=0, group=None):
1159 """
1160 Like _search_regex, but strips HTML tags and unescapes entities.
1161 """
1162 res = self._search_regex(pattern, string, name, default, fatal, flags, group)
1163 if res:
1164 return clean_html(res).strip()
1165 else:
1166 return res
1167
1168 def _get_netrc_login_info(self, netrc_machine=None):
1169 username = None
1170 password = None
1171 netrc_machine = netrc_machine or self._NETRC_MACHINE
1172
1173 if self.get_param('usenetrc', False):
1174 try:
1175 netrc_file = compat_expanduser(self.get_param('netrc_location') or '~')
1176 if os.path.isdir(netrc_file):
1177 netrc_file = os.path.join(netrc_file, '.netrc')
1178 info = netrc.netrc(file=netrc_file).authenticators(netrc_machine)
1179 if info is not None:
1180 username = info[0]
1181 password = info[2]
1182 else:
1183 raise netrc.NetrcParseError(
1184 'No authenticators for %s' % netrc_machine)
1185 except (IOError, netrc.NetrcParseError) as err:
1186 self.report_warning(
1187 'parsing .netrc: %s' % error_to_compat_str(err))
1188
1189 return username, password
1190
1191 def _get_login_info(self, username_option='username', password_option='password', netrc_machine=None):
1192 """
1193 Get the login info as (username, password)
1194 First look for the manually specified credentials using username_option
1195 and password_option as keys in params dictionary. If no such credentials
1196 available look in the netrc file using the netrc_machine or _NETRC_MACHINE
1197 value.
1198 If there's no info available, return (None, None)
1199 """
1200
1201 # Attempt to use provided username and password or .netrc data
1202 username = self.get_param(username_option)
1203 if username is not None:
1204 password = self.get_param(password_option)
1205 else:
1206 username, password = self._get_netrc_login_info(netrc_machine)
1207
1208 return username, password
1209
1210 def _get_tfa_info(self, note='two-factor verification code'):
1211 """
1212 Get the two-factor authentication info
1213 TODO - asking the user will be required for sms/phone verify
1214 currently just uses the command line option
1215 If there's no info available, return None
1216 """
1217
1218 tfa = self.get_param('twofactor')
1219 if tfa is not None:
1220 return tfa
1221
1222 return compat_getpass('Type %s and press [Return]: ' % note)
1223
1224 # Helper functions for extracting OpenGraph info
1225 @staticmethod
1226 def _og_regexes(prop):
1227 content_re = r'content=(?:"([^"]+?)"|\'([^\']+?)\'|\s*([^\s"\'=<>`]+?))'
1228 property_re = (r'(?:name|property)=(?:\'og[:-]%(prop)s\'|"og[:-]%(prop)s"|\s*og[:-]%(prop)s\b)'
1229 % {'prop': re.escape(prop)})
1230 template = r'<meta[^>]+?%s[^>]+?%s'
1231 return [
1232 template % (property_re, content_re),
1233 template % (content_re, property_re),
1234 ]
1235
1236 @staticmethod
1237 def _meta_regex(prop):
1238 return r'''(?isx)<meta
1239 (?=[^>]+(?:itemprop|name|property|id|http-equiv)=(["\']?)%s\1)
1240 [^>]+?content=(["\'])(?P<content>.*?)\2''' % re.escape(prop)
1241
1242 def _og_search_property(self, prop, html, name=None, **kargs):
1243 prop = variadic(prop)
1244 if name is None:
1245 name = 'OpenGraph %s' % prop[0]
1246 og_regexes = []
1247 for p in prop:
1248 og_regexes.extend(self._og_regexes(p))
1249 escaped = self._search_regex(og_regexes, html, name, flags=re.DOTALL, **kargs)
1250 if escaped is None:
1251 return None
1252 return unescapeHTML(escaped)
1253
1254 def _og_search_thumbnail(self, html, **kargs):
1255 return self._og_search_property('image', html, 'thumbnail URL', fatal=False, **kargs)
1256
1257 def _og_search_description(self, html, **kargs):
1258 return self._og_search_property('description', html, fatal=False, **kargs)
1259
1260 def _og_search_title(self, html, **kargs):
1261 return self._og_search_property('title', html, **kargs)
1262
1263 def _og_search_video_url(self, html, name='video url', secure=True, **kargs):
1264 regexes = self._og_regexes('video') + self._og_regexes('video:url')
1265 if secure:
1266 regexes = self._og_regexes('video:secure_url') + regexes
1267 return self._html_search_regex(regexes, html, name, **kargs)
1268
1269 def _og_search_url(self, html, **kargs):
1270 return self._og_search_property('url', html, **kargs)
1271
1272 def _html_search_meta(self, name, html, display_name=None, fatal=False, **kwargs):
1273 name = variadic(name)
1274 if display_name is None:
1275 display_name = name[0]
1276 return self._html_search_regex(
1277 [self._meta_regex(n) for n in name],
1278 html, display_name, fatal=fatal, group='content', **kwargs)
1279
1280 def _dc_search_uploader(self, html):
1281 return self._html_search_meta('dc.creator', html, 'uploader')
1282
1283 def _rta_search(self, html):
1284 # See http://www.rtalabel.org/index.php?content=howtofaq#single
1285 if re.search(r'(?ix)<meta\s+name="rating"\s+'
1286 r' content="RTA-5042-1996-1400-1577-RTA"',
1287 html):
1288 return 18
1289 return 0
1290
1291 def _media_rating_search(self, html):
1292 # See http://www.tjg-designs.com/WP/metadata-code-examples-adding-metadata-to-your-web-pages/
1293 rating = self._html_search_meta('rating', html)
1294
1295 if not rating:
1296 return None
1297
1298 RATING_TABLE = {
1299 'safe for kids': 0,
1300 'general': 8,
1301 '14 years': 14,
1302 'mature': 17,
1303 'restricted': 19,
1304 }
1305 return RATING_TABLE.get(rating.lower())
1306
1307 def _family_friendly_search(self, html):
1308 # See http://schema.org/VideoObject
1309 family_friendly = self._html_search_meta(
1310 'isFamilyFriendly', html, default=None)
1311
1312 if not family_friendly:
1313 return None
1314
1315 RATING_TABLE = {
1316 '1': 0,
1317 'true': 0,
1318 '0': 18,
1319 'false': 18,
1320 }
1321 return RATING_TABLE.get(family_friendly.lower())
1322
1323 def _twitter_search_player(self, html):
1324 return self._html_search_meta('twitter:player', html,
1325 'twitter card player')
1326
1327 def _search_json_ld(self, html, video_id, expected_type=None, **kwargs):
1328 json_ld_list = list(re.finditer(JSON_LD_RE, html))
1329 default = kwargs.get('default', NO_DEFAULT)
1330 # JSON-LD may be malformed and thus `fatal` should be respected.
1331 # At the same time `default` may be passed that assumes `fatal=False`
1332 # for _search_regex. Let's simulate the same behavior here as well.
1333 fatal = kwargs.get('fatal', True) if default is NO_DEFAULT else False
1334 json_ld = []
1335 for mobj in json_ld_list:
1336 json_ld_item = self._parse_json(
1337 mobj.group('json_ld'), video_id, fatal=fatal)
1338 if not json_ld_item:
1339 continue
1340 if isinstance(json_ld_item, dict):
1341 json_ld.append(json_ld_item)
1342 elif isinstance(json_ld_item, (list, tuple)):
1343 json_ld.extend(json_ld_item)
1344 if json_ld:
1345 json_ld = self._json_ld(json_ld, video_id, fatal=fatal, expected_type=expected_type)
1346 if json_ld:
1347 return json_ld
1348 if default is not NO_DEFAULT:
1349 return default
1350 elif fatal:
1351 raise RegexNotFoundError('Unable to extract JSON-LD')
1352 else:
1353 self.report_warning('unable to extract JSON-LD %s' % bug_reports_message())
1354 return {}
1355
1356 def _json_ld(self, json_ld, video_id, fatal=True, expected_type=None):
1357 if isinstance(json_ld, compat_str):
1358 json_ld = self._parse_json(json_ld, video_id, fatal=fatal)
1359 if not json_ld:
1360 return {}
1361 info = {}
1362 if not isinstance(json_ld, (list, tuple, dict)):
1363 return info
1364 if isinstance(json_ld, dict):
1365 json_ld = [json_ld]
1366
1367 INTERACTION_TYPE_MAP = {
1368 'CommentAction': 'comment',
1369 'AgreeAction': 'like',
1370 'DisagreeAction': 'dislike',
1371 'LikeAction': 'like',
1372 'DislikeAction': 'dislike',
1373 'ListenAction': 'view',
1374 'WatchAction': 'view',
1375 'ViewAction': 'view',
1376 }
1377
1378 def extract_interaction_type(e):
1379 interaction_type = e.get('interactionType')
1380 if isinstance(interaction_type, dict):
1381 interaction_type = interaction_type.get('@type')
1382 return str_or_none(interaction_type)
1383
1384 def extract_interaction_statistic(e):
1385 interaction_statistic = e.get('interactionStatistic')
1386 if isinstance(interaction_statistic, dict):
1387 interaction_statistic = [interaction_statistic]
1388 if not isinstance(interaction_statistic, list):
1389 return
1390 for is_e in interaction_statistic:
1391 if not isinstance(is_e, dict):
1392 continue
1393 if is_e.get('@type') != 'InteractionCounter':
1394 continue
1395 interaction_type = extract_interaction_type(is_e)
1396 if not interaction_type:
1397 continue
1398 # For interaction count some sites provide string instead of
1399 # an integer (as per spec) with non digit characters (e.g. ",")
1400 # so extracting count with more relaxed str_to_int
1401 interaction_count = str_to_int(is_e.get('userInteractionCount'))
1402 if interaction_count is None:
1403 continue
1404 count_kind = INTERACTION_TYPE_MAP.get(interaction_type.split('/')[-1])
1405 if not count_kind:
1406 continue
1407 count_key = '%s_count' % count_kind
1408 if info.get(count_key) is not None:
1409 continue
1410 info[count_key] = interaction_count
1411
1412 def extract_video_object(e):
1413 assert e['@type'] == 'VideoObject'
1414 author = e.get('author')
1415 info.update({
1416 'url': url_or_none(e.get('contentUrl')),
1417 'title': unescapeHTML(e.get('name')),
1418 'description': unescapeHTML(e.get('description')),
1419 'thumbnail': url_or_none(e.get('thumbnailUrl') or e.get('thumbnailURL')),
1420 'duration': parse_duration(e.get('duration')),
1421 'timestamp': unified_timestamp(e.get('uploadDate')),
1422 # author can be an instance of 'Organization' or 'Person' types.
1423 # both types can have 'name' property(inherited from 'Thing' type). [1]
1424 # however some websites are using 'Text' type instead.
1425 # 1. https://schema.org/VideoObject
1426 'uploader': author.get('name') if isinstance(author, dict) else author if isinstance(author, compat_str) else None,
1427 'filesize': float_or_none(e.get('contentSize')),
1428 'tbr': int_or_none(e.get('bitrate')),
1429 'width': int_or_none(e.get('width')),
1430 'height': int_or_none(e.get('height')),
1431 'view_count': int_or_none(e.get('interactionCount')),
1432 })
1433 extract_interaction_statistic(e)
1434
1435 for e in json_ld:
1436 if '@context' in e:
1437 item_type = e.get('@type')
1438 if expected_type is not None and expected_type != item_type:
1439 continue
1440 if item_type in ('TVEpisode', 'Episode'):
1441 episode_name = unescapeHTML(e.get('name'))
1442 info.update({
1443 'episode': episode_name,
1444 'episode_number': int_or_none(e.get('episodeNumber')),
1445 'description': unescapeHTML(e.get('description')),
1446 })
1447 if not info.get('title') and episode_name:
1448 info['title'] = episode_name
1449 part_of_season = e.get('partOfSeason')
1450 if isinstance(part_of_season, dict) and part_of_season.get('@type') in ('TVSeason', 'Season', 'CreativeWorkSeason'):
1451 info.update({
1452 'season': unescapeHTML(part_of_season.get('name')),
1453 'season_number': int_or_none(part_of_season.get('seasonNumber')),
1454 })
1455 part_of_series = e.get('partOfSeries') or e.get('partOfTVSeries')
1456 if isinstance(part_of_series, dict) and part_of_series.get('@type') in ('TVSeries', 'Series', 'CreativeWorkSeries'):
1457 info['series'] = unescapeHTML(part_of_series.get('name'))
1458 elif item_type == 'Movie':
1459 info.update({
1460 'title': unescapeHTML(e.get('name')),
1461 'description': unescapeHTML(e.get('description')),
1462 'duration': parse_duration(e.get('duration')),
1463 'timestamp': unified_timestamp(e.get('dateCreated')),
1464 })
1465 elif item_type in ('Article', 'NewsArticle'):
1466 info.update({
1467 'timestamp': parse_iso8601(e.get('datePublished')),
1468 'title': unescapeHTML(e.get('headline')),
1469 'description': unescapeHTML(e.get('articleBody')),
1470 })
1471 elif item_type == 'VideoObject':
1472 extract_video_object(e)
1473 if expected_type is None:
1474 continue
1475 else:
1476 break
1477 video = e.get('video')
1478 if isinstance(video, dict) and video.get('@type') == 'VideoObject':
1479 extract_video_object(video)
1480 if expected_type is None:
1481 continue
1482 else:
1483 break
1484 return dict((k, v) for k, v in info.items() if v is not None)
1485
1486 @staticmethod
1487 def _hidden_inputs(html):
1488 html = re.sub(r'<!--(?:(?!<!--).)*-->', '', html)
1489 hidden_inputs = {}
1490 for input in re.findall(r'(?i)(<input[^>]+>)', html):
1491 attrs = extract_attributes(input)
1492 if not input:
1493 continue
1494 if attrs.get('type') not in ('hidden', 'submit'):
1495 continue
1496 name = attrs.get('name') or attrs.get('id')
1497 value = attrs.get('value')
1498 if name and value is not None:
1499 hidden_inputs[name] = value
1500 return hidden_inputs
1501
1502 def _form_hidden_inputs(self, form_id, html):
1503 form = self._search_regex(
1504 r'(?is)<form[^>]+?id=(["\'])%s\1[^>]*>(?P<form>.+?)</form>' % form_id,
1505 html, '%s form' % form_id, group='form')
1506 return self._hidden_inputs(form)
1507
1508 class FormatSort:
1509 regex = r' *((?P<reverse>\+)?(?P<field>[a-zA-Z0-9_]+)((?P<separator>[~:])(?P<limit>.*?))?)? *$'
1510
1511 default = ('hidden', 'aud_or_vid', 'hasvid', 'ie_pref', 'lang', 'quality',
1512 'res', 'fps', 'codec:vp9.2', 'size', 'br', 'asr',
1513 'proto', 'ext', 'hasaud', 'source', 'format_id') # These must not be aliases
1514 ytdl_default = ('hasaud', 'lang', 'quality', 'tbr', 'filesize', 'vbr',
1515 'height', 'width', 'proto', 'vext', 'abr', 'aext',
1516 'fps', 'fs_approx', 'source', 'format_id')
1517
1518 settings = {
1519 'vcodec': {'type': 'ordered', 'regex': True,
1520 'order': ['av0?1', 'vp0?9.2', 'vp0?9', '[hx]265|he?vc?', '[hx]264|avc', 'vp0?8', 'mp4v|h263', 'theora', '', None, 'none']},
1521 'acodec': {'type': 'ordered', 'regex': True,
1522 'order': ['opus', 'vorbis', 'aac', 'mp?4a?', 'mp3', 'e?a?c-?3', 'dts', '', None, 'none']},
1523 'proto': {'type': 'ordered', 'regex': True, 'field': 'protocol',
1524 'order': ['(ht|f)tps', '(ht|f)tp$', 'm3u8.+', '.*dash', 'ws|websocket', '', 'mms|rtsp', 'none', 'f4']},
1525 'vext': {'type': 'ordered', 'field': 'video_ext',
1526 'order': ('mp4', 'webm', 'flv', '', 'none'),
1527 'order_free': ('webm', 'mp4', 'flv', '', 'none')},
1528 'aext': {'type': 'ordered', 'field': 'audio_ext',
1529 'order': ('m4a', 'aac', 'mp3', 'ogg', 'opus', 'webm', '', 'none'),
1530 'order_free': ('opus', 'ogg', 'webm', 'm4a', 'mp3', 'aac', '', 'none')},
1531 'hidden': {'visible': False, 'forced': True, 'type': 'extractor', 'max': -1000},
1532 'aud_or_vid': {'visible': False, 'forced': True, 'type': 'multiple',
1533 'field': ('vcodec', 'acodec'),
1534 'function': lambda it: int(any(v != 'none' for v in it))},
1535 'ie_pref': {'priority': True, 'type': 'extractor'},
1536 'hasvid': {'priority': True, 'field': 'vcodec', 'type': 'boolean', 'not_in_list': ('none',)},
1537 'hasaud': {'field': 'acodec', 'type': 'boolean', 'not_in_list': ('none',)},
1538 'lang': {'convert': 'ignore', 'field': 'language_preference'},
1539 'quality': {'convert': 'float_none', 'default': -1},
1540 'filesize': {'convert': 'bytes'},
1541 'fs_approx': {'convert': 'bytes', 'field': 'filesize_approx'},
1542 'id': {'convert': 'string', 'field': 'format_id'},
1543 'height': {'convert': 'float_none'},
1544 'width': {'convert': 'float_none'},
1545 'fps': {'convert': 'float_none'},
1546 'tbr': {'convert': 'float_none'},
1547 'vbr': {'convert': 'float_none'},
1548 'abr': {'convert': 'float_none'},
1549 'asr': {'convert': 'float_none'},
1550 'source': {'convert': 'ignore', 'field': 'source_preference'},
1551
1552 'codec': {'type': 'combined', 'field': ('vcodec', 'acodec')},
1553 'br': {'type': 'combined', 'field': ('tbr', 'vbr', 'abr'), 'same_limit': True},
1554 'size': {'type': 'combined', 'same_limit': True, 'field': ('filesize', 'fs_approx')},
1555 'ext': {'type': 'combined', 'field': ('vext', 'aext')},
1556 'res': {'type': 'multiple', 'field': ('height', 'width'),
1557 'function': lambda it: (lambda l: min(l) if l else 0)(tuple(filter(None, it)))},
1558
1559 # Most of these exist only for compatibility reasons
1560 'dimension': {'type': 'alias', 'field': 'res'},
1561 'resolution': {'type': 'alias', 'field': 'res'},
1562 'extension': {'type': 'alias', 'field': 'ext'},
1563 'bitrate': {'type': 'alias', 'field': 'br'},
1564 'total_bitrate': {'type': 'alias', 'field': 'tbr'},
1565 'video_bitrate': {'type': 'alias', 'field': 'vbr'},
1566 'audio_bitrate': {'type': 'alias', 'field': 'abr'},
1567 'framerate': {'type': 'alias', 'field': 'fps'},
1568 'language_preference': {'type': 'alias', 'field': 'lang'}, # not named as 'language' because such a field exists
1569 'protocol': {'type': 'alias', 'field': 'proto'},
1570 'source_preference': {'type': 'alias', 'field': 'source'},
1571 'filesize_approx': {'type': 'alias', 'field': 'fs_approx'},
1572 'filesize_estimate': {'type': 'alias', 'field': 'size'},
1573 'samplerate': {'type': 'alias', 'field': 'asr'},
1574 'video_ext': {'type': 'alias', 'field': 'vext'},
1575 'audio_ext': {'type': 'alias', 'field': 'aext'},
1576 'video_codec': {'type': 'alias', 'field': 'vcodec'},
1577 'audio_codec': {'type': 'alias', 'field': 'acodec'},
1578 'video': {'type': 'alias', 'field': 'hasvid'},
1579 'has_video': {'type': 'alias', 'field': 'hasvid'},
1580 'audio': {'type': 'alias', 'field': 'hasaud'},
1581 'has_audio': {'type': 'alias', 'field': 'hasaud'},
1582 'extractor': {'type': 'alias', 'field': 'ie_pref'},
1583 'preference': {'type': 'alias', 'field': 'ie_pref'},
1584 'extractor_preference': {'type': 'alias', 'field': 'ie_pref'},
1585 'format_id': {'type': 'alias', 'field': 'id'},
1586 }
1587
1588 _order = []
1589
1590 def _get_field_setting(self, field, key):
1591 if field not in self.settings:
1592 self.settings[field] = {}
1593 propObj = self.settings[field]
1594 if key not in propObj:
1595 type = propObj.get('type')
1596 if key == 'field':
1597 default = 'preference' if type == 'extractor' else (field,) if type in ('combined', 'multiple') else field
1598 elif key == 'convert':
1599 default = 'order' if type == 'ordered' else 'float_string' if field else 'ignore'
1600 else:
1601 default = {'type': 'field', 'visible': True, 'order': [], 'not_in_list': (None,)}.get(key, None)
1602 propObj[key] = default
1603 return propObj[key]
1604
1605 def _resolve_field_value(self, field, value, convertNone=False):
1606 if value is None:
1607 if not convertNone:
1608 return None
1609 else:
1610 value = value.lower()
1611 conversion = self._get_field_setting(field, 'convert')
1612 if conversion == 'ignore':
1613 return None
1614 if conversion == 'string':
1615 return value
1616 elif conversion == 'float_none':
1617 return float_or_none(value)
1618 elif conversion == 'bytes':
1619 return FileDownloader.parse_bytes(value)
1620 elif conversion == 'order':
1621 order_list = (self._use_free_order and self._get_field_setting(field, 'order_free')) or self._get_field_setting(field, 'order')
1622 use_regex = self._get_field_setting(field, 'regex')
1623 list_length = len(order_list)
1624 empty_pos = order_list.index('') if '' in order_list else list_length + 1
1625 if use_regex and value is not None:
1626 for i, regex in enumerate(order_list):
1627 if regex and re.match(regex, value):
1628 return list_length - i
1629 return list_length - empty_pos # not in list
1630 else: # not regex or value = None
1631 return list_length - (order_list.index(value) if value in order_list else empty_pos)
1632 else:
1633 if value.isnumeric():
1634 return float(value)
1635 else:
1636 self.settings[field]['convert'] = 'string'
1637 return value
1638
1639 def evaluate_params(self, params, sort_extractor):
1640 self._use_free_order = params.get('prefer_free_formats', False)
1641 self._sort_user = params.get('format_sort', [])
1642 self._sort_extractor = sort_extractor
1643
1644 def add_item(field, reverse, closest, limit_text):
1645 field = field.lower()
1646 if field in self._order:
1647 return
1648 self._order.append(field)
1649 limit = self._resolve_field_value(field, limit_text)
1650 data = {
1651 'reverse': reverse,
1652 'closest': False if limit is None else closest,
1653 'limit_text': limit_text,
1654 'limit': limit}
1655 if field in self.settings:
1656 self.settings[field].update(data)
1657 else:
1658 self.settings[field] = data
1659
1660 sort_list = (
1661 tuple(field for field in self.default if self._get_field_setting(field, 'forced'))
1662 + (tuple() if params.get('format_sort_force', False)
1663 else tuple(field for field in self.default if self._get_field_setting(field, 'priority')))
1664 + tuple(self._sort_user) + tuple(sort_extractor) + self.default)
1665
1666 for item in sort_list:
1667 match = re.match(self.regex, item)
1668 if match is None:
1669 raise ExtractorError('Invalid format sort string "%s" given by extractor' % item)
1670 field = match.group('field')
1671 if field is None:
1672 continue
1673 if self._get_field_setting(field, 'type') == 'alias':
1674 field = self._get_field_setting(field, 'field')
1675 reverse = match.group('reverse') is not None
1676 closest = match.group('separator') == '~'
1677 limit_text = match.group('limit')
1678
1679 has_limit = limit_text is not None
1680 has_multiple_fields = self._get_field_setting(field, 'type') == 'combined'
1681 has_multiple_limits = has_limit and has_multiple_fields and not self._get_field_setting(field, 'same_limit')
1682
1683 fields = self._get_field_setting(field, 'field') if has_multiple_fields else (field,)
1684 limits = limit_text.split(":") if has_multiple_limits else (limit_text,) if has_limit else tuple()
1685 limit_count = len(limits)
1686 for (i, f) in enumerate(fields):
1687 add_item(f, reverse, closest,
1688 limits[i] if i < limit_count
1689 else limits[0] if has_limit and not has_multiple_limits
1690 else None)
1691
1692 def print_verbose_info(self, write_debug):
1693 if self._sort_user:
1694 write_debug('Sort order given by user: %s' % ', '.join(self._sort_user))
1695 if self._sort_extractor:
1696 write_debug('Sort order given by extractor: %s' % ', '.join(self._sort_extractor))
1697 write_debug('Formats sorted by: %s' % ', '.join(['%s%s%s' % (
1698 '+' if self._get_field_setting(field, 'reverse') else '', field,
1699 '%s%s(%s)' % ('~' if self._get_field_setting(field, 'closest') else ':',
1700 self._get_field_setting(field, 'limit_text'),
1701 self._get_field_setting(field, 'limit'))
1702 if self._get_field_setting(field, 'limit_text') is not None else '')
1703 for field in self._order if self._get_field_setting(field, 'visible')]))
1704
1705 def _calculate_field_preference_from_value(self, format, field, type, value):
1706 reverse = self._get_field_setting(field, 'reverse')
1707 closest = self._get_field_setting(field, 'closest')
1708 limit = self._get_field_setting(field, 'limit')
1709
1710 if type == 'extractor':
1711 maximum = self._get_field_setting(field, 'max')
1712 if value is None or (maximum is not None and value >= maximum):
1713 value = -1
1714 elif type == 'boolean':
1715 in_list = self._get_field_setting(field, 'in_list')
1716 not_in_list = self._get_field_setting(field, 'not_in_list')
1717 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
1718 elif type == 'ordered':
1719 value = self._resolve_field_value(field, value, True)
1720
1721 # try to convert to number
1722 val_num = float_or_none(value, default=self._get_field_setting(field, 'default'))
1723 is_num = self._get_field_setting(field, 'convert') != 'string' and val_num is not None
1724 if is_num:
1725 value = val_num
1726
1727 return ((-10, 0) if value is None
1728 else (1, value, 0) if not is_num # if a field has mixed strings and numbers, strings are sorted higher
1729 else (0, -abs(value - limit), value - limit if reverse else limit - value) if closest
1730 else (0, value, 0) if not reverse and (limit is None or value <= limit)
1731 else (0, -value, 0) if limit is None or (reverse and value == limit) or value > limit
1732 else (-1, value, 0))
1733
1734 def _calculate_field_preference(self, format, field):
1735 type = self._get_field_setting(field, 'type') # extractor, boolean, ordered, field, multiple
1736 get_value = lambda f: format.get(self._get_field_setting(f, 'field'))
1737 if type == 'multiple':
1738 type = 'field' # Only 'field' is allowed in multiple for now
1739 actual_fields = self._get_field_setting(field, 'field')
1740
1741 value = self._get_field_setting(field, 'function')(get_value(f) for f in actual_fields)
1742 else:
1743 value = get_value(field)
1744 return self._calculate_field_preference_from_value(format, field, type, value)
1745
1746 def calculate_preference(self, format):
1747 # Determine missing protocol
1748 if not format.get('protocol'):
1749 format['protocol'] = determine_protocol(format)
1750
1751 # Determine missing ext
1752 if not format.get('ext') and 'url' in format:
1753 format['ext'] = determine_ext(format['url'])
1754 if format.get('vcodec') == 'none':
1755 format['audio_ext'] = format['ext'] if format.get('acodec') != 'none' else 'none'
1756 format['video_ext'] = 'none'
1757 else:
1758 format['video_ext'] = format['ext']
1759 format['audio_ext'] = 'none'
1760 # if format.get('preference') is None and format.get('ext') in ('f4f', 'f4m'): # Not supported?
1761 # format['preference'] = -1000
1762
1763 # Determine missing bitrates
1764 if format.get('tbr') is None:
1765 if format.get('vbr') is not None and format.get('abr') is not None:
1766 format['tbr'] = format.get('vbr', 0) + format.get('abr', 0)
1767 else:
1768 if format.get('vcodec') != "none" and format.get('vbr') is None:
1769 format['vbr'] = format.get('tbr') - format.get('abr', 0)
1770 if format.get('acodec') != "none" and format.get('abr') is None:
1771 format['abr'] = format.get('tbr') - format.get('vbr', 0)
1772
1773 return tuple(self._calculate_field_preference(format, field) for field in self._order)
1774
1775 def _sort_formats(self, formats, field_preference=[]):
1776 if not formats:
1777 return
1778 format_sort = self.FormatSort() # params and to_screen are taken from the downloader
1779 format_sort.evaluate_params(self._downloader.params, field_preference)
1780 if self.get_param('verbose', False):
1781 format_sort.print_verbose_info(self._downloader.write_debug)
1782 formats.sort(key=lambda f: format_sort.calculate_preference(f))
1783
1784 def _check_formats(self, formats, video_id):
1785 if formats:
1786 formats[:] = filter(
1787 lambda f: self._is_valid_url(
1788 f['url'], video_id,
1789 item='%s video format' % f.get('format_id') if f.get('format_id') else 'video'),
1790 formats)
1791
1792 @staticmethod
1793 def _remove_duplicate_formats(formats):
1794 format_urls = set()
1795 unique_formats = []
1796 for f in formats:
1797 if f['url'] not in format_urls:
1798 format_urls.add(f['url'])
1799 unique_formats.append(f)
1800 formats[:] = unique_formats
1801
1802 def _is_valid_url(self, url, video_id, item='video', headers={}):
1803 url = self._proto_relative_url(url, scheme='http:')
1804 # For now assume non HTTP(S) URLs always valid
1805 if not (url.startswith('http://') or url.startswith('https://')):
1806 return True
1807 try:
1808 self._request_webpage(url, video_id, 'Checking %s URL' % item, headers=headers)
1809 return True
1810 except ExtractorError as e:
1811 self.to_screen(
1812 '%s: %s URL is invalid, skipping: %s'
1813 % (video_id, item, error_to_compat_str(e.cause)))
1814 return False
1815
1816 def http_scheme(self):
1817 """ Either "http:" or "https:", depending on the user's preferences """
1818 return (
1819 'http:'
1820 if self.get_param('prefer_insecure', False)
1821 else 'https:')
1822
1823 def _proto_relative_url(self, url, scheme=None):
1824 if url is None:
1825 return url
1826 if url.startswith('//'):
1827 if scheme is None:
1828 scheme = self.http_scheme()
1829 return scheme + url
1830 else:
1831 return url
1832
1833 def _sleep(self, timeout, video_id, msg_template=None):
1834 if msg_template is None:
1835 msg_template = '%(video_id)s: Waiting for %(timeout)s seconds'
1836 msg = msg_template % {'video_id': video_id, 'timeout': timeout}
1837 self.to_screen(msg)
1838 time.sleep(timeout)
1839
1840 def _extract_f4m_formats(self, manifest_url, video_id, preference=None, quality=None, f4m_id=None,
1841 transform_source=lambda s: fix_xml_ampersands(s).strip(),
1842 fatal=True, m3u8_id=None, data=None, headers={}, query={}):
1843 manifest = self._download_xml(
1844 manifest_url, video_id, 'Downloading f4m manifest',
1845 'Unable to download f4m manifest',
1846 # Some manifests may be malformed, e.g. prosiebensat1 generated manifests
1847 # (see https://github.com/ytdl-org/youtube-dl/issues/6215#issuecomment-121704244)
1848 transform_source=transform_source,
1849 fatal=fatal, data=data, headers=headers, query=query)
1850
1851 if manifest is False:
1852 return []
1853
1854 return self._parse_f4m_formats(
1855 manifest, manifest_url, video_id, preference=preference, quality=quality, f4m_id=f4m_id,
1856 transform_source=transform_source, fatal=fatal, m3u8_id=m3u8_id)
1857
1858 def _parse_f4m_formats(self, manifest, manifest_url, video_id, preference=None, quality=None, f4m_id=None,
1859 transform_source=lambda s: fix_xml_ampersands(s).strip(),
1860 fatal=True, m3u8_id=None):
1861 if not isinstance(manifest, compat_etree_Element) and not fatal:
1862 return []
1863
1864 # currently yt-dlp cannot decode the playerVerificationChallenge as Akamai uses Adobe Alchemy
1865 akamai_pv = manifest.find('{http://ns.adobe.com/f4m/1.0}pv-2.0')
1866 if akamai_pv is not None and ';' in akamai_pv.text:
1867 playerVerificationChallenge = akamai_pv.text.split(';')[0]
1868 if playerVerificationChallenge.strip() != '':
1869 return []
1870
1871 formats = []
1872 manifest_version = '1.0'
1873 media_nodes = manifest.findall('{http://ns.adobe.com/f4m/1.0}media')
1874 if not media_nodes:
1875 manifest_version = '2.0'
1876 media_nodes = manifest.findall('{http://ns.adobe.com/f4m/2.0}media')
1877 # Remove unsupported DRM protected media from final formats
1878 # rendition (see https://github.com/ytdl-org/youtube-dl/issues/8573).
1879 media_nodes = remove_encrypted_media(media_nodes)
1880 if not media_nodes:
1881 return formats
1882
1883 manifest_base_url = get_base_url(manifest)
1884
1885 bootstrap_info = xpath_element(
1886 manifest, ['{http://ns.adobe.com/f4m/1.0}bootstrapInfo', '{http://ns.adobe.com/f4m/2.0}bootstrapInfo'],
1887 'bootstrap info', default=None)
1888
1889 vcodec = None
1890 mime_type = xpath_text(
1891 manifest, ['{http://ns.adobe.com/f4m/1.0}mimeType', '{http://ns.adobe.com/f4m/2.0}mimeType'],
1892 'base URL', default=None)
1893 if mime_type and mime_type.startswith('audio/'):
1894 vcodec = 'none'
1895
1896 for i, media_el in enumerate(media_nodes):
1897 tbr = int_or_none(media_el.attrib.get('bitrate'))
1898 width = int_or_none(media_el.attrib.get('width'))
1899 height = int_or_none(media_el.attrib.get('height'))
1900 format_id = '-'.join(filter(None, [f4m_id, compat_str(i if tbr is None else tbr)]))
1901 # If <bootstrapInfo> is present, the specified f4m is a
1902 # stream-level manifest, and only set-level manifests may refer to
1903 # external resources. See section 11.4 and section 4 of F4M spec
1904 if bootstrap_info is None:
1905 media_url = None
1906 # @href is introduced in 2.0, see section 11.6 of F4M spec
1907 if manifest_version == '2.0':
1908 media_url = media_el.attrib.get('href')
1909 if media_url is None:
1910 media_url = media_el.attrib.get('url')
1911 if not media_url:
1912 continue
1913 manifest_url = (
1914 media_url if media_url.startswith('http://') or media_url.startswith('https://')
1915 else ((manifest_base_url or '/'.join(manifest_url.split('/')[:-1])) + '/' + media_url))
1916 # If media_url is itself a f4m manifest do the recursive extraction
1917 # since bitrates in parent manifest (this one) and media_url manifest
1918 # may differ leading to inability to resolve the format by requested
1919 # bitrate in f4m downloader
1920 ext = determine_ext(manifest_url)
1921 if ext == 'f4m':
1922 f4m_formats = self._extract_f4m_formats(
1923 manifest_url, video_id, preference=preference, quality=quality, f4m_id=f4m_id,
1924 transform_source=transform_source, fatal=fatal)
1925 # Sometimes stream-level manifest contains single media entry that
1926 # does not contain any quality metadata (e.g. http://matchtv.ru/#live-player).
1927 # At the same time parent's media entry in set-level manifest may
1928 # contain it. We will copy it from parent in such cases.
1929 if len(f4m_formats) == 1:
1930 f = f4m_formats[0]
1931 f.update({
1932 'tbr': f.get('tbr') or tbr,
1933 'width': f.get('width') or width,
1934 'height': f.get('height') or height,
1935 'format_id': f.get('format_id') if not tbr else format_id,
1936 'vcodec': vcodec,
1937 })
1938 formats.extend(f4m_formats)
1939 continue
1940 elif ext == 'm3u8':
1941 formats.extend(self._extract_m3u8_formats(
1942 manifest_url, video_id, 'mp4', preference=preference,
1943 quality=quality, m3u8_id=m3u8_id, fatal=fatal))
1944 continue
1945 formats.append({
1946 'format_id': format_id,
1947 'url': manifest_url,
1948 'manifest_url': manifest_url,
1949 'ext': 'flv' if bootstrap_info is not None else None,
1950 'protocol': 'f4m',
1951 'tbr': tbr,
1952 'width': width,
1953 'height': height,
1954 'vcodec': vcodec,
1955 'preference': preference,
1956 'quality': quality,
1957 })
1958 return formats
1959
1960 def _m3u8_meta_format(self, m3u8_url, ext=None, preference=None, quality=None, m3u8_id=None):
1961 return {
1962 'format_id': '-'.join(filter(None, [m3u8_id, 'meta'])),
1963 'url': m3u8_url,
1964 'ext': ext,
1965 'protocol': 'm3u8',
1966 'preference': preference - 100 if preference else -100,
1967 'quality': quality,
1968 'resolution': 'multiple',
1969 'format_note': 'Quality selection URL',
1970 }
1971
1972 def _extract_m3u8_formats(self, *args, **kwargs):
1973 fmts, subs = self._extract_m3u8_formats_and_subtitles(*args, **kwargs)
1974 if subs:
1975 self.report_warning(bug_reports_message(
1976 "Ignoring subtitle tracks found in the HLS manifest; "
1977 "if any subtitle tracks are missing,"
1978 ), only_once=True)
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_warning(bug_reports_message(
2277 "Ignoring subtitle tracks found in the SMIL manifest; "
2278 "if any subtitle tracks are missing,"
2279 ), only_once=True)
2280 return fmts
2281
2282 def _extract_smil_info(self, smil_url, video_id, fatal=True, f4m_params=None):
2283 smil = self._download_smil(smil_url, video_id, fatal=fatal)
2284 if smil is False:
2285 return {}
2286 return self._parse_smil(smil, smil_url, video_id, f4m_params=f4m_params)
2287
2288 def _download_smil(self, smil_url, video_id, fatal=True, transform_source=None):
2289 return self._download_xml(
2290 smil_url, video_id, 'Downloading SMIL file',
2291 'Unable to download SMIL file', fatal=fatal, transform_source=transform_source)
2292
2293 def _parse_smil(self, smil, smil_url, video_id, f4m_params=None):
2294 namespace = self._parse_smil_namespace(smil)
2295
2296 formats = self._parse_smil_formats(
2297 smil, smil_url, video_id, namespace=namespace, f4m_params=f4m_params)
2298 subtitles = self._parse_smil_subtitles(smil, namespace=namespace)
2299
2300 video_id = os.path.splitext(url_basename(smil_url))[0]
2301 title = None
2302 description = None
2303 upload_date = None
2304 for meta in smil.findall(self._xpath_ns('./head/meta', namespace)):
2305 name = meta.attrib.get('name')
2306 content = meta.attrib.get('content')
2307 if not name or not content:
2308 continue
2309 if not title and name == 'title':
2310 title = content
2311 elif not description and name in ('description', 'abstract'):
2312 description = content
2313 elif not upload_date and name == 'date':
2314 upload_date = unified_strdate(content)
2315
2316 thumbnails = [{
2317 'id': image.get('type'),
2318 'url': image.get('src'),
2319 'width': int_or_none(image.get('width')),
2320 'height': int_or_none(image.get('height')),
2321 } for image in smil.findall(self._xpath_ns('.//image', namespace)) if image.get('src')]
2322
2323 return {
2324 'id': video_id,
2325 'title': title or video_id,
2326 'description': description,
2327 'upload_date': upload_date,
2328 'thumbnails': thumbnails,
2329 'formats': formats,
2330 'subtitles': subtitles,
2331 }
2332
2333 def _parse_smil_namespace(self, smil):
2334 return self._search_regex(
2335 r'(?i)^{([^}]+)?}smil$', smil.tag, 'namespace', default=None)
2336
2337 def _parse_smil_formats(self, smil, smil_url, video_id, namespace=None, f4m_params=None, transform_rtmp_url=None):
2338 base = smil_url
2339 for meta in smil.findall(self._xpath_ns('./head/meta', namespace)):
2340 b = meta.get('base') or meta.get('httpBase')
2341 if b:
2342 base = b
2343 break
2344
2345 formats = []
2346 rtmp_count = 0
2347 http_count = 0
2348 m3u8_count = 0
2349
2350 srcs = []
2351 media = smil.findall(self._xpath_ns('.//video', namespace)) + smil.findall(self._xpath_ns('.//audio', namespace))
2352 for medium in media:
2353 src = medium.get('src')
2354 if not src or src in srcs:
2355 continue
2356 srcs.append(src)
2357
2358 bitrate = float_or_none(medium.get('system-bitrate') or medium.get('systemBitrate'), 1000)
2359 filesize = int_or_none(medium.get('size') or medium.get('fileSize'))
2360 width = int_or_none(medium.get('width'))
2361 height = int_or_none(medium.get('height'))
2362 proto = medium.get('proto')
2363 ext = medium.get('ext')
2364 src_ext = determine_ext(src)
2365 streamer = medium.get('streamer') or base
2366
2367 if proto == 'rtmp' or streamer.startswith('rtmp'):
2368 rtmp_count += 1
2369 formats.append({
2370 'url': streamer,
2371 'play_path': src,
2372 'ext': 'flv',
2373 'format_id': 'rtmp-%d' % (rtmp_count if bitrate is None else bitrate),
2374 'tbr': bitrate,
2375 'filesize': filesize,
2376 'width': width,
2377 'height': height,
2378 })
2379 if transform_rtmp_url:
2380 streamer, src = transform_rtmp_url(streamer, src)
2381 formats[-1].update({
2382 'url': streamer,
2383 'play_path': src,
2384 })
2385 continue
2386
2387 src_url = src if src.startswith('http') else compat_urlparse.urljoin(base, src)
2388 src_url = src_url.strip()
2389
2390 if proto == 'm3u8' or src_ext == 'm3u8':
2391 m3u8_formats = self._extract_m3u8_formats(
2392 src_url, video_id, ext or 'mp4', m3u8_id='hls', fatal=False)
2393 if len(m3u8_formats) == 1:
2394 m3u8_count += 1
2395 m3u8_formats[0].update({
2396 'format_id': 'hls-%d' % (m3u8_count if bitrate is None else bitrate),
2397 'tbr': bitrate,
2398 'width': width,
2399 'height': height,
2400 })
2401 formats.extend(m3u8_formats)
2402 elif src_ext == 'f4m':
2403 f4m_url = src_url
2404 if not f4m_params:
2405 f4m_params = {
2406 'hdcore': '3.2.0',
2407 'plugin': 'flowplayer-3.2.0.1',
2408 }
2409 f4m_url += '&' if '?' in f4m_url else '?'
2410 f4m_url += compat_urllib_parse_urlencode(f4m_params)
2411 formats.extend(self._extract_f4m_formats(f4m_url, video_id, f4m_id='hds', fatal=False))
2412 elif src_ext == 'mpd':
2413 formats.extend(self._extract_mpd_formats(
2414 src_url, video_id, mpd_id='dash', fatal=False))
2415 elif re.search(r'\.ism/[Mm]anifest', src_url):
2416 formats.extend(self._extract_ism_formats(
2417 src_url, video_id, ism_id='mss', fatal=False))
2418 elif src_url.startswith('http') and self._is_valid_url(src, video_id):
2419 http_count += 1
2420 formats.append({
2421 'url': src_url,
2422 'ext': ext or src_ext or 'flv',
2423 'format_id': 'http-%d' % (bitrate or http_count),
2424 'tbr': bitrate,
2425 'filesize': filesize,
2426 'width': width,
2427 'height': height,
2428 })
2429
2430 return formats
2431
2432 def _parse_smil_subtitles(self, smil, namespace=None, subtitles_lang='en'):
2433 urls = []
2434 subtitles = {}
2435 for num, textstream in enumerate(smil.findall(self._xpath_ns('.//textstream', namespace))):
2436 src = textstream.get('src')
2437 if not src or src in urls:
2438 continue
2439 urls.append(src)
2440 ext = textstream.get('ext') or mimetype2ext(textstream.get('type')) or determine_ext(src)
2441 lang = textstream.get('systemLanguage') or textstream.get('systemLanguageName') or textstream.get('lang') or subtitles_lang
2442 subtitles.setdefault(lang, []).append({
2443 'url': src,
2444 'ext': ext,
2445 })
2446 return subtitles
2447
2448 def _extract_xspf_playlist(self, xspf_url, playlist_id, fatal=True):
2449 xspf = self._download_xml(
2450 xspf_url, playlist_id, 'Downloading xpsf playlist',
2451 'Unable to download xspf manifest', fatal=fatal)
2452 if xspf is False:
2453 return []
2454 return self._parse_xspf(
2455 xspf, playlist_id, xspf_url=xspf_url,
2456 xspf_base_url=base_url(xspf_url))
2457
2458 def _parse_xspf(self, xspf_doc, playlist_id, xspf_url=None, xspf_base_url=None):
2459 NS_MAP = {
2460 'xspf': 'http://xspf.org/ns/0/',
2461 's1': 'http://static.streamone.nl/player/ns/0',
2462 }
2463
2464 entries = []
2465 for track in xspf_doc.findall(xpath_with_ns('./xspf:trackList/xspf:track', NS_MAP)):
2466 title = xpath_text(
2467 track, xpath_with_ns('./xspf:title', NS_MAP), 'title', default=playlist_id)
2468 description = xpath_text(
2469 track, xpath_with_ns('./xspf:annotation', NS_MAP), 'description')
2470 thumbnail = xpath_text(
2471 track, xpath_with_ns('./xspf:image', NS_MAP), 'thumbnail')
2472 duration = float_or_none(
2473 xpath_text(track, xpath_with_ns('./xspf:duration', NS_MAP), 'duration'), 1000)
2474
2475 formats = []
2476 for location in track.findall(xpath_with_ns('./xspf:location', NS_MAP)):
2477 format_url = urljoin(xspf_base_url, location.text)
2478 if not format_url:
2479 continue
2480 formats.append({
2481 'url': format_url,
2482 'manifest_url': xspf_url,
2483 'format_id': location.get(xpath_with_ns('s1:label', NS_MAP)),
2484 'width': int_or_none(location.get(xpath_with_ns('s1:width', NS_MAP))),
2485 'height': int_or_none(location.get(xpath_with_ns('s1:height', NS_MAP))),
2486 })
2487 self._sort_formats(formats)
2488
2489 entries.append({
2490 'id': playlist_id,
2491 'title': title,
2492 'description': description,
2493 'thumbnail': thumbnail,
2494 'duration': duration,
2495 'formats': formats,
2496 })
2497 return entries
2498
2499 def _extract_mpd_formats(self, *args, **kwargs):
2500 fmts, subs = self._extract_mpd_formats_and_subtitles(*args, **kwargs)
2501 if subs:
2502 self.report_warning(bug_reports_message(
2503 "Ignoring subtitle tracks found in the DASH manifest; "
2504 "if any subtitle tracks are missing,"
2505 ), only_once=True)
2506 return fmts
2507
2508 def _extract_mpd_formats_and_subtitles(
2509 self, mpd_url, video_id, mpd_id=None, note=None, errnote=None,
2510 fatal=True, data=None, headers={}, query={}):
2511 res = self._download_xml_handle(
2512 mpd_url, video_id,
2513 note='Downloading MPD manifest' if note is None else note,
2514 errnote='Failed to download MPD manifest' if errnote is None else errnote,
2515 fatal=fatal, data=data, headers=headers, query=query)
2516 if res is False:
2517 return [], {}
2518 mpd_doc, urlh = res
2519 if mpd_doc is None:
2520 return [], {}
2521 mpd_base_url = base_url(urlh.geturl())
2522
2523 return self._parse_mpd_formats_and_subtitles(
2524 mpd_doc, mpd_id, mpd_base_url, mpd_url)
2525
2526 def _parse_mpd_formats(self, *args, **kwargs):
2527 fmts, subs = self._parse_mpd_formats_and_subtitles(*args, **kwargs)
2528 if subs:
2529 self.report_warning(bug_reports_message(
2530 "Ignoring subtitle tracks found in the DASH manifest; "
2531 "if any subtitle tracks are missing,"
2532 ), only_once=True)
2533 return fmts
2534
2535 def _parse_mpd_formats_and_subtitles(
2536 self, mpd_doc, mpd_id=None, mpd_base_url='', mpd_url=None):
2537 """
2538 Parse formats from MPD manifest.
2539 References:
2540 1. MPEG-DASH Standard, ISO/IEC 23009-1:2014(E),
2541 http://standards.iso.org/ittf/PubliclyAvailableStandards/c065274_ISO_IEC_23009-1_2014.zip
2542 2. https://en.wikipedia.org/wiki/Dynamic_Adaptive_Streaming_over_HTTP
2543 """
2544 if not self.get_param('dynamic_mpd', True):
2545 if mpd_doc.get('type') == 'dynamic':
2546 return [], {}
2547
2548 namespace = self._search_regex(r'(?i)^{([^}]+)?}MPD$', mpd_doc.tag, 'namespace', default=None)
2549
2550 def _add_ns(path):
2551 return self._xpath_ns(path, namespace)
2552
2553 def is_drm_protected(element):
2554 return element.find(_add_ns('ContentProtection')) is not None
2555
2556 def extract_multisegment_info(element, ms_parent_info):
2557 ms_info = ms_parent_info.copy()
2558
2559 # As per [1, 5.3.9.2.2] SegmentList and SegmentTemplate share some
2560 # common attributes and elements. We will only extract relevant
2561 # for us.
2562 def extract_common(source):
2563 segment_timeline = source.find(_add_ns('SegmentTimeline'))
2564 if segment_timeline is not None:
2565 s_e = segment_timeline.findall(_add_ns('S'))
2566 if s_e:
2567 ms_info['total_number'] = 0
2568 ms_info['s'] = []
2569 for s in s_e:
2570 r = int(s.get('r', 0))
2571 ms_info['total_number'] += 1 + r
2572 ms_info['s'].append({
2573 't': int(s.get('t', 0)),
2574 # @d is mandatory (see [1, 5.3.9.6.2, Table 17, page 60])
2575 'd': int(s.attrib['d']),
2576 'r': r,
2577 })
2578 start_number = source.get('startNumber')
2579 if start_number:
2580 ms_info['start_number'] = int(start_number)
2581 timescale = source.get('timescale')
2582 if timescale:
2583 ms_info['timescale'] = int(timescale)
2584 segment_duration = source.get('duration')
2585 if segment_duration:
2586 ms_info['segment_duration'] = float(segment_duration)
2587
2588 def extract_Initialization(source):
2589 initialization = source.find(_add_ns('Initialization'))
2590 if initialization is not None:
2591 ms_info['initialization_url'] = initialization.attrib['sourceURL']
2592
2593 segment_list = element.find(_add_ns('SegmentList'))
2594 if segment_list is not None:
2595 extract_common(segment_list)
2596 extract_Initialization(segment_list)
2597 segment_urls_e = segment_list.findall(_add_ns('SegmentURL'))
2598 if segment_urls_e:
2599 ms_info['segment_urls'] = [segment.attrib['media'] for segment in segment_urls_e]
2600 else:
2601 segment_template = element.find(_add_ns('SegmentTemplate'))
2602 if segment_template is not None:
2603 extract_common(segment_template)
2604 media = segment_template.get('media')
2605 if media:
2606 ms_info['media'] = media
2607 initialization = segment_template.get('initialization')
2608 if initialization:
2609 ms_info['initialization'] = initialization
2610 else:
2611 extract_Initialization(segment_template)
2612 return ms_info
2613
2614 mpd_duration = parse_duration(mpd_doc.get('mediaPresentationDuration'))
2615 formats, subtitles = [], {}
2616 stream_numbers = {'audio': 0, 'video': 0}
2617 for period in mpd_doc.findall(_add_ns('Period')):
2618 period_duration = parse_duration(period.get('duration')) or mpd_duration
2619 period_ms_info = extract_multisegment_info(period, {
2620 'start_number': 1,
2621 'timescale': 1,
2622 })
2623 for adaptation_set in period.findall(_add_ns('AdaptationSet')):
2624 adaption_set_ms_info = extract_multisegment_info(adaptation_set, period_ms_info)
2625 for representation in adaptation_set.findall(_add_ns('Representation')):
2626 representation_attrib = adaptation_set.attrib.copy()
2627 representation_attrib.update(representation.attrib)
2628 # According to [1, 5.3.7.2, Table 9, page 41], @mimeType is mandatory
2629 mime_type = representation_attrib['mimeType']
2630 content_type = representation_attrib.get('contentType', mime_type.split('/')[0])
2631
2632 codecs = representation_attrib.get('codecs', '')
2633 if content_type not in ('video', 'audio', 'text'):
2634 if mime_type == 'image/jpeg':
2635 content_type = mime_type
2636 elif codecs.split('.')[0] == 'stpp':
2637 content_type = 'text'
2638 else:
2639 self.report_warning('Unknown MIME type %s in DASH manifest' % mime_type)
2640 continue
2641
2642 base_url = ''
2643 for element in (representation, adaptation_set, period, mpd_doc):
2644 base_url_e = element.find(_add_ns('BaseURL'))
2645 if base_url_e is not None:
2646 base_url = base_url_e.text + base_url
2647 if re.match(r'^https?://', base_url):
2648 break
2649 if mpd_base_url and base_url.startswith('/'):
2650 base_url = compat_urlparse.urljoin(mpd_base_url, base_url)
2651 elif mpd_base_url and not re.match(r'^https?://', base_url):
2652 if not mpd_base_url.endswith('/'):
2653 mpd_base_url += '/'
2654 base_url = mpd_base_url + base_url
2655 representation_id = representation_attrib.get('id')
2656 lang = representation_attrib.get('lang')
2657 url_el = representation.find(_add_ns('BaseURL'))
2658 filesize = int_or_none(url_el.attrib.get('{http://youtube.com/yt/2012/10/10}contentLength') if url_el is not None else None)
2659 bandwidth = int_or_none(representation_attrib.get('bandwidth'))
2660 if representation_id is not None:
2661 format_id = representation_id
2662 else:
2663 format_id = content_type
2664 if mpd_id:
2665 format_id = mpd_id + '-' + format_id
2666 if content_type in ('video', 'audio'):
2667 f = {
2668 'format_id': format_id,
2669 'manifest_url': mpd_url,
2670 'ext': mimetype2ext(mime_type),
2671 'width': int_or_none(representation_attrib.get('width')),
2672 'height': int_or_none(representation_attrib.get('height')),
2673 'tbr': float_or_none(bandwidth, 1000),
2674 'asr': int_or_none(representation_attrib.get('audioSamplingRate')),
2675 'fps': int_or_none(representation_attrib.get('frameRate')),
2676 'language': lang if lang not in ('mul', 'und', 'zxx', 'mis') else None,
2677 'format_note': 'DASH %s' % content_type,
2678 'filesize': filesize,
2679 'container': mimetype2ext(mime_type) + '_dash',
2680 'manifest_stream_number': stream_numbers[content_type]
2681 }
2682 f.update(parse_codecs(codecs))
2683 stream_numbers[content_type] += 1
2684 elif content_type == 'text':
2685 f = {
2686 'ext': mimetype2ext(mime_type),
2687 'manifest_url': mpd_url,
2688 'filesize': filesize,
2689 }
2690 elif content_type == 'image/jpeg':
2691 # See test case in VikiIE
2692 # https://www.viki.com/videos/1175236v-choosing-spouse-by-lottery-episode-1
2693 f = {
2694 'format_id': format_id,
2695 'ext': 'mhtml',
2696 'manifest_url': mpd_url,
2697 'format_note': 'DASH storyboards (jpeg)',
2698 'acodec': 'none',
2699 'vcodec': 'none',
2700 }
2701 if is_drm_protected(adaptation_set) or is_drm_protected(representation):
2702 f['has_drm'] = True
2703 representation_ms_info = extract_multisegment_info(representation, adaption_set_ms_info)
2704
2705 def prepare_template(template_name, identifiers):
2706 tmpl = representation_ms_info[template_name]
2707 # First of, % characters outside $...$ templates
2708 # must be escaped by doubling for proper processing
2709 # by % operator string formatting used further (see
2710 # https://github.com/ytdl-org/youtube-dl/issues/16867).
2711 t = ''
2712 in_template = False
2713 for c in tmpl:
2714 t += c
2715 if c == '$':
2716 in_template = not in_template
2717 elif c == '%' and not in_template:
2718 t += c
2719 # Next, $...$ templates are translated to their
2720 # %(...) counterparts to be used with % operator
2721 if representation_id is not None:
2722 t = t.replace('$RepresentationID$', representation_id)
2723 t = re.sub(r'\$(%s)\$' % '|'.join(identifiers), r'%(\1)d', t)
2724 t = re.sub(r'\$(%s)%%([^$]+)\$' % '|'.join(identifiers), r'%(\1)\2', t)
2725 t.replace('$$', '$')
2726 return t
2727
2728 # @initialization is a regular template like @media one
2729 # so it should be handled just the same way (see
2730 # https://github.com/ytdl-org/youtube-dl/issues/11605)
2731 if 'initialization' in representation_ms_info:
2732 initialization_template = prepare_template(
2733 'initialization',
2734 # As per [1, 5.3.9.4.2, Table 15, page 54] $Number$ and
2735 # $Time$ shall not be included for @initialization thus
2736 # only $Bandwidth$ remains
2737 ('Bandwidth', ))
2738 representation_ms_info['initialization_url'] = initialization_template % {
2739 'Bandwidth': bandwidth,
2740 }
2741
2742 def location_key(location):
2743 return 'url' if re.match(r'^https?://', location) else 'path'
2744
2745 if 'segment_urls' not in representation_ms_info and 'media' in representation_ms_info:
2746
2747 media_template = prepare_template('media', ('Number', 'Bandwidth', 'Time'))
2748 media_location_key = location_key(media_template)
2749
2750 # As per [1, 5.3.9.4.4, Table 16, page 55] $Number$ and $Time$
2751 # can't be used at the same time
2752 if '%(Number' in media_template and 's' not in representation_ms_info:
2753 segment_duration = None
2754 if 'total_number' not in representation_ms_info and 'segment_duration' in representation_ms_info:
2755 segment_duration = float_or_none(representation_ms_info['segment_duration'], representation_ms_info['timescale'])
2756 representation_ms_info['total_number'] = int(math.ceil(float(period_duration) / segment_duration))
2757 representation_ms_info['fragments'] = [{
2758 media_location_key: media_template % {
2759 'Number': segment_number,
2760 'Bandwidth': bandwidth,
2761 },
2762 'duration': segment_duration,
2763 } for segment_number in range(
2764 representation_ms_info['start_number'],
2765 representation_ms_info['total_number'] + representation_ms_info['start_number'])]
2766 else:
2767 # $Number*$ or $Time$ in media template with S list available
2768 # Example $Number*$: http://www.svtplay.se/klipp/9023742/stopptid-om-bjorn-borg
2769 # Example $Time$: https://play.arkena.com/embed/avp/v2/player/media/b41dda37-d8e7-4d3f-b1b5-9a9db578bdfe/1/129411
2770 representation_ms_info['fragments'] = []
2771 segment_time = 0
2772 segment_d = None
2773 segment_number = representation_ms_info['start_number']
2774
2775 def add_segment_url():
2776 segment_url = media_template % {
2777 'Time': segment_time,
2778 'Bandwidth': bandwidth,
2779 'Number': segment_number,
2780 }
2781 representation_ms_info['fragments'].append({
2782 media_location_key: segment_url,
2783 'duration': float_or_none(segment_d, representation_ms_info['timescale']),
2784 })
2785
2786 for num, s in enumerate(representation_ms_info['s']):
2787 segment_time = s.get('t') or segment_time
2788 segment_d = s['d']
2789 add_segment_url()
2790 segment_number += 1
2791 for r in range(s.get('r', 0)):
2792 segment_time += segment_d
2793 add_segment_url()
2794 segment_number += 1
2795 segment_time += segment_d
2796 elif 'segment_urls' in representation_ms_info and 's' in representation_ms_info:
2797 # No media template
2798 # Example: https://www.youtube.com/watch?v=iXZV5uAYMJI
2799 # or any YouTube dashsegments video
2800 fragments = []
2801 segment_index = 0
2802 timescale = representation_ms_info['timescale']
2803 for s in representation_ms_info['s']:
2804 duration = float_or_none(s['d'], timescale)
2805 for r in range(s.get('r', 0) + 1):
2806 segment_uri = representation_ms_info['segment_urls'][segment_index]
2807 fragments.append({
2808 location_key(segment_uri): segment_uri,
2809 'duration': duration,
2810 })
2811 segment_index += 1
2812 representation_ms_info['fragments'] = fragments
2813 elif 'segment_urls' in representation_ms_info:
2814 # Segment URLs with no SegmentTimeline
2815 # Example: https://www.seznam.cz/zpravy/clanek/cesko-zasahne-vitr-o-sile-vichrice-muze-byt-i-zivotu-nebezpecny-39091
2816 # https://github.com/ytdl-org/youtube-dl/pull/14844
2817 fragments = []
2818 segment_duration = float_or_none(
2819 representation_ms_info['segment_duration'],
2820 representation_ms_info['timescale']) if 'segment_duration' in representation_ms_info else None
2821 for segment_url in representation_ms_info['segment_urls']:
2822 fragment = {
2823 location_key(segment_url): segment_url,
2824 }
2825 if segment_duration:
2826 fragment['duration'] = segment_duration
2827 fragments.append(fragment)
2828 representation_ms_info['fragments'] = fragments
2829 # If there is a fragments key available then we correctly recognized fragmented media.
2830 # Otherwise we will assume unfragmented media with direct access. Technically, such
2831 # assumption is not necessarily correct since we may simply have no support for
2832 # some forms of fragmented media renditions yet, but for now we'll use this fallback.
2833 if 'fragments' in representation_ms_info:
2834 f.update({
2835 # NB: mpd_url may be empty when MPD manifest is parsed from a string
2836 'url': mpd_url or base_url,
2837 'fragment_base_url': base_url,
2838 'fragments': [],
2839 'protocol': 'http_dash_segments' if mime_type != 'image/jpeg' else 'mhtml',
2840 })
2841 if 'initialization_url' in representation_ms_info:
2842 initialization_url = representation_ms_info['initialization_url']
2843 if not f.get('url'):
2844 f['url'] = initialization_url
2845 f['fragments'].append({location_key(initialization_url): initialization_url})
2846 f['fragments'].extend(representation_ms_info['fragments'])
2847 else:
2848 # Assuming direct URL to unfragmented media.
2849 f['url'] = base_url
2850 if content_type in ('video', 'audio') or mime_type == 'image/jpeg':
2851 formats.append(f)
2852 elif content_type == 'text':
2853 subtitles.setdefault(lang or 'und', []).append(f)
2854
2855 return formats, subtitles
2856
2857 def _extract_ism_formats(self, *args, **kwargs):
2858 fmts, subs = self._extract_ism_formats_and_subtitles(*args, **kwargs)
2859 if subs:
2860 self.report_warning(bug_reports_message(
2861 "Ignoring subtitle tracks found in the ISM manifest; "
2862 "if any subtitle tracks are missing,"
2863 ))
2864 return fmts
2865
2866 def _extract_ism_formats_and_subtitles(self, ism_url, video_id, ism_id=None, note=None, errnote=None, fatal=True, data=None, headers={}, query={}):
2867 res = self._download_xml_handle(
2868 ism_url, video_id,
2869 note='Downloading ISM manifest' if note is None else note,
2870 errnote='Failed to download ISM manifest' if errnote is None else errnote,
2871 fatal=fatal, data=data, headers=headers, query=query)
2872 if res is False:
2873 return [], {}
2874 ism_doc, urlh = res
2875 if ism_doc is None:
2876 return [], {}
2877
2878 return self._parse_ism_formats_and_subtitles(ism_doc, urlh.geturl(), ism_id)
2879
2880 def _parse_ism_formats_and_subtitles(self, ism_doc, ism_url, ism_id=None):
2881 """
2882 Parse formats from ISM manifest.
2883 References:
2884 1. [MS-SSTR]: Smooth Streaming Protocol,
2885 https://msdn.microsoft.com/en-us/library/ff469518.aspx
2886 """
2887 if ism_doc.get('IsLive') == 'TRUE':
2888 return [], {}
2889
2890 duration = int(ism_doc.attrib['Duration'])
2891 timescale = int_or_none(ism_doc.get('TimeScale')) or 10000000
2892
2893 formats = []
2894 subtitles = {}
2895 for stream in ism_doc.findall('StreamIndex'):
2896 stream_type = stream.get('Type')
2897 if stream_type not in ('video', 'audio', 'text'):
2898 continue
2899 url_pattern = stream.attrib['Url']
2900 stream_timescale = int_or_none(stream.get('TimeScale')) or timescale
2901 stream_name = stream.get('Name')
2902 stream_language = stream.get('Language', 'und')
2903 for track in stream.findall('QualityLevel'):
2904 fourcc = track.get('FourCC') or ('AACL' if track.get('AudioTag') == '255' else None)
2905 # TODO: add support for WVC1 and WMAP
2906 if fourcc not in ('H264', 'AVC1', 'AACL', 'TTML'):
2907 self.report_warning('%s is not a supported codec' % fourcc)
2908 continue
2909 tbr = int(track.attrib['Bitrate']) // 1000
2910 # [1] does not mention Width and Height attributes. However,
2911 # they're often present while MaxWidth and MaxHeight are
2912 # missing, so should be used as fallbacks
2913 width = int_or_none(track.get('MaxWidth') or track.get('Width'))
2914 height = int_or_none(track.get('MaxHeight') or track.get('Height'))
2915 sampling_rate = int_or_none(track.get('SamplingRate'))
2916
2917 track_url_pattern = re.sub(r'{[Bb]itrate}', track.attrib['Bitrate'], url_pattern)
2918 track_url_pattern = compat_urlparse.urljoin(ism_url, track_url_pattern)
2919
2920 fragments = []
2921 fragment_ctx = {
2922 'time': 0,
2923 }
2924 stream_fragments = stream.findall('c')
2925 for stream_fragment_index, stream_fragment in enumerate(stream_fragments):
2926 fragment_ctx['time'] = int_or_none(stream_fragment.get('t')) or fragment_ctx['time']
2927 fragment_repeat = int_or_none(stream_fragment.get('r')) or 1
2928 fragment_ctx['duration'] = int_or_none(stream_fragment.get('d'))
2929 if not fragment_ctx['duration']:
2930 try:
2931 next_fragment_time = int(stream_fragment[stream_fragment_index + 1].attrib['t'])
2932 except IndexError:
2933 next_fragment_time = duration
2934 fragment_ctx['duration'] = (next_fragment_time - fragment_ctx['time']) / fragment_repeat
2935 for _ in range(fragment_repeat):
2936 fragments.append({
2937 'url': re.sub(r'{start[ _]time}', compat_str(fragment_ctx['time']), track_url_pattern),
2938 'duration': fragment_ctx['duration'] / stream_timescale,
2939 })
2940 fragment_ctx['time'] += fragment_ctx['duration']
2941
2942 format_id = []
2943 if ism_id:
2944 format_id.append(ism_id)
2945 if stream_name:
2946 format_id.append(stream_name)
2947 format_id.append(compat_str(tbr))
2948
2949 if stream_type == 'text':
2950 subtitles.setdefault(stream_language, []).append({
2951 'ext': 'ismt',
2952 'protocol': 'ism',
2953 'url': ism_url,
2954 'manifest_url': ism_url,
2955 'fragments': fragments,
2956 '_download_params': {
2957 'stream_type': stream_type,
2958 'duration': duration,
2959 'timescale': stream_timescale,
2960 'fourcc': fourcc,
2961 'language': stream_language,
2962 'codec_private_data': track.get('CodecPrivateData'),
2963 }
2964 })
2965 elif stream_type in ('video', 'audio'):
2966 formats.append({
2967 'format_id': '-'.join(format_id),
2968 'url': ism_url,
2969 'manifest_url': ism_url,
2970 'ext': 'ismv' if stream_type == 'video' else 'isma',
2971 'width': width,
2972 'height': height,
2973 'tbr': tbr,
2974 'asr': sampling_rate,
2975 'vcodec': 'none' if stream_type == 'audio' else fourcc,
2976 'acodec': 'none' if stream_type == 'video' else fourcc,
2977 'protocol': 'ism',
2978 'fragments': fragments,
2979 'has_drm': ism_doc.find('Protection') is not None,
2980 '_download_params': {
2981 'stream_type': stream_type,
2982 'duration': duration,
2983 'timescale': stream_timescale,
2984 'width': width or 0,
2985 'height': height or 0,
2986 'fourcc': fourcc,
2987 'language': stream_language,
2988 'codec_private_data': track.get('CodecPrivateData'),
2989 'sampling_rate': sampling_rate,
2990 'channels': int_or_none(track.get('Channels', 2)),
2991 'bits_per_sample': int_or_none(track.get('BitsPerSample', 16)),
2992 'nal_unit_length_field': int_or_none(track.get('NALUnitLengthField', 4)),
2993 },
2994 })
2995 return formats, subtitles
2996
2997 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):
2998 def absolute_url(item_url):
2999 return urljoin(base_url, item_url)
3000
3001 def parse_content_type(content_type):
3002 if not content_type:
3003 return {}
3004 ctr = re.search(r'(?P<mimetype>[^/]+/[^;]+)(?:;\s*codecs="?(?P<codecs>[^"]+))?', content_type)
3005 if ctr:
3006 mimetype, codecs = ctr.groups()
3007 f = parse_codecs(codecs)
3008 f['ext'] = mimetype2ext(mimetype)
3009 return f
3010 return {}
3011
3012 def _media_formats(src, cur_media_type, type_info={}):
3013 full_url = absolute_url(src)
3014 ext = type_info.get('ext') or determine_ext(full_url)
3015 if ext == 'm3u8':
3016 is_plain_url = False
3017 formats = self._extract_m3u8_formats(
3018 full_url, video_id, ext='mp4',
3019 entry_protocol=m3u8_entry_protocol, m3u8_id=m3u8_id,
3020 preference=preference, quality=quality, fatal=False)
3021 elif ext == 'mpd':
3022 is_plain_url = False
3023 formats = self._extract_mpd_formats(
3024 full_url, video_id, mpd_id=mpd_id, fatal=False)
3025 else:
3026 is_plain_url = True
3027 formats = [{
3028 'url': full_url,
3029 'vcodec': 'none' if cur_media_type == 'audio' else None,
3030 }]
3031 return is_plain_url, formats
3032
3033 entries = []
3034 # amp-video and amp-audio are very similar to their HTML5 counterparts
3035 # so we wll include them right here (see
3036 # https://www.ampproject.org/docs/reference/components/amp-video)
3037 # For dl8-* tags see https://delight-vr.com/documentation/dl8-video/
3038 _MEDIA_TAG_NAME_RE = r'(?:(?:amp|dl8(?:-live)?)-)?(video|audio)'
3039 media_tags = [(media_tag, media_tag_name, media_type, '')
3040 for media_tag, media_tag_name, media_type
3041 in re.findall(r'(?s)(<(%s)[^>]*/>)' % _MEDIA_TAG_NAME_RE, webpage)]
3042 media_tags.extend(re.findall(
3043 # We only allow video|audio followed by a whitespace or '>'.
3044 # Allowing more characters may end up in significant slow down (see
3045 # https://github.com/ytdl-org/youtube-dl/issues/11979, example URL:
3046 # http://www.porntrex.com/maps/videositemap.xml).
3047 r'(?s)(<(?P<tag>%s)(?:\s+[^>]*)?>)(.*?)</(?P=tag)>' % _MEDIA_TAG_NAME_RE, webpage))
3048 for media_tag, _, media_type, media_content in media_tags:
3049 media_info = {
3050 'formats': [],
3051 'subtitles': {},
3052 }
3053 media_attributes = extract_attributes(media_tag)
3054 src = strip_or_none(media_attributes.get('src'))
3055 if src:
3056 _, formats = _media_formats(src, media_type)
3057 media_info['formats'].extend(formats)
3058 media_info['thumbnail'] = absolute_url(media_attributes.get('poster'))
3059 if media_content:
3060 for source_tag in re.findall(r'<source[^>]+>', media_content):
3061 s_attr = extract_attributes(source_tag)
3062 # data-video-src and data-src are non standard but seen
3063 # several times in the wild
3064 src = strip_or_none(dict_get(s_attr, ('src', 'data-video-src', 'data-src')))
3065 if not src:
3066 continue
3067 f = parse_content_type(s_attr.get('type'))
3068 is_plain_url, formats = _media_formats(src, media_type, f)
3069 if is_plain_url:
3070 # width, height, res, label and title attributes are
3071 # all not standard but seen several times in the wild
3072 labels = [
3073 s_attr.get(lbl)
3074 for lbl in ('label', 'title')
3075 if str_or_none(s_attr.get(lbl))
3076 ]
3077 width = int_or_none(s_attr.get('width'))
3078 height = (int_or_none(s_attr.get('height'))
3079 or int_or_none(s_attr.get('res')))
3080 if not width or not height:
3081 for lbl in labels:
3082 resolution = parse_resolution(lbl)
3083 if not resolution:
3084 continue
3085 width = width or resolution.get('width')
3086 height = height or resolution.get('height')
3087 for lbl in labels:
3088 tbr = parse_bitrate(lbl)
3089 if tbr:
3090 break
3091 else:
3092 tbr = None
3093 f.update({
3094 'width': width,
3095 'height': height,
3096 'tbr': tbr,
3097 'format_id': s_attr.get('label') or s_attr.get('title'),
3098 })
3099 f.update(formats[0])
3100 media_info['formats'].append(f)
3101 else:
3102 media_info['formats'].extend(formats)
3103 for track_tag in re.findall(r'<track[^>]+>', media_content):
3104 track_attributes = extract_attributes(track_tag)
3105 kind = track_attributes.get('kind')
3106 if not kind or kind in ('subtitles', 'captions'):
3107 src = strip_or_none(track_attributes.get('src'))
3108 if not src:
3109 continue
3110 lang = track_attributes.get('srclang') or track_attributes.get('lang') or track_attributes.get('label')
3111 media_info['subtitles'].setdefault(lang, []).append({
3112 'url': absolute_url(src),
3113 })
3114 for f in media_info['formats']:
3115 f.setdefault('http_headers', {})['Referer'] = base_url
3116 if media_info['formats'] or media_info['subtitles']:
3117 entries.append(media_info)
3118 return entries
3119
3120 def _extract_akamai_formats(self, *args, **kwargs):
3121 fmts, subs = self._extract_akamai_formats_and_subtitles(*args, **kwargs)
3122 if subs:
3123 self.report_warning(bug_reports_message(
3124 "Ignoring subtitle tracks found in the manifests; "
3125 "if any subtitle tracks are missing,"
3126 ))
3127 return fmts
3128
3129 def _extract_akamai_formats_and_subtitles(self, manifest_url, video_id, hosts={}):
3130 signed = 'hdnea=' in manifest_url
3131 if not signed:
3132 # https://learn.akamai.com/en-us/webhelp/media-services-on-demand/stream-packaging-user-guide/GUID-BE6C0F73-1E06-483B-B0EA-57984B91B7F9.html
3133 manifest_url = re.sub(
3134 r'(?:b=[\d,-]+|(?:__a__|attributes)=off|__b__=\d+)&?',
3135 '', manifest_url).strip('?')
3136
3137 formats = []
3138 subtitles = {}
3139
3140 hdcore_sign = 'hdcore=3.7.0'
3141 f4m_url = re.sub(r'(https?://[^/]+)/i/', r'\1/z/', manifest_url).replace('/master.m3u8', '/manifest.f4m')
3142 hds_host = hosts.get('hds')
3143 if hds_host:
3144 f4m_url = re.sub(r'(https?://)[^/]+', r'\1' + hds_host, f4m_url)
3145 if 'hdcore=' not in f4m_url:
3146 f4m_url += ('&' if '?' in f4m_url else '?') + hdcore_sign
3147 f4m_formats = self._extract_f4m_formats(
3148 f4m_url, video_id, f4m_id='hds', fatal=False)
3149 for entry in f4m_formats:
3150 entry.update({'extra_param_to_segment_url': hdcore_sign})
3151 formats.extend(f4m_formats)
3152
3153 m3u8_url = re.sub(r'(https?://[^/]+)/z/', r'\1/i/', manifest_url).replace('/manifest.f4m', '/master.m3u8')
3154 hls_host = hosts.get('hls')
3155 if hls_host:
3156 m3u8_url = re.sub(r'(https?://)[^/]+', r'\1' + hls_host, m3u8_url)
3157 m3u8_formats, m3u8_subtitles = self._extract_m3u8_formats_and_subtitles(
3158 m3u8_url, video_id, 'mp4', 'm3u8_native',
3159 m3u8_id='hls', fatal=False)
3160 formats.extend(m3u8_formats)
3161 subtitles = self._merge_subtitles(subtitles, m3u8_subtitles)
3162
3163 http_host = hosts.get('http')
3164 if http_host and m3u8_formats and not signed:
3165 REPL_REGEX = r'https?://[^/]+/i/([^,]+),([^/]+),([^/]+)\.csmil/.+'
3166 qualities = re.match(REPL_REGEX, m3u8_url).group(2).split(',')
3167 qualities_length = len(qualities)
3168 if len(m3u8_formats) in (qualities_length, qualities_length + 1):
3169 i = 0
3170 for f in m3u8_formats:
3171 if f['vcodec'] != 'none':
3172 for protocol in ('http', 'https'):
3173 http_f = f.copy()
3174 del http_f['manifest_url']
3175 http_url = re.sub(
3176 REPL_REGEX, protocol + r'://%s/\g<1>%s\3' % (http_host, qualities[i]), f['url'])
3177 http_f.update({
3178 'format_id': http_f['format_id'].replace('hls-', protocol + '-'),
3179 'url': http_url,
3180 'protocol': protocol,
3181 })
3182 formats.append(http_f)
3183 i += 1
3184
3185 return formats, subtitles
3186
3187 def _extract_wowza_formats(self, url, video_id, m3u8_entry_protocol='m3u8_native', skip_protocols=[]):
3188 query = compat_urlparse.urlparse(url).query
3189 url = re.sub(r'/(?:manifest|playlist|jwplayer)\.(?:m3u8|f4m|mpd|smil)', '', url)
3190 mobj = re.search(
3191 r'(?:(?:http|rtmp|rtsp)(?P<s>s)?:)?(?P<url>//[^?]+)', url)
3192 url_base = mobj.group('url')
3193 http_base_url = '%s%s:%s' % ('http', mobj.group('s') or '', url_base)
3194 formats = []
3195
3196 def manifest_url(manifest):
3197 m_url = '%s/%s' % (http_base_url, manifest)
3198 if query:
3199 m_url += '?%s' % query
3200 return m_url
3201
3202 if 'm3u8' not in skip_protocols:
3203 formats.extend(self._extract_m3u8_formats(
3204 manifest_url('playlist.m3u8'), video_id, 'mp4',
3205 m3u8_entry_protocol, m3u8_id='hls', fatal=False))
3206 if 'f4m' not in skip_protocols:
3207 formats.extend(self._extract_f4m_formats(
3208 manifest_url('manifest.f4m'),
3209 video_id, f4m_id='hds', fatal=False))
3210 if 'dash' not in skip_protocols:
3211 formats.extend(self._extract_mpd_formats(
3212 manifest_url('manifest.mpd'),
3213 video_id, mpd_id='dash', fatal=False))
3214 if re.search(r'(?:/smil:|\.smil)', url_base):
3215 if 'smil' not in skip_protocols:
3216 rtmp_formats = self._extract_smil_formats(
3217 manifest_url('jwplayer.smil'),
3218 video_id, fatal=False)
3219 for rtmp_format in rtmp_formats:
3220 rtsp_format = rtmp_format.copy()
3221 rtsp_format['url'] = '%s/%s' % (rtmp_format['url'], rtmp_format['play_path'])
3222 del rtsp_format['play_path']
3223 del rtsp_format['ext']
3224 rtsp_format.update({
3225 'url': rtsp_format['url'].replace('rtmp://', 'rtsp://'),
3226 'format_id': rtmp_format['format_id'].replace('rtmp', 'rtsp'),
3227 'protocol': 'rtsp',
3228 })
3229 formats.extend([rtmp_format, rtsp_format])
3230 else:
3231 for protocol in ('rtmp', 'rtsp'):
3232 if protocol not in skip_protocols:
3233 formats.append({
3234 'url': '%s:%s' % (protocol, url_base),
3235 'format_id': protocol,
3236 'protocol': protocol,
3237 })
3238 return formats
3239
3240 def _find_jwplayer_data(self, webpage, video_id=None, transform_source=js_to_json):
3241 mobj = re.search(
3242 r'(?s)jwplayer\((?P<quote>[\'"])[^\'" ]+(?P=quote)\)(?!</script>).*?\.setup\s*\((?P<options>[^)]+)\)',
3243 webpage)
3244 if mobj:
3245 try:
3246 jwplayer_data = self._parse_json(mobj.group('options'),
3247 video_id=video_id,
3248 transform_source=transform_source)
3249 except ExtractorError:
3250 pass
3251 else:
3252 if isinstance(jwplayer_data, dict):
3253 return jwplayer_data
3254
3255 def _extract_jwplayer_data(self, webpage, video_id, *args, **kwargs):
3256 jwplayer_data = self._find_jwplayer_data(
3257 webpage, video_id, transform_source=js_to_json)
3258 return self._parse_jwplayer_data(
3259 jwplayer_data, video_id, *args, **kwargs)
3260
3261 def _parse_jwplayer_data(self, jwplayer_data, video_id=None, require_title=True,
3262 m3u8_id=None, mpd_id=None, rtmp_params=None, base_url=None):
3263 # JWPlayer backward compatibility: flattened playlists
3264 # https://github.com/jwplayer/jwplayer/blob/v7.4.3/src/js/api/config.js#L81-L96
3265 if 'playlist' not in jwplayer_data:
3266 jwplayer_data = {'playlist': [jwplayer_data]}
3267
3268 entries = []
3269
3270 # JWPlayer backward compatibility: single playlist item
3271 # https://github.com/jwplayer/jwplayer/blob/v7.7.0/src/js/playlist/playlist.js#L10
3272 if not isinstance(jwplayer_data['playlist'], list):
3273 jwplayer_data['playlist'] = [jwplayer_data['playlist']]
3274
3275 for video_data in jwplayer_data['playlist']:
3276 # JWPlayer backward compatibility: flattened sources
3277 # https://github.com/jwplayer/jwplayer/blob/v7.4.3/src/js/playlist/item.js#L29-L35
3278 if 'sources' not in video_data:
3279 video_data['sources'] = [video_data]
3280
3281 this_video_id = video_id or video_data['mediaid']
3282
3283 formats = self._parse_jwplayer_formats(
3284 video_data['sources'], video_id=this_video_id, m3u8_id=m3u8_id,
3285 mpd_id=mpd_id, rtmp_params=rtmp_params, base_url=base_url)
3286
3287 subtitles = {}
3288 tracks = video_data.get('tracks')
3289 if tracks and isinstance(tracks, list):
3290 for track in tracks:
3291 if not isinstance(track, dict):
3292 continue
3293 track_kind = track.get('kind')
3294 if not track_kind or not isinstance(track_kind, compat_str):
3295 continue
3296 if track_kind.lower() not in ('captions', 'subtitles'):
3297 continue
3298 track_url = urljoin(base_url, track.get('file'))
3299 if not track_url:
3300 continue
3301 subtitles.setdefault(track.get('label') or 'en', []).append({
3302 'url': self._proto_relative_url(track_url)
3303 })
3304
3305 entry = {
3306 'id': this_video_id,
3307 'title': unescapeHTML(video_data['title'] if require_title else video_data.get('title')),
3308 'description': clean_html(video_data.get('description')),
3309 'thumbnail': urljoin(base_url, self._proto_relative_url(video_data.get('image'))),
3310 'timestamp': int_or_none(video_data.get('pubdate')),
3311 'duration': float_or_none(jwplayer_data.get('duration') or video_data.get('duration')),
3312 'subtitles': subtitles,
3313 }
3314 # https://github.com/jwplayer/jwplayer/blob/master/src/js/utils/validator.js#L32
3315 if len(formats) == 1 and re.search(r'^(?:http|//).*(?:youtube\.com|youtu\.be)/.+', formats[0]['url']):
3316 entry.update({
3317 '_type': 'url_transparent',
3318 'url': formats[0]['url'],
3319 })
3320 else:
3321 self._sort_formats(formats)
3322 entry['formats'] = formats
3323 entries.append(entry)
3324 if len(entries) == 1:
3325 return entries[0]
3326 else:
3327 return self.playlist_result(entries)
3328
3329 def _parse_jwplayer_formats(self, jwplayer_sources_data, video_id=None,
3330 m3u8_id=None, mpd_id=None, rtmp_params=None, base_url=None):
3331 urls = []
3332 formats = []
3333 for source in jwplayer_sources_data:
3334 if not isinstance(source, dict):
3335 continue
3336 source_url = urljoin(
3337 base_url, self._proto_relative_url(source.get('file')))
3338 if not source_url or source_url in urls:
3339 continue
3340 urls.append(source_url)
3341 source_type = source.get('type') or ''
3342 ext = mimetype2ext(source_type) or determine_ext(source_url)
3343 if source_type == 'hls' or ext == 'm3u8':
3344 formats.extend(self._extract_m3u8_formats(
3345 source_url, video_id, 'mp4', entry_protocol='m3u8_native',
3346 m3u8_id=m3u8_id, fatal=False))
3347 elif source_type == 'dash' or ext == 'mpd':
3348 formats.extend(self._extract_mpd_formats(
3349 source_url, video_id, mpd_id=mpd_id, fatal=False))
3350 elif ext == 'smil':
3351 formats.extend(self._extract_smil_formats(
3352 source_url, video_id, fatal=False))
3353 # https://github.com/jwplayer/jwplayer/blob/master/src/js/providers/default.js#L67
3354 elif source_type.startswith('audio') or ext in (
3355 'oga', 'aac', 'mp3', 'mpeg', 'vorbis'):
3356 formats.append({
3357 'url': source_url,
3358 'vcodec': 'none',
3359 'ext': ext,
3360 })
3361 else:
3362 height = int_or_none(source.get('height'))
3363 if height is None:
3364 # Often no height is provided but there is a label in
3365 # format like "1080p", "720p SD", or 1080.
3366 height = int_or_none(self._search_regex(
3367 r'^(\d{3,4})[pP]?(?:\b|$)', compat_str(source.get('label') or ''),
3368 'height', default=None))
3369 a_format = {
3370 'url': source_url,
3371 'width': int_or_none(source.get('width')),
3372 'height': height,
3373 'tbr': int_or_none(source.get('bitrate')),
3374 'ext': ext,
3375 }
3376 if source_url.startswith('rtmp'):
3377 a_format['ext'] = 'flv'
3378 # See com/longtailvideo/jwplayer/media/RTMPMediaProvider.as
3379 # of jwplayer.flash.swf
3380 rtmp_url_parts = re.split(
3381 r'((?:mp4|mp3|flv):)', source_url, 1)
3382 if len(rtmp_url_parts) == 3:
3383 rtmp_url, prefix, play_path = rtmp_url_parts
3384 a_format.update({
3385 'url': rtmp_url,
3386 'play_path': prefix + play_path,
3387 })
3388 if rtmp_params:
3389 a_format.update(rtmp_params)
3390 formats.append(a_format)
3391 return formats
3392
3393 def _live_title(self, name):
3394 """ Generate the title for a live video """
3395 now = datetime.datetime.now()
3396 now_str = now.strftime('%Y-%m-%d %H:%M')
3397 return name + ' ' + now_str
3398
3399 def _int(self, v, name, fatal=False, **kwargs):
3400 res = int_or_none(v, **kwargs)
3401 if 'get_attr' in kwargs:
3402 print(getattr(v, kwargs['get_attr']))
3403 if res is None:
3404 msg = 'Failed to extract %s: Could not parse value %r' % (name, v)
3405 if fatal:
3406 raise ExtractorError(msg)
3407 else:
3408 self.report_warning(msg)
3409 return res
3410
3411 def _float(self, v, name, fatal=False, **kwargs):
3412 res = float_or_none(v, **kwargs)
3413 if res is None:
3414 msg = 'Failed to extract %s: Could not parse value %r' % (name, v)
3415 if fatal:
3416 raise ExtractorError(msg)
3417 else:
3418 self.report_warning(msg)
3419 return res
3420
3421 def _set_cookie(self, domain, name, value, expire_time=None, port=None,
3422 path='/', secure=False, discard=False, rest={}, **kwargs):
3423 cookie = compat_cookiejar_Cookie(
3424 0, name, value, port, port is not None, domain, True,
3425 domain.startswith('.'), path, True, secure, expire_time,
3426 discard, None, None, rest)
3427 self._downloader.cookiejar.set_cookie(cookie)
3428
3429 def _get_cookies(self, url):
3430 """ Return a compat_cookies_SimpleCookie with the cookies for the url """
3431 req = sanitized_Request(url)
3432 self._downloader.cookiejar.add_cookie_header(req)
3433 return compat_cookies_SimpleCookie(req.get_header('Cookie'))
3434
3435 def _apply_first_set_cookie_header(self, url_handle, cookie):
3436 """
3437 Apply first Set-Cookie header instead of the last. Experimental.
3438
3439 Some sites (e.g. [1-3]) may serve two cookies under the same name
3440 in Set-Cookie header and expect the first (old) one to be set rather
3441 than second (new). However, as of RFC6265 the newer one cookie
3442 should be set into cookie store what actually happens.
3443 We will workaround this issue by resetting the cookie to
3444 the first one manually.
3445 1. https://new.vk.com/
3446 2. https://github.com/ytdl-org/youtube-dl/issues/9841#issuecomment-227871201
3447 3. https://learning.oreilly.com/
3448 """
3449 for header, cookies in url_handle.headers.items():
3450 if header.lower() != 'set-cookie':
3451 continue
3452 if sys.version_info[0] >= 3:
3453 cookies = cookies.encode('iso-8859-1')
3454 cookies = cookies.decode('utf-8')
3455 cookie_value = re.search(
3456 r'%s=(.+?);.*?\b[Dd]omain=(.+?)(?:[,;]|$)' % cookie, cookies)
3457 if cookie_value:
3458 value, domain = cookie_value.groups()
3459 self._set_cookie(domain, cookie, value)
3460 break
3461
3462 def get_testcases(self, include_onlymatching=False):
3463 t = getattr(self, '_TEST', None)
3464 if t:
3465 assert not hasattr(self, '_TESTS'), \
3466 '%s has _TEST and _TESTS' % type(self).__name__
3467 tests = [t]
3468 else:
3469 tests = getattr(self, '_TESTS', [])
3470 for t in tests:
3471 if not include_onlymatching and t.get('only_matching', False):
3472 continue
3473 t['name'] = type(self).__name__[:-len('IE')]
3474 yield t
3475
3476 def is_suitable(self, age_limit):
3477 """ Test whether the extractor is generally suitable for the given
3478 age limit (i.e. pornographic sites are not, all others usually are) """
3479
3480 any_restricted = False
3481 for tc in self.get_testcases(include_onlymatching=False):
3482 if tc.get('playlist', []):
3483 tc = tc['playlist'][0]
3484 is_restricted = age_restricted(
3485 tc.get('info_dict', {}).get('age_limit'), age_limit)
3486 if not is_restricted:
3487 return True
3488 any_restricted = any_restricted or is_restricted
3489 return not any_restricted
3490
3491 def extract_subtitles(self, *args, **kwargs):
3492 if (self.get_param('writesubtitles', False)
3493 or self.get_param('listsubtitles')):
3494 return self._get_subtitles(*args, **kwargs)
3495 return {}
3496
3497 def _get_subtitles(self, *args, **kwargs):
3498 raise NotImplementedError('This method must be implemented by subclasses')
3499
3500 @staticmethod
3501 def _merge_subtitle_items(subtitle_list1, subtitle_list2):
3502 """ Merge subtitle items for one language. Items with duplicated URLs
3503 will be dropped. """
3504 list1_urls = set([item['url'] for item in subtitle_list1])
3505 ret = list(subtitle_list1)
3506 ret.extend([item for item in subtitle_list2 if item['url'] not in list1_urls])
3507 return ret
3508
3509 @classmethod
3510 def _merge_subtitles(cls, *dicts, target=None):
3511 """ Merge subtitle dictionaries, language by language. """
3512 if target is None:
3513 target = {}
3514 for d in dicts:
3515 for lang, subs in d.items():
3516 target[lang] = cls._merge_subtitle_items(target.get(lang, []), subs)
3517 return target
3518
3519 def extract_automatic_captions(self, *args, **kwargs):
3520 if (self.get_param('writeautomaticsub', False)
3521 or self.get_param('listsubtitles')):
3522 return self._get_automatic_captions(*args, **kwargs)
3523 return {}
3524
3525 def _get_automatic_captions(self, *args, **kwargs):
3526 raise NotImplementedError('This method must be implemented by subclasses')
3527
3528 def mark_watched(self, *args, **kwargs):
3529 if not self.get_param('mark_watched', False):
3530 return
3531 if (self._get_login_info()[0] is not None
3532 or self.get_param('cookiefile')
3533 or self.get_param('cookiesfrombrowser')):
3534 self._mark_watched(*args, **kwargs)
3535
3536 def _mark_watched(self, *args, **kwargs):
3537 raise NotImplementedError('This method must be implemented by subclasses')
3538
3539 def geo_verification_headers(self):
3540 headers = {}
3541 geo_verification_proxy = self.get_param('geo_verification_proxy')
3542 if geo_verification_proxy:
3543 headers['Ytdl-request-proxy'] = geo_verification_proxy
3544 return headers
3545
3546 def _generic_id(self, url):
3547 return compat_urllib_parse_unquote(os.path.splitext(url.rstrip('/').split('/')[-1])[0])
3548
3549 def _generic_title(self, url):
3550 return compat_urllib_parse_unquote(os.path.splitext(url_basename(url))[0])
3551
3552 @staticmethod
3553 def _availability(is_private=None, needs_premium=None, needs_subscription=None, needs_auth=None, is_unlisted=None):
3554 all_known = all(map(
3555 lambda x: x is not None,
3556 (is_private, needs_premium, needs_subscription, needs_auth, is_unlisted)))
3557 return (
3558 'private' if is_private
3559 else 'premium_only' if needs_premium
3560 else 'subscriber_only' if needs_subscription
3561 else 'needs_auth' if needs_auth
3562 else 'unlisted' if is_unlisted
3563 else 'public' if all_known
3564 else None)
3565
3566 def _configuration_arg(self, key, default=NO_DEFAULT, casesense=False):
3567 '''
3568 @returns A list of values for the extractor argument given by "key"
3569 or "default" if no such key is present
3570 @param default The default value to return when the key is not present (default: [])
3571 @param casesense When false, the values are converted to lower case
3572 '''
3573 val = traverse_obj(
3574 self._downloader.params, ('extractor_args', self.ie_key().lower(), key))
3575 if val is None:
3576 return [] if default is NO_DEFAULT else default
3577 return list(val) if casesense else [x.lower() for x in val]
3578
3579
3580 class SearchInfoExtractor(InfoExtractor):
3581 """
3582 Base class for paged search queries extractors.
3583 They accept URLs in the format _SEARCH_KEY(|all|[0-9]):{query}
3584 Instances should define _SEARCH_KEY and _MAX_RESULTS.
3585 """
3586
3587 @classmethod
3588 def _make_valid_url(cls):
3589 return r'%s(?P<prefix>|[1-9][0-9]*|all):(?P<query>[\s\S]+)' % cls._SEARCH_KEY
3590
3591 @classmethod
3592 def suitable(cls, url):
3593 return re.match(cls._make_valid_url(), url) is not None
3594
3595 def _real_extract(self, query):
3596 mobj = re.match(self._make_valid_url(), query)
3597 if mobj is None:
3598 raise ExtractorError('Invalid search query "%s"' % query)
3599
3600 prefix = mobj.group('prefix')
3601 query = mobj.group('query')
3602 if prefix == '':
3603 return self._get_n_results(query, 1)
3604 elif prefix == 'all':
3605 return self._get_n_results(query, self._MAX_RESULTS)
3606 else:
3607 n = int(prefix)
3608 if n <= 0:
3609 raise ExtractorError('invalid download number %s for query "%s"' % (n, query))
3610 elif n > self._MAX_RESULTS:
3611 self.report_warning('%s returns max %i results (you requested %i)' % (self._SEARCH_KEY, self._MAX_RESULTS, n))
3612 n = self._MAX_RESULTS
3613 return self._get_n_results(query, n)
3614
3615 def _get_n_results(self, query, n):
3616 """Get a specified number of results for a query"""
3617 raise NotImplementedError('This method must be implemented by subclasses')
3618
3619 @property
3620 def SEARCH_KEY(self):
3621 return self._SEARCH_KEY