]> git.bitcoin.ninja Git - youtube-dl/blob - youtube_dl/extractor/common.py
[amcnetworks] fix typo
[youtube-dl] / youtube_dl / extractor / common.py
1 from __future__ import unicode_literals
2
3 import base64
4 import datetime
5 import hashlib
6 import json
7 import netrc
8 import os
9 import re
10 import socket
11 import sys
12 import time
13 import math
14
15 from ..compat import (
16     compat_cookiejar,
17     compat_cookies,
18     compat_etree_fromstring,
19     compat_getpass,
20     compat_http_client,
21     compat_os_name,
22     compat_str,
23     compat_urllib_error,
24     compat_urllib_parse_urlencode,
25     compat_urllib_request,
26     compat_urlparse,
27 )
28 from ..downloader.f4m import remove_encrypted_media
29 from ..utils import (
30     NO_DEFAULT,
31     age_restricted,
32     bug_reports_message,
33     clean_html,
34     compiled_regex_type,
35     determine_ext,
36     error_to_compat_str,
37     ExtractorError,
38     fix_xml_ampersands,
39     float_or_none,
40     int_or_none,
41     parse_iso8601,
42     RegexNotFoundError,
43     sanitize_filename,
44     sanitized_Request,
45     unescapeHTML,
46     unified_strdate,
47     unified_timestamp,
48     url_basename,
49     xpath_element,
50     xpath_text,
51     xpath_with_ns,
52     determine_protocol,
53     parse_duration,
54     mimetype2ext,
55     update_Request,
56     update_url_query,
57     parse_m3u8_attributes,
58     extract_attributes,
59     parse_codecs,
60 )
61
62
63 class InfoExtractor(object):
64     """Information Extractor class.
65
66     Information extractors are the classes that, given a URL, extract
67     information about the video (or videos) the URL refers to. This
68     information includes the real video URL, the video title, author and
69     others. The information is stored in a dictionary which is then
70     passed to the YoutubeDL. The YoutubeDL processes this
71     information possibly downloading the video to the file system, among
72     other possible outcomes.
73
74     The type field determines the type of the result.
75     By far the most common value (and the default if _type is missing) is
76     "video", which indicates a single video.
77
78     For a video, the dictionaries must include the following fields:
79
80     id:             Video identifier.
81     title:          Video title, unescaped.
82
83     Additionally, it must contain either a formats entry or a url one:
84
85     formats:        A list of dictionaries for each format available, ordered
86                     from worst to best quality.
87
88                     Potential fields:
89                     * url        Mandatory. The URL of the video file
90                     * ext        Will be calculated from URL if missing
91                     * format     A human-readable description of the format
92                                  ("mp4 container with h264/opus").
93                                  Calculated from the format_id, width, height.
94                                  and format_note fields if missing.
95                     * format_id  A short description of the format
96                                  ("mp4_h264_opus" or "19").
97                                 Technically optional, but strongly recommended.
98                     * format_note Additional info about the format
99                                  ("3D" or "DASH video")
100                     * width      Width of the video, if known
101                     * height     Height of the video, if known
102                     * resolution Textual description of width and height
103                     * tbr        Average bitrate of audio and video in KBit/s
104                     * abr        Average audio bitrate in KBit/s
105                     * acodec     Name of the audio codec in use
106                     * asr        Audio sampling rate in Hertz
107                     * vbr        Average video bitrate in KBit/s
108                     * fps        Frame rate
109                     * vcodec     Name of the video codec in use
110                     * container  Name of the container format
111                     * filesize   The number of bytes, if known in advance
112                     * filesize_approx  An estimate for the number of bytes
113                     * player_url SWF Player URL (used for rtmpdump).
114                     * protocol   The protocol that will be used for the actual
115                                  download, lower-case.
116                                  "http", "https", "rtsp", "rtmp", "rtmpe",
117                                  "m3u8", "m3u8_native" or "http_dash_segments".
118                     * preference Order number of this format. If this field is
119                                  present and not None, the formats get sorted
120                                  by this field, regardless of all other values.
121                                  -1 for default (order by other properties),
122                                  -2 or smaller for less than default.
123                                  < -1000 to hide the format (if there is
124                                     another one which is strictly better)
125                     * language   Language code, e.g. "de" or "en-US".
126                     * language_preference  Is this in the language mentioned in
127                                  the URL?
128                                  10 if it's what the URL is about,
129                                  -1 for default (don't know),
130                                  -10 otherwise, other values reserved for now.
131                     * quality    Order number of the video quality of this
132                                  format, irrespective of the file format.
133                                  -1 for default (order by other properties),
134                                  -2 or smaller for less than default.
135                     * source_preference  Order number for this video source
136                                   (quality takes higher priority)
137                                  -1 for default (order by other properties),
138                                  -2 or smaller for less than default.
139                     * http_headers  A dictionary of additional HTTP headers
140                                  to add to the request.
141                     * stretched_ratio  If given and not 1, indicates that the
142                                  video's pixels are not square.
143                                  width : height ratio as float.
144                     * no_resume  The server does not support resuming the
145                                  (HTTP or RTMP) download. Boolean.
146
147     url:            Final video URL.
148     ext:            Video filename extension.
149     format:         The video format, defaults to ext (used for --get-format)
150     player_url:     SWF Player URL (used for rtmpdump).
151
152     The following fields are optional:
153
154     alt_title:      A secondary title of the video.
155     display_id      An alternative identifier for the video, not necessarily
156                     unique, but available before title. Typically, id is
157                     something like "4234987", title "Dancing naked mole rats",
158                     and display_id "dancing-naked-mole-rats"
159     thumbnails:     A list of dictionaries, with the following entries:
160                         * "id" (optional, string) - Thumbnail format ID
161                         * "url"
162                         * "preference" (optional, int) - quality of the image
163                         * "width" (optional, int)
164                         * "height" (optional, int)
165                         * "resolution" (optional, string "{width}x{height"},
166                                         deprecated)
167                         * "filesize" (optional, int)
168     thumbnail:      Full URL to a video thumbnail image.
169     description:    Full video description.
170     uploader:       Full name of the video uploader.
171     license:        License name the video is licensed under.
172     creator:        The creator of the video.
173     release_date:   The date (YYYYMMDD) when the video was released.
174     timestamp:      UNIX timestamp of the moment the video became available.
175     upload_date:    Video upload date (YYYYMMDD).
176                     If not explicitly set, calculated from timestamp.
177     uploader_id:    Nickname or id of the video uploader.
178     uploader_url:   Full URL to a personal webpage of the video uploader.
179     location:       Physical location where the video was filmed.
180     subtitles:      The available subtitles as a dictionary in the format
181                     {language: subformats}. "subformats" is a list sorted from
182                     lower to higher preference, each element is a dictionary
183                     with the "ext" entry and one of:
184                         * "data": The subtitles file contents
185                         * "url": A URL pointing to the subtitles file
186                     "ext" will be calculated from URL if missing
187     automatic_captions: Like 'subtitles', used by the YoutubeIE for
188                     automatically generated captions
189     duration:       Length of the video in seconds, as an integer or float.
190     view_count:     How many users have watched the video on the platform.
191     like_count:     Number of positive ratings of the video
192     dislike_count:  Number of negative ratings of the video
193     repost_count:   Number of reposts of the video
194     average_rating: Average rating give by users, the scale used depends on the webpage
195     comment_count:  Number of comments on the video
196     comments:       A list of comments, each with one or more of the following
197                     properties (all but one of text or html optional):
198                         * "author" - human-readable name of the comment author
199                         * "author_id" - user ID of the comment author
200                         * "id" - Comment ID
201                         * "html" - Comment as HTML
202                         * "text" - Plain text of the comment
203                         * "timestamp" - UNIX timestamp of comment
204                         * "parent" - ID of the comment this one is replying to.
205                                      Set to "root" to indicate that this is a
206                                      comment to the original video.
207     age_limit:      Age restriction for the video, as an integer (years)
208     webpage_url:    The URL to the video webpage, if given to youtube-dl it
209                     should allow to get the same result again. (It will be set
210                     by YoutubeDL if it's missing)
211     categories:     A list of categories that the video falls in, for example
212                     ["Sports", "Berlin"]
213     tags:           A list of tags assigned to the video, e.g. ["sweden", "pop music"]
214     is_live:        True, False, or None (=unknown). Whether this video is a
215                     live stream that goes on instead of a fixed-length video.
216     start_time:     Time in seconds where the reproduction should start, as
217                     specified in the URL.
218     end_time:       Time in seconds where the reproduction should end, as
219                     specified in the URL.
220
221     The following fields should only be used when the video belongs to some logical
222     chapter or section:
223
224     chapter:        Name or title of the chapter the video belongs to.
225     chapter_number: Number of the chapter the video belongs to, as an integer.
226     chapter_id:     Id of the chapter the video belongs to, as a unicode string.
227
228     The following fields should only be used when the video is an episode of some
229     series or programme:
230
231     series:         Title of the series or programme the video episode belongs to.
232     season:         Title of the season the video episode belongs to.
233     season_number:  Number of the season the video episode belongs to, as an integer.
234     season_id:      Id of the season the video episode belongs to, as a unicode string.
235     episode:        Title of the video episode. Unlike mandatory video title field,
236                     this field should denote the exact title of the video episode
237                     without any kind of decoration.
238     episode_number: Number of the video episode within a season, as an integer.
239     episode_id:     Id of the video episode, as a unicode string.
240
241     The following fields should only be used when the media is a track or a part of
242     a music album:
243
244     track:          Title of the track.
245     track_number:   Number of the track within an album or a disc, as an integer.
246     track_id:       Id of the track (useful in case of custom indexing, e.g. 6.iii),
247                     as a unicode string.
248     artist:         Artist(s) of the track.
249     genre:          Genre(s) of the track.
250     album:          Title of the album the track belongs to.
251     album_type:     Type of the album (e.g. "Demo", "Full-length", "Split", "Compilation", etc).
252     album_artist:   List of all artists appeared on the album (e.g.
253                     "Ash Borer / Fell Voices" or "Various Artists", useful for splits
254                     and compilations).
255     disc_number:    Number of the disc or other physical medium the track belongs to,
256                     as an integer.
257     release_year:   Year (YYYY) when the album was released.
258
259     Unless mentioned otherwise, the fields should be Unicode strings.
260
261     Unless mentioned otherwise, None is equivalent to absence of information.
262
263
264     _type "playlist" indicates multiple videos.
265     There must be a key "entries", which is a list, an iterable, or a PagedList
266     object, each element of which is a valid dictionary by this specification.
267
268     Additionally, playlists can have "title", "description" and "id" attributes
269     with the same semantics as videos (see above).
270
271
272     _type "multi_video" indicates that there are multiple videos that
273     form a single show, for examples multiple acts of an opera or TV episode.
274     It must have an entries key like a playlist and contain all the keys
275     required for a video at the same time.
276
277
278     _type "url" indicates that the video must be extracted from another
279     location, possibly by a different extractor. Its only required key is:
280     "url" - the next URL to extract.
281     The key "ie_key" can be set to the class name (minus the trailing "IE",
282     e.g. "Youtube") if the extractor class is known in advance.
283     Additionally, the dictionary may have any properties of the resolved entity
284     known in advance, for example "title" if the title of the referred video is
285     known ahead of time.
286
287
288     _type "url_transparent" entities have the same specification as "url", but
289     indicate that the given additional information is more precise than the one
290     associated with the resolved URL.
291     This is useful when a site employs a video service that hosts the video and
292     its technical metadata, but that video service does not embed a useful
293     title, description etc.
294
295
296     Subclasses of this one should re-define the _real_initialize() and
297     _real_extract() methods and define a _VALID_URL regexp.
298     Probably, they should also be added to the list of extractors.
299
300     Finally, the _WORKING attribute should be set to False for broken IEs
301     in order to warn the users and skip the tests.
302     """
303
304     _ready = False
305     _downloader = None
306     _WORKING = True
307
308     def __init__(self, downloader=None):
309         """Constructor. Receives an optional downloader."""
310         self._ready = False
311         self.set_downloader(downloader)
312
313     @classmethod
314     def suitable(cls, url):
315         """Receives a URL and returns True if suitable for this IE."""
316
317         # This does not use has/getattr intentionally - we want to know whether
318         # we have cached the regexp for *this* class, whereas getattr would also
319         # match the superclass
320         if '_VALID_URL_RE' not in cls.__dict__:
321             cls._VALID_URL_RE = re.compile(cls._VALID_URL)
322         return cls._VALID_URL_RE.match(url) is not None
323
324     @classmethod
325     def _match_id(cls, url):
326         if '_VALID_URL_RE' not in cls.__dict__:
327             cls._VALID_URL_RE = re.compile(cls._VALID_URL)
328         m = cls._VALID_URL_RE.match(url)
329         assert m
330         return m.group('id')
331
332     @classmethod
333     def working(cls):
334         """Getter method for _WORKING."""
335         return cls._WORKING
336
337     def initialize(self):
338         """Initializes an instance (authentication, etc)."""
339         if not self._ready:
340             self._real_initialize()
341             self._ready = True
342
343     def extract(self, url):
344         """Extracts URL information and returns it in list of dicts."""
345         try:
346             self.initialize()
347             return self._real_extract(url)
348         except ExtractorError:
349             raise
350         except compat_http_client.IncompleteRead as e:
351             raise ExtractorError('A network error has occurred.', cause=e, expected=True)
352         except (KeyError, StopIteration) as e:
353             raise ExtractorError('An extractor error has occurred.', cause=e)
354
355     def set_downloader(self, downloader):
356         """Sets the downloader for this IE."""
357         self._downloader = downloader
358
359     def _real_initialize(self):
360         """Real initialization process. Redefine in subclasses."""
361         pass
362
363     def _real_extract(self, url):
364         """Real extraction process. Redefine in subclasses."""
365         pass
366
367     @classmethod
368     def ie_key(cls):
369         """A string for getting the InfoExtractor with get_info_extractor"""
370         return compat_str(cls.__name__[:-2])
371
372     @property
373     def IE_NAME(self):
374         return compat_str(type(self).__name__[:-2])
375
376     def _request_webpage(self, url_or_request, video_id, note=None, errnote=None, fatal=True, data=None, headers={}, query={}):
377         """ Returns the response handle """
378         if note is None:
379             self.report_download_webpage(video_id)
380         elif note is not False:
381             if video_id is None:
382                 self.to_screen('%s' % (note,))
383             else:
384                 self.to_screen('%s: %s' % (video_id, note))
385         if isinstance(url_or_request, compat_urllib_request.Request):
386             url_or_request = update_Request(
387                 url_or_request, data=data, headers=headers, query=query)
388         else:
389             if query:
390                 url_or_request = update_url_query(url_or_request, query)
391             if data is not None or headers:
392                 url_or_request = sanitized_Request(url_or_request, data, headers)
393         try:
394             return self._downloader.urlopen(url_or_request)
395         except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
396             if errnote is False:
397                 return False
398             if errnote is None:
399                 errnote = 'Unable to download webpage'
400
401             errmsg = '%s: %s' % (errnote, error_to_compat_str(err))
402             if fatal:
403                 raise ExtractorError(errmsg, sys.exc_info()[2], cause=err)
404             else:
405                 self._downloader.report_warning(errmsg)
406                 return False
407
408     def _download_webpage_handle(self, url_or_request, video_id, note=None, errnote=None, fatal=True, encoding=None, data=None, headers={}, query={}):
409         """ Returns a tuple (page content as string, URL handle) """
410         # Strip hashes from the URL (#1038)
411         if isinstance(url_or_request, (compat_str, str)):
412             url_or_request = url_or_request.partition('#')[0]
413
414         urlh = self._request_webpage(url_or_request, video_id, note, errnote, fatal, data=data, headers=headers, query=query)
415         if urlh is False:
416             assert not fatal
417             return False
418         content = self._webpage_read_content(urlh, url_or_request, video_id, note, errnote, fatal, encoding=encoding)
419         return (content, urlh)
420
421     @staticmethod
422     def _guess_encoding_from_content(content_type, webpage_bytes):
423         m = re.match(r'[a-zA-Z0-9_.-]+/[a-zA-Z0-9_.-]+\s*;\s*charset=(.+)', content_type)
424         if m:
425             encoding = m.group(1)
426         else:
427             m = re.search(br'<meta[^>]+charset=[\'"]?([^\'")]+)[ /\'">]',
428                           webpage_bytes[:1024])
429             if m:
430                 encoding = m.group(1).decode('ascii')
431             elif webpage_bytes.startswith(b'\xff\xfe'):
432                 encoding = 'utf-16'
433             else:
434                 encoding = 'utf-8'
435
436         return encoding
437
438     def _webpage_read_content(self, urlh, url_or_request, video_id, note=None, errnote=None, fatal=True, prefix=None, encoding=None):
439         content_type = urlh.headers.get('Content-Type', '')
440         webpage_bytes = urlh.read()
441         if prefix is not None:
442             webpage_bytes = prefix + webpage_bytes
443         if not encoding:
444             encoding = self._guess_encoding_from_content(content_type, webpage_bytes)
445         if self._downloader.params.get('dump_intermediate_pages', False):
446             try:
447                 url = url_or_request.get_full_url()
448             except AttributeError:
449                 url = url_or_request
450             self.to_screen('Dumping request to ' + url)
451             dump = base64.b64encode(webpage_bytes).decode('ascii')
452             self._downloader.to_screen(dump)
453         if self._downloader.params.get('write_pages', False):
454             try:
455                 url = url_or_request.get_full_url()
456             except AttributeError:
457                 url = url_or_request
458             basen = '%s_%s' % (video_id, url)
459             if len(basen) > 240:
460                 h = '___' + hashlib.md5(basen.encode('utf-8')).hexdigest()
461                 basen = basen[:240 - len(h)] + h
462             raw_filename = basen + '.dump'
463             filename = sanitize_filename(raw_filename, restricted=True)
464             self.to_screen('Saving request to ' + filename)
465             # Working around MAX_PATH limitation on Windows (see
466             # http://msdn.microsoft.com/en-us/library/windows/desktop/aa365247(v=vs.85).aspx)
467             if compat_os_name == 'nt':
468                 absfilepath = os.path.abspath(filename)
469                 if len(absfilepath) > 259:
470                     filename = '\\\\?\\' + absfilepath
471             with open(filename, 'wb') as outf:
472                 outf.write(webpage_bytes)
473
474         try:
475             content = webpage_bytes.decode(encoding, 'replace')
476         except LookupError:
477             content = webpage_bytes.decode('utf-8', 'replace')
478
479         if ('<title>Access to this site is blocked</title>' in content and
480                 'Websense' in content[:512]):
481             msg = 'Access to this webpage has been blocked by Websense filtering software in your network.'
482             blocked_iframe = self._html_search_regex(
483                 r'<iframe src="([^"]+)"', content,
484                 'Websense information URL', default=None)
485             if blocked_iframe:
486                 msg += ' Visit %s for more details' % blocked_iframe
487             raise ExtractorError(msg, expected=True)
488         if '<title>The URL you requested has been blocked</title>' in content[:512]:
489             msg = (
490                 'Access to this webpage has been blocked by Indian censorship. '
491                 'Use a VPN or proxy server (with --proxy) to route around it.')
492             block_msg = self._html_search_regex(
493                 r'</h1><p>(.*?)</p>',
494                 content, 'block message', default=None)
495             if block_msg:
496                 msg += ' (Message: "%s")' % block_msg.replace('\n', ' ')
497             raise ExtractorError(msg, expected=True)
498
499         return content
500
501     def _download_webpage(self, url_or_request, video_id, note=None, errnote=None, fatal=True, tries=1, timeout=5, encoding=None, data=None, headers={}, query={}):
502         """ Returns the data of the page as a string """
503         success = False
504         try_count = 0
505         while success is False:
506             try:
507                 res = self._download_webpage_handle(url_or_request, video_id, note, errnote, fatal, encoding=encoding, data=data, headers=headers, query=query)
508                 success = True
509             except compat_http_client.IncompleteRead as e:
510                 try_count += 1
511                 if try_count >= tries:
512                     raise e
513                 self._sleep(timeout, video_id)
514         if res is False:
515             return res
516         else:
517             content, _ = res
518             return content
519
520     def _download_xml(self, url_or_request, video_id,
521                       note='Downloading XML', errnote='Unable to download XML',
522                       transform_source=None, fatal=True, encoding=None, data=None, headers={}, query={}):
523         """Return the xml as an xml.etree.ElementTree.Element"""
524         xml_string = self._download_webpage(
525             url_or_request, video_id, note, errnote, fatal=fatal, encoding=encoding, data=data, headers=headers, query=query)
526         if xml_string is False:
527             return xml_string
528         if transform_source:
529             xml_string = transform_source(xml_string)
530         return compat_etree_fromstring(xml_string.encode('utf-8'))
531
532     def _download_json(self, url_or_request, video_id,
533                        note='Downloading JSON metadata',
534                        errnote='Unable to download JSON metadata',
535                        transform_source=None,
536                        fatal=True, encoding=None, data=None, headers={}, query={}):
537         json_string = self._download_webpage(
538             url_or_request, video_id, note, errnote, fatal=fatal,
539             encoding=encoding, data=data, headers=headers, query=query)
540         if (not fatal) and json_string is False:
541             return None
542         return self._parse_json(
543             json_string, video_id, transform_source=transform_source, fatal=fatal)
544
545     def _parse_json(self, json_string, video_id, transform_source=None, fatal=True):
546         if transform_source:
547             json_string = transform_source(json_string)
548         try:
549             return json.loads(json_string)
550         except ValueError as ve:
551             errmsg = '%s: Failed to parse JSON ' % video_id
552             if fatal:
553                 raise ExtractorError(errmsg, cause=ve)
554             else:
555                 self.report_warning(errmsg + str(ve))
556
557     def report_warning(self, msg, video_id=None):
558         idstr = '' if video_id is None else '%s: ' % video_id
559         self._downloader.report_warning(
560             '[%s] %s%s' % (self.IE_NAME, idstr, msg))
561
562     def to_screen(self, msg):
563         """Print msg to screen, prefixing it with '[ie_name]'"""
564         self._downloader.to_screen('[%s] %s' % (self.IE_NAME, msg))
565
566     def report_extraction(self, id_or_name):
567         """Report information extraction."""
568         self.to_screen('%s: Extracting information' % id_or_name)
569
570     def report_download_webpage(self, video_id):
571         """Report webpage download."""
572         self.to_screen('%s: Downloading webpage' % video_id)
573
574     def report_age_confirmation(self):
575         """Report attempt to confirm age."""
576         self.to_screen('Confirming age')
577
578     def report_login(self):
579         """Report attempt to log in."""
580         self.to_screen('Logging in')
581
582     @staticmethod
583     def raise_login_required(msg='This video is only available for registered users'):
584         raise ExtractorError(
585             '%s. Use --username and --password or --netrc to provide account credentials.' % msg,
586             expected=True)
587
588     @staticmethod
589     def raise_geo_restricted(msg='This video is not available from your location due to geo restriction'):
590         raise ExtractorError(
591             '%s. You might want to use --proxy to workaround.' % msg,
592             expected=True)
593
594     # Methods for following #608
595     @staticmethod
596     def url_result(url, ie=None, video_id=None, video_title=None):
597         """Returns a URL that points to a page that should be processed"""
598         # TODO: ie should be the class used for getting the info
599         video_info = {'_type': 'url',
600                       'url': url,
601                       'ie_key': ie}
602         if video_id is not None:
603             video_info['id'] = video_id
604         if video_title is not None:
605             video_info['title'] = video_title
606         return video_info
607
608     @staticmethod
609     def playlist_result(entries, playlist_id=None, playlist_title=None, playlist_description=None):
610         """Returns a playlist"""
611         video_info = {'_type': 'playlist',
612                       'entries': entries}
613         if playlist_id:
614             video_info['id'] = playlist_id
615         if playlist_title:
616             video_info['title'] = playlist_title
617         if playlist_description:
618             video_info['description'] = playlist_description
619         return video_info
620
621     def _search_regex(self, pattern, string, name, default=NO_DEFAULT, fatal=True, flags=0, group=None):
622         """
623         Perform a regex search on the given string, using a single or a list of
624         patterns returning the first matching group.
625         In case of failure return a default value or raise a WARNING or a
626         RegexNotFoundError, depending on fatal, specifying the field name.
627         """
628         if isinstance(pattern, (str, compat_str, compiled_regex_type)):
629             mobj = re.search(pattern, string, flags)
630         else:
631             for p in pattern:
632                 mobj = re.search(p, string, flags)
633                 if mobj:
634                     break
635
636         if not self._downloader.params.get('no_color') and compat_os_name != 'nt' and sys.stderr.isatty():
637             _name = '\033[0;34m%s\033[0m' % name
638         else:
639             _name = name
640
641         if mobj:
642             if group is None:
643                 # return the first matching group
644                 return next(g for g in mobj.groups() if g is not None)
645             else:
646                 return mobj.group(group)
647         elif default is not NO_DEFAULT:
648             return default
649         elif fatal:
650             raise RegexNotFoundError('Unable to extract %s' % _name)
651         else:
652             self._downloader.report_warning('unable to extract %s' % _name + bug_reports_message())
653             return None
654
655     def _html_search_regex(self, pattern, string, name, default=NO_DEFAULT, fatal=True, flags=0, group=None):
656         """
657         Like _search_regex, but strips HTML tags and unescapes entities.
658         """
659         res = self._search_regex(pattern, string, name, default, fatal, flags, group)
660         if res:
661             return clean_html(res).strip()
662         else:
663             return res
664
665     def _get_netrc_login_info(self, netrc_machine=None):
666         username = None
667         password = None
668         netrc_machine = netrc_machine or self._NETRC_MACHINE
669
670         if self._downloader.params.get('usenetrc', False):
671             try:
672                 info = netrc.netrc().authenticators(netrc_machine)
673                 if info is not None:
674                     username = info[0]
675                     password = info[2]
676                 else:
677                     raise netrc.NetrcParseError('No authenticators for %s' % netrc_machine)
678             except (IOError, netrc.NetrcParseError) as err:
679                 self._downloader.report_warning('parsing .netrc: %s' % error_to_compat_str(err))
680
681         return (username, password)
682
683     def _get_login_info(self):
684         """
685         Get the login info as (username, password)
686         It will look in the netrc file using the _NETRC_MACHINE value
687         If there's no info available, return (None, None)
688         """
689         if self._downloader is None:
690             return (None, None)
691
692         username = None
693         password = None
694         downloader_params = self._downloader.params
695
696         # Attempt to use provided username and password or .netrc data
697         if downloader_params.get('username') is not None:
698             username = downloader_params['username']
699             password = downloader_params['password']
700         else:
701             username, password = self._get_netrc_login_info()
702
703         return (username, password)
704
705     def _get_tfa_info(self, note='two-factor verification code'):
706         """
707         Get the two-factor authentication info
708         TODO - asking the user will be required for sms/phone verify
709         currently just uses the command line option
710         If there's no info available, return None
711         """
712         if self._downloader is None:
713             return None
714         downloader_params = self._downloader.params
715
716         if downloader_params.get('twofactor') is not None:
717             return downloader_params['twofactor']
718
719         return compat_getpass('Type %s and press [Return]: ' % note)
720
721     # Helper functions for extracting OpenGraph info
722     @staticmethod
723     def _og_regexes(prop):
724         content_re = r'content=(?:"([^"]+?)"|\'([^\']+?)\'|\s*([^\s"\'=<>`]+?))'
725         property_re = (r'(?:name|property)=(?:\'og:%(prop)s\'|"og:%(prop)s"|\s*og:%(prop)s\b)'
726                        % {'prop': re.escape(prop)})
727         template = r'<meta[^>]+?%s[^>]+?%s'
728         return [
729             template % (property_re, content_re),
730             template % (content_re, property_re),
731         ]
732
733     @staticmethod
734     def _meta_regex(prop):
735         return r'''(?isx)<meta
736                     (?=[^>]+(?:itemprop|name|property|id|http-equiv)=(["\']?)%s\1)
737                     [^>]+?content=(["\'])(?P<content>.*?)\2''' % re.escape(prop)
738
739     def _og_search_property(self, prop, html, name=None, **kargs):
740         if not isinstance(prop, (list, tuple)):
741             prop = [prop]
742         if name is None:
743             name = 'OpenGraph %s' % prop[0]
744         og_regexes = []
745         for p in prop:
746             og_regexes.extend(self._og_regexes(p))
747         escaped = self._search_regex(og_regexes, html, name, flags=re.DOTALL, **kargs)
748         if escaped is None:
749             return None
750         return unescapeHTML(escaped)
751
752     def _og_search_thumbnail(self, html, **kargs):
753         return self._og_search_property('image', html, 'thumbnail URL', fatal=False, **kargs)
754
755     def _og_search_description(self, html, **kargs):
756         return self._og_search_property('description', html, fatal=False, **kargs)
757
758     def _og_search_title(self, html, **kargs):
759         return self._og_search_property('title', html, **kargs)
760
761     def _og_search_video_url(self, html, name='video url', secure=True, **kargs):
762         regexes = self._og_regexes('video') + self._og_regexes('video:url')
763         if secure:
764             regexes = self._og_regexes('video:secure_url') + regexes
765         return self._html_search_regex(regexes, html, name, **kargs)
766
767     def _og_search_url(self, html, **kargs):
768         return self._og_search_property('url', html, **kargs)
769
770     def _html_search_meta(self, name, html, display_name=None, fatal=False, **kwargs):
771         if not isinstance(name, (list, tuple)):
772             name = [name]
773         if display_name is None:
774             display_name = name[0]
775         return self._html_search_regex(
776             [self._meta_regex(n) for n in name],
777             html, display_name, fatal=fatal, group='content', **kwargs)
778
779     def _dc_search_uploader(self, html):
780         return self._html_search_meta('dc.creator', html, 'uploader')
781
782     def _rta_search(self, html):
783         # See http://www.rtalabel.org/index.php?content=howtofaq#single
784         if re.search(r'(?ix)<meta\s+name="rating"\s+'
785                      r'     content="RTA-5042-1996-1400-1577-RTA"',
786                      html):
787             return 18
788         return 0
789
790     def _media_rating_search(self, html):
791         # See http://www.tjg-designs.com/WP/metadata-code-examples-adding-metadata-to-your-web-pages/
792         rating = self._html_search_meta('rating', html)
793
794         if not rating:
795             return None
796
797         RATING_TABLE = {
798             'safe for kids': 0,
799             'general': 8,
800             '14 years': 14,
801             'mature': 17,
802             'restricted': 19,
803         }
804         return RATING_TABLE.get(rating.lower())
805
806     def _family_friendly_search(self, html):
807         # See http://schema.org/VideoObject
808         family_friendly = self._html_search_meta('isFamilyFriendly', html)
809
810         if not family_friendly:
811             return None
812
813         RATING_TABLE = {
814             '1': 0,
815             'true': 0,
816             '0': 18,
817             'false': 18,
818         }
819         return RATING_TABLE.get(family_friendly.lower())
820
821     def _twitter_search_player(self, html):
822         return self._html_search_meta('twitter:player', html,
823                                       'twitter card player')
824
825     def _search_json_ld(self, html, video_id, expected_type=None, **kwargs):
826         json_ld = self._search_regex(
827             r'(?s)<script[^>]+type=(["\'])application/ld\+json\1[^>]*>(?P<json_ld>.+?)</script>',
828             html, 'JSON-LD', group='json_ld', **kwargs)
829         default = kwargs.get('default', NO_DEFAULT)
830         if not json_ld:
831             return default if default is not NO_DEFAULT else {}
832         # JSON-LD may be malformed and thus `fatal` should be respected.
833         # At the same time `default` may be passed that assumes `fatal=False`
834         # for _search_regex. Let's simulate the same behavior here as well.
835         fatal = kwargs.get('fatal', True) if default == NO_DEFAULT else False
836         return self._json_ld(json_ld, video_id, fatal=fatal, expected_type=expected_type)
837
838     def _json_ld(self, json_ld, video_id, fatal=True, expected_type=None):
839         if isinstance(json_ld, compat_str):
840             json_ld = self._parse_json(json_ld, video_id, fatal=fatal)
841         if not json_ld:
842             return {}
843         info = {}
844         if not isinstance(json_ld, (list, tuple, dict)):
845             return info
846         if isinstance(json_ld, dict):
847             json_ld = [json_ld]
848         for e in json_ld:
849             if e.get('@context') == 'http://schema.org':
850                 item_type = e.get('@type')
851                 if expected_type is not None and expected_type != item_type:
852                     return info
853                 if item_type == 'TVEpisode':
854                     info.update({
855                         'episode': unescapeHTML(e.get('name')),
856                         'episode_number': int_or_none(e.get('episodeNumber')),
857                         'description': unescapeHTML(e.get('description')),
858                     })
859                     part_of_season = e.get('partOfSeason')
860                     if isinstance(part_of_season, dict) and part_of_season.get('@type') == 'TVSeason':
861                         info['season_number'] = int_or_none(part_of_season.get('seasonNumber'))
862                     part_of_series = e.get('partOfSeries') or e.get('partOfTVSeries')
863                     if isinstance(part_of_series, dict) and part_of_series.get('@type') == 'TVSeries':
864                         info['series'] = unescapeHTML(part_of_series.get('name'))
865                 elif item_type == 'Article':
866                     info.update({
867                         'timestamp': parse_iso8601(e.get('datePublished')),
868                         'title': unescapeHTML(e.get('headline')),
869                         'description': unescapeHTML(e.get('articleBody')),
870                     })
871                 elif item_type == 'VideoObject':
872                     info.update({
873                         'url': e.get('contentUrl'),
874                         'title': unescapeHTML(e.get('name')),
875                         'description': unescapeHTML(e.get('description')),
876                         'thumbnail': e.get('thumbnailUrl'),
877                         'duration': parse_duration(e.get('duration')),
878                         'timestamp': unified_timestamp(e.get('uploadDate')),
879                         'filesize': float_or_none(e.get('contentSize')),
880                         'tbr': int_or_none(e.get('bitrate')),
881                         'width': int_or_none(e.get('width')),
882                         'height': int_or_none(e.get('height')),
883                     })
884                 break
885         return dict((k, v) for k, v in info.items() if v is not None)
886
887     @staticmethod
888     def _hidden_inputs(html):
889         html = re.sub(r'<!--(?:(?!<!--).)*-->', '', html)
890         hidden_inputs = {}
891         for input in re.findall(r'(?i)<input([^>]+)>', html):
892             if not re.search(r'type=(["\'])(?:hidden|submit)\1', input):
893                 continue
894             name = re.search(r'(?:name|id)=(["\'])(?P<value>.+?)\1', input)
895             if not name:
896                 continue
897             value = re.search(r'value=(["\'])(?P<value>.*?)\1', input)
898             if not value:
899                 continue
900             hidden_inputs[name.group('value')] = value.group('value')
901         return hidden_inputs
902
903     def _form_hidden_inputs(self, form_id, html):
904         form = self._search_regex(
905             r'(?is)<form[^>]+?id=(["\'])%s\1[^>]*>(?P<form>.+?)</form>' % form_id,
906             html, '%s form' % form_id, group='form')
907         return self._hidden_inputs(form)
908
909     def _sort_formats(self, formats, field_preference=None):
910         if not formats:
911             raise ExtractorError('No video formats found')
912
913         for f in formats:
914             # Automatically determine tbr when missing based on abr and vbr (improves
915             # formats sorting in some cases)
916             if 'tbr' not in f and f.get('abr') is not None and f.get('vbr') is not None:
917                 f['tbr'] = f['abr'] + f['vbr']
918
919         def _formats_key(f):
920             # TODO remove the following workaround
921             from ..utils import determine_ext
922             if not f.get('ext') and 'url' in f:
923                 f['ext'] = determine_ext(f['url'])
924
925             if isinstance(field_preference, (list, tuple)):
926                 return tuple(
927                     f.get(field)
928                     if f.get(field) is not None
929                     else ('' if field == 'format_id' else -1)
930                     for field in field_preference)
931
932             preference = f.get('preference')
933             if preference is None:
934                 preference = 0
935                 if f.get('ext') in ['f4f', 'f4m']:  # Not yet supported
936                     preference -= 0.5
937
938             protocol = f.get('protocol') or determine_protocol(f)
939             proto_preference = 0 if protocol in ['http', 'https'] else (-0.5 if protocol == 'rtsp' else -0.1)
940
941             if f.get('vcodec') == 'none':  # audio only
942                 preference -= 50
943                 if self._downloader.params.get('prefer_free_formats'):
944                     ORDER = ['aac', 'mp3', 'm4a', 'webm', 'ogg', 'opus']
945                 else:
946                     ORDER = ['webm', 'opus', 'ogg', 'mp3', 'aac', 'm4a']
947                 ext_preference = 0
948                 try:
949                     audio_ext_preference = ORDER.index(f['ext'])
950                 except ValueError:
951                     audio_ext_preference = -1
952             else:
953                 if f.get('acodec') == 'none':  # video only
954                     preference -= 40
955                 if self._downloader.params.get('prefer_free_formats'):
956                     ORDER = ['flv', 'mp4', 'webm']
957                 else:
958                     ORDER = ['webm', 'flv', 'mp4']
959                 try:
960                     ext_preference = ORDER.index(f['ext'])
961                 except ValueError:
962                     ext_preference = -1
963                 audio_ext_preference = 0
964
965             return (
966                 preference,
967                 f.get('language_preference') if f.get('language_preference') is not None else -1,
968                 f.get('quality') if f.get('quality') is not None else -1,
969                 f.get('tbr') if f.get('tbr') is not None else -1,
970                 f.get('filesize') if f.get('filesize') is not None else -1,
971                 f.get('vbr') if f.get('vbr') is not None else -1,
972                 f.get('height') if f.get('height') is not None else -1,
973                 f.get('width') if f.get('width') is not None else -1,
974                 proto_preference,
975                 ext_preference,
976                 f.get('abr') if f.get('abr') is not None else -1,
977                 audio_ext_preference,
978                 f.get('fps') if f.get('fps') is not None else -1,
979                 f.get('filesize_approx') if f.get('filesize_approx') is not None else -1,
980                 f.get('source_preference') if f.get('source_preference') is not None else -1,
981                 f.get('format_id') if f.get('format_id') is not None else '',
982             )
983         formats.sort(key=_formats_key)
984
985     def _check_formats(self, formats, video_id):
986         if formats:
987             formats[:] = filter(
988                 lambda f: self._is_valid_url(
989                     f['url'], video_id,
990                     item='%s video format' % f.get('format_id') if f.get('format_id') else 'video'),
991                 formats)
992
993     @staticmethod
994     def _remove_duplicate_formats(formats):
995         format_urls = set()
996         unique_formats = []
997         for f in formats:
998             if f['url'] not in format_urls:
999                 format_urls.add(f['url'])
1000                 unique_formats.append(f)
1001         formats[:] = unique_formats
1002
1003     def _is_valid_url(self, url, video_id, item='video'):
1004         url = self._proto_relative_url(url, scheme='http:')
1005         # For now assume non HTTP(S) URLs always valid
1006         if not (url.startswith('http://') or url.startswith('https://')):
1007             return True
1008         try:
1009             self._request_webpage(url, video_id, 'Checking %s URL' % item)
1010             return True
1011         except ExtractorError as e:
1012             if isinstance(e.cause, compat_urllib_error.URLError):
1013                 self.to_screen(
1014                     '%s: %s URL is invalid, skipping' % (video_id, item))
1015                 return False
1016             raise
1017
1018     def http_scheme(self):
1019         """ Either "http:" or "https:", depending on the user's preferences """
1020         return (
1021             'http:'
1022             if self._downloader.params.get('prefer_insecure', False)
1023             else 'https:')
1024
1025     def _proto_relative_url(self, url, scheme=None):
1026         if url is None:
1027             return url
1028         if url.startswith('//'):
1029             if scheme is None:
1030                 scheme = self.http_scheme()
1031             return scheme + url
1032         else:
1033             return url
1034
1035     def _sleep(self, timeout, video_id, msg_template=None):
1036         if msg_template is None:
1037             msg_template = '%(video_id)s: Waiting for %(timeout)s seconds'
1038         msg = msg_template % {'video_id': video_id, 'timeout': timeout}
1039         self.to_screen(msg)
1040         time.sleep(timeout)
1041
1042     def _extract_f4m_formats(self, manifest_url, video_id, preference=None, f4m_id=None,
1043                              transform_source=lambda s: fix_xml_ampersands(s).strip(),
1044                              fatal=True, m3u8_id=None):
1045         manifest = self._download_xml(
1046             manifest_url, video_id, 'Downloading f4m manifest',
1047             'Unable to download f4m manifest',
1048             # Some manifests may be malformed, e.g. prosiebensat1 generated manifests
1049             # (see https://github.com/rg3/youtube-dl/issues/6215#issuecomment-121704244)
1050             transform_source=transform_source,
1051             fatal=fatal)
1052
1053         if manifest is False:
1054             return []
1055
1056         return self._parse_f4m_formats(
1057             manifest, manifest_url, video_id, preference=preference, f4m_id=f4m_id,
1058             transform_source=transform_source, fatal=fatal, m3u8_id=m3u8_id)
1059
1060     def _parse_f4m_formats(self, manifest, manifest_url, video_id, preference=None, f4m_id=None,
1061                            transform_source=lambda s: fix_xml_ampersands(s).strip(),
1062                            fatal=True, m3u8_id=None):
1063         # currently youtube-dl cannot decode the playerVerificationChallenge as Akamai uses Adobe Alchemy
1064         akamai_pv = manifest.find('{http://ns.adobe.com/f4m/1.0}pv-2.0')
1065         if akamai_pv is not None and ';' in akamai_pv.text:
1066             playerVerificationChallenge = akamai_pv.text.split(';')[0]
1067             if playerVerificationChallenge.strip() != '':
1068                 return []
1069
1070         formats = []
1071         manifest_version = '1.0'
1072         media_nodes = manifest.findall('{http://ns.adobe.com/f4m/1.0}media')
1073         if not media_nodes:
1074             manifest_version = '2.0'
1075             media_nodes = manifest.findall('{http://ns.adobe.com/f4m/2.0}media')
1076         # Remove unsupported DRM protected media from final formats
1077         # rendition (see https://github.com/rg3/youtube-dl/issues/8573).
1078         media_nodes = remove_encrypted_media(media_nodes)
1079         if not media_nodes:
1080             return formats
1081         base_url = xpath_text(
1082             manifest, ['{http://ns.adobe.com/f4m/1.0}baseURL', '{http://ns.adobe.com/f4m/2.0}baseURL'],
1083             'base URL', default=None)
1084         if base_url:
1085             base_url = base_url.strip()
1086
1087         bootstrap_info = xpath_element(
1088             manifest, ['{http://ns.adobe.com/f4m/1.0}bootstrapInfo', '{http://ns.adobe.com/f4m/2.0}bootstrapInfo'],
1089             'bootstrap info', default=None)
1090
1091         for i, media_el in enumerate(media_nodes):
1092             tbr = int_or_none(media_el.attrib.get('bitrate'))
1093             width = int_or_none(media_el.attrib.get('width'))
1094             height = int_or_none(media_el.attrib.get('height'))
1095             format_id = '-'.join(filter(None, [f4m_id, compat_str(i if tbr is None else tbr)]))
1096             # If <bootstrapInfo> is present, the specified f4m is a
1097             # stream-level manifest, and only set-level manifests may refer to
1098             # external resources.  See section 11.4 and section 4 of F4M spec
1099             if bootstrap_info is None:
1100                 media_url = None
1101                 # @href is introduced in 2.0, see section 11.6 of F4M spec
1102                 if manifest_version == '2.0':
1103                     media_url = media_el.attrib.get('href')
1104                 if media_url is None:
1105                     media_url = media_el.attrib.get('url')
1106                 if not media_url:
1107                     continue
1108                 manifest_url = (
1109                     media_url if media_url.startswith('http://') or media_url.startswith('https://')
1110                     else ((base_url or '/'.join(manifest_url.split('/')[:-1])) + '/' + media_url))
1111                 # If media_url is itself a f4m manifest do the recursive extraction
1112                 # since bitrates in parent manifest (this one) and media_url manifest
1113                 # may differ leading to inability to resolve the format by requested
1114                 # bitrate in f4m downloader
1115                 ext = determine_ext(manifest_url)
1116                 if ext == 'f4m':
1117                     f4m_formats = self._extract_f4m_formats(
1118                         manifest_url, video_id, preference=preference, f4m_id=f4m_id,
1119                         transform_source=transform_source, fatal=fatal)
1120                     # Sometimes stream-level manifest contains single media entry that
1121                     # does not contain any quality metadata (e.g. http://matchtv.ru/#live-player).
1122                     # At the same time parent's media entry in set-level manifest may
1123                     # contain it. We will copy it from parent in such cases.
1124                     if len(f4m_formats) == 1:
1125                         f = f4m_formats[0]
1126                         f.update({
1127                             'tbr': f.get('tbr') or tbr,
1128                             'width': f.get('width') or width,
1129                             'height': f.get('height') or height,
1130                             'format_id': f.get('format_id') if not tbr else format_id,
1131                         })
1132                     formats.extend(f4m_formats)
1133                     continue
1134                 elif ext == 'm3u8':
1135                     formats.extend(self._extract_m3u8_formats(
1136                         manifest_url, video_id, 'mp4', preference=preference,
1137                         m3u8_id=m3u8_id, fatal=fatal))
1138                     continue
1139             formats.append({
1140                 'format_id': format_id,
1141                 'url': manifest_url,
1142                 'ext': 'flv' if bootstrap_info is not None else None,
1143                 'tbr': tbr,
1144                 'width': width,
1145                 'height': height,
1146                 'preference': preference,
1147             })
1148         return formats
1149
1150     def _m3u8_meta_format(self, m3u8_url, ext=None, preference=None, m3u8_id=None):
1151         return {
1152             'format_id': '-'.join(filter(None, [m3u8_id, 'meta'])),
1153             'url': m3u8_url,
1154             'ext': ext,
1155             'protocol': 'm3u8',
1156             'preference': preference - 100 if preference else -100,
1157             'resolution': 'multiple',
1158             'format_note': 'Quality selection URL',
1159         }
1160
1161     def _extract_m3u8_formats(self, m3u8_url, video_id, ext=None,
1162                               entry_protocol='m3u8', preference=None,
1163                               m3u8_id=None, note=None, errnote=None,
1164                               fatal=True, live=False):
1165
1166         formats = [self._m3u8_meta_format(m3u8_url, ext, preference, m3u8_id)]
1167
1168         format_url = lambda u: (
1169             u
1170             if re.match(r'^https?://', u)
1171             else compat_urlparse.urljoin(m3u8_url, u))
1172
1173         res = self._download_webpage_handle(
1174             m3u8_url, video_id,
1175             note=note or 'Downloading m3u8 information',
1176             errnote=errnote or 'Failed to download m3u8 information',
1177             fatal=fatal)
1178         if res is False:
1179             return []
1180         m3u8_doc, urlh = res
1181         m3u8_url = urlh.geturl()
1182
1183         # We should try extracting formats only from master playlists [1], i.e.
1184         # playlists that describe available qualities. On the other hand media
1185         # playlists [2] should be returned as is since they contain just the media
1186         # without qualities renditions.
1187         # Fortunately, master playlist can be easily distinguished from media
1188         # playlist based on particular tags availability. As of [1, 2] master
1189         # playlist tags MUST NOT appear in a media playist and vice versa.
1190         # As of [3] #EXT-X-TARGETDURATION tag is REQUIRED for every media playlist
1191         # and MUST NOT appear in master playlist thus we can clearly detect media
1192         # playlist with this criterion.
1193         # 1. https://tools.ietf.org/html/draft-pantos-http-live-streaming-17#section-4.3.4
1194         # 2. https://tools.ietf.org/html/draft-pantos-http-live-streaming-17#section-4.3.3
1195         # 3. https://tools.ietf.org/html/draft-pantos-http-live-streaming-17#section-4.3.3.1
1196         if '#EXT-X-TARGETDURATION' in m3u8_doc:  # media playlist, return as is
1197             return [{
1198                 'url': m3u8_url,
1199                 'format_id': m3u8_id,
1200                 'ext': ext,
1201                 'protocol': entry_protocol,
1202                 'preference': preference,
1203             }]
1204         last_info = None
1205         last_media = None
1206         for line in m3u8_doc.splitlines():
1207             if line.startswith('#EXT-X-STREAM-INF:'):
1208                 last_info = parse_m3u8_attributes(line)
1209             elif line.startswith('#EXT-X-MEDIA:'):
1210                 last_media = parse_m3u8_attributes(line)
1211             elif line.startswith('#') or not line.strip():
1212                 continue
1213             else:
1214                 if last_info is None:
1215                     formats.append({'url': format_url(line)})
1216                     continue
1217                 tbr = int_or_none(last_info.get('BANDWIDTH'), scale=1000)
1218                 format_id = []
1219                 if m3u8_id:
1220                     format_id.append(m3u8_id)
1221                 last_media_name = last_media.get('NAME') if last_media and last_media.get('TYPE') not in ('SUBTITLES', 'CLOSED-CAPTIONS') else None
1222                 # Despite specification does not mention NAME attribute for
1223                 # EXT-X-STREAM-INF it still sometimes may be present
1224                 stream_name = last_info.get('NAME') or last_media_name
1225                 # Bandwidth of live streams may differ over time thus making
1226                 # format_id unpredictable. So it's better to keep provided
1227                 # format_id intact.
1228                 if not live:
1229                     format_id.append(stream_name if stream_name else '%d' % (tbr if tbr else len(formats)))
1230                 f = {
1231                     'format_id': '-'.join(format_id),
1232                     'url': format_url(line.strip()),
1233                     'tbr': tbr,
1234                     'ext': ext,
1235                     'fps': float_or_none(last_info.get('FRAME-RATE')),
1236                     'protocol': entry_protocol,
1237                     'preference': preference,
1238                 }
1239                 resolution = last_info.get('RESOLUTION')
1240                 if resolution:
1241                     width_str, height_str = resolution.split('x')
1242                     f['width'] = int(width_str)
1243                     f['height'] = int(height_str)
1244                 # Unified Streaming Platform
1245                 mobj = re.search(
1246                     r'audio.*?(?:%3D|=)(\d+)(?:-video.*?(?:%3D|=)(\d+))?', f['url'])
1247                 if mobj:
1248                     abr, vbr = mobj.groups()
1249                     abr, vbr = float_or_none(abr, 1000), float_or_none(vbr, 1000)
1250                     f.update({
1251                         'vbr': vbr,
1252                         'abr': abr,
1253                     })
1254                 f.update(parse_codecs(last_info.get('CODECS')))
1255                 if last_media is not None:
1256                     f['m3u8_media'] = last_media
1257                     last_media = None
1258                 formats.append(f)
1259                 last_info = {}
1260         return formats
1261
1262     @staticmethod
1263     def _xpath_ns(path, namespace=None):
1264         if not namespace:
1265             return path
1266         out = []
1267         for c in path.split('/'):
1268             if not c or c == '.':
1269                 out.append(c)
1270             else:
1271                 out.append('{%s}%s' % (namespace, c))
1272         return '/'.join(out)
1273
1274     def _extract_smil_formats(self, smil_url, video_id, fatal=True, f4m_params=None, transform_source=None):
1275         smil = self._download_smil(smil_url, video_id, fatal=fatal, transform_source=transform_source)
1276
1277         if smil is False:
1278             assert not fatal
1279             return []
1280
1281         namespace = self._parse_smil_namespace(smil)
1282
1283         return self._parse_smil_formats(
1284             smil, smil_url, video_id, namespace=namespace, f4m_params=f4m_params)
1285
1286     def _extract_smil_info(self, smil_url, video_id, fatal=True, f4m_params=None):
1287         smil = self._download_smil(smil_url, video_id, fatal=fatal)
1288         if smil is False:
1289             return {}
1290         return self._parse_smil(smil, smil_url, video_id, f4m_params=f4m_params)
1291
1292     def _download_smil(self, smil_url, video_id, fatal=True, transform_source=None):
1293         return self._download_xml(
1294             smil_url, video_id, 'Downloading SMIL file',
1295             'Unable to download SMIL file', fatal=fatal, transform_source=transform_source)
1296
1297     def _parse_smil(self, smil, smil_url, video_id, f4m_params=None):
1298         namespace = self._parse_smil_namespace(smil)
1299
1300         formats = self._parse_smil_formats(
1301             smil, smil_url, video_id, namespace=namespace, f4m_params=f4m_params)
1302         subtitles = self._parse_smil_subtitles(smil, namespace=namespace)
1303
1304         video_id = os.path.splitext(url_basename(smil_url))[0]
1305         title = None
1306         description = None
1307         upload_date = None
1308         for meta in smil.findall(self._xpath_ns('./head/meta', namespace)):
1309             name = meta.attrib.get('name')
1310             content = meta.attrib.get('content')
1311             if not name or not content:
1312                 continue
1313             if not title and name == 'title':
1314                 title = content
1315             elif not description and name in ('description', 'abstract'):
1316                 description = content
1317             elif not upload_date and name == 'date':
1318                 upload_date = unified_strdate(content)
1319
1320         thumbnails = [{
1321             'id': image.get('type'),
1322             'url': image.get('src'),
1323             'width': int_or_none(image.get('width')),
1324             'height': int_or_none(image.get('height')),
1325         } for image in smil.findall(self._xpath_ns('.//image', namespace)) if image.get('src')]
1326
1327         return {
1328             'id': video_id,
1329             'title': title or video_id,
1330             'description': description,
1331             'upload_date': upload_date,
1332             'thumbnails': thumbnails,
1333             'formats': formats,
1334             'subtitles': subtitles,
1335         }
1336
1337     def _parse_smil_namespace(self, smil):
1338         return self._search_regex(
1339             r'(?i)^{([^}]+)?}smil$', smil.tag, 'namespace', default=None)
1340
1341     def _parse_smil_formats(self, smil, smil_url, video_id, namespace=None, f4m_params=None, transform_rtmp_url=None):
1342         base = smil_url
1343         for meta in smil.findall(self._xpath_ns('./head/meta', namespace)):
1344             b = meta.get('base') or meta.get('httpBase')
1345             if b:
1346                 base = b
1347                 break
1348
1349         formats = []
1350         rtmp_count = 0
1351         http_count = 0
1352         m3u8_count = 0
1353
1354         srcs = []
1355         media = smil.findall(self._xpath_ns('.//video', namespace)) + smil.findall(self._xpath_ns('.//audio', namespace))
1356         for medium in media:
1357             src = medium.get('src')
1358             if not src or src in srcs:
1359                 continue
1360             srcs.append(src)
1361
1362             bitrate = float_or_none(medium.get('system-bitrate') or medium.get('systemBitrate'), 1000)
1363             filesize = int_or_none(medium.get('size') or medium.get('fileSize'))
1364             width = int_or_none(medium.get('width'))
1365             height = int_or_none(medium.get('height'))
1366             proto = medium.get('proto')
1367             ext = medium.get('ext')
1368             src_ext = determine_ext(src)
1369             streamer = medium.get('streamer') or base
1370
1371             if proto == 'rtmp' or streamer.startswith('rtmp'):
1372                 rtmp_count += 1
1373                 formats.append({
1374                     'url': streamer,
1375                     'play_path': src,
1376                     'ext': 'flv',
1377                     'format_id': 'rtmp-%d' % (rtmp_count if bitrate is None else bitrate),
1378                     'tbr': bitrate,
1379                     'filesize': filesize,
1380                     'width': width,
1381                     'height': height,
1382                 })
1383                 if transform_rtmp_url:
1384                     streamer, src = transform_rtmp_url(streamer, src)
1385                     formats[-1].update({
1386                         'url': streamer,
1387                         'play_path': src,
1388                     })
1389                 continue
1390
1391             src_url = src if src.startswith('http') else compat_urlparse.urljoin(base, src)
1392             src_url = src_url.strip()
1393
1394             if proto == 'm3u8' or src_ext == 'm3u8':
1395                 m3u8_formats = self._extract_m3u8_formats(
1396                     src_url, video_id, ext or 'mp4', m3u8_id='hls', fatal=False)
1397                 if len(m3u8_formats) == 1:
1398                     m3u8_count += 1
1399                     m3u8_formats[0].update({
1400                         'format_id': 'hls-%d' % (m3u8_count if bitrate is None else bitrate),
1401                         'tbr': bitrate,
1402                         'width': width,
1403                         'height': height,
1404                     })
1405                 formats.extend(m3u8_formats)
1406                 continue
1407
1408             if src_ext == 'f4m':
1409                 f4m_url = src_url
1410                 if not f4m_params:
1411                     f4m_params = {
1412                         'hdcore': '3.2.0',
1413                         'plugin': 'flowplayer-3.2.0.1',
1414                     }
1415                 f4m_url += '&' if '?' in f4m_url else '?'
1416                 f4m_url += compat_urllib_parse_urlencode(f4m_params)
1417                 formats.extend(self._extract_f4m_formats(f4m_url, video_id, f4m_id='hds', fatal=False))
1418                 continue
1419
1420             if src_url.startswith('http') and self._is_valid_url(src, video_id):
1421                 http_count += 1
1422                 formats.append({
1423                     'url': src_url,
1424                     'ext': ext or src_ext or 'flv',
1425                     'format_id': 'http-%d' % (bitrate or http_count),
1426                     'tbr': bitrate,
1427                     'filesize': filesize,
1428                     'width': width,
1429                     'height': height,
1430                 })
1431                 continue
1432
1433         return formats
1434
1435     def _parse_smil_subtitles(self, smil, namespace=None, subtitles_lang='en'):
1436         urls = []
1437         subtitles = {}
1438         for num, textstream in enumerate(smil.findall(self._xpath_ns('.//textstream', namespace))):
1439             src = textstream.get('src')
1440             if not src or src in urls:
1441                 continue
1442             urls.append(src)
1443             ext = textstream.get('ext') or mimetype2ext(textstream.get('type')) or determine_ext(src)
1444             lang = textstream.get('systemLanguage') or textstream.get('systemLanguageName') or textstream.get('lang') or subtitles_lang
1445             subtitles.setdefault(lang, []).append({
1446                 'url': src,
1447                 'ext': ext,
1448             })
1449         return subtitles
1450
1451     def _extract_xspf_playlist(self, playlist_url, playlist_id, fatal=True):
1452         xspf = self._download_xml(
1453             playlist_url, playlist_id, 'Downloading xpsf playlist',
1454             'Unable to download xspf manifest', fatal=fatal)
1455         if xspf is False:
1456             return []
1457         return self._parse_xspf(xspf, playlist_id)
1458
1459     def _parse_xspf(self, playlist, playlist_id):
1460         NS_MAP = {
1461             'xspf': 'http://xspf.org/ns/0/',
1462             's1': 'http://static.streamone.nl/player/ns/0',
1463         }
1464
1465         entries = []
1466         for track in playlist.findall(xpath_with_ns('./xspf:trackList/xspf:track', NS_MAP)):
1467             title = xpath_text(
1468                 track, xpath_with_ns('./xspf:title', NS_MAP), 'title', default=playlist_id)
1469             description = xpath_text(
1470                 track, xpath_with_ns('./xspf:annotation', NS_MAP), 'description')
1471             thumbnail = xpath_text(
1472                 track, xpath_with_ns('./xspf:image', NS_MAP), 'thumbnail')
1473             duration = float_or_none(
1474                 xpath_text(track, xpath_with_ns('./xspf:duration', NS_MAP), 'duration'), 1000)
1475
1476             formats = [{
1477                 'url': location.text,
1478                 'format_id': location.get(xpath_with_ns('s1:label', NS_MAP)),
1479                 'width': int_or_none(location.get(xpath_with_ns('s1:width', NS_MAP))),
1480                 'height': int_or_none(location.get(xpath_with_ns('s1:height', NS_MAP))),
1481             } for location in track.findall(xpath_with_ns('./xspf:location', NS_MAP))]
1482             self._sort_formats(formats)
1483
1484             entries.append({
1485                 'id': playlist_id,
1486                 'title': title,
1487                 'description': description,
1488                 'thumbnail': thumbnail,
1489                 'duration': duration,
1490                 'formats': formats,
1491             })
1492         return entries
1493
1494     def _extract_mpd_formats(self, mpd_url, video_id, mpd_id=None, note=None, errnote=None, fatal=True, formats_dict={}):
1495         res = self._download_webpage_handle(
1496             mpd_url, video_id,
1497             note=note or 'Downloading MPD manifest',
1498             errnote=errnote or 'Failed to download MPD manifest',
1499             fatal=fatal)
1500         if res is False:
1501             return []
1502         mpd, urlh = res
1503         mpd_base_url = re.match(r'https?://.+/', urlh.geturl()).group()
1504
1505         return self._parse_mpd_formats(
1506             compat_etree_fromstring(mpd.encode('utf-8')), mpd_id, mpd_base_url, formats_dict=formats_dict)
1507
1508     def _parse_mpd_formats(self, mpd_doc, mpd_id=None, mpd_base_url='', formats_dict={}):
1509         """
1510         Parse formats from MPD manifest.
1511         References:
1512          1. MPEG-DASH Standard, ISO/IEC 23009-1:2014(E),
1513             http://standards.iso.org/ittf/PubliclyAvailableStandards/c065274_ISO_IEC_23009-1_2014.zip
1514          2. https://en.wikipedia.org/wiki/Dynamic_Adaptive_Streaming_over_HTTP
1515         """
1516         if mpd_doc.get('type') == 'dynamic':
1517             return []
1518
1519         namespace = self._search_regex(r'(?i)^{([^}]+)?}MPD$', mpd_doc.tag, 'namespace', default=None)
1520
1521         def _add_ns(path):
1522             return self._xpath_ns(path, namespace)
1523
1524         def is_drm_protected(element):
1525             return element.find(_add_ns('ContentProtection')) is not None
1526
1527         def extract_multisegment_info(element, ms_parent_info):
1528             ms_info = ms_parent_info.copy()
1529             segment_list = element.find(_add_ns('SegmentList'))
1530             if segment_list is not None:
1531                 segment_urls_e = segment_list.findall(_add_ns('SegmentURL'))
1532                 if segment_urls_e:
1533                     ms_info['segment_urls'] = [segment.attrib['media'] for segment in segment_urls_e]
1534                 initialization = segment_list.find(_add_ns('Initialization'))
1535                 if initialization is not None:
1536                     ms_info['initialization_url'] = initialization.attrib['sourceURL']
1537             else:
1538                 segment_template = element.find(_add_ns('SegmentTemplate'))
1539                 if segment_template is not None:
1540                     start_number = segment_template.get('startNumber')
1541                     if start_number:
1542                         ms_info['start_number'] = int(start_number)
1543                     segment_timeline = segment_template.find(_add_ns('SegmentTimeline'))
1544                     if segment_timeline is not None:
1545                         s_e = segment_timeline.findall(_add_ns('S'))
1546                         if s_e:
1547                             ms_info['total_number'] = 0
1548                             ms_info['s'] = []
1549                             for s in s_e:
1550                                 r = int(s.get('r', 0))
1551                                 ms_info['total_number'] += 1 + r
1552                                 ms_info['s'].append({
1553                                     't': int(s.get('t', 0)),
1554                                     # @d is mandatory (see [1, 5.3.9.6.2, Table 17, page 60])
1555                                     'd': int(s.attrib['d']),
1556                                     'r': r,
1557                                 })
1558                     else:
1559                         timescale = segment_template.get('timescale')
1560                         if timescale:
1561                             ms_info['timescale'] = int(timescale)
1562                         segment_duration = segment_template.get('duration')
1563                         if segment_duration:
1564                             ms_info['segment_duration'] = int(segment_duration)
1565                     media_template = segment_template.get('media')
1566                     if media_template:
1567                         ms_info['media_template'] = media_template
1568                     initialization = segment_template.get('initialization')
1569                     if initialization:
1570                         ms_info['initialization_url'] = initialization
1571                     else:
1572                         initialization = segment_template.find(_add_ns('Initialization'))
1573                         if initialization is not None:
1574                             ms_info['initialization_url'] = initialization.attrib['sourceURL']
1575             return ms_info
1576
1577         mpd_duration = parse_duration(mpd_doc.get('mediaPresentationDuration'))
1578         formats = []
1579         for period in mpd_doc.findall(_add_ns('Period')):
1580             period_duration = parse_duration(period.get('duration')) or mpd_duration
1581             period_ms_info = extract_multisegment_info(period, {
1582                 'start_number': 1,
1583                 'timescale': 1,
1584             })
1585             for adaptation_set in period.findall(_add_ns('AdaptationSet')):
1586                 if is_drm_protected(adaptation_set):
1587                     continue
1588                 adaption_set_ms_info = extract_multisegment_info(adaptation_set, period_ms_info)
1589                 for representation in adaptation_set.findall(_add_ns('Representation')):
1590                     if is_drm_protected(representation):
1591                         continue
1592                     representation_attrib = adaptation_set.attrib.copy()
1593                     representation_attrib.update(representation.attrib)
1594                     # According to [1, 5.3.7.2, Table 9, page 41], @mimeType is mandatory
1595                     mime_type = representation_attrib['mimeType']
1596                     content_type = mime_type.split('/')[0]
1597                     if content_type == 'text':
1598                         # TODO implement WebVTT downloading
1599                         pass
1600                     elif content_type == 'video' or content_type == 'audio':
1601                         base_url = ''
1602                         for element in (representation, adaptation_set, period, mpd_doc):
1603                             base_url_e = element.find(_add_ns('BaseURL'))
1604                             if base_url_e is not None:
1605                                 base_url = base_url_e.text + base_url
1606                                 if re.match(r'^https?://', base_url):
1607                                     break
1608                         if mpd_base_url and not re.match(r'^https?://', base_url):
1609                             if not mpd_base_url.endswith('/') and not base_url.startswith('/'):
1610                                 mpd_base_url += '/'
1611                             base_url = mpd_base_url + base_url
1612                         representation_id = representation_attrib.get('id')
1613                         lang = representation_attrib.get('lang')
1614                         url_el = representation.find(_add_ns('BaseURL'))
1615                         filesize = int_or_none(url_el.attrib.get('{http://youtube.com/yt/2012/10/10}contentLength') if url_el is not None else None)
1616                         f = {
1617                             'format_id': '%s-%s' % (mpd_id, representation_id) if mpd_id else representation_id,
1618                             'url': base_url,
1619                             'ext': mimetype2ext(mime_type),
1620                             'width': int_or_none(representation_attrib.get('width')),
1621                             'height': int_or_none(representation_attrib.get('height')),
1622                             'tbr': int_or_none(representation_attrib.get('bandwidth'), 1000),
1623                             'asr': int_or_none(representation_attrib.get('audioSamplingRate')),
1624                             'fps': int_or_none(representation_attrib.get('frameRate')),
1625                             'vcodec': 'none' if content_type == 'audio' else representation_attrib.get('codecs'),
1626                             'acodec': 'none' if content_type == 'video' else representation_attrib.get('codecs'),
1627                             'language': lang if lang not in ('mul', 'und', 'zxx', 'mis') else None,
1628                             'format_note': 'DASH %s' % content_type,
1629                             'filesize': filesize,
1630                         }
1631                         representation_ms_info = extract_multisegment_info(representation, adaption_set_ms_info)
1632                         if 'segment_urls' not in representation_ms_info and 'media_template' in representation_ms_info:
1633                             if 'total_number' not in representation_ms_info and 'segment_duration':
1634                                 segment_duration = float(representation_ms_info['segment_duration']) / float(representation_ms_info['timescale'])
1635                                 representation_ms_info['total_number'] = int(math.ceil(float(period_duration) / segment_duration))
1636                             media_template = representation_ms_info['media_template']
1637                             media_template = media_template.replace('$RepresentationID$', representation_id)
1638                             media_template = re.sub(r'\$(Number|Bandwidth|Time)\$', r'%(\1)d', media_template)
1639                             media_template = re.sub(r'\$(Number|Bandwidth|Time)%([^$]+)\$', r'%(\1)\2', media_template)
1640                             media_template.replace('$$', '$')
1641
1642                             # As per [1, 5.3.9.4.4, Table 16, page 55] $Number$ and $Time$
1643                             # can't be used at the same time
1644                             if '%(Number' in media_template:
1645                                 representation_ms_info['segment_urls'] = [
1646                                     media_template % {
1647                                         'Number': segment_number,
1648                                         'Bandwidth': representation_attrib.get('bandwidth'),
1649                                     }
1650                                     for segment_number in range(
1651                                         representation_ms_info['start_number'],
1652                                         representation_ms_info['total_number'] + representation_ms_info['start_number'])]
1653                             else:
1654                                 representation_ms_info['segment_urls'] = []
1655                                 segment_time = 0
1656
1657                                 def add_segment_url():
1658                                     representation_ms_info['segment_urls'].append(
1659                                         media_template % {
1660                                             'Time': segment_time,
1661                                             'Bandwidth': representation_attrib.get('bandwidth'),
1662                                         }
1663                                     )
1664
1665                                 for num, s in enumerate(representation_ms_info['s']):
1666                                     segment_time = s.get('t') or segment_time
1667                                     add_segment_url()
1668                                     for r in range(s.get('r', 0)):
1669                                         segment_time += s['d']
1670                                         add_segment_url()
1671                                     segment_time += s['d']
1672                         if 'segment_urls' in representation_ms_info:
1673                             f.update({
1674                                 'segment_urls': representation_ms_info['segment_urls'],
1675                                 'protocol': 'http_dash_segments',
1676                             })
1677                             if 'initialization_url' in representation_ms_info:
1678                                 initialization_url = representation_ms_info['initialization_url'].replace('$RepresentationID$', representation_id)
1679                                 f.update({
1680                                     'initialization_url': initialization_url,
1681                                 })
1682                                 if not f.get('url'):
1683                                     f['url'] = initialization_url
1684                         try:
1685                             existing_format = next(
1686                                 fo for fo in formats
1687                                 if fo['format_id'] == representation_id)
1688                         except StopIteration:
1689                             full_info = formats_dict.get(representation_id, {}).copy()
1690                             full_info.update(f)
1691                             formats.append(full_info)
1692                         else:
1693                             existing_format.update(f)
1694                     else:
1695                         self.report_warning('Unknown MIME type %s in DASH manifest' % mime_type)
1696         return formats
1697
1698     def _parse_html5_media_entries(self, base_url, webpage):
1699         def absolute_url(video_url):
1700             return compat_urlparse.urljoin(base_url, video_url)
1701
1702         def parse_content_type(content_type):
1703             if not content_type:
1704                 return {}
1705             ctr = re.search(r'(?P<mimetype>[^/]+/[^;]+)(?:;\s*codecs="?(?P<codecs>[^"]+))?', content_type)
1706             if ctr:
1707                 mimetype, codecs = ctr.groups()
1708                 f = parse_codecs(codecs)
1709                 f['ext'] = mimetype2ext(mimetype)
1710                 return f
1711             return {}
1712
1713         entries = []
1714         for media_tag, media_type, media_content in re.findall(r'(?s)(<(?P<tag>video|audio)[^>]*>)(.*?)</(?P=tag)>', webpage):
1715             media_info = {
1716                 'formats': [],
1717                 'subtitles': {},
1718             }
1719             media_attributes = extract_attributes(media_tag)
1720             src = media_attributes.get('src')
1721             if src:
1722                 media_info['formats'].append({
1723                     'url': absolute_url(src),
1724                     'vcodec': 'none' if media_type == 'audio' else None,
1725                 })
1726             media_info['thumbnail'] = media_attributes.get('poster')
1727             if media_content:
1728                 for source_tag in re.findall(r'<source[^>]+>', media_content):
1729                     source_attributes = extract_attributes(source_tag)
1730                     src = source_attributes.get('src')
1731                     if not src:
1732                         continue
1733                     f = parse_content_type(source_attributes.get('type'))
1734                     f.update({
1735                         'url': absolute_url(src),
1736                         'vcodec': 'none' if media_type == 'audio' else None,
1737                     })
1738                     media_info['formats'].append(f)
1739                 for track_tag in re.findall(r'<track[^>]+>', media_content):
1740                     track_attributes = extract_attributes(track_tag)
1741                     kind = track_attributes.get('kind')
1742                     if not kind or kind == 'subtitles':
1743                         src = track_attributes.get('src')
1744                         if not src:
1745                             continue
1746                         lang = track_attributes.get('srclang') or track_attributes.get('lang') or track_attributes.get('label')
1747                         media_info['subtitles'].setdefault(lang, []).append({
1748                             'url': absolute_url(src),
1749                         })
1750             if media_info['formats']:
1751                 entries.append(media_info)
1752         return entries
1753
1754     def _live_title(self, name):
1755         """ Generate the title for a live video """
1756         now = datetime.datetime.now()
1757         now_str = now.strftime('%Y-%m-%d %H:%M')
1758         return name + ' ' + now_str
1759
1760     def _int(self, v, name, fatal=False, **kwargs):
1761         res = int_or_none(v, **kwargs)
1762         if 'get_attr' in kwargs:
1763             print(getattr(v, kwargs['get_attr']))
1764         if res is None:
1765             msg = 'Failed to extract %s: Could not parse value %r' % (name, v)
1766             if fatal:
1767                 raise ExtractorError(msg)
1768             else:
1769                 self._downloader.report_warning(msg)
1770         return res
1771
1772     def _float(self, v, name, fatal=False, **kwargs):
1773         res = float_or_none(v, **kwargs)
1774         if res is None:
1775             msg = 'Failed to extract %s: Could not parse value %r' % (name, v)
1776             if fatal:
1777                 raise ExtractorError(msg)
1778             else:
1779                 self._downloader.report_warning(msg)
1780         return res
1781
1782     def _set_cookie(self, domain, name, value, expire_time=None):
1783         cookie = compat_cookiejar.Cookie(
1784             0, name, value, None, None, domain, None,
1785             None, '/', True, False, expire_time, '', None, None, None)
1786         self._downloader.cookiejar.set_cookie(cookie)
1787
1788     def _get_cookies(self, url):
1789         """ Return a compat_cookies.SimpleCookie with the cookies for the url """
1790         req = sanitized_Request(url)
1791         self._downloader.cookiejar.add_cookie_header(req)
1792         return compat_cookies.SimpleCookie(req.get_header('Cookie'))
1793
1794     def get_testcases(self, include_onlymatching=False):
1795         t = getattr(self, '_TEST', None)
1796         if t:
1797             assert not hasattr(self, '_TESTS'), \
1798                 '%s has _TEST and _TESTS' % type(self).__name__
1799             tests = [t]
1800         else:
1801             tests = getattr(self, '_TESTS', [])
1802         for t in tests:
1803             if not include_onlymatching and t.get('only_matching', False):
1804                 continue
1805             t['name'] = type(self).__name__[:-len('IE')]
1806             yield t
1807
1808     def is_suitable(self, age_limit):
1809         """ Test whether the extractor is generally suitable for the given
1810         age limit (i.e. pornographic sites are not, all others usually are) """
1811
1812         any_restricted = False
1813         for tc in self.get_testcases(include_onlymatching=False):
1814             if tc.get('playlist', []):
1815                 tc = tc['playlist'][0]
1816             is_restricted = age_restricted(
1817                 tc.get('info_dict', {}).get('age_limit'), age_limit)
1818             if not is_restricted:
1819                 return True
1820             any_restricted = any_restricted or is_restricted
1821         return not any_restricted
1822
1823     def extract_subtitles(self, *args, **kwargs):
1824         if (self._downloader.params.get('writesubtitles', False) or
1825                 self._downloader.params.get('listsubtitles')):
1826             return self._get_subtitles(*args, **kwargs)
1827         return {}
1828
1829     def _get_subtitles(self, *args, **kwargs):
1830         raise NotImplementedError('This method must be implemented by subclasses')
1831
1832     @staticmethod
1833     def _merge_subtitle_items(subtitle_list1, subtitle_list2):
1834         """ Merge subtitle items for one language. Items with duplicated URLs
1835         will be dropped. """
1836         list1_urls = set([item['url'] for item in subtitle_list1])
1837         ret = list(subtitle_list1)
1838         ret.extend([item for item in subtitle_list2 if item['url'] not in list1_urls])
1839         return ret
1840
1841     @classmethod
1842     def _merge_subtitles(cls, subtitle_dict1, subtitle_dict2):
1843         """ Merge two subtitle dictionaries, language by language. """
1844         ret = dict(subtitle_dict1)
1845         for lang in subtitle_dict2:
1846             ret[lang] = cls._merge_subtitle_items(subtitle_dict1.get(lang, []), subtitle_dict2[lang])
1847         return ret
1848
1849     def extract_automatic_captions(self, *args, **kwargs):
1850         if (self._downloader.params.get('writeautomaticsub', False) or
1851                 self._downloader.params.get('listsubtitles')):
1852             return self._get_automatic_captions(*args, **kwargs)
1853         return {}
1854
1855     def _get_automatic_captions(self, *args, **kwargs):
1856         raise NotImplementedError('This method must be implemented by subclasses')
1857
1858     def mark_watched(self, *args, **kwargs):
1859         if (self._downloader.params.get('mark_watched', False) and
1860                 (self._get_login_info()[0] is not None or
1861                     self._downloader.params.get('cookiefile') is not None)):
1862             self._mark_watched(*args, **kwargs)
1863
1864     def _mark_watched(self, *args, **kwargs):
1865         raise NotImplementedError('This method must be implemented by subclasses')
1866
1867     def geo_verification_headers(self):
1868         headers = {}
1869         geo_verification_proxy = self._downloader.params.get('geo_verification_proxy')
1870         if geo_verification_proxy:
1871             headers['Ytdl-request-proxy'] = geo_verification_proxy
1872         return headers
1873
1874
1875 class SearchInfoExtractor(InfoExtractor):
1876     """
1877     Base class for paged search queries extractors.
1878     They accept URLs in the format _SEARCH_KEY(|all|[0-9]):{query}
1879     Instances should define _SEARCH_KEY and _MAX_RESULTS.
1880     """
1881
1882     @classmethod
1883     def _make_valid_url(cls):
1884         return r'%s(?P<prefix>|[1-9][0-9]*|all):(?P<query>[\s\S]+)' % cls._SEARCH_KEY
1885
1886     @classmethod
1887     def suitable(cls, url):
1888         return re.match(cls._make_valid_url(), url) is not None
1889
1890     def _real_extract(self, query):
1891         mobj = re.match(self._make_valid_url(), query)
1892         if mobj is None:
1893             raise ExtractorError('Invalid search query "%s"' % query)
1894
1895         prefix = mobj.group('prefix')
1896         query = mobj.group('query')
1897         if prefix == '':
1898             return self._get_n_results(query, 1)
1899         elif prefix == 'all':
1900             return self._get_n_results(query, self._MAX_RESULTS)
1901         else:
1902             n = int(prefix)
1903             if n <= 0:
1904                 raise ExtractorError('invalid download number %s for query "%s"' % (n, query))
1905             elif n > self._MAX_RESULTS:
1906                 self._downloader.report_warning('%s returns max %i results (you requested %i)' % (self._SEARCH_KEY, self._MAX_RESULTS, n))
1907                 n = self._MAX_RESULTS
1908             return self._get_n_results(query, n)
1909
1910     def _get_n_results(self, query, n):
1911         """Get a specified number of results for a query"""
1912         raise NotImplementedError('This method must be implemented by subclasses')
1913
1914     @property
1915     def SEARCH_KEY(self):
1916         return self._SEARCH_KEY