]> jfr.im git - yt-dlp.git/blob - youtube_dlc/extractor/common.py
Fix some fields not sorting correctly
[yt-dlp.git] / youtube_dlc / extractor / common.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import base64
5 import datetime
6 import hashlib
7 import json
8 import netrc
9 import os
10 import random
11 import re
12 import socket
13 import ssl
14 import sys
15 import time
16 import math
17
18 from ..compat import (
19 compat_cookiejar_Cookie,
20 compat_cookies,
21 compat_etree_Element,
22 compat_etree_fromstring,
23 compat_getpass,
24 compat_integer_types,
25 compat_http_client,
26 compat_os_name,
27 compat_str,
28 compat_urllib_error,
29 compat_urllib_parse_unquote,
30 compat_urllib_parse_urlencode,
31 compat_urllib_request,
32 compat_urlparse,
33 compat_xml_parse_error,
34 )
35 from ..downloader import FileDownloader
36 from ..downloader.f4m import (
37 get_base_url,
38 remove_encrypted_media,
39 )
40 from ..utils import (
41 NO_DEFAULT,
42 age_restricted,
43 base_url,
44 bug_reports_message,
45 clean_html,
46 compiled_regex_type,
47 determine_ext,
48 determine_protocol,
49 dict_get,
50 error_to_compat_str,
51 ExtractorError,
52 extract_attributes,
53 fix_xml_ampersands,
54 float_or_none,
55 GeoRestrictedError,
56 GeoUtils,
57 int_or_none,
58 js_to_json,
59 JSON_LD_RE,
60 mimetype2ext,
61 orderedSet,
62 parse_bitrate,
63 parse_codecs,
64 parse_duration,
65 parse_iso8601,
66 parse_m3u8_attributes,
67 parse_resolution,
68 RegexNotFoundError,
69 sanitized_Request,
70 sanitize_filename,
71 str_or_none,
72 str_to_int,
73 strip_or_none,
74 unescapeHTML,
75 unified_strdate,
76 unified_timestamp,
77 update_Request,
78 update_url_query,
79 urljoin,
80 url_basename,
81 url_or_none,
82 xpath_element,
83 xpath_text,
84 xpath_with_ns,
85 )
86
87
88 class 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
95 passed to the YoutubeDL. The YoutubeDL processes this
96 information possibly downloading the video to the file system, among
97 other possible outcomes.
98
99 The type field determines the type of the result.
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:
104
105 id: Video identifier.
106 title: Video title, unescaped.
107
108 Additionally, it must contain either a formats entry or a url one:
109
110 formats: A list of dictionaries for each format available, ordered
111 from worst to best quality.
112
113 Potential fields:
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,
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
124 is parsed from a string (in case of
125 fragmented media)
126 for MSS - URL of the ISM manifest.
127 * manifest_url
128 The URL of the manifest file in case of
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.
134 * ext Will be calculated from URL if missing
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
140 ("mp4_h264_opus" or "19").
141 Technically optional, but strongly recommended.
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
146 * resolution Textual description of width and height
147 * tbr Average bitrate of audio and video in KBit/s
148 * abr Average audio bitrate in KBit/s
149 * acodec Name of the audio codec in use
150 * asr Audio sampling rate in Hertz
151 * vbr Average video bitrate in KBit/s
152 * fps Frame rate
153 * vcodec Name of the video codec in use
154 * container Name of the container format
155 * filesize The number of bytes, if known in advance
156 * filesize_approx An estimate for the number of bytes
157 * player_url SWF Player URL (used for rtmpdump).
158 * protocol The protocol that will be used for the actual
159 download, lower-case.
160 "http", "https", "rtsp", "rtmp", "rtmpe",
161 "m3u8", "m3u8_native" or "http_dash_segments".
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
175 * "duration" (optional, int or float)
176 * "filesize" (optional, int)
177 * preference Order number of this format. If this field is
178 present and not None, the formats get sorted
179 by this field, regardless of all other values.
180 -1 for default (order by other properties),
181 -2 or smaller for less than default.
182 < -1000 to hide the format (if there is
183 another one which is strictly better)
184 * language Language code, e.g. "de" or "en-US".
185 * language_preference Is this in the language mentioned in
186 the URL?
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.
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.
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.
198 * http_headers A dictionary of additional HTTP headers
199 to add to the request.
200 * stretched_ratio If given and not 1, indicates that the
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.
205 * downloader_options A dictionary of downloader options as
206 described in FileDownloader
207
208 url: Final video URL.
209 ext: Video filename extension.
210 format: The video format, defaults to ext (used for --get-format)
211 player_url: SWF Player URL (used for rtmpdump).
212
213 The following fields are optional:
214
215 alt_title: A secondary title of the video.
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"
220 thumbnails: A list of dictionaries, with the following entries:
221 * "id" (optional, string) - Thumbnail format ID
222 * "url"
223 * "preference" (optional, int) - quality of the image
224 * "width" (optional, int)
225 * "height" (optional, int)
226 * "resolution" (optional, string "{width}x{height}",
227 deprecated)
228 * "filesize" (optional, int)
229 thumbnail: Full URL to a video thumbnail image.
230 description: Full video description.
231 uploader: Full name of the video uploader.
232 license: License name the video is licensed under.
233 creator: The creator of the video.
234 release_date: The date (YYYYMMDD) when the video was released.
235 timestamp: UNIX timestamp of the moment the video became available.
236 upload_date: Video upload date (YYYYMMDD).
237 If not explicitly set, calculated from timestamp.
238 uploader_id: Nickname or id of the video uploader.
239 uploader_url: Full URL to a personal webpage of the video uploader.
240 channel: Full name of the channel the video is uploaded on.
241 Note that channel fields may or may not repeat uploader
242 fields. This depends on a particular extractor.
243 channel_id: Id of the channel.
244 channel_url: Full URL to a channel webpage.
245 location: Physical location where the video was filmed.
246 subtitles: The available subtitles as a dictionary in the format
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:
251 * "data": The subtitles file contents
252 * "url": A URL pointing to the subtitles file
253 "ext" will be calculated from URL if missing
254 automatic_captions: Like 'subtitles', used by the YoutubeIE for
255 automatically generated captions
256 duration: Length of the video in seconds, as an integer or float.
257 view_count: How many users have watched the video on the platform.
258 like_count: Number of positive ratings of the video
259 dislike_count: Number of negative ratings of the video
260 repost_count: Number of reposts of the video
261 average_rating: Average rating give by users, the scale used depends on the webpage
262 comment_count: Number of comments on the video
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.
274 age_limit: Age restriction for the video, as an integer (years)
275 webpage_url: The URL to the video webpage, if given to youtube-dlc it
276 should allow to get the same result again. (It will be set
277 by YoutubeDL if it's missing)
278 categories: A list of categories that the video falls in, for example
279 ["Sports", "Berlin"]
280 tags: A list of tags assigned to the video, e.g. ["sweden", "pop music"]
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.
283 start_time: Time in seconds where the reproduction should start, as
284 specified in the URL.
285 end_time: Time in seconds where the reproduction should end, as
286 specified in the URL.
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)
291
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.
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.
298
299 The following fields should only be used when the video is an episode of some
300 series, programme or podcast:
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.
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.
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.
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.
311
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
330 Unless mentioned otherwise, the fields should be Unicode strings.
331
332 Unless mentioned otherwise, None is equivalent to absence of information.
333
334
335 _type "playlist" indicates multiple videos.
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.
338
339 Additionally, playlists can have "id", "title", "description", "uploader",
340 "uploader_id", "uploader_url", "duration" attributes with the same semantics
341 as videos (see above).
342
343
344 _type "multi_video" indicates that there are multiple videos that
345 form a single show, for examples multiple acts of an opera or TV episode.
346 It must have an entries key like a playlist and contain all the keys
347 required for a video at the same time.
348
349
350 _type "url" indicates that the video must be extracted from another
351 location, possibly by a different extractor. Its only required key is:
352 "url" - the next URL to extract.
353 The key "ie_key" can be set to the class name (minus the trailing "IE",
354 e.g. "Youtube") if the extractor class is known in advance.
355 Additionally, the dictionary may have any properties of the resolved entity
356 known in advance, for example "title" if the title of the referred video is
357 known ahead of time.
358
359
360 _type "url_transparent" entities have the same specification as "url", but
361 indicate that the given additional information is more precise than the one
362 associated with the resolved URL.
363 This is useful when a site employs a video service that hosts the video and
364 its technical metadata, but that video service does not embed a useful
365 title, description etc.
366
367
368 Subclasses of this one should re-define the _real_initialize() and
369 _real_extract() methods and define a _VALID_URL regexp.
370 Probably, they should also be added to the list of extractors.
371
372 _GEO_BYPASS attribute may be set to False in order to disable
373 geo restriction bypass mechanisms for a particular extractor.
374 Though it won't disable explicit geo restriction bypass based on
375 country code provided with geo_bypass_country.
376
377 _GEO_COUNTRIES attribute may contain a list of presumably geo unrestricted
378 countries for this extractor. One of these countries will be used by
379 geo restriction bypass mechanism right away in order to bypass
380 geo restriction, of course, if the mechanism is not disabled.
381
382 _GEO_IP_BLOCKS attribute may contain a list of presumably geo unrestricted
383 IP blocks in CIDR notation for this extractor. One of these IP blocks
384 will be used by geo restriction bypass mechanism similarly
385 to _GEO_COUNTRIES.
386
387 Finally, the _WORKING attribute should be set to False for broken IEs
388 in order to warn the users and skip the tests.
389 """
390
391 _ready = False
392 _downloader = None
393 _x_forwarded_for_ip = None
394 _GEO_BYPASS = True
395 _GEO_COUNTRIES = None
396 _GEO_IP_BLOCKS = None
397 _WORKING = True
398
399 def __init__(self, downloader=None):
400 """Constructor. Receives an optional downloader."""
401 self._ready = False
402 self._x_forwarded_for_ip = None
403 self.set_downloader(downloader)
404
405 @classmethod
406 def suitable(cls, url):
407 """Receives a URL and returns True if suitable for this IE."""
408
409 # This does not use has/getattr intentionally - we want to know whether
410 # we have cached the regexp for *this* class, whereas getattr would also
411 # match the superclass
412 if '_VALID_URL_RE' not in cls.__dict__:
413 cls._VALID_URL_RE = re.compile(cls._VALID_URL)
414 return cls._VALID_URL_RE.match(url) is not None
415
416 @classmethod
417 def _match_id(cls, url):
418 if '_VALID_URL_RE' not in cls.__dict__:
419 cls._VALID_URL_RE = re.compile(cls._VALID_URL)
420 m = cls._VALID_URL_RE.match(url)
421 assert m
422 return compat_str(m.group('id'))
423
424 @classmethod
425 def working(cls):
426 """Getter method for _WORKING."""
427 return cls._WORKING
428
429 def initialize(self):
430 """Initializes an instance (authentication, etc)."""
431 self._initialize_geo_bypass({
432 'countries': self._GEO_COUNTRIES,
433 'ip_blocks': self._GEO_IP_BLOCKS,
434 })
435 if not self._ready:
436 self._real_initialize()
437 self._ready = True
438
439 def _initialize_geo_bypass(self, geo_bypass_context):
440 """
441 Initialize geo restriction bypass mechanism.
442
443 This method is used to initialize geo bypass mechanism based on faking
444 X-Forwarded-For HTTP header. A random country from provided country list
445 is selected and a random IP belonging to this country is generated. This
446 IP will be passed as X-Forwarded-For HTTP header in all subsequent
447 HTTP requests.
448
449 This method will be used for initial geo bypass mechanism initialization
450 during the instance initialization with _GEO_COUNTRIES and
451 _GEO_IP_BLOCKS.
452
453 You may also manually call it from extractor's code if geo bypass
454 information is not available beforehand (e.g. obtained during
455 extraction) or due to some other reason. In this case you should pass
456 this information in geo bypass context passed as first argument. It may
457 contain following fields:
458
459 countries: List of geo unrestricted countries (similar
460 to _GEO_COUNTRIES)
461 ip_blocks: List of geo unrestricted IP blocks in CIDR notation
462 (similar to _GEO_IP_BLOCKS)
463
464 """
465 if not self._x_forwarded_for_ip:
466
467 # Geo bypass mechanism is explicitly disabled by user
468 if not self._downloader.params.get('geo_bypass', True):
469 return
470
471 if not geo_bypass_context:
472 geo_bypass_context = {}
473
474 # Backward compatibility: previously _initialize_geo_bypass
475 # expected a list of countries, some 3rd party code may still use
476 # it this way
477 if isinstance(geo_bypass_context, (list, tuple)):
478 geo_bypass_context = {
479 'countries': geo_bypass_context,
480 }
481
482 # The whole point of geo bypass mechanism is to fake IP
483 # as X-Forwarded-For HTTP header based on some IP block or
484 # country code.
485
486 # Path 1: bypassing based on IP block in CIDR notation
487
488 # Explicit IP block specified by user, use it right away
489 # regardless of whether extractor is geo bypassable or not
490 ip_block = self._downloader.params.get('geo_bypass_ip_block', None)
491
492 # Otherwise use random IP block from geo bypass context but only
493 # if extractor is known as geo bypassable
494 if not ip_block:
495 ip_blocks = geo_bypass_context.get('ip_blocks')
496 if self._GEO_BYPASS and ip_blocks:
497 ip_block = random.choice(ip_blocks)
498
499 if ip_block:
500 self._x_forwarded_for_ip = GeoUtils.random_ipv4(ip_block)
501 if self._downloader.params.get('verbose', False):
502 self._downloader.to_screen(
503 '[debug] Using fake IP %s as X-Forwarded-For.'
504 % self._x_forwarded_for_ip)
505 return
506
507 # Path 2: bypassing based on country code
508
509 # Explicit country code specified by user, use it right away
510 # regardless of whether extractor is geo bypassable or not
511 country = self._downloader.params.get('geo_bypass_country', None)
512
513 # Otherwise use random country code from geo bypass context but
514 # only if extractor is known as geo bypassable
515 if not country:
516 countries = geo_bypass_context.get('countries')
517 if self._GEO_BYPASS and countries:
518 country = random.choice(countries)
519
520 if country:
521 self._x_forwarded_for_ip = GeoUtils.random_ipv4(country)
522 if self._downloader.params.get('verbose', False):
523 self._downloader.to_screen(
524 '[debug] Using fake IP %s (%s) as X-Forwarded-For.'
525 % (self._x_forwarded_for_ip, country.upper()))
526
527 def extract(self, url):
528 """Extracts URL information and returns it in list of dicts."""
529 try:
530 for _ in range(2):
531 try:
532 self.initialize()
533 ie_result = self._real_extract(url)
534 if self._x_forwarded_for_ip:
535 ie_result['__x_forwarded_for_ip'] = self._x_forwarded_for_ip
536 return ie_result
537 except GeoRestrictedError as e:
538 if self.__maybe_fake_ip_and_retry(e.countries):
539 continue
540 raise
541 except ExtractorError:
542 raise
543 except compat_http_client.IncompleteRead as e:
544 raise ExtractorError('A network error has occurred.', cause=e, expected=True)
545 except (KeyError, StopIteration) as e:
546 raise ExtractorError('An extractor error has occurred.', cause=e)
547
548 def __maybe_fake_ip_and_retry(self, countries):
549 if (not self._downloader.params.get('geo_bypass_country', None)
550 and self._GEO_BYPASS
551 and self._downloader.params.get('geo_bypass', True)
552 and not self._x_forwarded_for_ip
553 and countries):
554 country_code = random.choice(countries)
555 self._x_forwarded_for_ip = GeoUtils.random_ipv4(country_code)
556 if self._x_forwarded_for_ip:
557 self.report_warning(
558 'Video is geo restricted. Retrying extraction with fake IP %s (%s) as X-Forwarded-For.'
559 % (self._x_forwarded_for_ip, country_code.upper()))
560 return True
561 return False
562
563 def set_downloader(self, downloader):
564 """Sets the downloader for this IE."""
565 self._downloader = downloader
566
567 def _real_initialize(self):
568 """Real initialization process. Redefine in subclasses."""
569 pass
570
571 def _real_extract(self, url):
572 """Real extraction process. Redefine in subclasses."""
573 pass
574
575 @classmethod
576 def ie_key(cls):
577 """A string for getting the InfoExtractor with get_info_extractor"""
578 return compat_str(cls.__name__[:-2])
579
580 @property
581 def IE_NAME(self):
582 return compat_str(type(self).__name__[:-2])
583
584 @staticmethod
585 def __can_accept_status_code(err, expected_status):
586 assert isinstance(err, compat_urllib_error.HTTPError)
587 if expected_status is None:
588 return False
589 if isinstance(expected_status, compat_integer_types):
590 return err.code == expected_status
591 elif isinstance(expected_status, (list, tuple)):
592 return err.code in expected_status
593 elif callable(expected_status):
594 return expected_status(err.code) is True
595 else:
596 assert False
597
598 def _request_webpage(self, url_or_request, video_id, note=None, errnote=None, fatal=True, data=None, headers={}, query={}, expected_status=None):
599 """
600 Return the response handle.
601
602 See _download_webpage docstring for arguments specification.
603 """
604 if note is None:
605 self.report_download_webpage(video_id)
606 elif note is not False:
607 if video_id is None:
608 self.to_screen('%s' % (note,))
609 else:
610 self.to_screen('%s: %s' % (video_id, note))
611
612 # Some sites check X-Forwarded-For HTTP header in order to figure out
613 # the origin of the client behind proxy. This allows bypassing geo
614 # restriction by faking this header's value to IP that belongs to some
615 # geo unrestricted country. We will do so once we encounter any
616 # geo restriction error.
617 if self._x_forwarded_for_ip:
618 if 'X-Forwarded-For' not in headers:
619 headers['X-Forwarded-For'] = self._x_forwarded_for_ip
620
621 if isinstance(url_or_request, compat_urllib_request.Request):
622 url_or_request = update_Request(
623 url_or_request, data=data, headers=headers, query=query)
624 else:
625 if query:
626 url_or_request = update_url_query(url_or_request, query)
627 if data is not None or headers:
628 url_or_request = sanitized_Request(url_or_request, data, headers)
629 exceptions = [compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error]
630 if hasattr(ssl, 'CertificateError'):
631 exceptions.append(ssl.CertificateError)
632 try:
633 return self._downloader.urlopen(url_or_request)
634 except tuple(exceptions) as err:
635 if isinstance(err, compat_urllib_error.HTTPError):
636 if self.__can_accept_status_code(err, expected_status):
637 # Retain reference to error to prevent file object from
638 # being closed before it can be read. Works around the
639 # effects of <https://bugs.python.org/issue15002>
640 # introduced in Python 3.4.1.
641 err.fp._error = err
642 return err.fp
643
644 if errnote is False:
645 return False
646 if errnote is None:
647 errnote = 'Unable to download webpage'
648
649 errmsg = '%s: %s' % (errnote, error_to_compat_str(err))
650 if fatal:
651 raise ExtractorError(errmsg, sys.exc_info()[2], cause=err)
652 else:
653 self._downloader.report_warning(errmsg)
654 return False
655
656 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):
657 """
658 Return a tuple (page content as string, URL handle).
659
660 See _download_webpage docstring for arguments specification.
661 """
662 # Strip hashes from the URL (#1038)
663 if isinstance(url_or_request, (compat_str, str)):
664 url_or_request = url_or_request.partition('#')[0]
665
666 urlh = self._request_webpage(url_or_request, video_id, note, errnote, fatal, data=data, headers=headers, query=query, expected_status=expected_status)
667 if urlh is False:
668 assert not fatal
669 return False
670 content = self._webpage_read_content(urlh, url_or_request, video_id, note, errnote, fatal, encoding=encoding)
671 return (content, urlh)
672
673 @staticmethod
674 def _guess_encoding_from_content(content_type, webpage_bytes):
675 m = re.match(r'[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+\s*;\s*charset=(.+)', content_type)
676 if m:
677 encoding = m.group(1)
678 else:
679 m = re.search(br'<meta[^>]+charset=[\'"]?([^\'")]+)[ /\'">]',
680 webpage_bytes[:1024])
681 if m:
682 encoding = m.group(1).decode('ascii')
683 elif webpage_bytes.startswith(b'\xff\xfe'):
684 encoding = 'utf-16'
685 else:
686 encoding = 'utf-8'
687
688 return encoding
689
690 def __check_blocked(self, content):
691 first_block = content[:512]
692 if ('<title>Access to this site is blocked</title>' in content
693 and 'Websense' in first_block):
694 msg = 'Access to this webpage has been blocked by Websense filtering software in your network.'
695 blocked_iframe = self._html_search_regex(
696 r'<iframe src="([^"]+)"', content,
697 'Websense information URL', default=None)
698 if blocked_iframe:
699 msg += ' Visit %s for more details' % blocked_iframe
700 raise ExtractorError(msg, expected=True)
701 if '<title>The URL you requested has been blocked</title>' in first_block:
702 msg = (
703 'Access to this webpage has been blocked by Indian censorship. '
704 'Use a VPN or proxy server (with --proxy) to route around it.')
705 block_msg = self._html_search_regex(
706 r'</h1><p>(.*?)</p>',
707 content, 'block message', default=None)
708 if block_msg:
709 msg += ' (Message: "%s")' % block_msg.replace('\n', ' ')
710 raise ExtractorError(msg, expected=True)
711 if ('<title>TTK :: Доступ к ресурсу ограничен</title>' in content
712 and 'blocklist.rkn.gov.ru' in content):
713 raise ExtractorError(
714 'Access to this webpage has been blocked by decision of the Russian government. '
715 'Visit http://blocklist.rkn.gov.ru/ for a block reason.',
716 expected=True)
717
718 def _webpage_read_content(self, urlh, url_or_request, video_id, note=None, errnote=None, fatal=True, prefix=None, encoding=None):
719 content_type = urlh.headers.get('Content-Type', '')
720 webpage_bytes = urlh.read()
721 if prefix is not None:
722 webpage_bytes = prefix + webpage_bytes
723 if not encoding:
724 encoding = self._guess_encoding_from_content(content_type, webpage_bytes)
725 if self._downloader.params.get('dump_intermediate_pages', False):
726 self.to_screen('Dumping request to ' + urlh.geturl())
727 dump = base64.b64encode(webpage_bytes).decode('ascii')
728 self._downloader.to_screen(dump)
729 if self._downloader.params.get('write_pages', False):
730 basen = '%s_%s' % (video_id, urlh.geturl())
731 if len(basen) > 240:
732 h = '___' + hashlib.md5(basen.encode('utf-8')).hexdigest()
733 basen = basen[:240 - len(h)] + h
734 raw_filename = basen + '.dump'
735 filename = sanitize_filename(raw_filename, restricted=True)
736 self.to_screen('Saving request to ' + filename)
737 # Working around MAX_PATH limitation on Windows (see
738 # http://msdn.microsoft.com/en-us/library/windows/desktop/aa365247(v=vs.85).aspx)
739 if compat_os_name == 'nt':
740 absfilepath = os.path.abspath(filename)
741 if len(absfilepath) > 259:
742 filename = '\\\\?\\' + absfilepath
743 with open(filename, 'wb') as outf:
744 outf.write(webpage_bytes)
745
746 try:
747 content = webpage_bytes.decode(encoding, 'replace')
748 except LookupError:
749 content = webpage_bytes.decode('utf-8', 'replace')
750
751 self.__check_blocked(content)
752
753 return content
754
755 def _download_webpage(
756 self, url_or_request, video_id, note=None, errnote=None,
757 fatal=True, tries=1, timeout=5, encoding=None, data=None,
758 headers={}, query={}, expected_status=None):
759 """
760 Return the data of the page as a string.
761
762 Arguments:
763 url_or_request -- plain text URL as a string or
764 a compat_urllib_request.Requestobject
765 video_id -- Video/playlist/item identifier (string)
766
767 Keyword arguments:
768 note -- note printed before downloading (string)
769 errnote -- note printed in case of an error (string)
770 fatal -- flag denoting whether error should be considered fatal,
771 i.e. whether it should cause ExtractionError to be raised,
772 otherwise a warning will be reported and extraction continued
773 tries -- number of tries
774 timeout -- sleep interval between tries
775 encoding -- encoding for a page content decoding, guessed automatically
776 when not explicitly specified
777 data -- POST data (bytes)
778 headers -- HTTP headers (dict)
779 query -- URL query (dict)
780 expected_status -- allows to accept failed HTTP requests (non 2xx
781 status code) by explicitly specifying a set of accepted status
782 codes. Can be any of the following entities:
783 - an integer type specifying an exact failed status code to
784 accept
785 - a list or a tuple of integer types specifying a list of
786 failed status codes to accept
787 - a callable accepting an actual failed status code and
788 returning True if it should be accepted
789 Note that this argument does not affect success status codes (2xx)
790 which are always accepted.
791 """
792
793 success = False
794 try_count = 0
795 while success is False:
796 try:
797 res = self._download_webpage_handle(
798 url_or_request, video_id, note, errnote, fatal,
799 encoding=encoding, data=data, headers=headers, query=query,
800 expected_status=expected_status)
801 success = True
802 except compat_http_client.IncompleteRead as e:
803 try_count += 1
804 if try_count >= tries:
805 raise e
806 self._sleep(timeout, video_id)
807 if res is False:
808 return res
809 else:
810 content, _ = res
811 return content
812
813 def _download_xml_handle(
814 self, url_or_request, video_id, note='Downloading XML',
815 errnote='Unable to download XML', transform_source=None,
816 fatal=True, encoding=None, data=None, headers={}, query={},
817 expected_status=None):
818 """
819 Return a tuple (xml as an compat_etree_Element, URL handle).
820
821 See _download_webpage docstring for arguments specification.
822 """
823 res = self._download_webpage_handle(
824 url_or_request, video_id, note, errnote, fatal=fatal,
825 encoding=encoding, data=data, headers=headers, query=query,
826 expected_status=expected_status)
827 if res is False:
828 return res
829 xml_string, urlh = res
830 return self._parse_xml(
831 xml_string, video_id, transform_source=transform_source,
832 fatal=fatal), urlh
833
834 def _download_xml(
835 self, url_or_request, video_id,
836 note='Downloading XML', errnote='Unable to download XML',
837 transform_source=None, fatal=True, encoding=None,
838 data=None, headers={}, query={}, expected_status=None):
839 """
840 Return the xml as an compat_etree_Element.
841
842 See _download_webpage docstring for arguments specification.
843 """
844 res = self._download_xml_handle(
845 url_or_request, video_id, note=note, errnote=errnote,
846 transform_source=transform_source, fatal=fatal, encoding=encoding,
847 data=data, headers=headers, query=query,
848 expected_status=expected_status)
849 return res if res is False else res[0]
850
851 def _parse_xml(self, xml_string, video_id, transform_source=None, fatal=True):
852 if transform_source:
853 xml_string = transform_source(xml_string)
854 try:
855 return compat_etree_fromstring(xml_string.encode('utf-8'))
856 except compat_xml_parse_error as ve:
857 errmsg = '%s: Failed to parse XML ' % video_id
858 if fatal:
859 raise ExtractorError(errmsg, cause=ve)
860 else:
861 self.report_warning(errmsg + str(ve))
862
863 def _download_json_handle(
864 self, url_or_request, video_id, note='Downloading JSON metadata',
865 errnote='Unable to download JSON metadata', transform_source=None,
866 fatal=True, encoding=None, data=None, headers={}, query={},
867 expected_status=None):
868 """
869 Return a tuple (JSON object, URL handle).
870
871 See _download_webpage docstring for arguments specification.
872 """
873 res = self._download_webpage_handle(
874 url_or_request, video_id, note, errnote, fatal=fatal,
875 encoding=encoding, data=data, headers=headers, query=query,
876 expected_status=expected_status)
877 if res is False:
878 return res
879 json_string, urlh = res
880 return self._parse_json(
881 json_string, video_id, transform_source=transform_source,
882 fatal=fatal), urlh
883
884 def _download_json(
885 self, url_or_request, video_id, note='Downloading JSON metadata',
886 errnote='Unable to download JSON metadata', transform_source=None,
887 fatal=True, encoding=None, data=None, headers={}, query={},
888 expected_status=None):
889 """
890 Return the JSON object as a dict.
891
892 See _download_webpage docstring for arguments specification.
893 """
894 res = self._download_json_handle(
895 url_or_request, video_id, note=note, errnote=errnote,
896 transform_source=transform_source, fatal=fatal, encoding=encoding,
897 data=data, headers=headers, query=query,
898 expected_status=expected_status)
899 return res if res is False else res[0]
900
901 def _parse_json(self, json_string, video_id, transform_source=None, fatal=True):
902 if transform_source:
903 json_string = transform_source(json_string)
904 try:
905 return json.loads(json_string)
906 except ValueError as ve:
907 errmsg = '%s: Failed to parse JSON ' % video_id
908 if fatal:
909 raise ExtractorError(errmsg, cause=ve)
910 else:
911 self.report_warning(errmsg + str(ve))
912
913 def report_warning(self, msg, video_id=None):
914 idstr = '' if video_id is None else '%s: ' % video_id
915 self._downloader.report_warning(
916 '[%s] %s%s' % (self.IE_NAME, idstr, msg))
917
918 def to_screen(self, msg):
919 """Print msg to screen, prefixing it with '[ie_name]'"""
920 self._downloader.to_screen('[%s] %s' % (self.IE_NAME, msg))
921
922 def report_extraction(self, id_or_name):
923 """Report information extraction."""
924 self.to_screen('%s: Extracting information' % id_or_name)
925
926 def report_download_webpage(self, video_id):
927 """Report webpage download."""
928 self.to_screen('%s: Downloading webpage' % video_id)
929
930 def report_age_confirmation(self):
931 """Report attempt to confirm age."""
932 self.to_screen('Confirming age')
933
934 def report_login(self):
935 """Report attempt to log in."""
936 self.to_screen('Logging in')
937
938 @staticmethod
939 def raise_login_required(msg='This video is only available for registered users'):
940 raise ExtractorError(
941 '%s. Use --username and --password or --netrc to provide account credentials.' % msg,
942 expected=True)
943
944 @staticmethod
945 def raise_geo_restricted(msg='This video is not available from your location due to geo restriction', countries=None):
946 raise GeoRestrictedError(msg, countries=countries)
947
948 # Methods for following #608
949 @staticmethod
950 def url_result(url, ie=None, video_id=None, video_title=None):
951 """Returns a URL that points to a page that should be processed"""
952 # TODO: ie should be the class used for getting the info
953 video_info = {'_type': 'url',
954 'url': url,
955 'ie_key': ie}
956 if video_id is not None:
957 video_info['id'] = video_id
958 if video_title is not None:
959 video_info['title'] = video_title
960 return video_info
961
962 def playlist_from_matches(self, matches, playlist_id=None, playlist_title=None, getter=None, ie=None):
963 urls = orderedSet(
964 self.url_result(self._proto_relative_url(getter(m) if getter else m), ie)
965 for m in matches)
966 return self.playlist_result(
967 urls, playlist_id=playlist_id, playlist_title=playlist_title)
968
969 @staticmethod
970 def playlist_result(entries, playlist_id=None, playlist_title=None, playlist_description=None):
971 """Returns a playlist"""
972 video_info = {'_type': 'playlist',
973 'entries': entries}
974 if playlist_id:
975 video_info['id'] = playlist_id
976 if playlist_title:
977 video_info['title'] = playlist_title
978 if playlist_description:
979 video_info['description'] = playlist_description
980 return video_info
981
982 def _search_regex(self, pattern, string, name, default=NO_DEFAULT, fatal=True, flags=0, group=None):
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
987 RegexNotFoundError, depending on fatal, specifying the field name.
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)
994 if mobj:
995 break
996
997 if not self._downloader.params.get('no_color') and compat_os_name != 'nt' and sys.stderr.isatty():
998 _name = '\033[0;34m%s\033[0m' % name
999 else:
1000 _name = name
1001
1002 if mobj:
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)
1008 elif default is not NO_DEFAULT:
1009 return default
1010 elif fatal:
1011 raise RegexNotFoundError('Unable to extract %s' % _name)
1012 else:
1013 self._downloader.report_warning('unable to extract %s' % _name + bug_reports_message())
1014 return None
1015
1016 def _html_search_regex(self, pattern, string, name, default=NO_DEFAULT, fatal=True, flags=0, group=None):
1017 """
1018 Like _search_regex, but strips HTML tags and unescapes entities.
1019 """
1020 res = self._search_regex(pattern, string, name, default, fatal, flags, group)
1021 if res:
1022 return clean_html(res).strip()
1023 else:
1024 return res
1025
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:
1038 raise netrc.NetrcParseError(
1039 'No authenticators for %s' % netrc_machine)
1040 except (IOError, netrc.NetrcParseError) as err:
1041 self._downloader.report_warning(
1042 'parsing .netrc: %s' % error_to_compat_str(err))
1043
1044 return username, password
1045
1046 def _get_login_info(self, username_option='username', password_option='password', netrc_machine=None):
1047 """
1048 Get the login info as (username, password)
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.
1053 If there's no info available, return (None, None)
1054 """
1055 if self._downloader is None:
1056 return (None, None)
1057
1058 downloader_params = self._downloader.params
1059
1060 # Attempt to use provided username and password or .netrc data
1061 if downloader_params.get(username_option) is not None:
1062 username = downloader_params[username_option]
1063 password = downloader_params[password_option]
1064 else:
1065 username, password = self._get_netrc_login_info(netrc_machine)
1066
1067 return username, password
1068
1069 def _get_tfa_info(self, note='two-factor verification code'):
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:
1077 return None
1078 downloader_params = self._downloader.params
1079
1080 if downloader_params.get('twofactor') is not None:
1081 return downloader_params['twofactor']
1082
1083 return compat_getpass('Type %s and press [Return]: ' % note)
1084
1085 # Helper functions for extracting OpenGraph info
1086 @staticmethod
1087 def _og_regexes(prop):
1088 content_re = r'content=(?:"([^"]+?)"|\'([^\']+?)\'|\s*([^\s"\'=<>`]+?))'
1089 property_re = (r'(?:name|property)=(?:\'og[:-]%(prop)s\'|"og[:-]%(prop)s"|\s*og[:-]%(prop)s\b)'
1090 % {'prop': re.escape(prop)})
1091 template = r'<meta[^>]+?%s[^>]+?%s'
1092 return [
1093 template % (property_re, content_re),
1094 template % (content_re, property_re),
1095 ]
1096
1097 @staticmethod
1098 def _meta_regex(prop):
1099 return r'''(?isx)<meta
1100 (?=[^>]+(?:itemprop|name|property|id|http-equiv)=(["\']?)%s\1)
1101 [^>]+?content=(["\'])(?P<content>.*?)\2''' % re.escape(prop)
1102
1103 def _og_search_property(self, prop, html, name=None, **kargs):
1104 if not isinstance(prop, (list, tuple)):
1105 prop = [prop]
1106 if name is None:
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)
1112 if escaped is None:
1113 return None
1114 return unescapeHTML(escaped)
1115
1116 def _og_search_thumbnail(self, html, **kargs):
1117 return self._og_search_property('image', html, 'thumbnail URL', fatal=False, **kargs)
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
1125 def _og_search_video_url(self, html, name='video url', secure=True, **kargs):
1126 regexes = self._og_regexes('video') + self._og_regexes('video:url')
1127 if secure:
1128 regexes = self._og_regexes('video:secure_url') + regexes
1129 return self._html_search_regex(regexes, html, name, **kargs)
1130
1131 def _og_search_url(self, html, **kargs):
1132 return self._og_search_property('url', html, **kargs)
1133
1134 def _html_search_meta(self, name, html, display_name=None, fatal=False, **kwargs):
1135 if not isinstance(name, (list, tuple)):
1136 name = [name]
1137 if display_name is None:
1138 display_name = name[0]
1139 return self._html_search_regex(
1140 [self._meta_regex(n) for n in name],
1141 html, display_name, fatal=fatal, group='content', **kwargs)
1142
1143 def _dc_search_uploader(self, html):
1144 return self._html_search_meta('dc.creator', html, 'uploader')
1145
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
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 }
1168 return RATING_TABLE.get(rating.lower())
1169
1170 def _family_friendly_search(self, html):
1171 # See http://schema.org/VideoObject
1172 family_friendly = self._html_search_meta(
1173 'isFamilyFriendly', html, default=None)
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 }
1184 return RATING_TABLE.get(family_friendly.lower())
1185
1186 def _twitter_search_player(self, html):
1187 return self._html_search_meta('twitter:player', html,
1188 'twitter card player')
1189
1190 def _search_json_ld(self, html, video_id, expected_type=None, **kwargs):
1191 json_ld_list = list(re.finditer(JSON_LD_RE, html))
1192 default = kwargs.get('default', NO_DEFAULT)
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
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 {}
1218
1219 def _json_ld(self, json_ld, video_id, fatal=True, expected_type=None):
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 = {}
1225 if not isinstance(json_ld, (list, tuple, dict)):
1226 return info
1227 if isinstance(json_ld, dict):
1228 json_ld = [json_ld]
1229
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
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
1247 def extract_interaction_statistic(e):
1248 interaction_statistic = e.get('interactionStatistic')
1249 if isinstance(interaction_statistic, dict):
1250 interaction_statistic = [interaction_statistic]
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
1258 interaction_type = extract_interaction_type(is_e)
1259 if not interaction_type:
1260 continue
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'))
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
1275 def extract_video_object(e):
1276 assert e['@type'] == 'VideoObject'
1277 info.update({
1278 'url': url_or_none(e.get('contentUrl')),
1279 'title': unescapeHTML(e.get('name')),
1280 'description': unescapeHTML(e.get('description')),
1281 'thumbnail': url_or_none(e.get('thumbnailUrl') or e.get('thumbnailURL')),
1282 'duration': parse_duration(e.get('duration')),
1283 'timestamp': unified_timestamp(e.get('uploadDate')),
1284 'uploader': str_or_none(e.get('author')),
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')),
1289 'view_count': int_or_none(e.get('interactionCount')),
1290 })
1291 extract_interaction_statistic(e)
1292
1293 for e in json_ld:
1294 if '@context' in e:
1295 item_type = e.get('@type')
1296 if expected_type is not None and expected_type != item_type:
1297 continue
1298 if item_type in ('TVEpisode', 'Episode'):
1299 episode_name = unescapeHTML(e.get('name'))
1300 info.update({
1301 'episode': episode_name,
1302 'episode_number': int_or_none(e.get('episodeNumber')),
1303 'description': unescapeHTML(e.get('description')),
1304 })
1305 if not info.get('title') and episode_name:
1306 info['title'] = episode_name
1307 part_of_season = e.get('partOfSeason')
1308 if isinstance(part_of_season, dict) and part_of_season.get('@type') in ('TVSeason', 'Season', 'CreativeWorkSeason'):
1309 info.update({
1310 'season': unescapeHTML(part_of_season.get('name')),
1311 'season_number': int_or_none(part_of_season.get('seasonNumber')),
1312 })
1313 part_of_series = e.get('partOfSeries') or e.get('partOfTVSeries')
1314 if isinstance(part_of_series, dict) and part_of_series.get('@type') in ('TVSeries', 'Series', 'CreativeWorkSeries'):
1315 info['series'] = unescapeHTML(part_of_series.get('name'))
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 })
1323 elif item_type in ('Article', 'NewsArticle'):
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':
1330 extract_video_object(e)
1331 if expected_type is None:
1332 continue
1333 else:
1334 break
1335 video = e.get('video')
1336 if isinstance(video, dict) and video.get('@type') == 'VideoObject':
1337 extract_video_object(video)
1338 if expected_type is None:
1339 continue
1340 else:
1341 break
1342 return dict((k, v) for k, v in info.items() if v is not None)
1343
1344 @staticmethod
1345 def _hidden_inputs(html):
1346 html = re.sub(r'<!--(?:(?!<!--).)*-->', '', html)
1347 hidden_inputs = {}
1348 for input in re.findall(r'(?i)(<input[^>]+>)', html):
1349 attrs = extract_attributes(input)
1350 if not input:
1351 continue
1352 if attrs.get('type') not in ('hidden', 'submit'):
1353 continue
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
1358 return hidden_inputs
1359
1360 def _form_hidden_inputs(self, form_id, html):
1361 form = self._search_regex(
1362 r'(?is)<form[^>]+?id=(["\'])%s\1[^>]*>(?P<form>.+?)</form>' % form_id,
1363 html, '%s form' % form_id, group='form')
1364 return self._hidden_inputs(form)
1365
1366 class FormatSort:
1367 regex = r' *((?P<reverse>\+)?(?P<field>[a-zA-Z0-9_]+)((?P<seperator>[~:])(?P<limit>.*?))?)? *$'
1368
1369 default = ('hidden', 'has_video', 'extractor', 'lang', 'quality',
1370 'res', 'fps', 'codec', 'size', 'br', 'asr',
1371 'proto', 'ext', 'has_audio', 'source', 'format_id')
1372
1373 settings = {
1374 'vcodec': {'type': 'ordered', 'regex': True,
1375 'order': ['vp9', '(h265|he?vc?)', '(h264|avc)', 'vp8', '(mp4v|h263)', 'theora', '', None, 'none']},
1376 'acodec': {'type': 'ordered', 'regex': True,
1377 'order': ['opus', 'vorbis', 'aac', 'mp?4a?', 'mp3', 'e?a?c-?3', 'dts', '', None, 'none']},
1378 'proto': {'type': 'ordered', 'regex': True, 'field': 'protocol',
1379 'order': ['(ht|f)tps', '(ht|f)tp$', 'm3u8.+', 'm3u8', '.*dash', '', 'mms|rtsp', 'none', 'f4']},
1380 'vext': {'type': 'ordered', 'field': 'video_ext',
1381 'order': ('mp4', 'webm', 'flv', '', 'none'),
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},
1387 'ie_pref': {'priority': True, 'type': 'extractor', 'field': 'extractor_preference'},
1388 'hasvid': {'priority': True, 'field': 'vcodec', 'type': 'boolean', 'not_in_list': ('none',)},
1389 'hasaud': {'field': 'acodec', 'type': 'boolean', 'not_in_list': ('none',)},
1390 'lang': {'priority': True, 'convert': 'ignore', 'field': 'language_preference'},
1391 'quality': {'priority': True, 'convert': 'float_none'},
1392 'filesize': {'convert': 'bytes'},
1393 'fs_approx': {'convert': 'bytes', 'field': 'filesize_approx'},
1394 'id': {'convert': 'string', 'field': 'format_id'},
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'},
1402 'source': {'convert': 'ignore', 'field': 'source_preference'},
1403
1404 'codec': {'type': 'combined', 'field': ('vcodec', 'acodec')},
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'},
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'},
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'},
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'},
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 }
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'
1451 else:
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
1460 else:
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)
1487 else:
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))
1640
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
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
1659 def _is_valid_url(self, url, video_id, item='video', headers={}):
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
1664 try:
1665 self._request_webpage(url, video_id, 'Checking %s URL' % item, headers=headers)
1666 return True
1667 except ExtractorError as e:
1668 self.to_screen(
1669 '%s: %s URL is invalid, skipping: %s'
1670 % (video_id, item, error_to_compat_str(e.cause)))
1671 return False
1672
1673 def http_scheme(self):
1674 """ Either "http:" or "https:", depending on the user's preferences """
1675 return (
1676 'http:'
1677 if self._downloader.params.get('prefer_insecure', False)
1678 else 'https:')
1679
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
1690 def _sleep(self, timeout, video_id, msg_template=None):
1691 if msg_template is None:
1692 msg_template = '%(video_id)s: Waiting for %(timeout)s seconds'
1693 msg = msg_template % {'video_id': video_id, 'timeout': timeout}
1694 self.to_screen(msg)
1695 time.sleep(timeout)
1696
1697 def _extract_f4m_formats(self, manifest_url, video_id, preference=None, f4m_id=None,
1698 transform_source=lambda s: fix_xml_ampersands(s).strip(),
1699 fatal=True, m3u8_id=None, data=None, headers={}, query={}):
1700 manifest = self._download_xml(
1701 manifest_url, video_id, 'Downloading f4m manifest',
1702 'Unable to download f4m manifest',
1703 # Some manifests may be malformed, e.g. prosiebensat1 generated manifests
1704 # (see https://github.com/ytdl-org/youtube-dl/issues/6215#issuecomment-121704244)
1705 transform_source=transform_source,
1706 fatal=fatal, data=data, headers=headers, query=query)
1707
1708 if manifest is False:
1709 return []
1710
1711 return self._parse_f4m_formats(
1712 manifest, manifest_url, video_id, preference=preference, f4m_id=f4m_id,
1713 transform_source=transform_source, fatal=fatal, m3u8_id=m3u8_id)
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(),
1717 fatal=True, m3u8_id=None):
1718 if not isinstance(manifest, compat_etree_Element) and not fatal:
1719 return []
1720
1721 # currently youtube-dlc cannot decode the playerVerificationChallenge as Akamai uses Adobe Alchemy
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
1728 formats = []
1729 manifest_version = '1.0'
1730 media_nodes = manifest.findall('{http://ns.adobe.com/f4m/1.0}media')
1731 if not media_nodes:
1732 manifest_version = '2.0'
1733 media_nodes = manifest.findall('{http://ns.adobe.com/f4m/2.0}media')
1734 # Remove unsupported DRM protected media from final formats
1735 # rendition (see https://github.com/ytdl-org/youtube-dl/issues/8573).
1736 media_nodes = remove_encrypted_media(media_nodes)
1737 if not media_nodes:
1738 return formats
1739
1740 manifest_base_url = get_base_url(manifest)
1741
1742 bootstrap_info = xpath_element(
1743 manifest, ['{http://ns.adobe.com/f4m/1.0}bootstrapInfo', '{http://ns.adobe.com/f4m/2.0}bootstrapInfo'],
1744 'bootstrap info', default=None)
1745
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
1753 for i, media_el in enumerate(media_nodes):
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)]))
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')
1768 if not media_url:
1769 continue
1770 manifest_url = (
1771 media_url if media_url.startswith('http://') or media_url.startswith('https://')
1772 else ((manifest_base_url or '/'.join(manifest_url.split('/')[:-1])) + '/' + media_url))
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
1777 ext = determine_ext(manifest_url)
1778 if ext == 'f4m':
1779 f4m_formats = self._extract_f4m_formats(
1780 manifest_url, video_id, preference=preference, f4m_id=f4m_id,
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,
1793 'vcodec': vcodec,
1794 })
1795 formats.extend(f4m_formats)
1796 continue
1797 elif ext == 'm3u8':
1798 formats.extend(self._extract_m3u8_formats(
1799 manifest_url, video_id, 'mp4', preference=preference,
1800 m3u8_id=m3u8_id, fatal=fatal))
1801 continue
1802 formats.append({
1803 'format_id': format_id,
1804 'url': manifest_url,
1805 'manifest_url': manifest_url,
1806 'ext': 'flv' if bootstrap_info is not None else None,
1807 'protocol': 'f4m',
1808 'tbr': tbr,
1809 'width': width,
1810 'height': height,
1811 'vcodec': vcodec,
1812 'preference': preference,
1813 })
1814 return formats
1815
1816 def _m3u8_meta_format(self, m3u8_url, ext=None, preference=None, m3u8_id=None):
1817 return {
1818 'format_id': '-'.join(filter(None, [m3u8_id, 'meta'])),
1819 'url': m3u8_url,
1820 'ext': ext,
1821 'protocol': 'm3u8',
1822 'preference': preference - 100 if preference else -100,
1823 'resolution': 'multiple',
1824 'format_note': 'Quality selection URL',
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,
1830 fatal=True, live=False, data=None, headers={},
1831 query={}):
1832 res = self._download_webpage_handle(
1833 m3u8_url, video_id,
1834 note=note or 'Downloading m3u8 information',
1835 errnote=errnote or 'Failed to download m3u8 information',
1836 fatal=fatal, data=data, headers=headers, query=query)
1837
1838 if res is False:
1839 return []
1840
1841 m3u8_doc, urlh = res
1842 m3u8_url = urlh.geturl()
1843
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):
1851 if '#EXT-X-FAXS-CM:' in m3u8_doc: # Adobe Flash Access
1852 return []
1853
1854 if re.search(r'#EXT-X-SESSION-KEY:.*?URI="skd://', m3u8_doc): # Apple FairPlay
1855 return []
1856
1857 formats = []
1858
1859 format_url = lambda u: (
1860 u
1861 if re.match(r'^https?://', u)
1862 else compat_urlparse.urljoin(m3u8_url, u))
1863
1864 # References:
1865 # 1. https://tools.ietf.org/html/draft-pantos-http-live-streaming-21
1866 # 2. https://github.com/ytdl-org/youtube-dl/issues/12211
1867 # 3. https://github.com/ytdl-org/youtube-dl/issues/18923
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.
1873 # Fortunately, master playlist can be easily distinguished from media
1874 # playlist based on particular tags availability. As of [1, 4.3.3, 4.3.4]
1875 # master playlist tags MUST NOT appear in a media playlist and vice versa.
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
1880 if '#EXT-X-TARGETDURATION' in m3u8_doc: # media playlist, return as is
1881 return [{
1882 'url': m3u8_url,
1883 'format_id': m3u8_id,
1884 'ext': ext,
1885 'protocol': entry_protocol,
1886 'preference': preference,
1887 }]
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 = []
1904 for v in (m3u8_id, group_id, name):
1905 if v:
1906 format_id.append(v)
1907 f = {
1908 'format_id': '-'.join(format_id),
1909 'url': format_url(media_url),
1910 'manifest_url': m3u8_url,
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
1922 # EXT-X-STREAM-INF tag it still sometimes may be present (see [1]
1923 # or vidio test in TestInfoExtractor.test_parse_m3u8_formats)
1924 # 1. http://www.vidio.com/watch/165683-dj_ambred-booyah-live-2015
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
1939 # parse EXT-X-MEDIA tags before EXT-X-STREAM-INF in order to have the
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
1946 for line in m3u8_doc.splitlines():
1947 if line.startswith('#EXT-X-STREAM-INF:'):
1948 last_stream_inf = parse_m3u8_attributes(line)
1949 elif line.startswith('#') or not line.strip():
1950 continue
1951 else:
1952 tbr = float_or_none(
1953 last_stream_inf.get('AVERAGE-BANDWIDTH')
1954 or last_stream_inf.get('BANDWIDTH'), scale=1000)
1955 format_id = []
1956 if m3u8_id:
1957 format_id.append(m3u8_id)
1958 stream_name = build_stream_name()
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.
1962 if not live:
1963 format_id.append(stream_name if stream_name else '%d' % (tbr if tbr else len(formats)))
1964 manifest_url = format_url(line.strip())
1965 f = {
1966 'format_id': '-'.join(format_id),
1967 'url': manifest_url,
1968 'manifest_url': m3u8_url,
1969 'tbr': tbr,
1970 'ext': ext,
1971 'fps': float_or_none(last_stream_inf.get('FRAME-RATE')),
1972 'protocol': entry_protocol,
1973 'preference': preference,
1974 }
1975 resolution = last_stream_inf.get('RESOLUTION')
1976 if resolution:
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'))
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)
1987 f.update({
1988 'vbr': vbr,
1989 'abr': abr,
1990 })
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
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
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'
2009 formats.append(f)
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
2023 last_stream_inf = {}
2024 return formats
2025
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
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)
2040
2041 if smil is False:
2042 assert not fatal
2043 return []
2044
2045 namespace = self._parse_smil_namespace(smil)
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
2056 def _download_smil(self, smil_url, video_id, fatal=True, transform_source=None):
2057 return self._download_xml(
2058 smil_url, video_id, 'Downloading SMIL file',
2059 'Unable to download SMIL file', fatal=fatal, transform_source=transform_source)
2060
2061 def _parse_smil(self, smil, smil_url, video_id, f4m_params=None):
2062 namespace = self._parse_smil_namespace(smil)
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
2071 upload_date = None
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
2081 elif not upload_date and name == 'date':
2082 upload_date = unified_strdate(content)
2083
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
2091 return {
2092 'id': video_id,
2093 'title': title or video_id,
2094 'description': description,
2095 'upload_date': upload_date,
2096 'thumbnails': thumbnails,
2097 'formats': formats,
2098 'subtitles': subtitles,
2099 }
2100
2101 def _parse_smil_namespace(self, smil):
2102 return self._search_regex(
2103 r'(?i)^{([^}]+)?}smil$', smil.tag, 'namespace', default=None)
2104
2105 def _parse_smil_formats(self, smil, smil_url, video_id, namespace=None, f4m_params=None, transform_rtmp_url=None):
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
2112
2113 formats = []
2114 rtmp_count = 0
2115 http_count = 0
2116 m3u8_count = 0
2117
2118 srcs = []
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')
2122 if not src or src in srcs:
2123 continue
2124 srcs.append(src)
2125
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')
2132 src_ext = determine_ext(src)
2133 streamer = medium.get('streamer') or base
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 })
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 })
2153 continue
2154
2155 src_url = src if src.startswith('http') else compat_urlparse.urljoin(base, src)
2156 src_url = src_url.strip()
2157
2158 if proto == 'm3u8' or src_ext == 'm3u8':
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)
2170 elif src_ext == 'f4m':
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 '?'
2178 f4m_url += compat_urllib_parse_urlencode(f4m_params)
2179 formats.extend(self._extract_f4m_formats(f4m_url, video_id, f4m_id='hds', fatal=False))
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):
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 })
2197
2198 return formats
2199
2200 def _parse_smil_subtitles(self, smil, namespace=None, subtitles_lang='en'):
2201 urls = []
2202 subtitles = {}
2203 for num, textstream in enumerate(smil.findall(self._xpath_ns('.//textstream', namespace))):
2204 src = textstream.get('src')
2205 if not src or src in urls:
2206 continue
2207 urls.append(src)
2208 ext = textstream.get('ext') or mimetype2ext(textstream.get('type')) or determine_ext(src)
2209 lang = textstream.get('systemLanguage') or textstream.get('systemLanguageName') or textstream.get('lang') or subtitles_lang
2210 subtitles.setdefault(lang, []).append({
2211 'url': src,
2212 'ext': ext,
2213 })
2214 return subtitles
2215
2216 def _extract_xspf_playlist(self, xspf_url, playlist_id, fatal=True):
2217 xspf = self._download_xml(
2218 xspf_url, playlist_id, 'Downloading xpsf playlist',
2219 'Unable to download xspf manifest', fatal=fatal)
2220 if xspf is False:
2221 return []
2222 return self._parse_xspf(
2223 xspf, playlist_id, xspf_url=xspf_url,
2224 xspf_base_url=base_url(xspf_url))
2225
2226 def _parse_xspf(self, xspf_doc, playlist_id, xspf_url=None, xspf_base_url=None):
2227 NS_MAP = {
2228 'xspf': 'http://xspf.org/ns/0/',
2229 's1': 'http://static.streamone.nl/player/ns/0',
2230 }
2231
2232 entries = []
2233 for track in xspf_doc.findall(xpath_with_ns('./xspf:trackList/xspf:track', NS_MAP)):
2234 title = xpath_text(
2235 track, xpath_with_ns('./xspf:title', NS_MAP), 'title', default=playlist_id)
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
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 })
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
2267 def _extract_mpd_formats(self, mpd_url, video_id, mpd_id=None, note=None, errnote=None, fatal=True, formats_dict={}, data=None, headers={}, query={}):
2268 res = self._download_xml_handle(
2269 mpd_url, video_id,
2270 note=note or 'Downloading MPD manifest',
2271 errnote=errnote or 'Failed to download MPD manifest',
2272 fatal=fatal, data=data, headers=headers, query=query)
2273 if res is False:
2274 return []
2275 mpd_doc, urlh = res
2276 if mpd_doc is None:
2277 return []
2278 mpd_base_url = base_url(urlh.geturl())
2279
2280 return self._parse_mpd_formats(
2281 mpd_doc, mpd_id=mpd_id, mpd_base_url=mpd_base_url,
2282 formats_dict=formats_dict, mpd_url=mpd_url)
2283
2284 def _parse_mpd_formats(self, mpd_doc, mpd_id=None, mpd_base_url='', formats_dict={}, mpd_url=None):
2285 """
2286 Parse formats from MPD manifest.
2287 References:
2288 1. MPEG-DASH Standard, ISO/IEC 23009-1:2014(E),
2289 http://standards.iso.org/ittf/PubliclyAvailableStandards/c065274_ISO_IEC_23009-1_2014.zip
2290 2. https://en.wikipedia.org/wiki/Dynamic_Adaptive_Streaming_over_HTTP
2291 """
2292 if not self._downloader.params.get('dynamic_mpd'):
2293 if mpd_doc.get('type') == 'dynamic':
2294 return []
2295
2296 namespace = self._search_regex(r'(?i)^{([^}]+)?}MPD$', mpd_doc.tag, 'namespace', default=None)
2297
2298 def _add_ns(path):
2299 return self._xpath_ns(path, namespace)
2300
2301 def is_drm_protected(element):
2302 return element.find(_add_ns('ContentProtection')) is not None
2303
2304 def extract_multisegment_info(element, ms_parent_info):
2305 ms_info = ms_parent_info.copy()
2306
2307 # As per [1, 5.3.9.2.2] SegmentList and SegmentTemplate share some
2308 # common attributes and elements. We will only extract relevant
2309 # for us.
2310 def extract_common(source):
2311 segment_timeline = source.find(_add_ns('SegmentTimeline'))
2312 if segment_timeline is not None:
2313 s_e = segment_timeline.findall(_add_ns('S'))
2314 if s_e:
2315 ms_info['total_number'] = 0
2316 ms_info['s'] = []
2317 for s in s_e:
2318 r = int(s.get('r', 0))
2319 ms_info['total_number'] += 1 + r
2320 ms_info['s'].append({
2321 't': int(s.get('t', 0)),
2322 # @d is mandatory (see [1, 5.3.9.6.2, Table 17, page 60])
2323 'd': int(s.attrib['d']),
2324 'r': r,
2325 })
2326 start_number = source.get('startNumber')
2327 if start_number:
2328 ms_info['start_number'] = int(start_number)
2329 timescale = source.get('timescale')
2330 if timescale:
2331 ms_info['timescale'] = int(timescale)
2332 segment_duration = source.get('duration')
2333 if segment_duration:
2334 ms_info['segment_duration'] = float(segment_duration)
2335
2336 def extract_Initialization(source):
2337 initialization = source.find(_add_ns('Initialization'))
2338 if initialization is not None:
2339 ms_info['initialization_url'] = initialization.attrib['sourceURL']
2340
2341 segment_list = element.find(_add_ns('SegmentList'))
2342 if segment_list is not None:
2343 extract_common(segment_list)
2344 extract_Initialization(segment_list)
2345 segment_urls_e = segment_list.findall(_add_ns('SegmentURL'))
2346 if segment_urls_e:
2347 ms_info['segment_urls'] = [segment.attrib['media'] for segment in segment_urls_e]
2348 else:
2349 segment_template = element.find(_add_ns('SegmentTemplate'))
2350 if segment_template is not None:
2351 extract_common(segment_template)
2352 media = segment_template.get('media')
2353 if media:
2354 ms_info['media'] = media
2355 initialization = segment_template.get('initialization')
2356 if initialization:
2357 ms_info['initialization'] = initialization
2358 else:
2359 extract_Initialization(segment_template)
2360 return ms_info
2361
2362 mpd_duration = parse_duration(mpd_doc.get('mediaPresentationDuration'))
2363 formats = []
2364 for period in mpd_doc.findall(_add_ns('Period')):
2365 period_duration = parse_duration(period.get('duration')) or mpd_duration
2366 period_ms_info = extract_multisegment_info(period, {
2367 'start_number': 1,
2368 'timescale': 1,
2369 })
2370 for adaptation_set in period.findall(_add_ns('AdaptationSet')):
2371 if is_drm_protected(adaptation_set):
2372 continue
2373 adaption_set_ms_info = extract_multisegment_info(adaptation_set, period_ms_info)
2374 for representation in adaptation_set.findall(_add_ns('Representation')):
2375 if is_drm_protected(representation):
2376 continue
2377 representation_attrib = adaptation_set.attrib.copy()
2378 representation_attrib.update(representation.attrib)
2379 # According to [1, 5.3.7.2, Table 9, page 41], @mimeType is mandatory
2380 mime_type = representation_attrib['mimeType']
2381 content_type = mime_type.split('/')[0]
2382 if content_type == 'text':
2383 # TODO implement WebVTT downloading
2384 pass
2385 elif content_type in ('video', 'audio'):
2386 base_url = ''
2387 for element in (representation, adaptation_set, period, mpd_doc):
2388 base_url_e = element.find(_add_ns('BaseURL'))
2389 if base_url_e is not None:
2390 base_url = base_url_e.text + base_url
2391 if re.match(r'^https?://', base_url):
2392 break
2393 if mpd_base_url and not re.match(r'^https?://', base_url):
2394 if not mpd_base_url.endswith('/') and not base_url.startswith('/'):
2395 mpd_base_url += '/'
2396 base_url = mpd_base_url + base_url
2397 representation_id = representation_attrib.get('id')
2398 lang = representation_attrib.get('lang')
2399 url_el = representation.find(_add_ns('BaseURL'))
2400 filesize = int_or_none(url_el.attrib.get('{http://youtube.com/yt/2012/10/10}contentLength') if url_el is not None else None)
2401 bandwidth = int_or_none(representation_attrib.get('bandwidth'))
2402 f = {
2403 'format_id': '%s-%s' % (mpd_id, representation_id) if mpd_id else representation_id,
2404 'manifest_url': mpd_url,
2405 'ext': mimetype2ext(mime_type),
2406 'width': int_or_none(representation_attrib.get('width')),
2407 'height': int_or_none(representation_attrib.get('height')),
2408 'tbr': float_or_none(bandwidth, 1000),
2409 'asr': int_or_none(representation_attrib.get('audioSamplingRate')),
2410 'fps': int_or_none(representation_attrib.get('frameRate')),
2411 'language': lang if lang not in ('mul', 'und', 'zxx', 'mis') else None,
2412 'format_note': 'DASH %s' % content_type,
2413 'filesize': filesize,
2414 'container': mimetype2ext(mime_type) + '_dash',
2415 }
2416 f.update(parse_codecs(representation_attrib.get('codecs')))
2417 representation_ms_info = extract_multisegment_info(representation, adaption_set_ms_info)
2418
2419 def prepare_template(template_name, identifiers):
2420 tmpl = representation_ms_info[template_name]
2421 # First of, % characters outside $...$ templates
2422 # must be escaped by doubling for proper processing
2423 # by % operator string formatting used further (see
2424 # https://github.com/ytdl-org/youtube-dl/issues/16867).
2425 t = ''
2426 in_template = False
2427 for c in tmpl:
2428 t += c
2429 if c == '$':
2430 in_template = not in_template
2431 elif c == '%' and not in_template:
2432 t += c
2433 # Next, $...$ templates are translated to their
2434 # %(...) counterparts to be used with % operator
2435 t = t.replace('$RepresentationID$', representation_id)
2436 t = re.sub(r'\$(%s)\$' % '|'.join(identifiers), r'%(\1)d', t)
2437 t = re.sub(r'\$(%s)%%([^$]+)\$' % '|'.join(identifiers), r'%(\1)\2', t)
2438 t.replace('$$', '$')
2439 return t
2440
2441 # @initialization is a regular template like @media one
2442 # so it should be handled just the same way (see
2443 # https://github.com/ytdl-org/youtube-dl/issues/11605)
2444 if 'initialization' in representation_ms_info:
2445 initialization_template = prepare_template(
2446 'initialization',
2447 # As per [1, 5.3.9.4.2, Table 15, page 54] $Number$ and
2448 # $Time$ shall not be included for @initialization thus
2449 # only $Bandwidth$ remains
2450 ('Bandwidth', ))
2451 representation_ms_info['initialization_url'] = initialization_template % {
2452 'Bandwidth': bandwidth,
2453 }
2454
2455 def location_key(location):
2456 return 'url' if re.match(r'^https?://', location) else 'path'
2457
2458 if 'segment_urls' not in representation_ms_info and 'media' in representation_ms_info:
2459
2460 media_template = prepare_template('media', ('Number', 'Bandwidth', 'Time'))
2461 media_location_key = location_key(media_template)
2462
2463 # As per [1, 5.3.9.4.4, Table 16, page 55] $Number$ and $Time$
2464 # can't be used at the same time
2465 if '%(Number' in media_template and 's' not in representation_ms_info:
2466 segment_duration = None
2467 if 'total_number' not in representation_ms_info and 'segment_duration' in representation_ms_info:
2468 segment_duration = float_or_none(representation_ms_info['segment_duration'], representation_ms_info['timescale'])
2469 representation_ms_info['total_number'] = int(math.ceil(float(period_duration) / segment_duration))
2470 representation_ms_info['fragments'] = [{
2471 media_location_key: media_template % {
2472 'Number': segment_number,
2473 'Bandwidth': bandwidth,
2474 },
2475 'duration': segment_duration,
2476 } for segment_number in range(
2477 representation_ms_info['start_number'],
2478 representation_ms_info['total_number'] + representation_ms_info['start_number'])]
2479 else:
2480 # $Number*$ or $Time$ in media template with S list available
2481 # Example $Number*$: http://www.svtplay.se/klipp/9023742/stopptid-om-bjorn-borg
2482 # Example $Time$: https://play.arkena.com/embed/avp/v2/player/media/b41dda37-d8e7-4d3f-b1b5-9a9db578bdfe/1/129411
2483 representation_ms_info['fragments'] = []
2484 segment_time = 0
2485 segment_d = None
2486 segment_number = representation_ms_info['start_number']
2487
2488 def add_segment_url():
2489 segment_url = media_template % {
2490 'Time': segment_time,
2491 'Bandwidth': bandwidth,
2492 'Number': segment_number,
2493 }
2494 representation_ms_info['fragments'].append({
2495 media_location_key: segment_url,
2496 'duration': float_or_none(segment_d, representation_ms_info['timescale']),
2497 })
2498
2499 for num, s in enumerate(representation_ms_info['s']):
2500 segment_time = s.get('t') or segment_time
2501 segment_d = s['d']
2502 add_segment_url()
2503 segment_number += 1
2504 for r in range(s.get('r', 0)):
2505 segment_time += segment_d
2506 add_segment_url()
2507 segment_number += 1
2508 segment_time += segment_d
2509 elif 'segment_urls' in representation_ms_info and 's' in representation_ms_info:
2510 # No media template
2511 # Example: https://www.youtube.com/watch?v=iXZV5uAYMJI
2512 # or any YouTube dashsegments video
2513 fragments = []
2514 segment_index = 0
2515 timescale = representation_ms_info['timescale']
2516 for s in representation_ms_info['s']:
2517 duration = float_or_none(s['d'], timescale)
2518 for r in range(s.get('r', 0) + 1):
2519 segment_uri = representation_ms_info['segment_urls'][segment_index]
2520 fragments.append({
2521 location_key(segment_uri): segment_uri,
2522 'duration': duration,
2523 })
2524 segment_index += 1
2525 representation_ms_info['fragments'] = fragments
2526 elif 'segment_urls' in representation_ms_info:
2527 # Segment URLs with no SegmentTimeline
2528 # Example: https://www.seznam.cz/zpravy/clanek/cesko-zasahne-vitr-o-sile-vichrice-muze-byt-i-zivotu-nebezpecny-39091
2529 # https://github.com/ytdl-org/youtube-dl/pull/14844
2530 fragments = []
2531 segment_duration = float_or_none(
2532 representation_ms_info['segment_duration'],
2533 representation_ms_info['timescale']) if 'segment_duration' in representation_ms_info else None
2534 for segment_url in representation_ms_info['segment_urls']:
2535 fragment = {
2536 location_key(segment_url): segment_url,
2537 }
2538 if segment_duration:
2539 fragment['duration'] = segment_duration
2540 fragments.append(fragment)
2541 representation_ms_info['fragments'] = fragments
2542 # If there is a fragments key available then we correctly recognized fragmented media.
2543 # Otherwise we will assume unfragmented media with direct access. Technically, such
2544 # assumption is not necessarily correct since we may simply have no support for
2545 # some forms of fragmented media renditions yet, but for now we'll use this fallback.
2546 if 'fragments' in representation_ms_info:
2547 f.update({
2548 # NB: mpd_url may be empty when MPD manifest is parsed from a string
2549 'url': mpd_url or base_url,
2550 'fragment_base_url': base_url,
2551 'fragments': [],
2552 'protocol': 'http_dash_segments',
2553 })
2554 if 'initialization_url' in representation_ms_info:
2555 initialization_url = representation_ms_info['initialization_url']
2556 if not f.get('url'):
2557 f['url'] = initialization_url
2558 f['fragments'].append({location_key(initialization_url): initialization_url})
2559 f['fragments'].extend(representation_ms_info['fragments'])
2560 else:
2561 # Assuming direct URL to unfragmented media.
2562 f['url'] = base_url
2563
2564 # According to [1, 5.3.5.2, Table 7, page 35] @id of Representation
2565 # is not necessarily unique within a Period thus formats with
2566 # the same `format_id` are quite possible. There are numerous examples
2567 # of such manifests (see https://github.com/ytdl-org/youtube-dl/issues/15111,
2568 # https://github.com/ytdl-org/youtube-dl/issues/13919)
2569 full_info = formats_dict.get(representation_id, {}).copy()
2570 full_info.update(f)
2571 formats.append(full_info)
2572 else:
2573 self.report_warning('Unknown MIME type %s in DASH manifest' % mime_type)
2574 return formats
2575
2576 def _extract_ism_formats(self, ism_url, video_id, ism_id=None, note=None, errnote=None, fatal=True, data=None, headers={}, query={}):
2577 res = self._download_xml_handle(
2578 ism_url, video_id,
2579 note=note or 'Downloading ISM manifest',
2580 errnote=errnote or 'Failed to download ISM manifest',
2581 fatal=fatal, data=data, headers=headers, query=query)
2582 if res is False:
2583 return []
2584 ism_doc, urlh = res
2585 if ism_doc is None:
2586 return []
2587
2588 return self._parse_ism_formats(ism_doc, urlh.geturl(), ism_id)
2589
2590 def _parse_ism_formats(self, ism_doc, ism_url, ism_id=None):
2591 """
2592 Parse formats from ISM manifest.
2593 References:
2594 1. [MS-SSTR]: Smooth Streaming Protocol,
2595 https://msdn.microsoft.com/en-us/library/ff469518.aspx
2596 """
2597 if ism_doc.get('IsLive') == 'TRUE' or ism_doc.find('Protection') is not None:
2598 return []
2599
2600 duration = int(ism_doc.attrib['Duration'])
2601 timescale = int_or_none(ism_doc.get('TimeScale')) or 10000000
2602
2603 formats = []
2604 for stream in ism_doc.findall('StreamIndex'):
2605 stream_type = stream.get('Type')
2606 if stream_type not in ('video', 'audio'):
2607 continue
2608 url_pattern = stream.attrib['Url']
2609 stream_timescale = int_or_none(stream.get('TimeScale')) or timescale
2610 stream_name = stream.get('Name')
2611 for track in stream.findall('QualityLevel'):
2612 fourcc = track.get('FourCC', 'AACL' if track.get('AudioTag') == '255' else None)
2613 # TODO: add support for WVC1 and WMAP
2614 if fourcc not in ('H264', 'AVC1', 'AACL'):
2615 self.report_warning('%s is not a supported codec' % fourcc)
2616 continue
2617 tbr = int(track.attrib['Bitrate']) // 1000
2618 # [1] does not mention Width and Height attributes. However,
2619 # they're often present while MaxWidth and MaxHeight are
2620 # missing, so should be used as fallbacks
2621 width = int_or_none(track.get('MaxWidth') or track.get('Width'))
2622 height = int_or_none(track.get('MaxHeight') or track.get('Height'))
2623 sampling_rate = int_or_none(track.get('SamplingRate'))
2624
2625 track_url_pattern = re.sub(r'{[Bb]itrate}', track.attrib['Bitrate'], url_pattern)
2626 track_url_pattern = compat_urlparse.urljoin(ism_url, track_url_pattern)
2627
2628 fragments = []
2629 fragment_ctx = {
2630 'time': 0,
2631 }
2632 stream_fragments = stream.findall('c')
2633 for stream_fragment_index, stream_fragment in enumerate(stream_fragments):
2634 fragment_ctx['time'] = int_or_none(stream_fragment.get('t')) or fragment_ctx['time']
2635 fragment_repeat = int_or_none(stream_fragment.get('r')) or 1
2636 fragment_ctx['duration'] = int_or_none(stream_fragment.get('d'))
2637 if not fragment_ctx['duration']:
2638 try:
2639 next_fragment_time = int(stream_fragment[stream_fragment_index + 1].attrib['t'])
2640 except IndexError:
2641 next_fragment_time = duration
2642 fragment_ctx['duration'] = (next_fragment_time - fragment_ctx['time']) / fragment_repeat
2643 for _ in range(fragment_repeat):
2644 fragments.append({
2645 'url': re.sub(r'{start[ _]time}', compat_str(fragment_ctx['time']), track_url_pattern),
2646 'duration': fragment_ctx['duration'] / stream_timescale,
2647 })
2648 fragment_ctx['time'] += fragment_ctx['duration']
2649
2650 format_id = []
2651 if ism_id:
2652 format_id.append(ism_id)
2653 if stream_name:
2654 format_id.append(stream_name)
2655 format_id.append(compat_str(tbr))
2656
2657 formats.append({
2658 'format_id': '-'.join(format_id),
2659 'url': ism_url,
2660 'manifest_url': ism_url,
2661 'ext': 'ismv' if stream_type == 'video' else 'isma',
2662 'width': width,
2663 'height': height,
2664 'tbr': tbr,
2665 'asr': sampling_rate,
2666 'vcodec': 'none' if stream_type == 'audio' else fourcc,
2667 'acodec': 'none' if stream_type == 'video' else fourcc,
2668 'protocol': 'ism',
2669 'fragments': fragments,
2670 '_download_params': {
2671 'duration': duration,
2672 'timescale': stream_timescale,
2673 'width': width or 0,
2674 'height': height or 0,
2675 'fourcc': fourcc,
2676 'codec_private_data': track.get('CodecPrivateData'),
2677 'sampling_rate': sampling_rate,
2678 'channels': int_or_none(track.get('Channels', 2)),
2679 'bits_per_sample': int_or_none(track.get('BitsPerSample', 16)),
2680 'nal_unit_length_field': int_or_none(track.get('NALUnitLengthField', 4)),
2681 },
2682 })
2683 return formats
2684
2685 def _parse_html5_media_entries(self, base_url, webpage, video_id, m3u8_id=None, m3u8_entry_protocol='m3u8', mpd_id=None, preference=None):
2686 def absolute_url(item_url):
2687 return urljoin(base_url, item_url)
2688
2689 def parse_content_type(content_type):
2690 if not content_type:
2691 return {}
2692 ctr = re.search(r'(?P<mimetype>[^/]+/[^;]+)(?:;\s*codecs="?(?P<codecs>[^"]+))?', content_type)
2693 if ctr:
2694 mimetype, codecs = ctr.groups()
2695 f = parse_codecs(codecs)
2696 f['ext'] = mimetype2ext(mimetype)
2697 return f
2698 return {}
2699
2700 def _media_formats(src, cur_media_type, type_info={}):
2701 full_url = absolute_url(src)
2702 ext = type_info.get('ext') or determine_ext(full_url)
2703 if ext == 'm3u8':
2704 is_plain_url = False
2705 formats = self._extract_m3u8_formats(
2706 full_url, video_id, ext='mp4',
2707 entry_protocol=m3u8_entry_protocol, m3u8_id=m3u8_id,
2708 preference=preference, fatal=False)
2709 elif ext == 'mpd':
2710 is_plain_url = False
2711 formats = self._extract_mpd_formats(
2712 full_url, video_id, mpd_id=mpd_id, fatal=False)
2713 else:
2714 is_plain_url = True
2715 formats = [{
2716 'url': full_url,
2717 'vcodec': 'none' if cur_media_type == 'audio' else None,
2718 }]
2719 return is_plain_url, formats
2720
2721 entries = []
2722 # amp-video and amp-audio are very similar to their HTML5 counterparts
2723 # so we wll include them right here (see
2724 # https://www.ampproject.org/docs/reference/components/amp-video)
2725 # For dl8-* tags see https://delight-vr.com/documentation/dl8-video/
2726 _MEDIA_TAG_NAME_RE = r'(?:(?:amp|dl8(?:-live)?)-)?(video|audio)'
2727 media_tags = [(media_tag, media_tag_name, media_type, '')
2728 for media_tag, media_tag_name, media_type
2729 in re.findall(r'(?s)(<(%s)[^>]*/>)' % _MEDIA_TAG_NAME_RE, webpage)]
2730 media_tags.extend(re.findall(
2731 # We only allow video|audio followed by a whitespace or '>'.
2732 # Allowing more characters may end up in significant slow down (see
2733 # https://github.com/ytdl-org/youtube-dl/issues/11979, example URL:
2734 # http://www.porntrex.com/maps/videositemap.xml).
2735 r'(?s)(<(?P<tag>%s)(?:\s+[^>]*)?>)(.*?)</(?P=tag)>' % _MEDIA_TAG_NAME_RE, webpage))
2736 for media_tag, _, media_type, media_content in media_tags:
2737 media_info = {
2738 'formats': [],
2739 'subtitles': {},
2740 }
2741 media_attributes = extract_attributes(media_tag)
2742 src = strip_or_none(media_attributes.get('src'))
2743 if src:
2744 _, formats = _media_formats(src, media_type)
2745 media_info['formats'].extend(formats)
2746 media_info['thumbnail'] = absolute_url(media_attributes.get('poster'))
2747 if media_content:
2748 for source_tag in re.findall(r'<source[^>]+>', media_content):
2749 s_attr = extract_attributes(source_tag)
2750 # data-video-src and data-src are non standard but seen
2751 # several times in the wild
2752 src = strip_or_none(dict_get(s_attr, ('src', 'data-video-src', 'data-src')))
2753 if not src:
2754 continue
2755 f = parse_content_type(s_attr.get('type'))
2756 is_plain_url, formats = _media_formats(src, media_type, f)
2757 if is_plain_url:
2758 # width, height, res, label and title attributes are
2759 # all not standard but seen several times in the wild
2760 labels = [
2761 s_attr.get(lbl)
2762 for lbl in ('label', 'title')
2763 if str_or_none(s_attr.get(lbl))
2764 ]
2765 width = int_or_none(s_attr.get('width'))
2766 height = (int_or_none(s_attr.get('height'))
2767 or int_or_none(s_attr.get('res')))
2768 if not width or not height:
2769 for lbl in labels:
2770 resolution = parse_resolution(lbl)
2771 if not resolution:
2772 continue
2773 width = width or resolution.get('width')
2774 height = height or resolution.get('height')
2775 for lbl in labels:
2776 tbr = parse_bitrate(lbl)
2777 if tbr:
2778 break
2779 else:
2780 tbr = None
2781 f.update({
2782 'width': width,
2783 'height': height,
2784 'tbr': tbr,
2785 'format_id': s_attr.get('label') or s_attr.get('title'),
2786 })
2787 f.update(formats[0])
2788 media_info['formats'].append(f)
2789 else:
2790 media_info['formats'].extend(formats)
2791 for track_tag in re.findall(r'<track[^>]+>', media_content):
2792 track_attributes = extract_attributes(track_tag)
2793 kind = track_attributes.get('kind')
2794 if not kind or kind in ('subtitles', 'captions'):
2795 src = strip_or_none(track_attributes.get('src'))
2796 if not src:
2797 continue
2798 lang = track_attributes.get('srclang') or track_attributes.get('lang') or track_attributes.get('label')
2799 media_info['subtitles'].setdefault(lang, []).append({
2800 'url': absolute_url(src),
2801 })
2802 for f in media_info['formats']:
2803 f.setdefault('http_headers', {})['Referer'] = base_url
2804 if media_info['formats'] or media_info['subtitles']:
2805 entries.append(media_info)
2806 return entries
2807
2808 def _extract_akamai_formats(self, manifest_url, video_id, hosts={}):
2809 signed = 'hdnea=' in manifest_url
2810 if not signed:
2811 # https://learn.akamai.com/en-us/webhelp/media-services-on-demand/stream-packaging-user-guide/GUID-BE6C0F73-1E06-483B-B0EA-57984B91B7F9.html
2812 manifest_url = re.sub(
2813 r'(?:b=[\d,-]+|(?:__a__|attributes)=off|__b__=\d+)&?',
2814 '', manifest_url).strip('?')
2815
2816 formats = []
2817
2818 hdcore_sign = 'hdcore=3.7.0'
2819 f4m_url = re.sub(r'(https?://[^/]+)/i/', r'\1/z/', manifest_url).replace('/master.m3u8', '/manifest.f4m')
2820 hds_host = hosts.get('hds')
2821 if hds_host:
2822 f4m_url = re.sub(r'(https?://)[^/]+', r'\1' + hds_host, f4m_url)
2823 if 'hdcore=' not in f4m_url:
2824 f4m_url += ('&' if '?' in f4m_url else '?') + hdcore_sign
2825 f4m_formats = self._extract_f4m_formats(
2826 f4m_url, video_id, f4m_id='hds', fatal=False)
2827 for entry in f4m_formats:
2828 entry.update({'extra_param_to_segment_url': hdcore_sign})
2829 formats.extend(f4m_formats)
2830
2831 m3u8_url = re.sub(r'(https?://[^/]+)/z/', r'\1/i/', manifest_url).replace('/manifest.f4m', '/master.m3u8')
2832 hls_host = hosts.get('hls')
2833 if hls_host:
2834 m3u8_url = re.sub(r'(https?://)[^/]+', r'\1' + hls_host, m3u8_url)
2835 m3u8_formats = self._extract_m3u8_formats(
2836 m3u8_url, video_id, 'mp4', 'm3u8_native',
2837 m3u8_id='hls', fatal=False)
2838 formats.extend(m3u8_formats)
2839
2840 http_host = hosts.get('http')
2841 if http_host and m3u8_formats and not signed:
2842 REPL_REGEX = r'https?://[^/]+/i/([^,]+),([^/]+),([^/]+)\.csmil/.+'
2843 qualities = re.match(REPL_REGEX, m3u8_url).group(2).split(',')
2844 qualities_length = len(qualities)
2845 if len(m3u8_formats) in (qualities_length, qualities_length + 1):
2846 i = 0
2847 for f in m3u8_formats:
2848 if f['vcodec'] != 'none':
2849 for protocol in ('http', 'https'):
2850 http_f = f.copy()
2851 del http_f['manifest_url']
2852 http_url = re.sub(
2853 REPL_REGEX, protocol + r'://%s/\g<1>%s\3' % (http_host, qualities[i]), f['url'])
2854 http_f.update({
2855 'format_id': http_f['format_id'].replace('hls-', protocol + '-'),
2856 'url': http_url,
2857 'protocol': protocol,
2858 })
2859 formats.append(http_f)
2860 i += 1
2861
2862 return formats
2863
2864 def _extract_wowza_formats(self, url, video_id, m3u8_entry_protocol='m3u8_native', skip_protocols=[]):
2865 query = compat_urlparse.urlparse(url).query
2866 url = re.sub(r'/(?:manifest|playlist|jwplayer)\.(?:m3u8|f4m|mpd|smil)', '', url)
2867 mobj = re.search(
2868 r'(?:(?:http|rtmp|rtsp)(?P<s>s)?:)?(?P<url>//[^?]+)', url)
2869 url_base = mobj.group('url')
2870 http_base_url = '%s%s:%s' % ('http', mobj.group('s') or '', url_base)
2871 formats = []
2872
2873 def manifest_url(manifest):
2874 m_url = '%s/%s' % (http_base_url, manifest)
2875 if query:
2876 m_url += '?%s' % query
2877 return m_url
2878
2879 if 'm3u8' not in skip_protocols:
2880 formats.extend(self._extract_m3u8_formats(
2881 manifest_url('playlist.m3u8'), video_id, 'mp4',
2882 m3u8_entry_protocol, m3u8_id='hls', fatal=False))
2883 if 'f4m' not in skip_protocols:
2884 formats.extend(self._extract_f4m_formats(
2885 manifest_url('manifest.f4m'),
2886 video_id, f4m_id='hds', fatal=False))
2887 if 'dash' not in skip_protocols:
2888 formats.extend(self._extract_mpd_formats(
2889 manifest_url('manifest.mpd'),
2890 video_id, mpd_id='dash', fatal=False))
2891 if re.search(r'(?:/smil:|\.smil)', url_base):
2892 if 'smil' not in skip_protocols:
2893 rtmp_formats = self._extract_smil_formats(
2894 manifest_url('jwplayer.smil'),
2895 video_id, fatal=False)
2896 for rtmp_format in rtmp_formats:
2897 rtsp_format = rtmp_format.copy()
2898 rtsp_format['url'] = '%s/%s' % (rtmp_format['url'], rtmp_format['play_path'])
2899 del rtsp_format['play_path']
2900 del rtsp_format['ext']
2901 rtsp_format.update({
2902 'url': rtsp_format['url'].replace('rtmp://', 'rtsp://'),
2903 'format_id': rtmp_format['format_id'].replace('rtmp', 'rtsp'),
2904 'protocol': 'rtsp',
2905 })
2906 formats.extend([rtmp_format, rtsp_format])
2907 else:
2908 for protocol in ('rtmp', 'rtsp'):
2909 if protocol not in skip_protocols:
2910 formats.append({
2911 'url': '%s:%s' % (protocol, url_base),
2912 'format_id': protocol,
2913 'protocol': protocol,
2914 })
2915 return formats
2916
2917 def _find_jwplayer_data(self, webpage, video_id=None, transform_source=js_to_json):
2918 mobj = re.search(
2919 r'(?s)jwplayer\((?P<quote>[\'"])[^\'" ]+(?P=quote)\)(?!</script>).*?\.setup\s*\((?P<options>[^)]+)\)',
2920 webpage)
2921 if mobj:
2922 try:
2923 jwplayer_data = self._parse_json(mobj.group('options'),
2924 video_id=video_id,
2925 transform_source=transform_source)
2926 except ExtractorError:
2927 pass
2928 else:
2929 if isinstance(jwplayer_data, dict):
2930 return jwplayer_data
2931
2932 def _extract_jwplayer_data(self, webpage, video_id, *args, **kwargs):
2933 jwplayer_data = self._find_jwplayer_data(
2934 webpage, video_id, transform_source=js_to_json)
2935 return self._parse_jwplayer_data(
2936 jwplayer_data, video_id, *args, **kwargs)
2937
2938 def _parse_jwplayer_data(self, jwplayer_data, video_id=None, require_title=True,
2939 m3u8_id=None, mpd_id=None, rtmp_params=None, base_url=None):
2940 # JWPlayer backward compatibility: flattened playlists
2941 # https://github.com/jwplayer/jwplayer/blob/v7.4.3/src/js/api/config.js#L81-L96
2942 if 'playlist' not in jwplayer_data:
2943 jwplayer_data = {'playlist': [jwplayer_data]}
2944
2945 entries = []
2946
2947 # JWPlayer backward compatibility: single playlist item
2948 # https://github.com/jwplayer/jwplayer/blob/v7.7.0/src/js/playlist/playlist.js#L10
2949 if not isinstance(jwplayer_data['playlist'], list):
2950 jwplayer_data['playlist'] = [jwplayer_data['playlist']]
2951
2952 for video_data in jwplayer_data['playlist']:
2953 # JWPlayer backward compatibility: flattened sources
2954 # https://github.com/jwplayer/jwplayer/blob/v7.4.3/src/js/playlist/item.js#L29-L35
2955 if 'sources' not in video_data:
2956 video_data['sources'] = [video_data]
2957
2958 this_video_id = video_id or video_data['mediaid']
2959
2960 formats = self._parse_jwplayer_formats(
2961 video_data['sources'], video_id=this_video_id, m3u8_id=m3u8_id,
2962 mpd_id=mpd_id, rtmp_params=rtmp_params, base_url=base_url)
2963
2964 subtitles = {}
2965 tracks = video_data.get('tracks')
2966 if tracks and isinstance(tracks, list):
2967 for track in tracks:
2968 if not isinstance(track, dict):
2969 continue
2970 track_kind = track.get('kind')
2971 if not track_kind or not isinstance(track_kind, compat_str):
2972 continue
2973 if track_kind.lower() not in ('captions', 'subtitles'):
2974 continue
2975 track_url = urljoin(base_url, track.get('file'))
2976 if not track_url:
2977 continue
2978 subtitles.setdefault(track.get('label') or 'en', []).append({
2979 'url': self._proto_relative_url(track_url)
2980 })
2981
2982 entry = {
2983 'id': this_video_id,
2984 'title': unescapeHTML(video_data['title'] if require_title else video_data.get('title')),
2985 'description': clean_html(video_data.get('description')),
2986 'thumbnail': urljoin(base_url, self._proto_relative_url(video_data.get('image'))),
2987 'timestamp': int_or_none(video_data.get('pubdate')),
2988 'duration': float_or_none(jwplayer_data.get('duration') or video_data.get('duration')),
2989 'subtitles': subtitles,
2990 }
2991 # https://github.com/jwplayer/jwplayer/blob/master/src/js/utils/validator.js#L32
2992 if len(formats) == 1 and re.search(r'^(?:http|//).*(?:youtube\.com|youtu\.be)/.+', formats[0]['url']):
2993 entry.update({
2994 '_type': 'url_transparent',
2995 'url': formats[0]['url'],
2996 })
2997 else:
2998 self._sort_formats(formats)
2999 entry['formats'] = formats
3000 entries.append(entry)
3001 if len(entries) == 1:
3002 return entries[0]
3003 else:
3004 return self.playlist_result(entries)
3005
3006 def _parse_jwplayer_formats(self, jwplayer_sources_data, video_id=None,
3007 m3u8_id=None, mpd_id=None, rtmp_params=None, base_url=None):
3008 urls = []
3009 formats = []
3010 for source in jwplayer_sources_data:
3011 if not isinstance(source, dict):
3012 continue
3013 source_url = urljoin(
3014 base_url, self._proto_relative_url(source.get('file')))
3015 if not source_url or source_url in urls:
3016 continue
3017 urls.append(source_url)
3018 source_type = source.get('type') or ''
3019 ext = mimetype2ext(source_type) or determine_ext(source_url)
3020 if source_type == 'hls' or ext == 'm3u8':
3021 formats.extend(self._extract_m3u8_formats(
3022 source_url, video_id, 'mp4', entry_protocol='m3u8_native',
3023 m3u8_id=m3u8_id, fatal=False))
3024 elif source_type == 'dash' or ext == 'mpd':
3025 formats.extend(self._extract_mpd_formats(
3026 source_url, video_id, mpd_id=mpd_id, fatal=False))
3027 elif ext == 'smil':
3028 formats.extend(self._extract_smil_formats(
3029 source_url, video_id, fatal=False))
3030 # https://github.com/jwplayer/jwplayer/blob/master/src/js/providers/default.js#L67
3031 elif source_type.startswith('audio') or ext in (
3032 'oga', 'aac', 'mp3', 'mpeg', 'vorbis'):
3033 formats.append({
3034 'url': source_url,
3035 'vcodec': 'none',
3036 'ext': ext,
3037 })
3038 else:
3039 height = int_or_none(source.get('height'))
3040 if height is None:
3041 # Often no height is provided but there is a label in
3042 # format like "1080p", "720p SD", or 1080.
3043 height = int_or_none(self._search_regex(
3044 r'^(\d{3,4})[pP]?(?:\b|$)', compat_str(source.get('label') or ''),
3045 'height', default=None))
3046 a_format = {
3047 'url': source_url,
3048 'width': int_or_none(source.get('width')),
3049 'height': height,
3050 'tbr': int_or_none(source.get('bitrate')),
3051 'ext': ext,
3052 }
3053 if source_url.startswith('rtmp'):
3054 a_format['ext'] = 'flv'
3055 # See com/longtailvideo/jwplayer/media/RTMPMediaProvider.as
3056 # of jwplayer.flash.swf
3057 rtmp_url_parts = re.split(
3058 r'((?:mp4|mp3|flv):)', source_url, 1)
3059 if len(rtmp_url_parts) == 3:
3060 rtmp_url, prefix, play_path = rtmp_url_parts
3061 a_format.update({
3062 'url': rtmp_url,
3063 'play_path': prefix + play_path,
3064 })
3065 if rtmp_params:
3066 a_format.update(rtmp_params)
3067 formats.append(a_format)
3068 return formats
3069
3070 def _live_title(self, name):
3071 """ Generate the title for a live video """
3072 now = datetime.datetime.now()
3073 now_str = now.strftime('%Y-%m-%d %H:%M')
3074 return name + ' ' + now_str
3075
3076 def _int(self, v, name, fatal=False, **kwargs):
3077 res = int_or_none(v, **kwargs)
3078 if 'get_attr' in kwargs:
3079 print(getattr(v, kwargs['get_attr']))
3080 if res is None:
3081 msg = 'Failed to extract %s: Could not parse value %r' % (name, v)
3082 if fatal:
3083 raise ExtractorError(msg)
3084 else:
3085 self._downloader.report_warning(msg)
3086 return res
3087
3088 def _float(self, v, name, fatal=False, **kwargs):
3089 res = float_or_none(v, **kwargs)
3090 if res is None:
3091 msg = 'Failed to extract %s: Could not parse value %r' % (name, v)
3092 if fatal:
3093 raise ExtractorError(msg)
3094 else:
3095 self._downloader.report_warning(msg)
3096 return res
3097
3098 def _set_cookie(self, domain, name, value, expire_time=None, port=None,
3099 path='/', secure=False, discard=False, rest={}, **kwargs):
3100 cookie = compat_cookiejar_Cookie(
3101 0, name, value, port, port is not None, domain, True,
3102 domain.startswith('.'), path, True, secure, expire_time,
3103 discard, None, None, rest)
3104 self._downloader.cookiejar.set_cookie(cookie)
3105
3106 def _get_cookies(self, url):
3107 """ Return a compat_cookies.SimpleCookie with the cookies for the url """
3108 req = sanitized_Request(url)
3109 self._downloader.cookiejar.add_cookie_header(req)
3110 return compat_cookies.SimpleCookie(req.get_header('Cookie'))
3111
3112 def _apply_first_set_cookie_header(self, url_handle, cookie):
3113 """
3114 Apply first Set-Cookie header instead of the last. Experimental.
3115
3116 Some sites (e.g. [1-3]) may serve two cookies under the same name
3117 in Set-Cookie header and expect the first (old) one to be set rather
3118 than second (new). However, as of RFC6265 the newer one cookie
3119 should be set into cookie store what actually happens.
3120 We will workaround this issue by resetting the cookie to
3121 the first one manually.
3122 1. https://new.vk.com/
3123 2. https://github.com/ytdl-org/youtube-dl/issues/9841#issuecomment-227871201
3124 3. https://learning.oreilly.com/
3125 """
3126 for header, cookies in url_handle.headers.items():
3127 if header.lower() != 'set-cookie':
3128 continue
3129 if sys.version_info[0] >= 3:
3130 cookies = cookies.encode('iso-8859-1')
3131 cookies = cookies.decode('utf-8')
3132 cookie_value = re.search(
3133 r'%s=(.+?);.*?\b[Dd]omain=(.+?)(?:[,;]|$)' % cookie, cookies)
3134 if cookie_value:
3135 value, domain = cookie_value.groups()
3136 self._set_cookie(domain, cookie, value)
3137 break
3138
3139 def get_testcases(self, include_onlymatching=False):
3140 t = getattr(self, '_TEST', None)
3141 if t:
3142 assert not hasattr(self, '_TESTS'), \
3143 '%s has _TEST and _TESTS' % type(self).__name__
3144 tests = [t]
3145 else:
3146 tests = getattr(self, '_TESTS', [])
3147 for t in tests:
3148 if not include_onlymatching and t.get('only_matching', False):
3149 continue
3150 t['name'] = type(self).__name__[:-len('IE')]
3151 yield t
3152
3153 def is_suitable(self, age_limit):
3154 """ Test whether the extractor is generally suitable for the given
3155 age limit (i.e. pornographic sites are not, all others usually are) """
3156
3157 any_restricted = False
3158 for tc in self.get_testcases(include_onlymatching=False):
3159 if tc.get('playlist', []):
3160 tc = tc['playlist'][0]
3161 is_restricted = age_restricted(
3162 tc.get('info_dict', {}).get('age_limit'), age_limit)
3163 if not is_restricted:
3164 return True
3165 any_restricted = any_restricted or is_restricted
3166 return not any_restricted
3167
3168 def extract_subtitles(self, *args, **kwargs):
3169 if (self._downloader.params.get('writesubtitles', False)
3170 or self._downloader.params.get('listsubtitles')):
3171 return self._get_subtitles(*args, **kwargs)
3172 return {}
3173
3174 def _get_subtitles(self, *args, **kwargs):
3175 raise NotImplementedError('This method must be implemented by subclasses')
3176
3177 @staticmethod
3178 def _merge_subtitle_items(subtitle_list1, subtitle_list2):
3179 """ Merge subtitle items for one language. Items with duplicated URLs
3180 will be dropped. """
3181 list1_urls = set([item['url'] for item in subtitle_list1])
3182 ret = list(subtitle_list1)
3183 ret.extend([item for item in subtitle_list2 if item['url'] not in list1_urls])
3184 return ret
3185
3186 @classmethod
3187 def _merge_subtitles(cls, subtitle_dict1, subtitle_dict2):
3188 """ Merge two subtitle dictionaries, language by language. """
3189 ret = dict(subtitle_dict1)
3190 for lang in subtitle_dict2:
3191 ret[lang] = cls._merge_subtitle_items(subtitle_dict1.get(lang, []), subtitle_dict2[lang])
3192 return ret
3193
3194 def extract_automatic_captions(self, *args, **kwargs):
3195 if (self._downloader.params.get('writeautomaticsub', False)
3196 or self._downloader.params.get('listsubtitles')):
3197 return self._get_automatic_captions(*args, **kwargs)
3198 return {}
3199
3200 def _get_automatic_captions(self, *args, **kwargs):
3201 raise NotImplementedError('This method must be implemented by subclasses')
3202
3203 def mark_watched(self, *args, **kwargs):
3204 if (self._downloader.params.get('mark_watched', False)
3205 and (self._get_login_info()[0] is not None
3206 or self._downloader.params.get('cookiefile') is not None)):
3207 self._mark_watched(*args, **kwargs)
3208
3209 def _mark_watched(self, *args, **kwargs):
3210 raise NotImplementedError('This method must be implemented by subclasses')
3211
3212 def geo_verification_headers(self):
3213 headers = {}
3214 geo_verification_proxy = self._downloader.params.get('geo_verification_proxy')
3215 if geo_verification_proxy:
3216 headers['Ytdl-request-proxy'] = geo_verification_proxy
3217 return headers
3218
3219 def _generic_id(self, url):
3220 return compat_urllib_parse_unquote(os.path.splitext(url.rstrip('/').split('/')[-1])[0])
3221
3222 def _generic_title(self, url):
3223 return compat_urllib_parse_unquote(os.path.splitext(url_basename(url))[0])
3224
3225
3226 class SearchInfoExtractor(InfoExtractor):
3227 """
3228 Base class for paged search queries extractors.
3229 They accept URLs in the format _SEARCH_KEY(|all|[0-9]):{query}
3230 Instances should define _SEARCH_KEY and _MAX_RESULTS.
3231 """
3232
3233 @classmethod
3234 def _make_valid_url(cls):
3235 return r'%s(?P<prefix>|[1-9][0-9]*|all):(?P<query>[\s\S]+)' % cls._SEARCH_KEY
3236
3237 @classmethod
3238 def suitable(cls, url):
3239 return re.match(cls._make_valid_url(), url) is not None
3240
3241 def _real_extract(self, query):
3242 mobj = re.match(self._make_valid_url(), query)
3243 if mobj is None:
3244 raise ExtractorError('Invalid search query "%s"' % query)
3245
3246 prefix = mobj.group('prefix')
3247 query = mobj.group('query')
3248 if prefix == '':
3249 return self._get_n_results(query, 1)
3250 elif prefix == 'all':
3251 return self._get_n_results(query, self._MAX_RESULTS)
3252 else:
3253 n = int(prefix)
3254 if n <= 0:
3255 raise ExtractorError('invalid download number %s for query "%s"' % (n, query))
3256 elif n > self._MAX_RESULTS:
3257 self._downloader.report_warning('%s returns max %i results (you requested %i)' % (self._SEARCH_KEY, self._MAX_RESULTS, n))
3258 n = self._MAX_RESULTS
3259 return self._get_n_results(query, n)
3260
3261 def _get_n_results(self, query, n):
3262 """Get a specified number of results for a query"""
3263 raise NotImplementedError('This method must be implemented by subclasses')
3264
3265 @property
3266 def SEARCH_KEY(self):
3267 return self._SEARCH_KEY