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