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