Merge pull request #12909 from remitamine/raw-sub
[youtube-dl] / youtube_dl / YoutubeDL.py
1 #!/usr/bin/env python
2 # coding: utf-8
3
4 from __future__ import absolute_import, unicode_literals
5
6 import collections
7 import contextlib
8 import copy
9 import datetime
10 import errno
11 import fileinput
12 import io
13 import itertools
14 import json
15 import locale
16 import operator
17 import os
18 import platform
19 import re
20 import shutil
21 import subprocess
22 import socket
23 import sys
24 import time
25 import tokenize
26 import traceback
27 import random
28
29 from string import ascii_letters
30
31 from .compat import (
32     compat_basestring,
33     compat_cookiejar,
34     compat_get_terminal_size,
35     compat_http_client,
36     compat_kwargs,
37     compat_numeric_types,
38     compat_os_name,
39     compat_str,
40     compat_tokenize_tokenize,
41     compat_urllib_error,
42     compat_urllib_request,
43     compat_urllib_request_DataHandler,
44 )
45 from .utils import (
46     age_restricted,
47     args_to_str,
48     ContentTooShortError,
49     date_from_str,
50     DateRange,
51     DEFAULT_OUTTMPL,
52     determine_ext,
53     determine_protocol,
54     DownloadError,
55     encode_compat_str,
56     encodeFilename,
57     error_to_compat_str,
58     expand_path,
59     ExtractorError,
60     format_bytes,
61     formatSeconds,
62     GeoRestrictedError,
63     int_or_none,
64     ISO3166Utils,
65     locked_file,
66     make_HTTPS_handler,
67     MaxDownloadsReached,
68     PagedList,
69     parse_filesize,
70     PerRequestProxyHandler,
71     platform_name,
72     PostProcessingError,
73     preferredencoding,
74     prepend_extension,
75     register_socks_protocols,
76     render_table,
77     replace_extension,
78     SameFileError,
79     sanitize_filename,
80     sanitize_path,
81     sanitize_url,
82     sanitized_Request,
83     std_headers,
84     subtitles_filename,
85     UnavailableVideoError,
86     url_basename,
87     version_tuple,
88     write_json_file,
89     write_string,
90     YoutubeDLCookieProcessor,
91     YoutubeDLHandler,
92 )
93 from .cache import Cache
94 from .extractor import get_info_extractor, gen_extractor_classes, _LAZY_LOADER
95 from .downloader import get_suitable_downloader
96 from .downloader.rtmp import rtmpdump_version
97 from .postprocessor import (
98     FFmpegFixupM3u8PP,
99     FFmpegFixupM4aPP,
100     FFmpegFixupStretchedPP,
101     FFmpegMergerPP,
102     FFmpegPostProcessor,
103     get_postprocessor,
104 )
105 from .version import __version__
106
107 if compat_os_name == 'nt':
108     import ctypes
109
110
111 class YoutubeDL(object):
112     """YoutubeDL class.
113
114     YoutubeDL objects are the ones responsible of downloading the
115     actual video file and writing it to disk if the user has requested
116     it, among some other tasks. In most cases there should be one per
117     program. As, given a video URL, the downloader doesn't know how to
118     extract all the needed information, task that InfoExtractors do, it
119     has to pass the URL to one of them.
120
121     For this, YoutubeDL objects have a method that allows
122     InfoExtractors to be registered in a given order. When it is passed
123     a URL, the YoutubeDL object handles it to the first InfoExtractor it
124     finds that reports being able to handle it. The InfoExtractor extracts
125     all the information about the video or videos the URL refers to, and
126     YoutubeDL process the extracted information, possibly using a File
127     Downloader to download the video.
128
129     YoutubeDL objects accept a lot of parameters. In order not to saturate
130     the object constructor with arguments, it receives a dictionary of
131     options instead. These options are available through the params
132     attribute for the InfoExtractors to use. The YoutubeDL also
133     registers itself as the downloader in charge for the InfoExtractors
134     that are added to it, so this is a "mutual registration".
135
136     Available options:
137
138     username:          Username for authentication purposes.
139     password:          Password for authentication purposes.
140     videopassword:     Password for accessing a video.
141     ap_mso:            Adobe Pass multiple-system operator identifier.
142     ap_username:       Multiple-system operator account username.
143     ap_password:       Multiple-system operator account password.
144     usenetrc:          Use netrc for authentication instead.
145     verbose:           Print additional info to stdout.
146     quiet:             Do not print messages to stdout.
147     no_warnings:       Do not print out anything for warnings.
148     forceurl:          Force printing final URL.
149     forcetitle:        Force printing title.
150     forceid:           Force printing ID.
151     forcethumbnail:    Force printing thumbnail URL.
152     forcedescription:  Force printing description.
153     forcefilename:     Force printing final filename.
154     forceduration:     Force printing duration.
155     forcejson:         Force printing info_dict as JSON.
156     dump_single_json:  Force printing the info_dict of the whole playlist
157                        (or video) as a single JSON line.
158     simulate:          Do not download the video files.
159     format:            Video format code. See options.py for more information.
160     outtmpl:           Template for output names.
161     restrictfilenames: Do not allow "&" and spaces in file names
162     ignoreerrors:      Do not stop on download errors.
163     force_generic_extractor: Force downloader to use the generic extractor
164     nooverwrites:      Prevent overwriting files.
165     playliststart:     Playlist item to start at.
166     playlistend:       Playlist item to end at.
167     playlist_items:    Specific indices of playlist to download.
168     playlistreverse:   Download playlist items in reverse order.
169     playlistrandom:    Download playlist items in random order.
170     matchtitle:        Download only matching titles.
171     rejecttitle:       Reject downloads for matching titles.
172     logger:            Log messages to a logging.Logger instance.
173     logtostderr:       Log messages to stderr instead of stdout.
174     writedescription:  Write the video description to a .description file
175     writeinfojson:     Write the video description to a .info.json file
176     writeannotations:  Write the video annotations to a .annotations.xml file
177     writethumbnail:    Write the thumbnail image to a file
178     write_all_thumbnails:  Write all thumbnail formats to files
179     writesubtitles:    Write the video subtitles to a file
180     writeautomaticsub: Write the automatically generated subtitles to a file
181     allsubtitles:      Downloads all the subtitles of the video
182                        (requires writesubtitles or writeautomaticsub)
183     listsubtitles:     Lists all available subtitles for the video
184     subtitlesformat:   The format code for subtitles
185     subtitleslangs:    List of languages of the subtitles to download
186     keepvideo:         Keep the video file after post-processing
187     daterange:         A DateRange object, download only if the upload_date is in the range.
188     skip_download:     Skip the actual download of the video file
189     cachedir:          Location of the cache files in the filesystem.
190                        False to disable filesystem cache.
191     noplaylist:        Download single video instead of a playlist if in doubt.
192     age_limit:         An integer representing the user's age in years.
193                        Unsuitable videos for the given age are skipped.
194     min_views:         An integer representing the minimum view count the video
195                        must have in order to not be skipped.
196                        Videos without view count information are always
197                        downloaded. None for no limit.
198     max_views:         An integer representing the maximum view count.
199                        Videos that are more popular than that are not
200                        downloaded.
201                        Videos without view count information are always
202                        downloaded. None for no limit.
203     download_archive:  File name of a file where all downloads are recorded.
204                        Videos already present in the file are not downloaded
205                        again.
206     cookiefile:        File name where cookies should be read from and dumped to.
207     nocheckcertificate:Do not verify SSL certificates
208     prefer_insecure:   Use HTTP instead of HTTPS to retrieve information.
209                        At the moment, this is only supported by YouTube.
210     proxy:             URL of the proxy server to use
211     geo_verification_proxy:  URL of the proxy to use for IP address verification
212                        on geo-restricted sites. (Experimental)
213     socket_timeout:    Time to wait for unresponsive hosts, in seconds
214     bidi_workaround:   Work around buggy terminals without bidirectional text
215                        support, using fridibi
216     debug_printtraffic:Print out sent and received HTTP traffic
217     include_ads:       Download ads as well
218     default_search:    Prepend this string if an input url is not valid.
219                        'auto' for elaborate guessing
220     encoding:          Use this encoding instead of the system-specified.
221     extract_flat:      Do not resolve URLs, return the immediate result.
222                        Pass in 'in_playlist' to only show this behavior for
223                        playlist items.
224     postprocessors:    A list of dictionaries, each with an entry
225                        * key:  The name of the postprocessor. See
226                                youtube_dl/postprocessor/__init__.py for a list.
227                        as well as any further keyword arguments for the
228                        postprocessor.
229     progress_hooks:    A list of functions that get called on download
230                        progress, with a dictionary with the entries
231                        * status: One of "downloading", "error", or "finished".
232                                  Check this first and ignore unknown values.
233
234                        If status is one of "downloading", or "finished", the
235                        following properties may also be present:
236                        * filename: The final filename (always present)
237                        * tmpfilename: The filename we're currently writing to
238                        * downloaded_bytes: Bytes on disk
239                        * total_bytes: Size of the whole file, None if unknown
240                        * total_bytes_estimate: Guess of the eventual file size,
241                                                None if unavailable.
242                        * elapsed: The number of seconds since download started.
243                        * eta: The estimated time in seconds, None if unknown
244                        * speed: The download speed in bytes/second, None if
245                                 unknown
246                        * fragment_index: The counter of the currently
247                                          downloaded video fragment.
248                        * fragment_count: The number of fragments (= individual
249                                          files that will be merged)
250
251                        Progress hooks are guaranteed to be called at least once
252                        (with status "finished") if the download is successful.
253     merge_output_format: Extension to use when merging formats.
254     fixup:             Automatically correct known faults of the file.
255                        One of:
256                        - "never": do nothing
257                        - "warn": only emit a warning
258                        - "detect_or_warn": check whether we can do anything
259                                            about it, warn otherwise (default)
260     source_address:    (Experimental) Client-side IP address to bind to.
261     call_home:         Boolean, true iff we are allowed to contact the
262                        youtube-dl servers for debugging.
263     sleep_interval:    Number of seconds to sleep before each download when
264                        used alone or a lower bound of a range for randomized
265                        sleep before each download (minimum possible number
266                        of seconds to sleep) when used along with
267                        max_sleep_interval.
268     max_sleep_interval:Upper bound of a range for randomized sleep before each
269                        download (maximum possible number of seconds to sleep).
270                        Must only be used along with sleep_interval.
271                        Actual sleep time will be a random float from range
272                        [sleep_interval; max_sleep_interval].
273     listformats:       Print an overview of available video formats and exit.
274     list_thumbnails:   Print a table of all thumbnails and exit.
275     match_filter:      A function that gets called with the info_dict of
276                        every video.
277                        If it returns a message, the video is ignored.
278                        If it returns None, the video is downloaded.
279                        match_filter_func in utils.py is one example for this.
280     no_color:          Do not emit color codes in output.
281     geo_bypass:        Bypass geographic restriction via faking X-Forwarded-For
282                        HTTP header (experimental)
283     geo_bypass_country:
284                        Two-letter ISO 3166-2 country code that will be used for
285                        explicit geographic restriction bypassing via faking
286                        X-Forwarded-For HTTP header (experimental)
287
288     The following options determine which downloader is picked:
289     external_downloader: Executable of the external downloader to call.
290                        None or unset for standard (built-in) downloader.
291     hls_prefer_native: Use the native HLS downloader instead of ffmpeg/avconv
292                        if True, otherwise use ffmpeg/avconv if False, otherwise
293                        use downloader suggested by extractor if None.
294
295     The following parameters are not used by YoutubeDL itself, they are used by
296     the downloader (see youtube_dl/downloader/common.py):
297     nopart, updatetime, buffersize, ratelimit, min_filesize, max_filesize, test,
298     noresizebuffer, retries, continuedl, noprogress, consoletitle,
299     xattr_set_filesize, external_downloader_args, hls_use_mpegts.
300
301     The following options are used by the post processors:
302     prefer_ffmpeg:     If True, use ffmpeg instead of avconv if both are available,
303                        otherwise prefer avconv.
304     postprocessor_args: A list of additional command-line arguments for the
305                         postprocessor.
306     """
307
308     _NUMERIC_FIELDS = set((
309         'width', 'height', 'tbr', 'abr', 'asr', 'vbr', 'fps', 'filesize', 'filesize_approx',
310         'timestamp', 'upload_year', 'upload_month', 'upload_day',
311         'duration', 'view_count', 'like_count', 'dislike_count', 'repost_count',
312         'average_rating', 'comment_count', 'age_limit',
313         'start_time', 'end_time',
314         'chapter_number', 'season_number', 'episode_number',
315         'track_number', 'disc_number', 'release_year',
316         'playlist_index',
317     ))
318
319     params = None
320     _ies = []
321     _pps = []
322     _download_retcode = None
323     _num_downloads = None
324     _screen_file = None
325
326     def __init__(self, params=None, auto_init=True):
327         """Create a FileDownloader object with the given options."""
328         if params is None:
329             params = {}
330         self._ies = []
331         self._ies_instances = {}
332         self._pps = []
333         self._progress_hooks = []
334         self._download_retcode = 0
335         self._num_downloads = 0
336         self._screen_file = [sys.stdout, sys.stderr][params.get('logtostderr', False)]
337         self._err_file = sys.stderr
338         self.params = {
339             # Default parameters
340             'nocheckcertificate': False,
341         }
342         self.params.update(params)
343         self.cache = Cache(self)
344
345         def check_deprecated(param, option, suggestion):
346             if self.params.get(param) is not None:
347                 self.report_warning(
348                     '%s is deprecated. Use %s instead.' % (option, suggestion))
349                 return True
350             return False
351
352         if check_deprecated('cn_verification_proxy', '--cn-verification-proxy', '--geo-verification-proxy'):
353             if self.params.get('geo_verification_proxy') is None:
354                 self.params['geo_verification_proxy'] = self.params['cn_verification_proxy']
355
356         check_deprecated('autonumber_size', '--autonumber-size', 'output template with %(autonumber)0Nd, where N in the number of digits')
357         check_deprecated('autonumber', '--auto-number', '-o "%(autonumber)s-%(title)s.%(ext)s"')
358         check_deprecated('usetitle', '--title', '-o "%(title)s-%(id)s.%(ext)s"')
359
360         if params.get('bidi_workaround', False):
361             try:
362                 import pty
363                 master, slave = pty.openpty()
364                 width = compat_get_terminal_size().columns
365                 if width is None:
366                     width_args = []
367                 else:
368                     width_args = ['-w', str(width)]
369                 sp_kwargs = dict(
370                     stdin=subprocess.PIPE,
371                     stdout=slave,
372                     stderr=self._err_file)
373                 try:
374                     self._output_process = subprocess.Popen(
375                         ['bidiv'] + width_args, **sp_kwargs
376                     )
377                 except OSError:
378                     self._output_process = subprocess.Popen(
379                         ['fribidi', '-c', 'UTF-8'] + width_args, **sp_kwargs)
380                 self._output_channel = os.fdopen(master, 'rb')
381             except OSError as ose:
382                 if ose.errno == errno.ENOENT:
383                     self.report_warning('Could not find fribidi executable, ignoring --bidi-workaround . Make sure that  fribidi  is an executable file in one of the directories in your $PATH.')
384                 else:
385                     raise
386
387         if (sys.platform != 'win32' and
388                 sys.getfilesystemencoding() in ['ascii', 'ANSI_X3.4-1968'] and
389                 not params.get('restrictfilenames', False)):
390             # Unicode filesystem API will throw errors (#1474, #13027)
391             self.report_warning(
392                 'Assuming --restrict-filenames since file system encoding '
393                 'cannot encode all characters. '
394                 'Set the LC_ALL environment variable to fix this.')
395             self.params['restrictfilenames'] = True
396
397         if isinstance(params.get('outtmpl'), bytes):
398             self.report_warning(
399                 'Parameter outtmpl is bytes, but should be a unicode string. '
400                 'Put  from __future__ import unicode_literals  at the top of your code file or consider switching to Python 3.x.')
401
402         self._setup_opener()
403
404         if auto_init:
405             self.print_debug_header()
406             self.add_default_info_extractors()
407
408         for pp_def_raw in self.params.get('postprocessors', []):
409             pp_class = get_postprocessor(pp_def_raw['key'])
410             pp_def = dict(pp_def_raw)
411             del pp_def['key']
412             pp = pp_class(self, **compat_kwargs(pp_def))
413             self.add_post_processor(pp)
414
415         for ph in self.params.get('progress_hooks', []):
416             self.add_progress_hook(ph)
417
418         register_socks_protocols()
419
420     def warn_if_short_id(self, argv):
421         # short YouTube ID starting with dash?
422         idxs = [
423             i for i, a in enumerate(argv)
424             if re.match(r'^-[0-9A-Za-z_-]{10}$', a)]
425         if idxs:
426             correct_argv = (
427                 ['youtube-dl'] +
428                 [a for i, a in enumerate(argv) if i not in idxs] +
429                 ['--'] + [argv[i] for i in idxs]
430             )
431             self.report_warning(
432                 'Long argument string detected. '
433                 'Use -- to separate parameters and URLs, like this:\n%s\n' %
434                 args_to_str(correct_argv))
435
436     def add_info_extractor(self, ie):
437         """Add an InfoExtractor object to the end of the list."""
438         self._ies.append(ie)
439         if not isinstance(ie, type):
440             self._ies_instances[ie.ie_key()] = ie
441             ie.set_downloader(self)
442
443     def get_info_extractor(self, ie_key):
444         """
445         Get an instance of an IE with name ie_key, it will try to get one from
446         the _ies list, if there's no instance it will create a new one and add
447         it to the extractor list.
448         """
449         ie = self._ies_instances.get(ie_key)
450         if ie is None:
451             ie = get_info_extractor(ie_key)()
452             self.add_info_extractor(ie)
453         return ie
454
455     def add_default_info_extractors(self):
456         """
457         Add the InfoExtractors returned by gen_extractors to the end of the list
458         """
459         for ie in gen_extractor_classes():
460             self.add_info_extractor(ie)
461
462     def add_post_processor(self, pp):
463         """Add a PostProcessor object to the end of the chain."""
464         self._pps.append(pp)
465         pp.set_downloader(self)
466
467     def add_progress_hook(self, ph):
468         """Add the progress hook (currently only for the file downloader)"""
469         self._progress_hooks.append(ph)
470
471     def _bidi_workaround(self, message):
472         if not hasattr(self, '_output_channel'):
473             return message
474
475         assert hasattr(self, '_output_process')
476         assert isinstance(message, compat_str)
477         line_count = message.count('\n') + 1
478         self._output_process.stdin.write((message + '\n').encode('utf-8'))
479         self._output_process.stdin.flush()
480         res = ''.join(self._output_channel.readline().decode('utf-8')
481                       for _ in range(line_count))
482         return res[:-len('\n')]
483
484     def to_screen(self, message, skip_eol=False):
485         """Print message to stdout if not in quiet mode."""
486         return self.to_stdout(message, skip_eol, check_quiet=True)
487
488     def _write_string(self, s, out=None):
489         write_string(s, out=out, encoding=self.params.get('encoding'))
490
491     def to_stdout(self, message, skip_eol=False, check_quiet=False):
492         """Print message to stdout if not in quiet mode."""
493         if self.params.get('logger'):
494             self.params['logger'].debug(message)
495         elif not check_quiet or not self.params.get('quiet', False):
496             message = self._bidi_workaround(message)
497             terminator = ['\n', ''][skip_eol]
498             output = message + terminator
499
500             self._write_string(output, self._screen_file)
501
502     def to_stderr(self, message):
503         """Print message to stderr."""
504         assert isinstance(message, compat_str)
505         if self.params.get('logger'):
506             self.params['logger'].error(message)
507         else:
508             message = self._bidi_workaround(message)
509             output = message + '\n'
510             self._write_string(output, self._err_file)
511
512     def to_console_title(self, message):
513         if not self.params.get('consoletitle', False):
514             return
515         if compat_os_name == 'nt':
516             if ctypes.windll.kernel32.GetConsoleWindow():
517                 # c_wchar_p() might not be necessary if `message` is
518                 # already of type unicode()
519                 ctypes.windll.kernel32.SetConsoleTitleW(ctypes.c_wchar_p(message))
520         elif 'TERM' in os.environ:
521             self._write_string('\033]0;%s\007' % message, self._screen_file)
522
523     def save_console_title(self):
524         if not self.params.get('consoletitle', False):
525             return
526         if compat_os_name != 'nt' and 'TERM' in os.environ:
527             # Save the title on stack
528             self._write_string('\033[22;0t', self._screen_file)
529
530     def restore_console_title(self):
531         if not self.params.get('consoletitle', False):
532             return
533         if compat_os_name != 'nt' and 'TERM' in os.environ:
534             # Restore the title from stack
535             self._write_string('\033[23;0t', self._screen_file)
536
537     def __enter__(self):
538         self.save_console_title()
539         return self
540
541     def __exit__(self, *args):
542         self.restore_console_title()
543
544         if self.params.get('cookiefile') is not None:
545             self.cookiejar.save()
546
547     def trouble(self, message=None, tb=None):
548         """Determine action to take when a download problem appears.
549
550         Depending on if the downloader has been configured to ignore
551         download errors or not, this method may throw an exception or
552         not when errors are found, after printing the message.
553
554         tb, if given, is additional traceback information.
555         """
556         if message is not None:
557             self.to_stderr(message)
558         if self.params.get('verbose'):
559             if tb is None:
560                 if sys.exc_info()[0]:  # if .trouble has been called from an except block
561                     tb = ''
562                     if hasattr(sys.exc_info()[1], 'exc_info') and sys.exc_info()[1].exc_info[0]:
563                         tb += ''.join(traceback.format_exception(*sys.exc_info()[1].exc_info))
564                     tb += encode_compat_str(traceback.format_exc())
565                 else:
566                     tb_data = traceback.format_list(traceback.extract_stack())
567                     tb = ''.join(tb_data)
568             self.to_stderr(tb)
569         if not self.params.get('ignoreerrors', False):
570             if sys.exc_info()[0] and hasattr(sys.exc_info()[1], 'exc_info') and sys.exc_info()[1].exc_info[0]:
571                 exc_info = sys.exc_info()[1].exc_info
572             else:
573                 exc_info = sys.exc_info()
574             raise DownloadError(message, exc_info)
575         self._download_retcode = 1
576
577     def report_warning(self, message):
578         '''
579         Print the message to stderr, it will be prefixed with 'WARNING:'
580         If stderr is a tty file the 'WARNING:' will be colored
581         '''
582         if self.params.get('logger') is not None:
583             self.params['logger'].warning(message)
584         else:
585             if self.params.get('no_warnings'):
586                 return
587             if not self.params.get('no_color') and self._err_file.isatty() and compat_os_name != 'nt':
588                 _msg_header = '\033[0;33mWARNING:\033[0m'
589             else:
590                 _msg_header = 'WARNING:'
591             warning_message = '%s %s' % (_msg_header, message)
592             self.to_stderr(warning_message)
593
594     def report_error(self, message, tb=None):
595         '''
596         Do the same as trouble, but prefixes the message with 'ERROR:', colored
597         in red if stderr is a tty file.
598         '''
599         if not self.params.get('no_color') and self._err_file.isatty() and compat_os_name != 'nt':
600             _msg_header = '\033[0;31mERROR:\033[0m'
601         else:
602             _msg_header = 'ERROR:'
603         error_message = '%s %s' % (_msg_header, message)
604         self.trouble(error_message, tb)
605
606     def report_file_already_downloaded(self, file_name):
607         """Report file has already been fully downloaded."""
608         try:
609             self.to_screen('[download] %s has already been downloaded' % file_name)
610         except UnicodeEncodeError:
611             self.to_screen('[download] The file has already been downloaded')
612
613     def prepare_filename(self, info_dict):
614         """Generate the output filename."""
615         try:
616             template_dict = dict(info_dict)
617
618             template_dict['epoch'] = int(time.time())
619             autonumber_size = self.params.get('autonumber_size')
620             if autonumber_size is None:
621                 autonumber_size = 5
622             template_dict['autonumber'] = self.params.get('autonumber_start', 1) - 1 + self._num_downloads
623             if template_dict.get('resolution') is None:
624                 if template_dict.get('width') and template_dict.get('height'):
625                     template_dict['resolution'] = '%dx%d' % (template_dict['width'], template_dict['height'])
626                 elif template_dict.get('height'):
627                     template_dict['resolution'] = '%sp' % template_dict['height']
628                 elif template_dict.get('width'):
629                     template_dict['resolution'] = '%dx?' % template_dict['width']
630
631             sanitize = lambda k, v: sanitize_filename(
632                 compat_str(v),
633                 restricted=self.params.get('restrictfilenames'),
634                 is_id=(k == 'id' or k.endswith('_id')))
635             template_dict = dict((k, v if isinstance(v, compat_numeric_types) else sanitize(k, v))
636                                  for k, v in template_dict.items()
637                                  if v is not None and not isinstance(v, (list, tuple, dict)))
638             template_dict = collections.defaultdict(lambda: 'NA', template_dict)
639
640             outtmpl = self.params.get('outtmpl', DEFAULT_OUTTMPL)
641
642             # For fields playlist_index and autonumber convert all occurrences
643             # of %(field)s to %(field)0Nd for backward compatibility
644             field_size_compat_map = {
645                 'playlist_index': len(str(template_dict['n_entries'])),
646                 'autonumber': autonumber_size,
647             }
648             FIELD_SIZE_COMPAT_RE = r'(?<!%)%\((?P<field>autonumber|playlist_index)\)s'
649             mobj = re.search(FIELD_SIZE_COMPAT_RE, outtmpl)
650             if mobj:
651                 outtmpl = re.sub(
652                     FIELD_SIZE_COMPAT_RE,
653                     r'%%(\1)0%dd' % field_size_compat_map[mobj.group('field')],
654                     outtmpl)
655
656             # Missing numeric fields used together with integer presentation types
657             # in format specification will break the argument substitution since
658             # string 'NA' is returned for missing fields. We will patch output
659             # template for missing fields to meet string presentation type.
660             for numeric_field in self._NUMERIC_FIELDS:
661                 if numeric_field not in template_dict:
662                     # As of [1] format syntax is:
663                     #  %[mapping_key][conversion_flags][minimum_width][.precision][length_modifier]type
664                     # 1. https://docs.python.org/2/library/stdtypes.html#string-formatting
665                     FORMAT_RE = r'''(?x)
666                         (?<!%)
667                         %
668                         \({0}\)  # mapping key
669                         (?:[#0\-+ ]+)?  # conversion flags (optional)
670                         (?:\d+)?  # minimum field width (optional)
671                         (?:\.\d+)?  # precision (optional)
672                         [hlL]?  # length modifier (optional)
673                         [diouxXeEfFgGcrs%]  # conversion type
674                     '''
675                     outtmpl = re.sub(
676                         FORMAT_RE.format(numeric_field),
677                         r'%({0})s'.format(numeric_field), outtmpl)
678
679             # expand_path translates '%%' into '%' and '$$' into '$'
680             # correspondingly that is not what we want since we need to keep
681             # '%%' intact for template dict substitution step. Working around
682             # with boundary-alike separator hack.
683             sep = ''.join([random.choice(ascii_letters) for _ in range(32)])
684             outtmpl = outtmpl.replace('%%', '%{0}%'.format(sep)).replace('$$', '${0}$'.format(sep))
685
686             # outtmpl should be expand_path'ed before template dict substitution
687             # because meta fields may contain env variables we don't want to
688             # be expanded. For example, for outtmpl "%(title)s.%(ext)s" and
689             # title "Hello $PATH", we don't want `$PATH` to be expanded.
690             filename = expand_path(outtmpl).replace(sep, '') % template_dict
691
692             # Temporary fix for #4787
693             # 'Treat' all problem characters by passing filename through preferredencoding
694             # to workaround encoding issues with subprocess on python2 @ Windows
695             if sys.version_info < (3, 0) and sys.platform == 'win32':
696                 filename = encodeFilename(filename, True).decode(preferredencoding())
697             return sanitize_path(filename)
698         except ValueError as err:
699             self.report_error('Error in output template: ' + str(err) + ' (encoding: ' + repr(preferredencoding()) + ')')
700             return None
701
702     def _match_entry(self, info_dict, incomplete):
703         """ Returns None iff the file should be downloaded """
704
705         video_title = info_dict.get('title', info_dict.get('id', 'video'))
706         if 'title' in info_dict:
707             # This can happen when we're just evaluating the playlist
708             title = info_dict['title']
709             matchtitle = self.params.get('matchtitle', False)
710             if matchtitle:
711                 if not re.search(matchtitle, title, re.IGNORECASE):
712                     return '"' + title + '" title did not match pattern "' + matchtitle + '"'
713             rejecttitle = self.params.get('rejecttitle', False)
714             if rejecttitle:
715                 if re.search(rejecttitle, title, re.IGNORECASE):
716                     return '"' + title + '" title matched reject pattern "' + rejecttitle + '"'
717         date = info_dict.get('upload_date')
718         if date is not None:
719             dateRange = self.params.get('daterange', DateRange())
720             if date not in dateRange:
721                 return '%s upload date is not in range %s' % (date_from_str(date).isoformat(), dateRange)
722         view_count = info_dict.get('view_count')
723         if view_count is not None:
724             min_views = self.params.get('min_views')
725             if min_views is not None and view_count < min_views:
726                 return 'Skipping %s, because it has not reached minimum view count (%d/%d)' % (video_title, view_count, min_views)
727             max_views = self.params.get('max_views')
728             if max_views is not None and view_count > max_views:
729                 return 'Skipping %s, because it has exceeded the maximum view count (%d/%d)' % (video_title, view_count, max_views)
730         if age_restricted(info_dict.get('age_limit'), self.params.get('age_limit')):
731             return 'Skipping "%s" because it is age restricted' % video_title
732         if self.in_download_archive(info_dict):
733             return '%s has already been recorded in archive' % video_title
734
735         if not incomplete:
736             match_filter = self.params.get('match_filter')
737             if match_filter is not None:
738                 ret = match_filter(info_dict)
739                 if ret is not None:
740                     return ret
741
742         return None
743
744     @staticmethod
745     def add_extra_info(info_dict, extra_info):
746         '''Set the keys from extra_info in info dict if they are missing'''
747         for key, value in extra_info.items():
748             info_dict.setdefault(key, value)
749
750     def extract_info(self, url, download=True, ie_key=None, extra_info={},
751                      process=True, force_generic_extractor=False):
752         '''
753         Returns a list with a dictionary for each video we find.
754         If 'download', also downloads the videos.
755         extra_info is a dict containing the extra values to add to each result
756         '''
757
758         if not ie_key and force_generic_extractor:
759             ie_key = 'Generic'
760
761         if ie_key:
762             ies = [self.get_info_extractor(ie_key)]
763         else:
764             ies = self._ies
765
766         for ie in ies:
767             if not ie.suitable(url):
768                 continue
769
770             ie = self.get_info_extractor(ie.ie_key())
771             if not ie.working():
772                 self.report_warning('The program functionality for this site has been marked as broken, '
773                                     'and will probably not work.')
774
775             try:
776                 ie_result = ie.extract(url)
777                 if ie_result is None:  # Finished already (backwards compatibility; listformats and friends should be moved here)
778                     break
779                 if isinstance(ie_result, list):
780                     # Backwards compatibility: old IE result format
781                     ie_result = {
782                         '_type': 'compat_list',
783                         'entries': ie_result,
784                     }
785                 self.add_default_extra_info(ie_result, ie, url)
786                 if process:
787                     return self.process_ie_result(ie_result, download, extra_info)
788                 else:
789                     return ie_result
790             except GeoRestrictedError as e:
791                 msg = e.msg
792                 if e.countries:
793                     msg += '\nThis video is available in %s.' % ', '.join(
794                         map(ISO3166Utils.short2full, e.countries))
795                 msg += '\nYou might want to use a VPN or a proxy server (with --proxy) to workaround.'
796                 self.report_error(msg)
797                 break
798             except ExtractorError as e:  # An error we somewhat expected
799                 self.report_error(compat_str(e), e.format_traceback())
800                 break
801             except MaxDownloadsReached:
802                 raise
803             except Exception as e:
804                 if self.params.get('ignoreerrors', False):
805                     self.report_error(error_to_compat_str(e), tb=encode_compat_str(traceback.format_exc()))
806                     break
807                 else:
808                     raise
809         else:
810             self.report_error('no suitable InfoExtractor for URL %s' % url)
811
812     def add_default_extra_info(self, ie_result, ie, url):
813         self.add_extra_info(ie_result, {
814             'extractor': ie.IE_NAME,
815             'webpage_url': url,
816             'webpage_url_basename': url_basename(url),
817             'extractor_key': ie.ie_key(),
818         })
819
820     def process_ie_result(self, ie_result, download=True, extra_info={}):
821         """
822         Take the result of the ie(may be modified) and resolve all unresolved
823         references (URLs, playlist items).
824
825         It will also download the videos if 'download'.
826         Returns the resolved ie_result.
827         """
828         result_type = ie_result.get('_type', 'video')
829
830         if result_type in ('url', 'url_transparent'):
831             ie_result['url'] = sanitize_url(ie_result['url'])
832             extract_flat = self.params.get('extract_flat', False)
833             if ((extract_flat == 'in_playlist' and 'playlist' in extra_info) or
834                     extract_flat is True):
835                 if self.params.get('forcejson', False):
836                     self.to_stdout(json.dumps(ie_result))
837                 return ie_result
838
839         if result_type == 'video':
840             self.add_extra_info(ie_result, extra_info)
841             return self.process_video_result(ie_result, download=download)
842         elif result_type == 'url':
843             # We have to add extra_info to the results because it may be
844             # contained in a playlist
845             return self.extract_info(ie_result['url'],
846                                      download,
847                                      ie_key=ie_result.get('ie_key'),
848                                      extra_info=extra_info)
849         elif result_type == 'url_transparent':
850             # Use the information from the embedding page
851             info = self.extract_info(
852                 ie_result['url'], ie_key=ie_result.get('ie_key'),
853                 extra_info=extra_info, download=False, process=False)
854
855             # extract_info may return None when ignoreerrors is enabled and
856             # extraction failed with an error, don't crash and return early
857             # in this case
858             if not info:
859                 return info
860
861             force_properties = dict(
862                 (k, v) for k, v in ie_result.items() if v is not None)
863             for f in ('_type', 'url', 'id', 'extractor', 'extractor_key', 'ie_key'):
864                 if f in force_properties:
865                     del force_properties[f]
866             new_result = info.copy()
867             new_result.update(force_properties)
868
869             # Extracted info may not be a video result (i.e.
870             # info.get('_type', 'video') != video) but rather an url or
871             # url_transparent. In such cases outer metadata (from ie_result)
872             # should be propagated to inner one (info). For this to happen
873             # _type of info should be overridden with url_transparent. This
874             # fixes issue from https://github.com/rg3/youtube-dl/pull/11163.
875             if new_result.get('_type') == 'url':
876                 new_result['_type'] = 'url_transparent'
877
878             return self.process_ie_result(
879                 new_result, download=download, extra_info=extra_info)
880         elif result_type in ('playlist', 'multi_video'):
881             # We process each entry in the playlist
882             playlist = ie_result.get('title') or ie_result.get('id')
883             self.to_screen('[download] Downloading playlist: %s' % playlist)
884
885             playlist_results = []
886
887             playliststart = self.params.get('playliststart', 1) - 1
888             playlistend = self.params.get('playlistend')
889             # For backwards compatibility, interpret -1 as whole list
890             if playlistend == -1:
891                 playlistend = None
892
893             playlistitems_str = self.params.get('playlist_items')
894             playlistitems = None
895             if playlistitems_str is not None:
896                 def iter_playlistitems(format):
897                     for string_segment in format.split(','):
898                         if '-' in string_segment:
899                             start, end = string_segment.split('-')
900                             for item in range(int(start), int(end) + 1):
901                                 yield int(item)
902                         else:
903                             yield int(string_segment)
904                 playlistitems = iter_playlistitems(playlistitems_str)
905
906             ie_entries = ie_result['entries']
907             if isinstance(ie_entries, list):
908                 n_all_entries = len(ie_entries)
909                 if playlistitems:
910                     entries = [
911                         ie_entries[i - 1] for i in playlistitems
912                         if -n_all_entries <= i - 1 < n_all_entries]
913                 else:
914                     entries = ie_entries[playliststart:playlistend]
915                 n_entries = len(entries)
916                 self.to_screen(
917                     '[%s] playlist %s: Collected %d video ids (downloading %d of them)' %
918                     (ie_result['extractor'], playlist, n_all_entries, n_entries))
919             elif isinstance(ie_entries, PagedList):
920                 if playlistitems:
921                     entries = []
922                     for item in playlistitems:
923                         entries.extend(ie_entries.getslice(
924                             item - 1, item
925                         ))
926                 else:
927                     entries = ie_entries.getslice(
928                         playliststart, playlistend)
929                 n_entries = len(entries)
930                 self.to_screen(
931                     '[%s] playlist %s: Downloading %d videos' %
932                     (ie_result['extractor'], playlist, n_entries))
933             else:  # iterable
934                 if playlistitems:
935                     entry_list = list(ie_entries)
936                     entries = [entry_list[i - 1] for i in playlistitems]
937                 else:
938                     entries = list(itertools.islice(
939                         ie_entries, playliststart, playlistend))
940                 n_entries = len(entries)
941                 self.to_screen(
942                     '[%s] playlist %s: Downloading %d videos' %
943                     (ie_result['extractor'], playlist, n_entries))
944
945             if self.params.get('playlistreverse', False):
946                 entries = entries[::-1]
947
948             if self.params.get('playlistrandom', False):
949                 random.shuffle(entries)
950
951             x_forwarded_for = ie_result.get('__x_forwarded_for_ip')
952
953             for i, entry in enumerate(entries, 1):
954                 self.to_screen('[download] Downloading video %s of %s' % (i, n_entries))
955                 # This __x_forwarded_for_ip thing is a bit ugly but requires
956                 # minimal changes
957                 if x_forwarded_for:
958                     entry['__x_forwarded_for_ip'] = x_forwarded_for
959                 extra = {
960                     'n_entries': n_entries,
961                     'playlist': playlist,
962                     'playlist_id': ie_result.get('id'),
963                     'playlist_title': ie_result.get('title'),
964                     'playlist_index': i + playliststart,
965                     'extractor': ie_result['extractor'],
966                     'webpage_url': ie_result['webpage_url'],
967                     'webpage_url_basename': url_basename(ie_result['webpage_url']),
968                     'extractor_key': ie_result['extractor_key'],
969                 }
970
971                 reason = self._match_entry(entry, incomplete=True)
972                 if reason is not None:
973                     self.to_screen('[download] ' + reason)
974                     continue
975
976                 entry_result = self.process_ie_result(entry,
977                                                       download=download,
978                                                       extra_info=extra)
979                 playlist_results.append(entry_result)
980             ie_result['entries'] = playlist_results
981             self.to_screen('[download] Finished downloading playlist: %s' % playlist)
982             return ie_result
983         elif result_type == 'compat_list':
984             self.report_warning(
985                 'Extractor %s returned a compat_list result. '
986                 'It needs to be updated.' % ie_result.get('extractor'))
987
988             def _fixup(r):
989                 self.add_extra_info(
990                     r,
991                     {
992                         'extractor': ie_result['extractor'],
993                         'webpage_url': ie_result['webpage_url'],
994                         'webpage_url_basename': url_basename(ie_result['webpage_url']),
995                         'extractor_key': ie_result['extractor_key'],
996                     }
997                 )
998                 return r
999             ie_result['entries'] = [
1000                 self.process_ie_result(_fixup(r), download, extra_info)
1001                 for r in ie_result['entries']
1002             ]
1003             return ie_result
1004         else:
1005             raise Exception('Invalid result type: %s' % result_type)
1006
1007     def _build_format_filter(self, filter_spec):
1008         " Returns a function to filter the formats according to the filter_spec "
1009
1010         OPERATORS = {
1011             '<': operator.lt,
1012             '<=': operator.le,
1013             '>': operator.gt,
1014             '>=': operator.ge,
1015             '=': operator.eq,
1016             '!=': operator.ne,
1017         }
1018         operator_rex = re.compile(r'''(?x)\s*
1019             (?P<key>width|height|tbr|abr|vbr|asr|filesize|fps)
1020             \s*(?P<op>%s)(?P<none_inclusive>\s*\?)?\s*
1021             (?P<value>[0-9.]+(?:[kKmMgGtTpPeEzZyY]i?[Bb]?)?)
1022             $
1023             ''' % '|'.join(map(re.escape, OPERATORS.keys())))
1024         m = operator_rex.search(filter_spec)
1025         if m:
1026             try:
1027                 comparison_value = int(m.group('value'))
1028             except ValueError:
1029                 comparison_value = parse_filesize(m.group('value'))
1030                 if comparison_value is None:
1031                     comparison_value = parse_filesize(m.group('value') + 'B')
1032                 if comparison_value is None:
1033                     raise ValueError(
1034                         'Invalid value %r in format specification %r' % (
1035                             m.group('value'), filter_spec))
1036             op = OPERATORS[m.group('op')]
1037
1038         if not m:
1039             STR_OPERATORS = {
1040                 '=': operator.eq,
1041                 '!=': operator.ne,
1042                 '^=': lambda attr, value: attr.startswith(value),
1043                 '$=': lambda attr, value: attr.endswith(value),
1044                 '*=': lambda attr, value: value in attr,
1045             }
1046             str_operator_rex = re.compile(r'''(?x)
1047                 \s*(?P<key>ext|acodec|vcodec|container|protocol|format_id)
1048                 \s*(?P<op>%s)(?P<none_inclusive>\s*\?)?
1049                 \s*(?P<value>[a-zA-Z0-9._-]+)
1050                 \s*$
1051                 ''' % '|'.join(map(re.escape, STR_OPERATORS.keys())))
1052             m = str_operator_rex.search(filter_spec)
1053             if m:
1054                 comparison_value = m.group('value')
1055                 op = STR_OPERATORS[m.group('op')]
1056
1057         if not m:
1058             raise ValueError('Invalid filter specification %r' % filter_spec)
1059
1060         def _filter(f):
1061             actual_value = f.get(m.group('key'))
1062             if actual_value is None:
1063                 return m.group('none_inclusive')
1064             return op(actual_value, comparison_value)
1065         return _filter
1066
1067     def _default_format_spec(self, info_dict, download=True):
1068         req_format_list = []
1069
1070         def can_have_partial_formats():
1071             if self.params.get('simulate', False):
1072                 return True
1073             if not download:
1074                 return True
1075             if self.params.get('outtmpl', DEFAULT_OUTTMPL) == '-':
1076                 return False
1077             if info_dict.get('is_live'):
1078                 return False
1079             merger = FFmpegMergerPP(self)
1080             return merger.available and merger.can_merge()
1081         if can_have_partial_formats():
1082             req_format_list.append('bestvideo+bestaudio')
1083         req_format_list.append('best')
1084         return '/'.join(req_format_list)
1085
1086     def build_format_selector(self, format_spec):
1087         def syntax_error(note, start):
1088             message = (
1089                 'Invalid format specification: '
1090                 '{0}\n\t{1}\n\t{2}^'.format(note, format_spec, ' ' * start[1]))
1091             return SyntaxError(message)
1092
1093         PICKFIRST = 'PICKFIRST'
1094         MERGE = 'MERGE'
1095         SINGLE = 'SINGLE'
1096         GROUP = 'GROUP'
1097         FormatSelector = collections.namedtuple('FormatSelector', ['type', 'selector', 'filters'])
1098
1099         def _parse_filter(tokens):
1100             filter_parts = []
1101             for type, string, start, _, _ in tokens:
1102                 if type == tokenize.OP and string == ']':
1103                     return ''.join(filter_parts)
1104                 else:
1105                     filter_parts.append(string)
1106
1107         def _remove_unused_ops(tokens):
1108             # Remove operators that we don't use and join them with the surrounding strings
1109             # for example: 'mp4' '-' 'baseline' '-' '16x9' is converted to 'mp4-baseline-16x9'
1110             ALLOWED_OPS = ('/', '+', ',', '(', ')')
1111             last_string, last_start, last_end, last_line = None, None, None, None
1112             for type, string, start, end, line in tokens:
1113                 if type == tokenize.OP and string == '[':
1114                     if last_string:
1115                         yield tokenize.NAME, last_string, last_start, last_end, last_line
1116                         last_string = None
1117                     yield type, string, start, end, line
1118                     # everything inside brackets will be handled by _parse_filter
1119                     for type, string, start, end, line in tokens:
1120                         yield type, string, start, end, line
1121                         if type == tokenize.OP and string == ']':
1122                             break
1123                 elif type == tokenize.OP and string in ALLOWED_OPS:
1124                     if last_string:
1125                         yield tokenize.NAME, last_string, last_start, last_end, last_line
1126                         last_string = None
1127                     yield type, string, start, end, line
1128                 elif type in [tokenize.NAME, tokenize.NUMBER, tokenize.OP]:
1129                     if not last_string:
1130                         last_string = string
1131                         last_start = start
1132                         last_end = end
1133                     else:
1134                         last_string += string
1135             if last_string:
1136                 yield tokenize.NAME, last_string, last_start, last_end, last_line
1137
1138         def _parse_format_selection(tokens, inside_merge=False, inside_choice=False, inside_group=False):
1139             selectors = []
1140             current_selector = None
1141             for type, string, start, _, _ in tokens:
1142                 # ENCODING is only defined in python 3.x
1143                 if type == getattr(tokenize, 'ENCODING', None):
1144                     continue
1145                 elif type in [tokenize.NAME, tokenize.NUMBER]:
1146                     current_selector = FormatSelector(SINGLE, string, [])
1147                 elif type == tokenize.OP:
1148                     if string == ')':
1149                         if not inside_group:
1150                             # ')' will be handled by the parentheses group
1151                             tokens.restore_last_token()
1152                         break
1153                     elif inside_merge and string in ['/', ',']:
1154                         tokens.restore_last_token()
1155                         break
1156                     elif inside_choice and string == ',':
1157                         tokens.restore_last_token()
1158                         break
1159                     elif string == ',':
1160                         if not current_selector:
1161                             raise syntax_error('"," must follow a format selector', start)
1162                         selectors.append(current_selector)
1163                         current_selector = None
1164                     elif string == '/':
1165                         if not current_selector:
1166                             raise syntax_error('"/" must follow a format selector', start)
1167                         first_choice = current_selector
1168                         second_choice = _parse_format_selection(tokens, inside_choice=True)
1169                         current_selector = FormatSelector(PICKFIRST, (first_choice, second_choice), [])
1170                     elif string == '[':
1171                         if not current_selector:
1172                             current_selector = FormatSelector(SINGLE, 'best', [])
1173                         format_filter = _parse_filter(tokens)
1174                         current_selector.filters.append(format_filter)
1175                     elif string == '(':
1176                         if current_selector:
1177                             raise syntax_error('Unexpected "("', start)
1178                         group = _parse_format_selection(tokens, inside_group=True)
1179                         current_selector = FormatSelector(GROUP, group, [])
1180                     elif string == '+':
1181                         video_selector = current_selector
1182                         audio_selector = _parse_format_selection(tokens, inside_merge=True)
1183                         if not video_selector or not audio_selector:
1184                             raise syntax_error('"+" must be between two format selectors', start)
1185                         current_selector = FormatSelector(MERGE, (video_selector, audio_selector), [])
1186                     else:
1187                         raise syntax_error('Operator not recognized: "{0}"'.format(string), start)
1188                 elif type == tokenize.ENDMARKER:
1189                     break
1190             if current_selector:
1191                 selectors.append(current_selector)
1192             return selectors
1193
1194         def _build_selector_function(selector):
1195             if isinstance(selector, list):
1196                 fs = [_build_selector_function(s) for s in selector]
1197
1198                 def selector_function(ctx):
1199                     for f in fs:
1200                         for format in f(ctx):
1201                             yield format
1202                 return selector_function
1203             elif selector.type == GROUP:
1204                 selector_function = _build_selector_function(selector.selector)
1205             elif selector.type == PICKFIRST:
1206                 fs = [_build_selector_function(s) for s in selector.selector]
1207
1208                 def selector_function(ctx):
1209                     for f in fs:
1210                         picked_formats = list(f(ctx))
1211                         if picked_formats:
1212                             return picked_formats
1213                     return []
1214             elif selector.type == SINGLE:
1215                 format_spec = selector.selector
1216
1217                 def selector_function(ctx):
1218                     formats = list(ctx['formats'])
1219                     if not formats:
1220                         return
1221                     if format_spec == 'all':
1222                         for f in formats:
1223                             yield f
1224                     elif format_spec in ['best', 'worst', None]:
1225                         format_idx = 0 if format_spec == 'worst' else -1
1226                         audiovideo_formats = [
1227                             f for f in formats
1228                             if f.get('vcodec') != 'none' and f.get('acodec') != 'none']
1229                         if audiovideo_formats:
1230                             yield audiovideo_formats[format_idx]
1231                         # for extractors with incomplete formats (audio only (soundcloud)
1232                         # or video only (imgur)) we will fallback to best/worst
1233                         # {video,audio}-only format
1234                         elif ctx['incomplete_formats']:
1235                             yield formats[format_idx]
1236                     elif format_spec == 'bestaudio':
1237                         audio_formats = [
1238                             f for f in formats
1239                             if f.get('vcodec') == 'none']
1240                         if audio_formats:
1241                             yield audio_formats[-1]
1242                     elif format_spec == 'worstaudio':
1243                         audio_formats = [
1244                             f for f in formats
1245                             if f.get('vcodec') == 'none']
1246                         if audio_formats:
1247                             yield audio_formats[0]
1248                     elif format_spec == 'bestvideo':
1249                         video_formats = [
1250                             f for f in formats
1251                             if f.get('acodec') == 'none']
1252                         if video_formats:
1253                             yield video_formats[-1]
1254                     elif format_spec == 'worstvideo':
1255                         video_formats = [
1256                             f for f in formats
1257                             if f.get('acodec') == 'none']
1258                         if video_formats:
1259                             yield video_formats[0]
1260                     else:
1261                         extensions = ['mp4', 'flv', 'webm', '3gp', 'm4a', 'mp3', 'ogg', 'aac', 'wav']
1262                         if format_spec in extensions:
1263                             filter_f = lambda f: f['ext'] == format_spec
1264                         else:
1265                             filter_f = lambda f: f['format_id'] == format_spec
1266                         matches = list(filter(filter_f, formats))
1267                         if matches:
1268                             yield matches[-1]
1269             elif selector.type == MERGE:
1270                 def _merge(formats_info):
1271                     format_1, format_2 = [f['format_id'] for f in formats_info]
1272                     # The first format must contain the video and the
1273                     # second the audio
1274                     if formats_info[0].get('vcodec') == 'none':
1275                         self.report_error('The first format must '
1276                                           'contain the video, try using '
1277                                           '"-f %s+%s"' % (format_2, format_1))
1278                         return
1279                     # Formats must be opposite (video+audio)
1280                     if formats_info[0].get('acodec') == 'none' and formats_info[1].get('acodec') == 'none':
1281                         self.report_error(
1282                             'Both formats %s and %s are video-only, you must specify "-f video+audio"'
1283                             % (format_1, format_2))
1284                         return
1285                     output_ext = (
1286                         formats_info[0]['ext']
1287                         if self.params.get('merge_output_format') is None
1288                         else self.params['merge_output_format'])
1289                     return {
1290                         'requested_formats': formats_info,
1291                         'format': '%s+%s' % (formats_info[0].get('format'),
1292                                              formats_info[1].get('format')),
1293                         'format_id': '%s+%s' % (formats_info[0].get('format_id'),
1294                                                 formats_info[1].get('format_id')),
1295                         'width': formats_info[0].get('width'),
1296                         'height': formats_info[0].get('height'),
1297                         'resolution': formats_info[0].get('resolution'),
1298                         'fps': formats_info[0].get('fps'),
1299                         'vcodec': formats_info[0].get('vcodec'),
1300                         'vbr': formats_info[0].get('vbr'),
1301                         'stretched_ratio': formats_info[0].get('stretched_ratio'),
1302                         'acodec': formats_info[1].get('acodec'),
1303                         'abr': formats_info[1].get('abr'),
1304                         'ext': output_ext,
1305                     }
1306                 video_selector, audio_selector = map(_build_selector_function, selector.selector)
1307
1308                 def selector_function(ctx):
1309                     for pair in itertools.product(
1310                             video_selector(copy.deepcopy(ctx)), audio_selector(copy.deepcopy(ctx))):
1311                         yield _merge(pair)
1312
1313             filters = [self._build_format_filter(f) for f in selector.filters]
1314
1315             def final_selector(ctx):
1316                 ctx_copy = copy.deepcopy(ctx)
1317                 for _filter in filters:
1318                     ctx_copy['formats'] = list(filter(_filter, ctx_copy['formats']))
1319                 return selector_function(ctx_copy)
1320             return final_selector
1321
1322         stream = io.BytesIO(format_spec.encode('utf-8'))
1323         try:
1324             tokens = list(_remove_unused_ops(compat_tokenize_tokenize(stream.readline)))
1325         except tokenize.TokenError:
1326             raise syntax_error('Missing closing/opening brackets or parenthesis', (0, len(format_spec)))
1327
1328         class TokenIterator(object):
1329             def __init__(self, tokens):
1330                 self.tokens = tokens
1331                 self.counter = 0
1332
1333             def __iter__(self):
1334                 return self
1335
1336             def __next__(self):
1337                 if self.counter >= len(self.tokens):
1338                     raise StopIteration()
1339                 value = self.tokens[self.counter]
1340                 self.counter += 1
1341                 return value
1342
1343             next = __next__
1344
1345             def restore_last_token(self):
1346                 self.counter -= 1
1347
1348         parsed_selector = _parse_format_selection(iter(TokenIterator(tokens)))
1349         return _build_selector_function(parsed_selector)
1350
1351     def _calc_headers(self, info_dict):
1352         res = std_headers.copy()
1353
1354         add_headers = info_dict.get('http_headers')
1355         if add_headers:
1356             res.update(add_headers)
1357
1358         cookies = self._calc_cookies(info_dict)
1359         if cookies:
1360             res['Cookie'] = cookies
1361
1362         if 'X-Forwarded-For' not in res:
1363             x_forwarded_for_ip = info_dict.get('__x_forwarded_for_ip')
1364             if x_forwarded_for_ip:
1365                 res['X-Forwarded-For'] = x_forwarded_for_ip
1366
1367         return res
1368
1369     def _calc_cookies(self, info_dict):
1370         pr = sanitized_Request(info_dict['url'])
1371         self.cookiejar.add_cookie_header(pr)
1372         return pr.get_header('Cookie')
1373
1374     def process_video_result(self, info_dict, download=True):
1375         assert info_dict.get('_type', 'video') == 'video'
1376
1377         if 'id' not in info_dict:
1378             raise ExtractorError('Missing "id" field in extractor result')
1379         if 'title' not in info_dict:
1380             raise ExtractorError('Missing "title" field in extractor result')
1381
1382         def report_force_conversion(field, field_not, conversion):
1383             self.report_warning(
1384                 '"%s" field is not %s - forcing %s conversion, there is an error in extractor'
1385                 % (field, field_not, conversion))
1386
1387         def sanitize_string_field(info, string_field):
1388             field = info.get(string_field)
1389             if field is None or isinstance(field, compat_str):
1390                 return
1391             report_force_conversion(string_field, 'a string', 'string')
1392             info[string_field] = compat_str(field)
1393
1394         def sanitize_numeric_fields(info):
1395             for numeric_field in self._NUMERIC_FIELDS:
1396                 field = info.get(numeric_field)
1397                 if field is None or isinstance(field, compat_numeric_types):
1398                     continue
1399                 report_force_conversion(numeric_field, 'numeric', 'int')
1400                 info[numeric_field] = int_or_none(field)
1401
1402         sanitize_string_field(info_dict, 'id')
1403         sanitize_numeric_fields(info_dict)
1404
1405         if 'playlist' not in info_dict:
1406             # It isn't part of a playlist
1407             info_dict['playlist'] = None
1408             info_dict['playlist_index'] = None
1409
1410         thumbnails = info_dict.get('thumbnails')
1411         if thumbnails is None:
1412             thumbnail = info_dict.get('thumbnail')
1413             if thumbnail:
1414                 info_dict['thumbnails'] = thumbnails = [{'url': thumbnail}]
1415         if thumbnails:
1416             thumbnails.sort(key=lambda t: (
1417                 t.get('preference') if t.get('preference') is not None else -1,
1418                 t.get('width') if t.get('width') is not None else -1,
1419                 t.get('height') if t.get('height') is not None else -1,
1420                 t.get('id') if t.get('id') is not None else '', t.get('url')))
1421             for i, t in enumerate(thumbnails):
1422                 t['url'] = sanitize_url(t['url'])
1423                 if t.get('width') and t.get('height'):
1424                     t['resolution'] = '%dx%d' % (t['width'], t['height'])
1425                 if t.get('id') is None:
1426                     t['id'] = '%d' % i
1427
1428         if self.params.get('list_thumbnails'):
1429             self.list_thumbnails(info_dict)
1430             return
1431
1432         thumbnail = info_dict.get('thumbnail')
1433         if thumbnail:
1434             info_dict['thumbnail'] = sanitize_url(thumbnail)
1435         elif thumbnails:
1436             info_dict['thumbnail'] = thumbnails[-1]['url']
1437
1438         if 'display_id' not in info_dict and 'id' in info_dict:
1439             info_dict['display_id'] = info_dict['id']
1440
1441         if info_dict.get('upload_date') is None and info_dict.get('timestamp') is not None:
1442             # Working around out-of-range timestamp values (e.g. negative ones on Windows,
1443             # see http://bugs.python.org/issue1646728)
1444             try:
1445                 upload_date = datetime.datetime.utcfromtimestamp(info_dict['timestamp'])
1446                 info_dict['upload_date'] = upload_date.strftime('%Y%m%d')
1447             except (ValueError, OverflowError, OSError):
1448                 pass
1449
1450         # Auto generate title fields corresponding to the *_number fields when missing
1451         # in order to always have clean titles. This is very common for TV series.
1452         for field in ('chapter', 'season', 'episode'):
1453             if info_dict.get('%s_number' % field) is not None and not info_dict.get(field):
1454                 info_dict[field] = '%s %d' % (field.capitalize(), info_dict['%s_number' % field])
1455
1456         subtitles = info_dict.get('subtitles')
1457         if subtitles:
1458             for _, subtitle in subtitles.items():
1459                 for subtitle_format in subtitle:
1460                     if subtitle_format.get('url'):
1461                         subtitle_format['url'] = sanitize_url(subtitle_format['url'])
1462                     if subtitle_format.get('ext') is None:
1463                         subtitle_format['ext'] = determine_ext(subtitle_format['url']).lower()
1464
1465         if self.params.get('listsubtitles', False):
1466             if 'automatic_captions' in info_dict:
1467                 self.list_subtitles(info_dict['id'], info_dict.get('automatic_captions'), 'automatic captions')
1468             self.list_subtitles(info_dict['id'], subtitles, 'subtitles')
1469             return
1470         info_dict['requested_subtitles'] = self.process_subtitles(
1471             info_dict['id'], subtitles,
1472             info_dict.get('automatic_captions'))
1473
1474         # We now pick which formats have to be downloaded
1475         if info_dict.get('formats') is None:
1476             # There's only one format available
1477             formats = [info_dict]
1478         else:
1479             formats = info_dict['formats']
1480
1481         if not formats:
1482             raise ExtractorError('No video formats found!')
1483
1484         def is_wellformed(f):
1485             url = f.get('url')
1486             if not url:
1487                 self.report_warning(
1488                     '"url" field is missing or empty - skipping format, '
1489                     'there is an error in extractor')
1490                 return False
1491             if isinstance(url, bytes):
1492                 sanitize_string_field(f, 'url')
1493             return True
1494
1495         # Filter out malformed formats for better extraction robustness
1496         formats = list(filter(is_wellformed, formats))
1497
1498         formats_dict = {}
1499
1500         # We check that all the formats have the format and format_id fields
1501         for i, format in enumerate(formats):
1502             sanitize_string_field(format, 'format_id')
1503             sanitize_numeric_fields(format)
1504             format['url'] = sanitize_url(format['url'])
1505             if not format.get('format_id'):
1506                 format['format_id'] = compat_str(i)
1507             else:
1508                 # Sanitize format_id from characters used in format selector expression
1509                 format['format_id'] = re.sub(r'[\s,/+\[\]()]', '_', format['format_id'])
1510             format_id = format['format_id']
1511             if format_id not in formats_dict:
1512                 formats_dict[format_id] = []
1513             formats_dict[format_id].append(format)
1514
1515         # Make sure all formats have unique format_id
1516         for format_id, ambiguous_formats in formats_dict.items():
1517             if len(ambiguous_formats) > 1:
1518                 for i, format in enumerate(ambiguous_formats):
1519                     format['format_id'] = '%s-%d' % (format_id, i)
1520
1521         for i, format in enumerate(formats):
1522             if format.get('format') is None:
1523                 format['format'] = '{id} - {res}{note}'.format(
1524                     id=format['format_id'],
1525                     res=self.format_resolution(format),
1526                     note=' ({0})'.format(format['format_note']) if format.get('format_note') is not None else '',
1527                 )
1528             # Automatically determine file extension if missing
1529             if format.get('ext') is None:
1530                 format['ext'] = determine_ext(format['url']).lower()
1531             # Automatically determine protocol if missing (useful for format
1532             # selection purposes)
1533             if format.get('protocol') is None:
1534                 format['protocol'] = determine_protocol(format)
1535             # Add HTTP headers, so that external programs can use them from the
1536             # json output
1537             full_format_info = info_dict.copy()
1538             full_format_info.update(format)
1539             format['http_headers'] = self._calc_headers(full_format_info)
1540         # Remove private housekeeping stuff
1541         if '__x_forwarded_for_ip' in info_dict:
1542             del info_dict['__x_forwarded_for_ip']
1543
1544         # TODO Central sorting goes here
1545
1546         if formats[0] is not info_dict:
1547             # only set the 'formats' fields if the original info_dict list them
1548             # otherwise we end up with a circular reference, the first (and unique)
1549             # element in the 'formats' field in info_dict is info_dict itself,
1550             # which can't be exported to json
1551             info_dict['formats'] = formats
1552         if self.params.get('listformats'):
1553             self.list_formats(info_dict)
1554             return
1555
1556         req_format = self.params.get('format')
1557         if req_format is None:
1558             req_format = self._default_format_spec(info_dict, download=download)
1559             if self.params.get('verbose'):
1560                 self.to_stdout('[debug] Default format spec: %s' % req_format)
1561
1562         format_selector = self.build_format_selector(req_format)
1563
1564         # While in format selection we may need to have an access to the original
1565         # format set in order to calculate some metrics or do some processing.
1566         # For now we need to be able to guess whether original formats provided
1567         # by extractor are incomplete or not (i.e. whether extractor provides only
1568         # video-only or audio-only formats) for proper formats selection for
1569         # extractors with such incomplete formats (see
1570         # https://github.com/rg3/youtube-dl/pull/5556).
1571         # Since formats may be filtered during format selection and may not match
1572         # the original formats the results may be incorrect. Thus original formats
1573         # or pre-calculated metrics should be passed to format selection routines
1574         # as well.
1575         # We will pass a context object containing all necessary additional data
1576         # instead of just formats.
1577         # This fixes incorrect format selection issue (see
1578         # https://github.com/rg3/youtube-dl/issues/10083).
1579         incomplete_formats = (
1580             # All formats are video-only or
1581             all(f.get('vcodec') != 'none' and f.get('acodec') == 'none' for f in formats) or
1582             # all formats are audio-only
1583             all(f.get('vcodec') == 'none' and f.get('acodec') != 'none' for f in formats))
1584
1585         ctx = {
1586             'formats': formats,
1587             'incomplete_formats': incomplete_formats,
1588         }
1589
1590         formats_to_download = list(format_selector(ctx))
1591         if not formats_to_download:
1592             raise ExtractorError('requested format not available',
1593                                  expected=True)
1594
1595         if download:
1596             if len(formats_to_download) > 1:
1597                 self.to_screen('[info] %s: downloading video in %s formats' % (info_dict['id'], len(formats_to_download)))
1598             for format in formats_to_download:
1599                 new_info = dict(info_dict)
1600                 new_info.update(format)
1601                 self.process_info(new_info)
1602         # We update the info dict with the best quality format (backwards compatibility)
1603         info_dict.update(formats_to_download[-1])
1604         return info_dict
1605
1606     def process_subtitles(self, video_id, normal_subtitles, automatic_captions):
1607         """Select the requested subtitles and their format"""
1608         available_subs = {}
1609         if normal_subtitles and self.params.get('writesubtitles'):
1610             available_subs.update(normal_subtitles)
1611         if automatic_captions and self.params.get('writeautomaticsub'):
1612             for lang, cap_info in automatic_captions.items():
1613                 if lang not in available_subs:
1614                     available_subs[lang] = cap_info
1615
1616         if (not self.params.get('writesubtitles') and not
1617                 self.params.get('writeautomaticsub') or not
1618                 available_subs):
1619             return None
1620
1621         if self.params.get('allsubtitles', False):
1622             requested_langs = available_subs.keys()
1623         else:
1624             if self.params.get('subtitleslangs', False):
1625                 requested_langs = self.params.get('subtitleslangs')
1626             elif 'en' in available_subs:
1627                 requested_langs = ['en']
1628             else:
1629                 requested_langs = [list(available_subs.keys())[0]]
1630
1631         formats_query = self.params.get('subtitlesformat', 'best')
1632         formats_preference = formats_query.split('/') if formats_query else []
1633         subs = {}
1634         for lang in requested_langs:
1635             formats = available_subs.get(lang)
1636             if formats is None:
1637                 self.report_warning('%s subtitles not available for %s' % (lang, video_id))
1638                 continue
1639             for ext in formats_preference:
1640                 if ext == 'best':
1641                     f = formats[-1]
1642                     break
1643                 matches = list(filter(lambda f: f['ext'] == ext, formats))
1644                 if matches:
1645                     f = matches[-1]
1646                     break
1647             else:
1648                 f = formats[-1]
1649                 self.report_warning(
1650                     'No subtitle format found matching "%s" for language %s, '
1651                     'using %s' % (formats_query, lang, f['ext']))
1652             subs[lang] = f
1653         return subs
1654
1655     def process_info(self, info_dict):
1656         """Process a single resolved IE result."""
1657
1658         assert info_dict.get('_type', 'video') == 'video'
1659
1660         max_downloads = self.params.get('max_downloads')
1661         if max_downloads is not None:
1662             if self._num_downloads >= int(max_downloads):
1663                 raise MaxDownloadsReached()
1664
1665         info_dict['fulltitle'] = info_dict['title']
1666         if len(info_dict['title']) > 200:
1667             info_dict['title'] = info_dict['title'][:197] + '...'
1668
1669         if 'format' not in info_dict:
1670             info_dict['format'] = info_dict['ext']
1671
1672         reason = self._match_entry(info_dict, incomplete=False)
1673         if reason is not None:
1674             self.to_screen('[download] ' + reason)
1675             return
1676
1677         self._num_downloads += 1
1678
1679         info_dict['_filename'] = filename = self.prepare_filename(info_dict)
1680
1681         # Forced printings
1682         if self.params.get('forcetitle', False):
1683             self.to_stdout(info_dict['fulltitle'])
1684         if self.params.get('forceid', False):
1685             self.to_stdout(info_dict['id'])
1686         if self.params.get('forceurl', False):
1687             if info_dict.get('requested_formats') is not None:
1688                 for f in info_dict['requested_formats']:
1689                     self.to_stdout(f['url'] + f.get('play_path', ''))
1690             else:
1691                 # For RTMP URLs, also include the playpath
1692                 self.to_stdout(info_dict['url'] + info_dict.get('play_path', ''))
1693         if self.params.get('forcethumbnail', False) and info_dict.get('thumbnail') is not None:
1694             self.to_stdout(info_dict['thumbnail'])
1695         if self.params.get('forcedescription', False) and info_dict.get('description') is not None:
1696             self.to_stdout(info_dict['description'])
1697         if self.params.get('forcefilename', False) and filename is not None:
1698             self.to_stdout(filename)
1699         if self.params.get('forceduration', False) and info_dict.get('duration') is not None:
1700             self.to_stdout(formatSeconds(info_dict['duration']))
1701         if self.params.get('forceformat', False):
1702             self.to_stdout(info_dict['format'])
1703         if self.params.get('forcejson', False):
1704             self.to_stdout(json.dumps(info_dict))
1705
1706         # Do nothing else if in simulate mode
1707         if self.params.get('simulate', False):
1708             return
1709
1710         if filename is None:
1711             return
1712
1713         def ensure_dir_exists(path):
1714             try:
1715                 dn = os.path.dirname(path)
1716                 if dn and not os.path.exists(dn):
1717                     os.makedirs(dn)
1718                 return True
1719             except (OSError, IOError) as err:
1720                 self.report_error('unable to create directory ' + error_to_compat_str(err))
1721                 return False
1722
1723         if not ensure_dir_exists(sanitize_path(encodeFilename(filename))):
1724             return
1725
1726         if self.params.get('writedescription', False):
1727             descfn = replace_extension(filename, 'description', info_dict.get('ext'))
1728             if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(descfn)):
1729                 self.to_screen('[info] Video description is already present')
1730             elif info_dict.get('description') is None:
1731                 self.report_warning('There\'s no description to write.')
1732             else:
1733                 try:
1734                     self.to_screen('[info] Writing video description to: ' + descfn)
1735                     with io.open(encodeFilename(descfn), 'w', encoding='utf-8') as descfile:
1736                         descfile.write(info_dict['description'])
1737                 except (OSError, IOError):
1738                     self.report_error('Cannot write description file ' + descfn)
1739                     return
1740
1741         if self.params.get('writeannotations', False):
1742             annofn = replace_extension(filename, 'annotations.xml', info_dict.get('ext'))
1743             if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(annofn)):
1744                 self.to_screen('[info] Video annotations are already present')
1745             else:
1746                 try:
1747                     self.to_screen('[info] Writing video annotations to: ' + annofn)
1748                     with io.open(encodeFilename(annofn), 'w', encoding='utf-8') as annofile:
1749                         annofile.write(info_dict['annotations'])
1750                 except (KeyError, TypeError):
1751                     self.report_warning('There are no annotations to write.')
1752                 except (OSError, IOError):
1753                     self.report_error('Cannot write annotations file: ' + annofn)
1754                     return
1755
1756         subtitles_are_requested = any([self.params.get('writesubtitles', False),
1757                                        self.params.get('writeautomaticsub')])
1758
1759         if subtitles_are_requested and info_dict.get('requested_subtitles'):
1760             # subtitles download errors are already managed as troubles in relevant IE
1761             # that way it will silently go on when used with unsupporting IE
1762             subtitles = info_dict['requested_subtitles']
1763             ie = self.get_info_extractor(info_dict['extractor_key'])
1764             for sub_lang, sub_info in subtitles.items():
1765                 sub_format = sub_info['ext']
1766                 sub_filename = subtitles_filename(filename, sub_lang, sub_format)
1767                 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(sub_filename)):
1768                     self.to_screen('[info] Video subtitle %s.%s is already present' % (sub_lang, sub_format))
1769                 else:
1770                     self.to_screen('[info] Writing video subtitles to: ' + sub_filename)
1771                     if sub_info.get('data') is not None:
1772                         try:
1773                             # Use newline='' to prevent conversion of newline characters
1774                             # See https://github.com/rg3/youtube-dl/issues/10268
1775                             with io.open(encodeFilename(sub_filename), 'w', encoding='utf-8', newline='') as subfile:
1776                                 subfile.write(sub_info['data'])
1777                         except (OSError, IOError):
1778                             self.report_error('Cannot write subtitles file ' + sub_filename)
1779                             return
1780                     else:
1781                         try:
1782                             sub_data = ie._request_webpage(
1783                                 sub_info['url'], info_dict['id'], note=False).read()
1784                             with io.open(encodeFilename(sub_filename), 'wb') as subfile:
1785                                 subfile.write(sub_data)
1786                         except (ExtractorError, IOError, OSError, ValueError) as err:
1787                             self.report_warning('Unable to download subtitle for "%s": %s' %
1788                                                 (sub_lang, error_to_compat_str(err)))
1789                             continue
1790
1791         if self.params.get('writeinfojson', False):
1792             infofn = replace_extension(filename, 'info.json', info_dict.get('ext'))
1793             if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(infofn)):
1794                 self.to_screen('[info] Video description metadata is already present')
1795             else:
1796                 self.to_screen('[info] Writing video description metadata as JSON to: ' + infofn)
1797                 try:
1798                     write_json_file(self.filter_requested_info(info_dict), infofn)
1799                 except (OSError, IOError):
1800                     self.report_error('Cannot write metadata to JSON file ' + infofn)
1801                     return
1802
1803         self._write_thumbnails(info_dict, filename)
1804
1805         if not self.params.get('skip_download', False):
1806             try:
1807                 def dl(name, info):
1808                     fd = get_suitable_downloader(info, self.params)(self, self.params)
1809                     for ph in self._progress_hooks:
1810                         fd.add_progress_hook(ph)
1811                     if self.params.get('verbose'):
1812                         self.to_stdout('[debug] Invoking downloader on %r' % info.get('url'))
1813                     return fd.download(name, info)
1814
1815                 if info_dict.get('requested_formats') is not None:
1816                     downloaded = []
1817                     success = True
1818                     merger = FFmpegMergerPP(self)
1819                     if not merger.available:
1820                         postprocessors = []
1821                         self.report_warning('You have requested multiple '
1822                                             'formats but ffmpeg or avconv are not installed.'
1823                                             ' The formats won\'t be merged.')
1824                     else:
1825                         postprocessors = [merger]
1826
1827                     def compatible_formats(formats):
1828                         video, audio = formats
1829                         # Check extension
1830                         video_ext, audio_ext = audio.get('ext'), video.get('ext')
1831                         if video_ext and audio_ext:
1832                             COMPATIBLE_EXTS = (
1833                                 ('mp3', 'mp4', 'm4a', 'm4p', 'm4b', 'm4r', 'm4v', 'ismv', 'isma'),
1834                                 ('webm')
1835                             )
1836                             for exts in COMPATIBLE_EXTS:
1837                                 if video_ext in exts and audio_ext in exts:
1838                                     return True
1839                         # TODO: Check acodec/vcodec
1840                         return False
1841
1842                     filename_real_ext = os.path.splitext(filename)[1][1:]
1843                     filename_wo_ext = (
1844                         os.path.splitext(filename)[0]
1845                         if filename_real_ext == info_dict['ext']
1846                         else filename)
1847                     requested_formats = info_dict['requested_formats']
1848                     if self.params.get('merge_output_format') is None and not compatible_formats(requested_formats):
1849                         info_dict['ext'] = 'mkv'
1850                         self.report_warning(
1851                             'Requested formats are incompatible for merge and will be merged into mkv.')
1852                     # Ensure filename always has a correct extension for successful merge
1853                     filename = '%s.%s' % (filename_wo_ext, info_dict['ext'])
1854                     if os.path.exists(encodeFilename(filename)):
1855                         self.to_screen(
1856                             '[download] %s has already been downloaded and '
1857                             'merged' % filename)
1858                     else:
1859                         for f in requested_formats:
1860                             new_info = dict(info_dict)
1861                             new_info.update(f)
1862                             fname = prepend_extension(
1863                                 self.prepare_filename(new_info),
1864                                 'f%s' % f['format_id'], new_info['ext'])
1865                             if not ensure_dir_exists(fname):
1866                                 return
1867                             downloaded.append(fname)
1868                             partial_success = dl(fname, new_info)
1869                             success = success and partial_success
1870                         info_dict['__postprocessors'] = postprocessors
1871                         info_dict['__files_to_merge'] = downloaded
1872                 else:
1873                     # Just a single file
1874                     success = dl(filename, info_dict)
1875             except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
1876                 self.report_error('unable to download video data: %s' % error_to_compat_str(err))
1877                 return
1878             except (OSError, IOError) as err:
1879                 raise UnavailableVideoError(err)
1880             except (ContentTooShortError, ) as err:
1881                 self.report_error('content too short (expected %s bytes and served %s)' % (err.expected, err.downloaded))
1882                 return
1883
1884             if success and filename != '-':
1885                 # Fixup content
1886                 fixup_policy = self.params.get('fixup')
1887                 if fixup_policy is None:
1888                     fixup_policy = 'detect_or_warn'
1889
1890                 INSTALL_FFMPEG_MESSAGE = 'Install ffmpeg or avconv to fix this automatically.'
1891
1892                 stretched_ratio = info_dict.get('stretched_ratio')
1893                 if stretched_ratio is not None and stretched_ratio != 1:
1894                     if fixup_policy == 'warn':
1895                         self.report_warning('%s: Non-uniform pixel ratio (%s)' % (
1896                             info_dict['id'], stretched_ratio))
1897                     elif fixup_policy == 'detect_or_warn':
1898                         stretched_pp = FFmpegFixupStretchedPP(self)
1899                         if stretched_pp.available:
1900                             info_dict.setdefault('__postprocessors', [])
1901                             info_dict['__postprocessors'].append(stretched_pp)
1902                         else:
1903                             self.report_warning(
1904                                 '%s: Non-uniform pixel ratio (%s). %s'
1905                                 % (info_dict['id'], stretched_ratio, INSTALL_FFMPEG_MESSAGE))
1906                     else:
1907                         assert fixup_policy in ('ignore', 'never')
1908
1909                 if (info_dict.get('requested_formats') is None and
1910                         info_dict.get('container') == 'm4a_dash'):
1911                     if fixup_policy == 'warn':
1912                         self.report_warning(
1913                             '%s: writing DASH m4a. '
1914                             'Only some players support this container.'
1915                             % info_dict['id'])
1916                     elif fixup_policy == 'detect_or_warn':
1917                         fixup_pp = FFmpegFixupM4aPP(self)
1918                         if fixup_pp.available:
1919                             info_dict.setdefault('__postprocessors', [])
1920                             info_dict['__postprocessors'].append(fixup_pp)
1921                         else:
1922                             self.report_warning(
1923                                 '%s: writing DASH m4a. '
1924                                 'Only some players support this container. %s'
1925                                 % (info_dict['id'], INSTALL_FFMPEG_MESSAGE))
1926                     else:
1927                         assert fixup_policy in ('ignore', 'never')
1928
1929                 if (info_dict.get('protocol') == 'm3u8_native' or
1930                         info_dict.get('protocol') == 'm3u8' and
1931                         self.params.get('hls_prefer_native')):
1932                     if fixup_policy == 'warn':
1933                         self.report_warning('%s: malformed AAC bitstream detected.' % (
1934                             info_dict['id']))
1935                     elif fixup_policy == 'detect_or_warn':
1936                         fixup_pp = FFmpegFixupM3u8PP(self)
1937                         if fixup_pp.available:
1938                             info_dict.setdefault('__postprocessors', [])
1939                             info_dict['__postprocessors'].append(fixup_pp)
1940                         else:
1941                             self.report_warning(
1942                                 '%s: malformed AAC bitstream detected. %s'
1943                                 % (info_dict['id'], INSTALL_FFMPEG_MESSAGE))
1944                     else:
1945                         assert fixup_policy in ('ignore', 'never')
1946
1947                 try:
1948                     self.post_process(filename, info_dict)
1949                 except (PostProcessingError) as err:
1950                     self.report_error('postprocessing: %s' % str(err))
1951                     return
1952                 self.record_download_archive(info_dict)
1953
1954     def download(self, url_list):
1955         """Download a given list of URLs."""
1956         outtmpl = self.params.get('outtmpl', DEFAULT_OUTTMPL)
1957         if (len(url_list) > 1 and
1958                 outtmpl != '-' and
1959                 '%' not in outtmpl and
1960                 self.params.get('max_downloads') != 1):
1961             raise SameFileError(outtmpl)
1962
1963         for url in url_list:
1964             try:
1965                 # It also downloads the videos
1966                 res = self.extract_info(
1967                     url, force_generic_extractor=self.params.get('force_generic_extractor', False))
1968             except UnavailableVideoError:
1969                 self.report_error('unable to download video')
1970             except MaxDownloadsReached:
1971                 self.to_screen('[info] Maximum number of downloaded files reached.')
1972                 raise
1973             else:
1974                 if self.params.get('dump_single_json', False):
1975                     self.to_stdout(json.dumps(res))
1976
1977         return self._download_retcode
1978
1979     def download_with_info_file(self, info_filename):
1980         with contextlib.closing(fileinput.FileInput(
1981                 [info_filename], mode='r',
1982                 openhook=fileinput.hook_encoded('utf-8'))) as f:
1983             # FileInput doesn't have a read method, we can't call json.load
1984             info = self.filter_requested_info(json.loads('\n'.join(f)))
1985         try:
1986             self.process_ie_result(info, download=True)
1987         except DownloadError:
1988             webpage_url = info.get('webpage_url')
1989             if webpage_url is not None:
1990                 self.report_warning('The info failed to download, trying with "%s"' % webpage_url)
1991                 return self.download([webpage_url])
1992             else:
1993                 raise
1994         return self._download_retcode
1995
1996     @staticmethod
1997     def filter_requested_info(info_dict):
1998         return dict(
1999             (k, v) for k, v in info_dict.items()
2000             if k not in ['requested_formats', 'requested_subtitles'])
2001
2002     def post_process(self, filename, ie_info):
2003         """Run all the postprocessors on the given file."""
2004         info = dict(ie_info)
2005         info['filepath'] = filename
2006         pps_chain = []
2007         if ie_info.get('__postprocessors') is not None:
2008             pps_chain.extend(ie_info['__postprocessors'])
2009         pps_chain.extend(self._pps)
2010         for pp in pps_chain:
2011             files_to_delete = []
2012             try:
2013                 files_to_delete, info = pp.run(info)
2014             except PostProcessingError as e:
2015                 self.report_error(e.msg)
2016             if files_to_delete and not self.params.get('keepvideo', False):
2017                 for old_filename in files_to_delete:
2018                     self.to_screen('Deleting original file %s (pass -k to keep)' % old_filename)
2019                     try:
2020                         os.remove(encodeFilename(old_filename))
2021                     except (IOError, OSError):
2022                         self.report_warning('Unable to remove downloaded original file')
2023
2024     def _make_archive_id(self, info_dict):
2025         # Future-proof against any change in case
2026         # and backwards compatibility with prior versions
2027         extractor = info_dict.get('extractor_key')
2028         if extractor is None:
2029             if 'id' in info_dict:
2030                 extractor = info_dict.get('ie_key')  # key in a playlist
2031         if extractor is None:
2032             return None  # Incomplete video information
2033         return extractor.lower() + ' ' + info_dict['id']
2034
2035     def in_download_archive(self, info_dict):
2036         fn = self.params.get('download_archive')
2037         if fn is None:
2038             return False
2039
2040         vid_id = self._make_archive_id(info_dict)
2041         if vid_id is None:
2042             return False  # Incomplete video information
2043
2044         try:
2045             with locked_file(fn, 'r', encoding='utf-8') as archive_file:
2046                 for line in archive_file:
2047                     if line.strip() == vid_id:
2048                         return True
2049         except IOError as ioe:
2050             if ioe.errno != errno.ENOENT:
2051                 raise
2052         return False
2053
2054     def record_download_archive(self, info_dict):
2055         fn = self.params.get('download_archive')
2056         if fn is None:
2057             return
2058         vid_id = self._make_archive_id(info_dict)
2059         assert vid_id
2060         with locked_file(fn, 'a', encoding='utf-8') as archive_file:
2061             archive_file.write(vid_id + '\n')
2062
2063     @staticmethod
2064     def format_resolution(format, default='unknown'):
2065         if format.get('vcodec') == 'none':
2066             return 'audio only'
2067         if format.get('resolution') is not None:
2068             return format['resolution']
2069         if format.get('height') is not None:
2070             if format.get('width') is not None:
2071                 res = '%sx%s' % (format['width'], format['height'])
2072             else:
2073                 res = '%sp' % format['height']
2074         elif format.get('width') is not None:
2075             res = '%dx?' % format['width']
2076         else:
2077             res = default
2078         return res
2079
2080     def _format_note(self, fdict):
2081         res = ''
2082         if fdict.get('ext') in ['f4f', 'f4m']:
2083             res += '(unsupported) '
2084         if fdict.get('language'):
2085             if res:
2086                 res += ' '
2087             res += '[%s] ' % fdict['language']
2088         if fdict.get('format_note') is not None:
2089             res += fdict['format_note'] + ' '
2090         if fdict.get('tbr') is not None:
2091             res += '%4dk ' % fdict['tbr']
2092         if fdict.get('container') is not None:
2093             if res:
2094                 res += ', '
2095             res += '%s container' % fdict['container']
2096         if (fdict.get('vcodec') is not None and
2097                 fdict.get('vcodec') != 'none'):
2098             if res:
2099                 res += ', '
2100             res += fdict['vcodec']
2101             if fdict.get('vbr') is not None:
2102                 res += '@'
2103         elif fdict.get('vbr') is not None and fdict.get('abr') is not None:
2104             res += 'video@'
2105         if fdict.get('vbr') is not None:
2106             res += '%4dk' % fdict['vbr']
2107         if fdict.get('fps') is not None:
2108             if res:
2109                 res += ', '
2110             res += '%sfps' % fdict['fps']
2111         if fdict.get('acodec') is not None:
2112             if res:
2113                 res += ', '
2114             if fdict['acodec'] == 'none':
2115                 res += 'video only'
2116             else:
2117                 res += '%-5s' % fdict['acodec']
2118         elif fdict.get('abr') is not None:
2119             if res:
2120                 res += ', '
2121             res += 'audio'
2122         if fdict.get('abr') is not None:
2123             res += '@%3dk' % fdict['abr']
2124         if fdict.get('asr') is not None:
2125             res += ' (%5dHz)' % fdict['asr']
2126         if fdict.get('filesize') is not None:
2127             if res:
2128                 res += ', '
2129             res += format_bytes(fdict['filesize'])
2130         elif fdict.get('filesize_approx') is not None:
2131             if res:
2132                 res += ', '
2133             res += '~' + format_bytes(fdict['filesize_approx'])
2134         return res
2135
2136     def list_formats(self, info_dict):
2137         formats = info_dict.get('formats', [info_dict])
2138         table = [
2139             [f['format_id'], f['ext'], self.format_resolution(f), self._format_note(f)]
2140             for f in formats
2141             if f.get('preference') is None or f['preference'] >= -1000]
2142         if len(formats) > 1:
2143             table[-1][-1] += (' ' if table[-1][-1] else '') + '(best)'
2144
2145         header_line = ['format code', 'extension', 'resolution', 'note']
2146         self.to_screen(
2147             '[info] Available formats for %s:\n%s' %
2148             (info_dict['id'], render_table(header_line, table)))
2149
2150     def list_thumbnails(self, info_dict):
2151         thumbnails = info_dict.get('thumbnails')
2152         if not thumbnails:
2153             self.to_screen('[info] No thumbnails present for %s' % info_dict['id'])
2154             return
2155
2156         self.to_screen(
2157             '[info] Thumbnails for %s:' % info_dict['id'])
2158         self.to_screen(render_table(
2159             ['ID', 'width', 'height', 'URL'],
2160             [[t['id'], t.get('width', 'unknown'), t.get('height', 'unknown'), t['url']] for t in thumbnails]))
2161
2162     def list_subtitles(self, video_id, subtitles, name='subtitles'):
2163         if not subtitles:
2164             self.to_screen('%s has no %s' % (video_id, name))
2165             return
2166         self.to_screen(
2167             'Available %s for %s:' % (name, video_id))
2168         self.to_screen(render_table(
2169             ['Language', 'formats'],
2170             [[lang, ', '.join(f['ext'] for f in reversed(formats))]
2171                 for lang, formats in subtitles.items()]))
2172
2173     def urlopen(self, req):
2174         """ Start an HTTP download """
2175         if isinstance(req, compat_basestring):
2176             req = sanitized_Request(req)
2177         return self._opener.open(req, timeout=self._socket_timeout)
2178
2179     def print_debug_header(self):
2180         if not self.params.get('verbose'):
2181             return
2182
2183         if type('') is not compat_str:
2184             # Python 2.6 on SLES11 SP1 (https://github.com/rg3/youtube-dl/issues/3326)
2185             self.report_warning(
2186                 'Your Python is broken! Update to a newer and supported version')
2187
2188         stdout_encoding = getattr(
2189             sys.stdout, 'encoding', 'missing (%s)' % type(sys.stdout).__name__)
2190         encoding_str = (
2191             '[debug] Encodings: locale %s, fs %s, out %s, pref %s\n' % (
2192                 locale.getpreferredencoding(),
2193                 sys.getfilesystemencoding(),
2194                 stdout_encoding,
2195                 self.get_encoding()))
2196         write_string(encoding_str, encoding=None)
2197
2198         self._write_string('[debug] youtube-dl version ' + __version__ + '\n')
2199         if _LAZY_LOADER:
2200             self._write_string('[debug] Lazy loading extractors enabled' + '\n')
2201         try:
2202             sp = subprocess.Popen(
2203                 ['git', 'rev-parse', '--short', 'HEAD'],
2204                 stdout=subprocess.PIPE, stderr=subprocess.PIPE,
2205                 cwd=os.path.dirname(os.path.abspath(__file__)))
2206             out, err = sp.communicate()
2207             out = out.decode().strip()
2208             if re.match('[0-9a-f]+', out):
2209                 self._write_string('[debug] Git HEAD: ' + out + '\n')
2210         except Exception:
2211             try:
2212                 sys.exc_clear()
2213             except Exception:
2214                 pass
2215         self._write_string('[debug] Python version %s - %s\n' % (
2216             platform.python_version(), platform_name()))
2217
2218         exe_versions = FFmpegPostProcessor.get_versions(self)
2219         exe_versions['rtmpdump'] = rtmpdump_version()
2220         exe_str = ', '.join(
2221             '%s %s' % (exe, v)
2222             for exe, v in sorted(exe_versions.items())
2223             if v
2224         )
2225         if not exe_str:
2226             exe_str = 'none'
2227         self._write_string('[debug] exe versions: %s\n' % exe_str)
2228
2229         proxy_map = {}
2230         for handler in self._opener.handlers:
2231             if hasattr(handler, 'proxies'):
2232                 proxy_map.update(handler.proxies)
2233         self._write_string('[debug] Proxy map: ' + compat_str(proxy_map) + '\n')
2234
2235         if self.params.get('call_home', False):
2236             ipaddr = self.urlopen('https://yt-dl.org/ip').read().decode('utf-8')
2237             self._write_string('[debug] Public IP address: %s\n' % ipaddr)
2238             latest_version = self.urlopen(
2239                 'https://yt-dl.org/latest/version').read().decode('utf-8')
2240             if version_tuple(latest_version) > version_tuple(__version__):
2241                 self.report_warning(
2242                     'You are using an outdated version (newest version: %s)! '
2243                     'See https://yt-dl.org/update if you need help updating.' %
2244                     latest_version)
2245
2246     def _setup_opener(self):
2247         timeout_val = self.params.get('socket_timeout')
2248         self._socket_timeout = 600 if timeout_val is None else float(timeout_val)
2249
2250         opts_cookiefile = self.params.get('cookiefile')
2251         opts_proxy = self.params.get('proxy')
2252
2253         if opts_cookiefile is None:
2254             self.cookiejar = compat_cookiejar.CookieJar()
2255         else:
2256             opts_cookiefile = expand_path(opts_cookiefile)
2257             self.cookiejar = compat_cookiejar.MozillaCookieJar(
2258                 opts_cookiefile)
2259             if os.access(opts_cookiefile, os.R_OK):
2260                 self.cookiejar.load()
2261
2262         cookie_processor = YoutubeDLCookieProcessor(self.cookiejar)
2263         if opts_proxy is not None:
2264             if opts_proxy == '':
2265                 proxies = {}
2266             else:
2267                 proxies = {'http': opts_proxy, 'https': opts_proxy}
2268         else:
2269             proxies = compat_urllib_request.getproxies()
2270             # Set HTTPS proxy to HTTP one if given (https://github.com/rg3/youtube-dl/issues/805)
2271             if 'http' in proxies and 'https' not in proxies:
2272                 proxies['https'] = proxies['http']
2273         proxy_handler = PerRequestProxyHandler(proxies)
2274
2275         debuglevel = 1 if self.params.get('debug_printtraffic') else 0
2276         https_handler = make_HTTPS_handler(self.params, debuglevel=debuglevel)
2277         ydlh = YoutubeDLHandler(self.params, debuglevel=debuglevel)
2278         data_handler = compat_urllib_request_DataHandler()
2279
2280         # When passing our own FileHandler instance, build_opener won't add the
2281         # default FileHandler and allows us to disable the file protocol, which
2282         # can be used for malicious purposes (see
2283         # https://github.com/rg3/youtube-dl/issues/8227)
2284         file_handler = compat_urllib_request.FileHandler()
2285
2286         def file_open(*args, **kwargs):
2287             raise compat_urllib_error.URLError('file:// scheme is explicitly disabled in youtube-dl for security reasons')
2288         file_handler.file_open = file_open
2289
2290         opener = compat_urllib_request.build_opener(
2291             proxy_handler, https_handler, cookie_processor, ydlh, data_handler, file_handler)
2292
2293         # Delete the default user-agent header, which would otherwise apply in
2294         # cases where our custom HTTP handler doesn't come into play
2295         # (See https://github.com/rg3/youtube-dl/issues/1309 for details)
2296         opener.addheaders = []
2297         self._opener = opener
2298
2299     def encode(self, s):
2300         if isinstance(s, bytes):
2301             return s  # Already encoded
2302
2303         try:
2304             return s.encode(self.get_encoding())
2305         except UnicodeEncodeError as err:
2306             err.reason = err.reason + '. Check your system encoding configuration or use the --encoding option.'
2307             raise
2308
2309     def get_encoding(self):
2310         encoding = self.params.get('encoding')
2311         if encoding is None:
2312             encoding = preferredencoding()
2313         return encoding
2314
2315     def _write_thumbnails(self, info_dict, filename):
2316         if self.params.get('writethumbnail', False):
2317             thumbnails = info_dict.get('thumbnails')
2318             if thumbnails:
2319                 thumbnails = [thumbnails[-1]]
2320         elif self.params.get('write_all_thumbnails', False):
2321             thumbnails = info_dict.get('thumbnails')
2322         else:
2323             return
2324
2325         if not thumbnails:
2326             # No thumbnails present, so return immediately
2327             return
2328
2329         for t in thumbnails:
2330             thumb_ext = determine_ext(t['url'], 'jpg')
2331             suffix = '_%s' % t['id'] if len(thumbnails) > 1 else ''
2332             thumb_display_id = '%s ' % t['id'] if len(thumbnails) > 1 else ''
2333             t['filename'] = thumb_filename = os.path.splitext(filename)[0] + suffix + '.' + thumb_ext
2334
2335             if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(thumb_filename)):
2336                 self.to_screen('[%s] %s: Thumbnail %sis already present' %
2337                                (info_dict['extractor'], info_dict['id'], thumb_display_id))
2338             else:
2339                 self.to_screen('[%s] %s: Downloading thumbnail %s...' %
2340                                (info_dict['extractor'], info_dict['id'], thumb_display_id))
2341                 try:
2342                     uf = self.urlopen(t['url'])
2343                     with open(encodeFilename(thumb_filename), 'wb') as thumbf:
2344                         shutil.copyfileobj(uf, thumbf)
2345                     self.to_screen('[%s] %s: Writing thumbnail %sto: %s' %
2346                                    (info_dict['extractor'], info_dict['id'], thumb_display_id, thumb_filename))
2347                 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
2348                     self.report_warning('Unable to download thumbnail "%s": %s' %
2349                                         (t['url'], error_to_compat_str(err)))