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