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