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