]> jfr.im git - yt-dlp.git/blame - youtube_dl/YoutubeDL.py
[utils,common] Merge format_srt_time and _subtitles_timecode
[yt-dlp.git] / youtube_dl / YoutubeDL.py
CommitLineData
8222d8de
JMF
1#!/usr/bin/env python
2# -*- coding: utf-8 -*-
3
6febd1c1 4from __future__ import absolute_import, unicode_literals
8222d8de 5
26e63931 6import collections
31bd3925 7import contextlib
9d2ecdbc 8import datetime
c1c9a79c 9import errno
31bd3925 10import fileinput
8222d8de 11import io
b82f815f 12import itertools
8694c600 13import json
62fec3b2 14import locale
083c9df9 15import operator
8222d8de 16import os
dca08720 17import platform
8222d8de
JMF
18import re
19import shutil
dca08720 20import subprocess
8222d8de
JMF
21import socket
22import sys
23import time
24import traceback
25
1e5b9a95
PH
26if os.name == 'nt':
27 import ctypes
28
8c25f81b 29from .compat import (
8f9312c3 30 compat_basestring,
dca08720 31 compat_cookiejar,
4644ac55 32 compat_expanduser,
003c69a8 33 compat_get_terminal_size,
ce02ed60 34 compat_http_client,
4f026faf 35 compat_kwargs,
ce02ed60
PH
36 compat_str,
37 compat_urllib_error,
38 compat_urllib_request,
8c25f81b
PH
39)
40from .utils import (
d05cfe06 41 escape_url,
ce02ed60
PH
42 ContentTooShortError,
43 date_from_str,
44 DateRange,
acd69589 45 DEFAULT_OUTTMPL,
ce02ed60
PH
46 determine_ext,
47 DownloadError,
48 encodeFilename,
49 ExtractorError,
02dbf93f 50 format_bytes,
525ef922 51 formatSeconds,
ce02ed60 52 locked_file,
dca08720 53 make_HTTPS_handler,
ce02ed60 54 MaxDownloadsReached,
b7ab0590 55 PagedList,
083c9df9 56 parse_filesize,
91410c9b 57 PerRequestProxyHandler,
ce02ed60 58 PostProcessingError,
dca08720 59 platform_name,
ce02ed60 60 preferredencoding,
cfb56d1a 61 render_table,
ce02ed60
PH
62 SameFileError,
63 sanitize_filename,
1bb5c511 64 sanitize_path,
e5660ee6 65 std_headers,
ce02ed60 66 subtitles_filename,
ce02ed60 67 UnavailableVideoError,
29eb5174 68 url_basename,
58b1f00d 69 version_tuple,
ce02ed60
PH
70 write_json_file,
71 write_string,
dca08720 72 YoutubeDLHandler,
6350728b 73 prepend_extension,
b29e0000 74 replace_extension,
7d4111ed 75 args_to_str,
05900629 76 age_restricted,
ce02ed60 77)
a0e07d31 78from .cache import Cache
023fa8c4 79from .extractor import get_info_extractor, gen_extractors
3bc2ddcc 80from .downloader import get_suitable_downloader
4c83c967 81from .downloader.rtmp import rtmpdump_version
4f026faf 82from .postprocessor import (
62cd676c 83 FFmpegFixupM4aPP,
6271f1ca 84 FFmpegFixupStretchedPP,
4f026faf
PH
85 FFmpegMergerPP,
86 FFmpegPostProcessor,
87 get_postprocessor,
88)
dca08720 89from .version import __version__
8222d8de
JMF
90
91
92class YoutubeDL(object):
93 """YoutubeDL class.
94
95 YoutubeDL objects are the ones responsible of downloading the
96 actual video file and writing it to disk if the user has requested
97 it, among some other tasks. In most cases there should be one per
98 program. As, given a video URL, the downloader doesn't know how to
99 extract all the needed information, task that InfoExtractors do, it
100 has to pass the URL to one of them.
101
102 For this, YoutubeDL objects have a method that allows
103 InfoExtractors to be registered in a given order. When it is passed
104 a URL, the YoutubeDL object handles it to the first InfoExtractor it
105 finds that reports being able to handle it. The InfoExtractor extracts
106 all the information about the video or videos the URL refers to, and
107 YoutubeDL process the extracted information, possibly using a File
108 Downloader to download the video.
109
110 YoutubeDL objects accept a lot of parameters. In order not to saturate
111 the object constructor with arguments, it receives a dictionary of
112 options instead. These options are available through the params
113 attribute for the InfoExtractors to use. The YoutubeDL also
114 registers itself as the downloader in charge for the InfoExtractors
115 that are added to it, so this is a "mutual registration".
116
117 Available options:
118
119 username: Username for authentication purposes.
120 password: Password for authentication purposes.
c6c19746 121 videopassword: Password for acces a video.
8222d8de
JMF
122 usenetrc: Use netrc for authentication instead.
123 verbose: Print additional info to stdout.
124 quiet: Do not print messages to stdout.
ad8915b7 125 no_warnings: Do not print out anything for warnings.
8222d8de
JMF
126 forceurl: Force printing final URL.
127 forcetitle: Force printing title.
128 forceid: Force printing ID.
129 forcethumbnail: Force printing thumbnail URL.
130 forcedescription: Force printing description.
131 forcefilename: Force printing final filename.
525ef922 132 forceduration: Force printing duration.
8694c600 133 forcejson: Force printing info_dict as JSON.
63e0be34
PH
134 dump_single_json: Force printing the info_dict of the whole playlist
135 (or video) as a single JSON line.
8222d8de 136 simulate: Do not download the video files.
d8600787 137 format: Video format code. See options.py for more information.
8222d8de
JMF
138 outtmpl: Template for output names.
139 restrictfilenames: Do not allow "&" and spaces in file names
140 ignoreerrors: Do not stop on download errors.
141 nooverwrites: Prevent overwriting files.
142 playliststart: Playlist item to start at.
143 playlistend: Playlist item to end at.
c14e88f0 144 playlist_items: Specific indices of playlist to download.
ff815fe6 145 playlistreverse: Download playlist items in reverse order.
8222d8de
JMF
146 matchtitle: Download only matching titles.
147 rejecttitle: Reject downloads for matching titles.
8bf9319e 148 logger: Log messages to a logging.Logger instance.
8222d8de
JMF
149 logtostderr: Log messages to stderr instead of stdout.
150 writedescription: Write the video description to a .description file
151 writeinfojson: Write the video description to a .info.json file
1fb07d10 152 writeannotations: Write the video annotations to a .annotations.xml file
8222d8de 153 writethumbnail: Write the thumbnail image to a file
ec82d85a 154 write_all_thumbnails: Write all thumbnail formats to files
8222d8de 155 writesubtitles: Write the video subtitles to a file
b004821f 156 writeautomaticsub: Write the automatic subtitles to a file
8222d8de 157 allsubtitles: Downloads all the subtitles of the video
0b7f3118 158 (requires writesubtitles or writeautomaticsub)
8222d8de 159 listsubtitles: Lists all available subtitles for the video
a504ced0 160 subtitlesformat: The format code for subtitles
aa6a10c4 161 subtitleslangs: List of languages of the subtitles to download
8222d8de
JMF
162 keepvideo: Keep the video file after post-processing
163 daterange: A DateRange object, download only if the upload_date is in the range.
164 skip_download: Skip the actual download of the video file
c35f9e72 165 cachedir: Location of the cache files in the filesystem.
a0e07d31 166 False to disable filesystem cache.
47192f92 167 noplaylist: Download single video instead of a playlist if in doubt.
8dbe9899
PH
168 age_limit: An integer representing the user's age in years.
169 Unsuitable videos for the given age are skipped.
5fe18bdb
PH
170 min_views: An integer representing the minimum view count the video
171 must have in order to not be skipped.
172 Videos without view count information are always
173 downloaded. None for no limit.
174 max_views: An integer representing the maximum view count.
175 Videos that are more popular than that are not
176 downloaded.
177 Videos without view count information are always
178 downloaded. None for no limit.
179 download_archive: File name of a file where all downloads are recorded.
c1c9a79c
PH
180 Videos already present in the file are not downloaded
181 again.
dca08720 182 cookiefile: File name where cookies should be read from and dumped to.
a1ee09e8 183 nocheckcertificate:Do not verify SSL certificates
7e8c0af0
PH
184 prefer_insecure: Use HTTP instead of HTTPS to retrieve information.
185 At the moment, this is only supported by YouTube.
a1ee09e8 186 proxy: URL of the proxy server to use
91410c9b
PH
187 cn_verification_proxy: URL of the proxy to use for IP address verification
188 on Chinese sites. (Experimental)
e344693b 189 socket_timeout: Time to wait for unresponsive hosts, in seconds
0783b09b
PH
190 bidi_workaround: Work around buggy terminals without bidirectional text
191 support, using fridibi
a0ddb8a2 192 debug_printtraffic:Print out sent and received HTTP traffic
7b0817e8 193 include_ads: Download ads as well
04b4d394
PH
194 default_search: Prepend this string if an input url is not valid.
195 'auto' for elaborate guessing
62fec3b2 196 encoding: Use this encoding instead of the system-specified.
e8ee972c 197 extract_flat: Do not resolve URLs, return the immediate result.
057a5206
PH
198 Pass in 'in_playlist' to only show this behavior for
199 playlist items.
4f026faf 200 postprocessors: A list of dictionaries, each with an entry
71b640cc
PH
201 * key: The name of the postprocessor. See
202 youtube_dl/postprocessor/__init__.py for a list.
4f026faf
PH
203 as well as any further keyword arguments for the
204 postprocessor.
71b640cc
PH
205 progress_hooks: A list of functions that get called on download
206 progress, with a dictionary with the entries
5cda4eda 207 * status: One of "downloading", "error", or "finished".
ee69b99a 208 Check this first and ignore unknown values.
71b640cc 209
5cda4eda 210 If status is one of "downloading", or "finished", the
ee69b99a
PH
211 following properties may also be present:
212 * filename: The final filename (always present)
5cda4eda 213 * tmpfilename: The filename we're currently writing to
71b640cc
PH
214 * downloaded_bytes: Bytes on disk
215 * total_bytes: Size of the whole file, None if unknown
5cda4eda
PH
216 * total_bytes_estimate: Guess of the eventual file size,
217 None if unavailable.
218 * elapsed: The number of seconds since download started.
71b640cc
PH
219 * eta: The estimated time in seconds, None if unknown
220 * speed: The download speed in bytes/second, None if
221 unknown
5cda4eda
PH
222 * fragment_index: The counter of the currently
223 downloaded video fragment.
224 * fragment_count: The number of fragments (= individual
225 files that will be merged)
71b640cc
PH
226
227 Progress hooks are guaranteed to be called at least once
228 (with status "finished") if the download is successful.
45598f15 229 merge_output_format: Extension to use when merging formats.
6271f1ca
PH
230 fixup: Automatically correct known faults of the file.
231 One of:
232 - "never": do nothing
233 - "warn": only emit a warning
234 - "detect_or_warn": check whether we can do anything
62cd676c 235 about it, warn otherwise (default)
be4a824d 236 source_address: (Experimental) Client-side IP address to bind to.
8bfa7545
PH
237 call_home: Boolean, true iff we are allowed to contact the
238 youtube-dl servers for debugging.
5f0d813d 239 sleep_interval: Number of seconds to sleep before each download.
cfb56d1a
PH
240 listformats: Print an overview of available video formats and exit.
241 list_thumbnails: Print a table of all thumbnails and exit.
347de493
PH
242 match_filter: A function that gets called with the info_dict of
243 every video.
244 If it returns a message, the video is ignored.
245 If it returns None, the video is downloaded.
246 match_filter_func in utils.py is one example for this.
7e5db8c9 247 no_color: Do not emit color codes in output.
71b640cc 248
85729c51
PH
249 The following options determine which downloader is picked:
250 external_downloader: Executable of the external downloader to call.
251 None or unset for standard (built-in) downloader.
252 hls_prefer_native: Use the native HLS downloader instead of ffmpeg/avconv.
fe7e0c98 253
8222d8de 254 The following parameters are not used by YoutubeDL itself, they are used by
c75f0b36 255 the downloader (see youtube_dl/downloader/common.py):
8222d8de 256 nopart, updatetime, buffersize, ratelimit, min_filesize, max_filesize, test,
881e6a1f 257 noresizebuffer, retries, continuedl, noprogress, consoletitle,
c75f0b36 258 xattr_set_filesize, external_downloader_args.
76b1bd67
JMF
259
260 The following options are used by the post processors:
261 prefer_ffmpeg: If True, use ffmpeg instead of avconv if both are available,
262 otherwise prefer avconv.
8222d8de
JMF
263 """
264
265 params = None
266 _ies = []
267 _pps = []
268 _download_retcode = None
269 _num_downloads = None
270 _screen_file = None
271
3511266b 272 def __init__(self, params=None, auto_init=True):
8222d8de 273 """Create a FileDownloader object with the given options."""
e9f9a10f
JMF
274 if params is None:
275 params = {}
8222d8de 276 self._ies = []
56c73665 277 self._ies_instances = {}
8222d8de 278 self._pps = []
933605d7 279 self._progress_hooks = []
8222d8de
JMF
280 self._download_retcode = 0
281 self._num_downloads = 0
282 self._screen_file = [sys.stdout, sys.stderr][params.get('logtostderr', False)]
0783b09b 283 self._err_file = sys.stderr
e9f9a10f 284 self.params = params
a0e07d31 285 self.cache = Cache(self)
34308b30 286
0783b09b 287 if params.get('bidi_workaround', False):
1c088fa8
PH
288 try:
289 import pty
290 master, slave = pty.openpty()
003c69a8 291 width = compat_get_terminal_size().columns
1c088fa8
PH
292 if width is None:
293 width_args = []
294 else:
295 width_args = ['-w', str(width)]
5d681e96 296 sp_kwargs = dict(
1c088fa8
PH
297 stdin=subprocess.PIPE,
298 stdout=slave,
299 stderr=self._err_file)
5d681e96
PH
300 try:
301 self._output_process = subprocess.Popen(
302 ['bidiv'] + width_args, **sp_kwargs
303 )
304 except OSError:
5d681e96
PH
305 self._output_process = subprocess.Popen(
306 ['fribidi', '-c', 'UTF-8'] + width_args, **sp_kwargs)
307 self._output_channel = os.fdopen(master, 'rb')
1c088fa8
PH
308 except OSError as ose:
309 if ose.errno == 2:
6febd1c1 310 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.')
1c088fa8
PH
311 else:
312 raise
0783b09b 313
34308b30 314 if (sys.version_info >= (3,) and sys.platform != 'win32' and
8fb3ac36
PH
315 sys.getfilesystemencoding() in ['ascii', 'ANSI_X3.4-1968'] and
316 not params.get('restrictfilenames', False)):
34308b30
PH
317 # On Python 3, the Unicode filesystem API will throw errors (#1474)
318 self.report_warning(
6febd1c1 319 'Assuming --restrict-filenames since file system encoding '
1b725173 320 'cannot encode all characters. '
6febd1c1 321 'Set the LC_ALL environment variable to fix this.')
4a98cdbf 322 self.params['restrictfilenames'] = True
34308b30 323
486dd09e
PH
324 if isinstance(params.get('outtmpl'), bytes):
325 self.report_warning(
326 'Parameter outtmpl is bytes, but should be a unicode string. '
327 'Put from __future__ import unicode_literals at the top of your code file or consider switching to Python 3.x.')
328
dca08720
PH
329 self._setup_opener()
330
3511266b
PH
331 if auto_init:
332 self.print_debug_header()
333 self.add_default_info_extractors()
334
4f026faf
PH
335 for pp_def_raw in self.params.get('postprocessors', []):
336 pp_class = get_postprocessor(pp_def_raw['key'])
337 pp_def = dict(pp_def_raw)
338 del pp_def['key']
339 pp = pp_class(self, **compat_kwargs(pp_def))
340 self.add_post_processor(pp)
341
71b640cc
PH
342 for ph in self.params.get('progress_hooks', []):
343 self.add_progress_hook(ph)
344
7d4111ed
PH
345 def warn_if_short_id(self, argv):
346 # short YouTube ID starting with dash?
347 idxs = [
348 i for i, a in enumerate(argv)
349 if re.match(r'^-[0-9A-Za-z_-]{10}$', a)]
350 if idxs:
351 correct_argv = (
352 ['youtube-dl'] +
353 [a for i, a in enumerate(argv) if i not in idxs] +
354 ['--'] + [argv[i] for i in idxs]
355 )
356 self.report_warning(
357 'Long argument string detected. '
358 'Use -- to separate parameters and URLs, like this:\n%s\n' %
359 args_to_str(correct_argv))
360
8222d8de
JMF
361 def add_info_extractor(self, ie):
362 """Add an InfoExtractor object to the end of the list."""
363 self._ies.append(ie)
56c73665 364 self._ies_instances[ie.ie_key()] = ie
8222d8de
JMF
365 ie.set_downloader(self)
366
56c73665
JMF
367 def get_info_extractor(self, ie_key):
368 """
369 Get an instance of an IE with name ie_key, it will try to get one from
370 the _ies list, if there's no instance it will create a new one and add
371 it to the extractor list.
372 """
373 ie = self._ies_instances.get(ie_key)
374 if ie is None:
375 ie = get_info_extractor(ie_key)()
376 self.add_info_extractor(ie)
377 return ie
378
023fa8c4
JMF
379 def add_default_info_extractors(self):
380 """
381 Add the InfoExtractors returned by gen_extractors to the end of the list
382 """
383 for ie in gen_extractors():
384 self.add_info_extractor(ie)
385
8222d8de
JMF
386 def add_post_processor(self, pp):
387 """Add a PostProcessor object to the end of the chain."""
388 self._pps.append(pp)
389 pp.set_downloader(self)
390
933605d7
JMF
391 def add_progress_hook(self, ph):
392 """Add the progress hook (currently only for the file downloader)"""
393 self._progress_hooks.append(ph)
8ab470f1 394
1c088fa8 395 def _bidi_workaround(self, message):
5d681e96 396 if not hasattr(self, '_output_channel'):
1c088fa8
PH
397 return message
398
5d681e96 399 assert hasattr(self, '_output_process')
11b85ce6 400 assert isinstance(message, compat_str)
6febd1c1
PH
401 line_count = message.count('\n') + 1
402 self._output_process.stdin.write((message + '\n').encode('utf-8'))
5d681e96 403 self._output_process.stdin.flush()
6febd1c1 404 res = ''.join(self._output_channel.readline().decode('utf-8')
9e1a5b84 405 for _ in range(line_count))
6febd1c1 406 return res[:-len('\n')]
1c088fa8 407
8222d8de 408 def to_screen(self, message, skip_eol=False):
0783b09b
PH
409 """Print message to stdout if not in quiet mode."""
410 return self.to_stdout(message, skip_eol, check_quiet=True)
411
734f90bb 412 def _write_string(self, s, out=None):
b58ddb32 413 write_string(s, out=out, encoding=self.params.get('encoding'))
734f90bb 414
0783b09b 415 def to_stdout(self, message, skip_eol=False, check_quiet=False):
8222d8de 416 """Print message to stdout if not in quiet mode."""
8bf9319e 417 if self.params.get('logger'):
43afe285 418 self.params['logger'].debug(message)
0783b09b 419 elif not check_quiet or not self.params.get('quiet', False):
1c088fa8 420 message = self._bidi_workaround(message)
6febd1c1 421 terminator = ['\n', ''][skip_eol]
8222d8de 422 output = message + terminator
1c088fa8 423
734f90bb 424 self._write_string(output, self._screen_file)
8222d8de
JMF
425
426 def to_stderr(self, message):
427 """Print message to stderr."""
11b85ce6 428 assert isinstance(message, compat_str)
8bf9319e 429 if self.params.get('logger'):
43afe285
IB
430 self.params['logger'].error(message)
431 else:
1c088fa8 432 message = self._bidi_workaround(message)
6febd1c1 433 output = message + '\n'
734f90bb 434 self._write_string(output, self._err_file)
8222d8de 435
1e5b9a95
PH
436 def to_console_title(self, message):
437 if not self.params.get('consoletitle', False):
438 return
439 if os.name == 'nt' and ctypes.windll.kernel32.GetConsoleWindow():
440 # c_wchar_p() might not be necessary if `message` is
441 # already of type unicode()
442 ctypes.windll.kernel32.SetConsoleTitleW(ctypes.c_wchar_p(message))
443 elif 'TERM' in os.environ:
734f90bb 444 self._write_string('\033]0;%s\007' % message, self._screen_file)
1e5b9a95 445
bdde425c
PH
446 def save_console_title(self):
447 if not self.params.get('consoletitle', False):
448 return
449 if 'TERM' in os.environ:
efd6c574 450 # Save the title on stack
734f90bb 451 self._write_string('\033[22;0t', self._screen_file)
bdde425c
PH
452
453 def restore_console_title(self):
454 if not self.params.get('consoletitle', False):
455 return
456 if 'TERM' in os.environ:
efd6c574 457 # Restore the title from stack
734f90bb 458 self._write_string('\033[23;0t', self._screen_file)
bdde425c
PH
459
460 def __enter__(self):
461 self.save_console_title()
462 return self
463
464 def __exit__(self, *args):
465 self.restore_console_title()
f89197d7 466
dca08720
PH
467 if self.params.get('cookiefile') is not None:
468 self.cookiejar.save()
bdde425c 469
8222d8de
JMF
470 def trouble(self, message=None, tb=None):
471 """Determine action to take when a download problem appears.
472
473 Depending on if the downloader has been configured to ignore
474 download errors or not, this method may throw an exception or
475 not when errors are found, after printing the message.
476
477 tb, if given, is additional traceback information.
478 """
479 if message is not None:
480 self.to_stderr(message)
481 if self.params.get('verbose'):
482 if tb is None:
483 if sys.exc_info()[0]: # if .trouble has been called from an except block
6febd1c1 484 tb = ''
8222d8de 485 if hasattr(sys.exc_info()[1], 'exc_info') and sys.exc_info()[1].exc_info[0]:
6febd1c1 486 tb += ''.join(traceback.format_exception(*sys.exc_info()[1].exc_info))
8222d8de
JMF
487 tb += compat_str(traceback.format_exc())
488 else:
489 tb_data = traceback.format_list(traceback.extract_stack())
6febd1c1 490 tb = ''.join(tb_data)
8222d8de
JMF
491 self.to_stderr(tb)
492 if not self.params.get('ignoreerrors', False):
493 if sys.exc_info()[0] and hasattr(sys.exc_info()[1], 'exc_info') and sys.exc_info()[1].exc_info[0]:
494 exc_info = sys.exc_info()[1].exc_info
495 else:
496 exc_info = sys.exc_info()
497 raise DownloadError(message, exc_info)
498 self._download_retcode = 1
499
500 def report_warning(self, message):
501 '''
502 Print the message to stderr, it will be prefixed with 'WARNING:'
503 If stderr is a tty file the 'WARNING:' will be colored
504 '''
6d07ce01
JMF
505 if self.params.get('logger') is not None:
506 self.params['logger'].warning(message)
8222d8de 507 else:
ad8915b7
PH
508 if self.params.get('no_warnings'):
509 return
7e5db8c9 510 if not self.params.get('no_color') and self._err_file.isatty() and os.name != 'nt':
6d07ce01
JMF
511 _msg_header = '\033[0;33mWARNING:\033[0m'
512 else:
513 _msg_header = 'WARNING:'
514 warning_message = '%s %s' % (_msg_header, message)
515 self.to_stderr(warning_message)
8222d8de
JMF
516
517 def report_error(self, message, tb=None):
518 '''
519 Do the same as trouble, but prefixes the message with 'ERROR:', colored
520 in red if stderr is a tty file.
521 '''
7e5db8c9 522 if not self.params.get('no_color') and self._err_file.isatty() and os.name != 'nt':
6febd1c1 523 _msg_header = '\033[0;31mERROR:\033[0m'
8222d8de 524 else:
6febd1c1
PH
525 _msg_header = 'ERROR:'
526 error_message = '%s %s' % (_msg_header, message)
8222d8de
JMF
527 self.trouble(error_message, tb)
528
8222d8de
JMF
529 def report_file_already_downloaded(self, file_name):
530 """Report file has already been fully downloaded."""
531 try:
6febd1c1 532 self.to_screen('[download] %s has already been downloaded' % file_name)
ce02ed60 533 except UnicodeEncodeError:
6febd1c1 534 self.to_screen('[download] The file has already been downloaded')
8222d8de 535
8222d8de
JMF
536 def prepare_filename(self, info_dict):
537 """Generate the output filename."""
538 try:
539 template_dict = dict(info_dict)
540
541 template_dict['epoch'] = int(time.time())
542 autonumber_size = self.params.get('autonumber_size')
543 if autonumber_size is None:
544 autonumber_size = 5
6febd1c1 545 autonumber_templ = '%0' + str(autonumber_size) + 'd'
8222d8de 546 template_dict['autonumber'] = autonumber_templ % self._num_downloads
702665c0 547 if template_dict.get('playlist_index') is not None:
c6b4132a 548 template_dict['playlist_index'] = '%0*d' % (len(str(template_dict['n_entries'])), template_dict['playlist_index'])
17b75c0d
PH
549 if template_dict.get('resolution') is None:
550 if template_dict.get('width') and template_dict.get('height'):
551 template_dict['resolution'] = '%dx%d' % (template_dict['width'], template_dict['height'])
552 elif template_dict.get('height'):
805ef3c6 553 template_dict['resolution'] = '%sp' % template_dict['height']
17b75c0d 554 elif template_dict.get('width'):
805ef3c6 555 template_dict['resolution'] = '?x%d' % template_dict['width']
8222d8de 556
586a91b6 557 sanitize = lambda k, v: sanitize_filename(
45598aab 558 compat_str(v),
1bb5c511 559 restricted=self.params.get('restrictfilenames'),
6febd1c1 560 is_id=(k == 'id'))
586a91b6 561 template_dict = dict((k, sanitize(k, v))
45598aab
PH
562 for k, v in template_dict.items()
563 if v is not None)
6febd1c1 564 template_dict = collections.defaultdict(lambda: 'NA', template_dict)
8222d8de 565
1bb5c511 566 outtmpl = sanitize_path(self.params.get('outtmpl', DEFAULT_OUTTMPL))
4644ac55 567 tmpl = compat_expanduser(outtmpl)
586a91b6 568 filename = tmpl % template_dict
3a0d2f52
S
569 # Temporary fix for #4787
570 # 'Treat' all problem characters by passing filename through preferredencoding
571 # to workaround encoding issues with subprocess on python2 @ Windows
572 if sys.version_info < (3, 0) and sys.platform == 'win32':
573 filename = encodeFilename(filename, True).decode(preferredencoding())
8222d8de 574 return filename
8222d8de 575 except ValueError as err:
6febd1c1 576 self.report_error('Error in output template: ' + str(err) + ' (encoding: ' + repr(preferredencoding()) + ')')
8222d8de
JMF
577 return None
578
442c37b7 579 def _match_entry(self, info_dict, incomplete):
8222d8de
JMF
580 """ Returns None iff the file should be downloaded """
581
6febd1c1 582 video_title = info_dict.get('title', info_dict.get('id', 'video'))
7012b23c
PH
583 if 'title' in info_dict:
584 # This can happen when we're just evaluating the playlist
585 title = info_dict['title']
586 matchtitle = self.params.get('matchtitle', False)
587 if matchtitle:
588 if not re.search(matchtitle, title, re.IGNORECASE):
6febd1c1 589 return '"' + title + '" title did not match pattern "' + matchtitle + '"'
7012b23c
PH
590 rejecttitle = self.params.get('rejecttitle', False)
591 if rejecttitle:
592 if re.search(rejecttitle, title, re.IGNORECASE):
6febd1c1 593 return '"' + title + '" title matched reject pattern "' + rejecttitle + '"'
8222d8de
JMF
594 date = info_dict.get('upload_date', None)
595 if date is not None:
596 dateRange = self.params.get('daterange', DateRange())
597 if date not in dateRange:
6febd1c1 598 return '%s upload date is not in range %s' % (date_from_str(date).isoformat(), dateRange)
5fe18bdb
PH
599 view_count = info_dict.get('view_count', None)
600 if view_count is not None:
601 min_views = self.params.get('min_views')
602 if min_views is not None and view_count < min_views:
6febd1c1 603 return 'Skipping %s, because it has not reached minimum view count (%d/%d)' % (video_title, view_count, min_views)
5fe18bdb
PH
604 max_views = self.params.get('max_views')
605 if max_views is not None and view_count > max_views:
6febd1c1 606 return 'Skipping %s, because it has exceeded the maximum view count (%d/%d)' % (video_title, view_count, max_views)
05900629 607 if age_restricted(info_dict.get('age_limit'), self.params.get('age_limit')):
347de493 608 return 'Skipping "%s" because it is age restricted' % video_title
c1c9a79c 609 if self.in_download_archive(info_dict):
6febd1c1 610 return '%s has already been recorded in archive' % video_title
347de493 611
442c37b7
PH
612 if not incomplete:
613 match_filter = self.params.get('match_filter')
614 if match_filter is not None:
615 ret = match_filter(info_dict)
616 if ret is not None:
617 return ret
347de493 618
8222d8de 619 return None
fe7e0c98 620
b6c45014
JMF
621 @staticmethod
622 def add_extra_info(info_dict, extra_info):
623 '''Set the keys from extra_info in info dict if they are missing'''
624 for key, value in extra_info.items():
625 info_dict.setdefault(key, value)
626
7fc3fa05
PH
627 def extract_info(self, url, download=True, ie_key=None, extra_info={},
628 process=True):
8222d8de
JMF
629 '''
630 Returns a list with a dictionary for each video we find.
631 If 'download', also downloads the videos.
632 extra_info is a dict containing the extra values to add to each result
613b2d9d 633 '''
fe7e0c98 634
8222d8de 635 if ie_key:
56c73665 636 ies = [self.get_info_extractor(ie_key)]
8222d8de
JMF
637 else:
638 ies = self._ies
639
640 for ie in ies:
641 if not ie.suitable(url):
642 continue
643
644 if not ie.working():
6febd1c1
PH
645 self.report_warning('The program functionality for this site has been marked as broken, '
646 'and will probably not work.')
8222d8de
JMF
647
648 try:
649 ie_result = ie.extract(url)
5f6a1245 650 if ie_result is None: # Finished already (backwards compatibility; listformats and friends should be moved here)
8222d8de
JMF
651 break
652 if isinstance(ie_result, list):
653 # Backwards compatibility: old IE result format
8222d8de
JMF
654 ie_result = {
655 '_type': 'compat_list',
656 'entries': ie_result,
657 }
ea38e55f 658 self.add_default_extra_info(ie_result, ie, url)
7fc3fa05
PH
659 if process:
660 return self.process_ie_result(ie_result, download, extra_info)
661 else:
662 return ie_result
5f6a1245 663 except ExtractorError as de: # An error we somewhat expected
8222d8de
JMF
664 self.report_error(compat_str(de), de.format_traceback())
665 break
d3e5bbf4
PH
666 except MaxDownloadsReached:
667 raise
8222d8de
JMF
668 except Exception as e:
669 if self.params.get('ignoreerrors', False):
670 self.report_error(compat_str(e), tb=compat_str(traceback.format_exc()))
671 break
672 else:
673 raise
674 else:
1a489545 675 self.report_error('no suitable InfoExtractor for URL %s' % url)
fe7e0c98 676
ea38e55f
PH
677 def add_default_extra_info(self, ie_result, ie, url):
678 self.add_extra_info(ie_result, {
679 'extractor': ie.IE_NAME,
680 'webpage_url': url,
681 'webpage_url_basename': url_basename(url),
682 'extractor_key': ie.ie_key(),
683 })
684
8222d8de
JMF
685 def process_ie_result(self, ie_result, download=True, extra_info={}):
686 """
687 Take the result of the ie(may be modified) and resolve all unresolved
688 references (URLs, playlist items).
689
690 It will also download the videos if 'download'.
691 Returns the resolved ie_result.
692 """
693
e8ee972c
PH
694 result_type = ie_result.get('_type', 'video')
695
057a5206
PH
696 if result_type in ('url', 'url_transparent'):
697 extract_flat = self.params.get('extract_flat', False)
698 if ((extract_flat == 'in_playlist' and 'playlist' in extra_info) or
699 extract_flat is True):
057a5206
PH
700 if self.params.get('forcejson', False):
701 self.to_stdout(json.dumps(ie_result))
e8ee972c
PH
702 return ie_result
703
8222d8de 704 if result_type == 'video':
b6c45014 705 self.add_extra_info(ie_result, extra_info)
feee2ecf 706 return self.process_video_result(ie_result, download=download)
8222d8de
JMF
707 elif result_type == 'url':
708 # We have to add extra_info to the results because it may be
709 # contained in a playlist
710 return self.extract_info(ie_result['url'],
711 download,
712 ie_key=ie_result.get('ie_key'),
713 extra_info=extra_info)
7fc3fa05
PH
714 elif result_type == 'url_transparent':
715 # Use the information from the embedding page
716 info = self.extract_info(
717 ie_result['url'], ie_key=ie_result.get('ie_key'),
718 extra_info=extra_info, download=False, process=False)
719
412c617d
PH
720 force_properties = dict(
721 (k, v) for k, v in ie_result.items() if v is not None)
722 for f in ('_type', 'url'):
723 if f in force_properties:
724 del force_properties[f]
725 new_result = info.copy()
726 new_result.update(force_properties)
7fc3fa05
PH
727
728 assert new_result.get('_type') != 'url_transparent'
7fc3fa05
PH
729
730 return self.process_ie_result(
731 new_result, download=download, extra_info=extra_info)
42e12102 732 elif result_type == 'playlist' or result_type == 'multi_video':
8222d8de
JMF
733 # We process each entry in the playlist
734 playlist = ie_result.get('title', None) or ie_result.get('id', None)
6febd1c1 735 self.to_screen('[download] Downloading playlist: %s' % playlist)
8222d8de
JMF
736
737 playlist_results = []
738
8222d8de 739 playliststart = self.params.get('playliststart', 1) - 1
a19fd00c
PH
740 playlistend = self.params.get('playlistend', None)
741 # For backwards compatibility, interpret -1 as whole list
8222d8de 742 if playlistend == -1:
a19fd00c 743 playlistend = None
8222d8de 744
c14e88f0
PH
745 playlistitems_str = self.params.get('playlist_items', None)
746 playlistitems = None
747 if playlistitems_str is not None:
748 def iter_playlistitems(format):
749 for string_segment in format.split(','):
750 if '-' in string_segment:
751 start, end = string_segment.split('-')
752 for item in range(int(start), int(end) + 1):
753 yield int(item)
754 else:
755 yield int(string_segment)
756 playlistitems = iter_playlistitems(playlistitems_str)
757
b82f815f
PH
758 ie_entries = ie_result['entries']
759 if isinstance(ie_entries, list):
760 n_all_entries = len(ie_entries)
c14e88f0
PH
761 if playlistitems:
762 entries = [ie_entries[i - 1] for i in playlistitems]
763 else:
764 entries = ie_entries[playliststart:playlistend]
b7ab0590
PH
765 n_entries = len(entries)
766 self.to_screen(
767 "[%s] playlist %s: Collected %d video ids (downloading %d of them)" %
768 (ie_result['extractor'], playlist, n_all_entries, n_entries))
b82f815f 769 elif isinstance(ie_entries, PagedList):
c14e88f0
PH
770 if playlistitems:
771 entries = []
772 for item in playlistitems:
773 entries.extend(ie_entries.getslice(
774 item - 1, item
775 ))
776 else:
777 entries = ie_entries.getslice(
778 playliststart, playlistend)
b7ab0590
PH
779 n_entries = len(entries)
780 self.to_screen(
781 "[%s] playlist %s: Downloading %d videos" %
782 (ie_result['extractor'], playlist, n_entries))
b82f815f 783 else: # iterable
c14e88f0
PH
784 if playlistitems:
785 entry_list = list(ie_entries)
786 entries = [entry_list[i - 1] for i in playlistitems]
787 else:
788 entries = list(itertools.islice(
789 ie_entries, playliststart, playlistend))
b82f815f
PH
790 n_entries = len(entries)
791 self.to_screen(
792 "[%s] playlist %s: Downloading %d videos" %
793 (ie_result['extractor'], playlist, n_entries))
8222d8de 794
ff815fe6
MS
795 if self.params.get('playlistreverse', False):
796 entries = entries[::-1]
797
fe7e0c98 798 for i, entry in enumerate(entries, 1):
734ea11e 799 self.to_screen('[download] Downloading video %s of %s' % (i, n_entries))
8222d8de 800 extra = {
c6b4132a 801 'n_entries': n_entries,
fe7e0c98 802 'playlist': playlist,
a1cf99d0
PH
803 'playlist_id': ie_result.get('id'),
804 'playlist_title': ie_result.get('title'),
fe7e0c98 805 'playlist_index': i + playliststart,
b6c45014 806 'extractor': ie_result['extractor'],
9103bbc5 807 'webpage_url': ie_result['webpage_url'],
29eb5174 808 'webpage_url_basename': url_basename(ie_result['webpage_url']),
be97abc2 809 'extractor_key': ie_result['extractor_key'],
fe7e0c98 810 }
7012b23c 811
442c37b7 812 reason = self._match_entry(entry, incomplete=True)
7012b23c 813 if reason is not None:
6febd1c1 814 self.to_screen('[download] ' + reason)
7012b23c
PH
815 continue
816
8222d8de
JMF
817 entry_result = self.process_ie_result(entry,
818 download=download,
819 extra_info=extra)
820 playlist_results.append(entry_result)
821 ie_result['entries'] = playlist_results
822 return ie_result
823 elif result_type == 'compat_list':
c9bf4114
PH
824 self.report_warning(
825 'Extractor %s returned a compat_list result. '
826 'It needs to be updated.' % ie_result.get('extractor'))
5f6a1245 827
8222d8de 828 def _fixup(r):
9e1a5b84
JW
829 self.add_extra_info(
830 r,
9103bbc5
JMF
831 {
832 'extractor': ie_result['extractor'],
833 'webpage_url': ie_result['webpage_url'],
29eb5174 834 'webpage_url_basename': url_basename(ie_result['webpage_url']),
be97abc2 835 'extractor_key': ie_result['extractor_key'],
9e1a5b84
JW
836 }
837 )
8222d8de
JMF
838 return r
839 ie_result['entries'] = [
b6c45014 840 self.process_ie_result(_fixup(r), download, extra_info)
8222d8de
JMF
841 for r in ie_result['entries']
842 ]
843 return ie_result
844 else:
845 raise Exception('Invalid result type: %s' % result_type)
846
083c9df9
PH
847 def _apply_format_filter(self, format_spec, available_formats):
848 " Returns a tuple of the remaining format_spec and filtered formats "
849
850 OPERATORS = {
851 '<': operator.lt,
852 '<=': operator.le,
853 '>': operator.gt,
854 '>=': operator.ge,
855 '=': operator.eq,
856 '!=': operator.ne,
857 }
858 operator_rex = re.compile(r'''(?x)\s*\[
2ec19e95 859 (?P<key>width|height|tbr|abr|vbr|asr|filesize|fps)
083c9df9
PH
860 \s*(?P<op>%s)(?P<none_inclusive>\s*\?)?\s*
861 (?P<value>[0-9.]+(?:[kKmMgGtTpPeEzZyY]i?[Bb]?)?)
862 \]$
863 ''' % '|'.join(map(re.escape, OPERATORS.keys())))
864 m = operator_rex.search(format_spec)
9ddb6925
S
865 if m:
866 try:
867 comparison_value = int(m.group('value'))
868 except ValueError:
869 comparison_value = parse_filesize(m.group('value'))
870 if comparison_value is None:
871 comparison_value = parse_filesize(m.group('value') + 'B')
872 if comparison_value is None:
873 raise ValueError(
874 'Invalid value %r in format specification %r' % (
875 m.group('value'), format_spec))
876 op = OPERATORS[m.group('op')]
877
083c9df9 878 if not m:
9ddb6925
S
879 STR_OPERATORS = {
880 '=': operator.eq,
881 '!=': operator.ne,
882 }
883 str_operator_rex = re.compile(r'''(?x)\s*\[
884 \s*(?P<key>ext|acodec|vcodec|container|protocol)
885 \s*(?P<op>%s)(?P<none_inclusive>\s*\?)?
886 \s*(?P<value>[a-zA-Z0-9_-]+)
887 \s*\]$
888 ''' % '|'.join(map(re.escape, STR_OPERATORS.keys())))
889 m = str_operator_rex.search(format_spec)
890 if m:
891 comparison_value = m.group('value')
892 op = STR_OPERATORS[m.group('op')]
083c9df9 893
9ddb6925
S
894 if not m:
895 raise ValueError('Invalid format specification %r' % format_spec)
083c9df9
PH
896
897 def _filter(f):
898 actual_value = f.get(m.group('key'))
899 if actual_value is None:
900 return m.group('none_inclusive')
901 return op(actual_value, comparison_value)
902 new_formats = [f for f in available_formats if _filter(f)]
903
904 new_format_spec = format_spec[:-len(m.group(0))]
905 if not new_format_spec:
906 new_format_spec = 'best'
907
908 return (new_format_spec, new_formats)
909
a9c58ad9 910 def select_format(self, format_spec, available_formats):
083c9df9
PH
911 while format_spec.endswith(']'):
912 format_spec, available_formats = self._apply_format_filter(
913 format_spec, available_formats)
914 if not available_formats:
915 return None
916
e01c56f9
S
917 if format_spec in ['best', 'worst', None]:
918 format_idx = 0 if format_spec == 'worst' else -1
8dd54188
JMF
919 audiovideo_formats = [
920 f for f in available_formats
921 if f.get('vcodec') != 'none' and f.get('acodec') != 'none']
922 if audiovideo_formats:
e01c56f9
S
923 return audiovideo_formats[format_idx]
924 # for audio only urls, select the best/worst audio format
8dd54188 925 elif all(f.get('acodec') != 'none' for f in available_formats):
e01c56f9 926 return available_formats[format_idx]
ba7678f9
PH
927 elif format_spec == 'bestaudio':
928 audio_formats = [
929 f for f in available_formats
930 if f.get('vcodec') == 'none']
931 if audio_formats:
932 return audio_formats[-1]
933 elif format_spec == 'worstaudio':
934 audio_formats = [
935 f for f in available_formats
936 if f.get('vcodec') == 'none']
937 if audio_formats:
938 return audio_formats[0]
bc6d5978
JMF
939 elif format_spec == 'bestvideo':
940 video_formats = [
941 f for f in available_formats
942 if f.get('acodec') == 'none']
943 if video_formats:
944 return video_formats[-1]
945 elif format_spec == 'worstvideo':
946 video_formats = [
947 f for f in available_formats
948 if f.get('acodec') == 'none']
949 if video_formats:
950 return video_formats[0]
a9c58ad9 951 else:
0217c783 952 extensions = ['mp4', 'flv', 'webm', '3gp', 'm4a', 'mp3', 'ogg', 'aac', 'wav']
49e86983
JMF
953 if format_spec in extensions:
954 filter_f = lambda f: f['ext'] == format_spec
955 else:
956 filter_f = lambda f: f['format_id'] == format_spec
fe7e0c98 957 matches = list(filter(filter_f, available_formats))
a9c58ad9
JMF
958 if matches:
959 return matches[-1]
960 return None
961
e5660ee6
JMF
962 def _calc_headers(self, info_dict):
963 res = std_headers.copy()
964
965 add_headers = info_dict.get('http_headers')
966 if add_headers:
967 res.update(add_headers)
968
969 cookies = self._calc_cookies(info_dict)
970 if cookies:
971 res['Cookie'] = cookies
972
973 return res
974
975 def _calc_cookies(self, info_dict):
662435f7 976 pr = compat_urllib_request.Request(info_dict['url'])
e5660ee6 977 self.cookiejar.add_cookie_header(pr)
662435f7 978 return pr.get_header('Cookie')
e5660ee6 979
dd82ffea
JMF
980 def process_video_result(self, info_dict, download=True):
981 assert info_dict.get('_type', 'video') == 'video'
982
bec1fad2
PH
983 if 'id' not in info_dict:
984 raise ExtractorError('Missing "id" field in extractor result')
985 if 'title' not in info_dict:
986 raise ExtractorError('Missing "title" field in extractor result')
987
dd82ffea
JMF
988 if 'playlist' not in info_dict:
989 # It isn't part of a playlist
990 info_dict['playlist'] = None
991 info_dict['playlist_index'] = None
992
d5519808 993 thumbnails = info_dict.get('thumbnails')
cfb56d1a
PH
994 if thumbnails is None:
995 thumbnail = info_dict.get('thumbnail')
996 if thumbnail:
a7a14d95 997 info_dict['thumbnails'] = thumbnails = [{'url': thumbnail}]
d5519808 998 if thumbnails:
be6d7229 999 thumbnails.sort(key=lambda t: (
cfb56d1a
PH
1000 t.get('preference'), t.get('width'), t.get('height'),
1001 t.get('id'), t.get('url')))
f6c24009 1002 for i, t in enumerate(thumbnails):
d5519808
PH
1003 if 'width' in t and 'height' in t:
1004 t['resolution'] = '%dx%d' % (t['width'], t['height'])
f6c24009
PH
1005 if t.get('id') is None:
1006 t['id'] = '%d' % i
d5519808
PH
1007
1008 if thumbnails and 'thumbnail' not in info_dict:
1009 info_dict['thumbnail'] = thumbnails[-1]['url']
1010
c9ae7b95 1011 if 'display_id' not in info_dict and 'id' in info_dict:
0afef30b
PH
1012 info_dict['display_id'] = info_dict['id']
1013
955c4514 1014 if info_dict.get('upload_date') is None and info_dict.get('timestamp') is not None:
706d7d4e
S
1015 # Working around negative timestamps in Windows
1016 # (see http://bugs.python.org/issue1646728)
1017 if info_dict['timestamp'] < 0 and os.name == 'nt':
1018 info_dict['timestamp'] = 0
9d2ecdbc 1019 upload_date = datetime.datetime.utcfromtimestamp(
955c4514 1020 info_dict['timestamp'])
9d2ecdbc
PH
1021 info_dict['upload_date'] = upload_date.strftime('%Y%m%d')
1022
a504ced0 1023 if self.params.get('listsubtitles', False):
360e1ca5
JMF
1024 if 'automatic_captions' in info_dict:
1025 self.list_subtitles(info_dict['id'], info_dict.get('automatic_captions'), 'automatic captions')
1026 self.list_subtitles(info_dict['id'], info_dict.get('subtitles'), 'subtitles')
a504ced0 1027 return
360e1ca5
JMF
1028 info_dict['requested_subtitles'] = self.process_subtitles(
1029 info_dict['id'], info_dict.get('subtitles'),
1030 info_dict.get('automatic_captions'))
a504ced0 1031
6ff000b8 1032 # This extractors handle format selection themselves
6febd1c1 1033 if info_dict['extractor'] in ['Youku']:
12893efe
JMF
1034 if download:
1035 self.process_info(info_dict)
6ff000b8
JMF
1036 return info_dict
1037
dd82ffea
JMF
1038 # We now pick which formats have to be downloaded
1039 if info_dict.get('formats') is None:
1040 # There's only one format available
1041 formats = [info_dict]
1042 else:
1043 formats = info_dict['formats']
1044
db95dc13
PH
1045 if not formats:
1046 raise ExtractorError('No video formats found!')
1047
dd82ffea 1048 # We check that all the formats have the format and format_id fields
db95dc13 1049 for i, format in enumerate(formats):
bec1fad2
PH
1050 if 'url' not in format:
1051 raise ExtractorError('Missing "url" key in result (index %d)' % i)
1052
dd82ffea 1053 if format.get('format_id') is None:
8016c922 1054 format['format_id'] = compat_str(i)
8c51aa65 1055 if format.get('format') is None:
6febd1c1 1056 format['format'] = '{id} - {res}{note}'.format(
8c51aa65
JMF
1057 id=format['format_id'],
1058 res=self.format_resolution(format),
6febd1c1 1059 note=' ({0})'.format(format['format_note']) if format.get('format_note') is not None else '',
8c51aa65 1060 )
c1002e96
PH
1061 # Automatically determine file extension if missing
1062 if 'ext' not in format:
cce929ea 1063 format['ext'] = determine_ext(format['url']).lower()
e5660ee6
JMF
1064 # Add HTTP headers, so that external programs can use them from the
1065 # json output
1066 full_format_info = info_dict.copy()
1067 full_format_info.update(format)
1068 format['http_headers'] = self._calc_headers(full_format_info)
dd82ffea 1069
4bcc7bd1 1070 # TODO Central sorting goes here
99e206d5 1071
f89197d7 1072 if formats[0] is not info_dict:
b3d9ef88
JMF
1073 # only set the 'formats' fields if the original info_dict list them
1074 # otherwise we end up with a circular reference, the first (and unique)
f89197d7 1075 # element in the 'formats' field in info_dict is info_dict itself,
b3d9ef88
JMF
1076 # wich can't be exported to json
1077 info_dict['formats'] = formats
cfb56d1a 1078 if self.params.get('listformats'):
bfaae0a7 1079 self.list_formats(info_dict)
1080 return
cfb56d1a
PH
1081 if self.params.get('list_thumbnails'):
1082 self.list_thumbnails(info_dict)
1083 return
bfaae0a7 1084
de3ef3ed 1085 req_format = self.params.get('format')
a9c58ad9 1086 if req_format is None:
feccf29c 1087 req_format_list = []
3799834d
S
1088 if (self.params.get('outtmpl', DEFAULT_OUTTMPL) != '-'
1089 and info_dict['extractor'] in ['youtube', 'ted']
1090 and FFmpegMergerPP(self).available):
feccf29c
S
1091 req_format_list.append('bestvideo+bestaudio')
1092 req_format_list.append('best')
1093 req_format = '/'.join(req_format_list)
dd82ffea 1094 formats_to_download = []
3a77719c 1095 if req_format == 'all':
dd82ffea
JMF
1096 formats_to_download = formats
1097 else:
1de33faf
PH
1098 for rfstr in req_format.split(','):
1099 # We can accept formats requested in the format: 34/5/best, we pick
1100 # the first that is available, starting from left
1101 req_formats = rfstr.split('/')
1102 for rf in req_formats:
1103 if re.match(r'.+?\+.+?', rf) is not None:
1104 # Two formats have been requested like '137+139'
1105 format_1, format_2 = rf.split('+')
1106 formats_info = (self.select_format(format_1, formats),
9e1a5b84 1107 self.select_format(format_2, formats))
1de33faf 1108 if all(formats_info):
c2954908
JMF
1109 # The first format must contain the video and the
1110 # second the audio
1111 if formats_info[0].get('vcodec') == 'none':
1112 self.report_error('The first format must '
9e1a5b84
JW
1113 'contain the video, try using '
1114 '"-f %s+%s"' % (format_2, format_1))
c2954908 1115 return
45598f15
PH
1116 output_ext = (
1117 formats_info[0]['ext']
1118 if self.params.get('merge_output_format') is None
1119 else self.params['merge_output_format'])
1de33faf
PH
1120 selected_format = {
1121 'requested_formats': formats_info,
300753a0
S
1122 'format': '%s+%s' % (formats_info[0].get('format'),
1123 formats_info[1].get('format')),
6d593c32
S
1124 'format_id': '%s+%s' % (formats_info[0].get('format_id'),
1125 formats_info[1].get('format_id')),
f90ad273
PH
1126 'width': formats_info[0].get('width'),
1127 'height': formats_info[0].get('height'),
1128 'resolution': formats_info[0].get('resolution'),
1129 'fps': formats_info[0].get('fps'),
1130 'vcodec': formats_info[0].get('vcodec'),
1131 'vbr': formats_info[0].get('vbr'),
6271f1ca 1132 'stretched_ratio': formats_info[0].get('stretched_ratio'),
f90ad273
PH
1133 'acodec': formats_info[1].get('acodec'),
1134 'abr': formats_info[1].get('abr'),
45598f15 1135 'ext': output_ext,
1de33faf
PH
1136 }
1137 else:
1138 selected_format = None
6350728b 1139 else:
1de33faf
PH
1140 selected_format = self.select_format(rf, formats)
1141 if selected_format is not None:
1142 formats_to_download.append(selected_format)
1143 break
dd82ffea 1144 if not formats_to_download:
6febd1c1 1145 raise ExtractorError('requested format not available',
78a3a9f8 1146 expected=True)
dd82ffea
JMF
1147
1148 if download:
1149 if len(formats_to_download) > 1:
6febd1c1 1150 self.to_screen('[info] %s: downloading video in %s formats' % (info_dict['id'], len(formats_to_download)))
dd82ffea
JMF
1151 for format in formats_to_download:
1152 new_info = dict(info_dict)
1153 new_info.update(format)
1154 self.process_info(new_info)
1155 # We update the info dict with the best quality format (backwards compatibility)
1156 info_dict.update(formats_to_download[-1])
1157 return info_dict
1158
98c70d6f 1159 def process_subtitles(self, video_id, normal_subtitles, automatic_captions):
a504ced0 1160 """Select the requested subtitles and their format"""
98c70d6f
JMF
1161 available_subs = {}
1162 if normal_subtitles and self.params.get('writesubtitles'):
1163 available_subs.update(normal_subtitles)
1164 if automatic_captions and self.params.get('writeautomaticsub'):
1165 for lang, cap_info in automatic_captions.items():
360e1ca5
JMF
1166 if lang not in available_subs:
1167 available_subs[lang] = cap_info
1168
4d171848
JMF
1169 if (not self.params.get('writesubtitles') and not
1170 self.params.get('writeautomaticsub') or not
1171 available_subs):
1172 return None
a504ced0
JMF
1173
1174 if self.params.get('allsubtitles', False):
1175 requested_langs = available_subs.keys()
1176 else:
1177 if self.params.get('subtitleslangs', False):
1178 requested_langs = self.params.get('subtitleslangs')
1179 elif 'en' in available_subs:
1180 requested_langs = ['en']
1181 else:
1182 requested_langs = [list(available_subs.keys())[0]]
1183
1184 formats_query = self.params.get('subtitlesformat', 'best')
1185 formats_preference = formats_query.split('/') if formats_query else []
1186 subs = {}
1187 for lang in requested_langs:
1188 formats = available_subs.get(lang)
1189 if formats is None:
1190 self.report_warning('%s subtitles not available for %s' % (lang, video_id))
1191 continue
a504ced0
JMF
1192 for ext in formats_preference:
1193 if ext == 'best':
1194 f = formats[-1]
1195 break
1196 matches = list(filter(lambda f: f['ext'] == ext, formats))
1197 if matches:
1198 f = matches[-1]
1199 break
1200 else:
1201 f = formats[-1]
1202 self.report_warning(
1203 'No subtitle format found matching "%s" for language %s, '
1204 'using %s' % (formats_query, lang, f['ext']))
1205 subs[lang] = f
1206 return subs
1207
8222d8de
JMF
1208 def process_info(self, info_dict):
1209 """Process a single resolved IE result."""
1210
1211 assert info_dict.get('_type', 'video') == 'video'
fd288278
PH
1212
1213 max_downloads = self.params.get('max_downloads')
1214 if max_downloads is not None:
1215 if self._num_downloads >= int(max_downloads):
1216 raise MaxDownloadsReached()
8222d8de
JMF
1217
1218 info_dict['fulltitle'] = info_dict['title']
1219 if len(info_dict['title']) > 200:
6febd1c1 1220 info_dict['title'] = info_dict['title'][:197] + '...'
8222d8de 1221
11b85ce6 1222 if 'format' not in info_dict:
8222d8de
JMF
1223 info_dict['format'] = info_dict['ext']
1224
442c37b7 1225 reason = self._match_entry(info_dict, incomplete=False)
8222d8de 1226 if reason is not None:
6febd1c1 1227 self.to_screen('[download] ' + reason)
8222d8de
JMF
1228 return
1229
fd288278 1230 self._num_downloads += 1
8222d8de 1231
e72c7e41 1232 info_dict['_filename'] = filename = self.prepare_filename(info_dict)
8222d8de
JMF
1233
1234 # Forced printings
1235 if self.params.get('forcetitle', False):
0783b09b 1236 self.to_stdout(info_dict['fulltitle'])
8222d8de 1237 if self.params.get('forceid', False):
0783b09b 1238 self.to_stdout(info_dict['id'])
8222d8de 1239 if self.params.get('forceurl', False):
16ae61f6 1240 if info_dict.get('requested_formats') is not None:
1241 for f in info_dict['requested_formats']:
1242 self.to_stdout(f['url'] + f.get('play_path', ''))
1243 else:
1244 # For RTMP URLs, also include the playpath
1245 self.to_stdout(info_dict['url'] + info_dict.get('play_path', ''))
216d71d0 1246 if self.params.get('forcethumbnail', False) and info_dict.get('thumbnail') is not None:
0783b09b 1247 self.to_stdout(info_dict['thumbnail'])
216d71d0 1248 if self.params.get('forcedescription', False) and info_dict.get('description') is not None:
0783b09b 1249 self.to_stdout(info_dict['description'])
8222d8de 1250 if self.params.get('forcefilename', False) and filename is not None:
0783b09b 1251 self.to_stdout(filename)
525ef922
PH
1252 if self.params.get('forceduration', False) and info_dict.get('duration') is not None:
1253 self.to_stdout(formatSeconds(info_dict['duration']))
8222d8de 1254 if self.params.get('forceformat', False):
0783b09b 1255 self.to_stdout(info_dict['format'])
9d153818 1256 if self.params.get('forcejson', False):
0783b09b 1257 self.to_stdout(json.dumps(info_dict))
8222d8de
JMF
1258
1259 # Do nothing else if in simulate mode
1260 if self.params.get('simulate', False):
1261 return
1262
1263 if filename is None:
1264 return
1265
1266 try:
e5a11a22 1267 dn = os.path.dirname(sanitize_path(encodeFilename(filename)))
d26e981d 1268 if dn and not os.path.exists(dn):
8222d8de
JMF
1269 os.makedirs(dn)
1270 except (OSError, IOError) as err:
6febd1c1 1271 self.report_error('unable to create directory ' + compat_str(err))
8222d8de
JMF
1272 return
1273
1274 if self.params.get('writedescription', False):
2699da80 1275 descfn = replace_extension(filename, 'description', info_dict.get('ext'))
7b6fefc9 1276 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(descfn)):
6febd1c1 1277 self.to_screen('[info] Video description is already present')
f00fd51d
JMF
1278 elif info_dict.get('description') is None:
1279 self.report_warning('There\'s no description to write.')
7b6fefc9
PH
1280 else:
1281 try:
6febd1c1 1282 self.to_screen('[info] Writing video description to: ' + descfn)
7b6fefc9
PH
1283 with io.open(encodeFilename(descfn), 'w', encoding='utf-8') as descfile:
1284 descfile.write(info_dict['description'])
7b6fefc9 1285 except (OSError, IOError):
6febd1c1 1286 self.report_error('Cannot write description file ' + descfn)
7b6fefc9 1287 return
8222d8de 1288
1fb07d10 1289 if self.params.get('writeannotations', False):
98727e12 1290 annofn = replace_extension(filename, 'annotations.xml', info_dict.get('ext'))
7b6fefc9 1291 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(annofn)):
6febd1c1 1292 self.to_screen('[info] Video annotations are already present')
7b6fefc9
PH
1293 else:
1294 try:
6febd1c1 1295 self.to_screen('[info] Writing video annotations to: ' + annofn)
7b6fefc9
PH
1296 with io.open(encodeFilename(annofn), 'w', encoding='utf-8') as annofile:
1297 annofile.write(info_dict['annotations'])
1298 except (KeyError, TypeError):
6febd1c1 1299 self.report_warning('There are no annotations to write.')
7b6fefc9 1300 except (OSError, IOError):
6febd1c1 1301 self.report_error('Cannot write annotations file: ' + annofn)
7b6fefc9 1302 return
1fb07d10 1303
c4a91be7 1304 subtitles_are_requested = any([self.params.get('writesubtitles', False),
0b7f3118 1305 self.params.get('writeautomaticsub')])
c4a91be7 1306
c84dd8a9 1307 if subtitles_are_requested and info_dict.get('requested_subtitles'):
8222d8de
JMF
1308 # subtitles download errors are already managed as troubles in relevant IE
1309 # that way it will silently go on when used with unsupporting IE
c84dd8a9 1310 subtitles = info_dict['requested_subtitles']
0f2c0d33 1311 ie = self.get_info_extractor(info_dict['extractor_key'])
a504ced0
JMF
1312 for sub_lang, sub_info in subtitles.items():
1313 sub_format = sub_info['ext']
1314 if sub_info.get('data') is not None:
1315 sub_data = sub_info['data']
1316 else:
1317 try:
0f2c0d33
JMF
1318 sub_data = ie._download_webpage(
1319 sub_info['url'], info_dict['id'], note=False)
1320 except ExtractorError as err:
a504ced0 1321 self.report_warning('Unable to download subtitle for "%s": %s' %
0f2c0d33 1322 (sub_lang, compat_str(err.cause)))
a504ced0 1323 continue
8222d8de 1324 try:
d4051a8e 1325 sub_filename = subtitles_filename(filename, sub_lang, sub_format)
7b6fefc9 1326 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(sub_filename)):
6febd1c1 1327 self.to_screen('[info] Video subtitle %s.%s is already_present' % (sub_lang, sub_format))
7b6fefc9 1328 else:
6febd1c1 1329 self.to_screen('[info] Writing video subtitles to: ' + sub_filename)
7b6fefc9 1330 with io.open(encodeFilename(sub_filename), 'w', encoding='utf-8') as subfile:
a504ced0 1331 subfile.write(sub_data)
8222d8de 1332 except (OSError, IOError):
e4db1951 1333 self.report_error('Cannot write subtitles file ' + sub_filename)
8222d8de
JMF
1334 return
1335
8222d8de 1336 if self.params.get('writeinfojson', False):
b29e0000 1337 infofn = replace_extension(filename, 'info.json', info_dict.get('ext'))
7b6fefc9 1338 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(infofn)):
6febd1c1 1339 self.to_screen('[info] Video description metadata is already present')
7b6fefc9 1340 else:
6febd1c1 1341 self.to_screen('[info] Writing video description metadata as JSON to: ' + infofn)
7b6fefc9 1342 try:
cb202fd2 1343 write_json_file(self.filter_requested_info(info_dict), infofn)
7b6fefc9 1344 except (OSError, IOError):
6febd1c1 1345 self.report_error('Cannot write metadata to JSON file ' + infofn)
7b6fefc9 1346 return
8222d8de 1347
ec82d85a 1348 self._write_thumbnails(info_dict, filename)
8222d8de
JMF
1349
1350 if not self.params.get('skip_download', False):
4340deca
P
1351 try:
1352 def dl(name, info):
a055469f 1353 fd = get_suitable_downloader(info, self.params)(self, self.params)
4340deca
P
1354 for ph in self._progress_hooks:
1355 fd.add_progress_hook(ph)
1356 if self.params.get('verbose'):
1357 self.to_stdout('[debug] Invoking downloader on %r' % info.get('url'))
1358 return fd.download(name, info)
ee69b99a 1359
4340deca
P
1360 if info_dict.get('requested_formats') is not None:
1361 downloaded = []
1362 success = True
d47aeb22 1363 merger = FFmpegMergerPP(self)
f740fae2 1364 if not merger.available:
4340deca
P
1365 postprocessors = []
1366 self.report_warning('You have requested multiple '
1367 'formats but ffmpeg or avconv are not installed.'
1368 ' The formats won\'t be merged')
6350728b 1369 else:
4340deca 1370 postprocessors = [merger]
81cd954a
S
1371
1372 def compatible_formats(formats):
1373 video, audio = formats
1374 # Check extension
1375 video_ext, audio_ext = audio.get('ext'), video.get('ext')
1376 if video_ext and audio_ext:
1377 COMPATIBLE_EXTS = (
6728187a 1378 ('mp3', 'mp4', 'm4a', 'm4p', 'm4b', 'm4r', 'm4v'),
81cd954a
S
1379 ('webm')
1380 )
1381 for exts in COMPATIBLE_EXTS:
1382 if video_ext in exts and audio_ext in exts:
1383 return True
1384 # TODO: Check acodec/vcodec
1385 return False
1386
38c6902b
S
1387 filename_real_ext = os.path.splitext(filename)[1][1:]
1388 filename_wo_ext = (
1389 os.path.splitext(filename)[0]
1390 if filename_real_ext == info_dict['ext']
1391 else filename)
81cd954a 1392 requested_formats = info_dict['requested_formats']
c0dea0a7 1393 if self.params.get('merge_output_format') is None and not compatible_formats(requested_formats):
38c6902b 1394 info_dict['ext'] = 'mkv'
5456d78f 1395 self.report_warning('You have requested formats incompatible for merge. '
81cd954a 1396 'The formats will be merged into mkv')
38c6902b
S
1397 # Ensure filename always has a correct extension for successful merge
1398 filename = '%s.%s' % (filename_wo_ext, info_dict['ext'])
5b5fbc08
JMF
1399 if os.path.exists(encodeFilename(filename)):
1400 self.to_screen(
1401 '[download] %s has already been downloaded and '
1402 'merged' % filename)
1403 else:
81cd954a 1404 for f in requested_formats:
5b5fbc08
JMF
1405 new_info = dict(info_dict)
1406 new_info.update(f)
1407 fname = self.prepare_filename(new_info)
666a9a2b 1408 fname = prepend_extension(fname, 'f%s' % f['format_id'], new_info['ext'])
5b5fbc08
JMF
1409 downloaded.append(fname)
1410 partial_success = dl(fname, new_info)
1411 success = success and partial_success
1412 info_dict['__postprocessors'] = postprocessors
1413 info_dict['__files_to_merge'] = downloaded
4340deca
P
1414 else:
1415 # Just a single file
1416 success = dl(filename, info_dict)
1417 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
1418 self.report_error('unable to download video data: %s' % str(err))
1419 return
1420 except (OSError, IOError) as err:
1421 raise UnavailableVideoError(err)
1422 except (ContentTooShortError, ) as err:
1423 self.report_error('content too short (expected %s bytes and served %s)' % (err.expected, err.downloaded))
1424 return
8222d8de
JMF
1425
1426 if success:
6271f1ca 1427 # Fixup content
62cd676c
PH
1428 fixup_policy = self.params.get('fixup')
1429 if fixup_policy is None:
1430 fixup_policy = 'detect_or_warn'
1431
6271f1ca
PH
1432 stretched_ratio = info_dict.get('stretched_ratio')
1433 if stretched_ratio is not None and stretched_ratio != 1:
6271f1ca
PH
1434 if fixup_policy == 'warn':
1435 self.report_warning('%s: Non-uniform pixel ratio (%s)' % (
1436 info_dict['id'], stretched_ratio))
1437 elif fixup_policy == 'detect_or_warn':
1438 stretched_pp = FFmpegFixupStretchedPP(self)
1439 if stretched_pp.available:
1440 info_dict.setdefault('__postprocessors', [])
1441 info_dict['__postprocessors'].append(stretched_pp)
1442 else:
1443 self.report_warning(
1444 '%s: Non-uniform pixel ratio (%s). Install ffmpeg or avconv to fix this automatically.' % (
1445 info_dict['id'], stretched_ratio))
1446 else:
62cd676c
PH
1447 assert fixup_policy in ('ignore', 'never')
1448
1449 if info_dict.get('requested_formats') is None and info_dict.get('container') == 'm4a_dash':
1450 if fixup_policy == 'warn':
1451 self.report_warning('%s: writing DASH m4a. Only some players support this container.' % (
1452 info_dict['id']))
1453 elif fixup_policy == 'detect_or_warn':
1454 fixup_pp = FFmpegFixupM4aPP(self)
1455 if fixup_pp.available:
1456 info_dict.setdefault('__postprocessors', [])
1457 info_dict['__postprocessors'].append(fixup_pp)
1458 else:
1459 self.report_warning(
1460 '%s: writing DASH m4a. Only some players support this container. Install ffmpeg or avconv to fix this automatically.' % (
1461 info_dict['id']))
1462 else:
1463 assert fixup_policy in ('ignore', 'never')
6271f1ca 1464
8222d8de
JMF
1465 try:
1466 self.post_process(filename, info_dict)
1467 except (PostProcessingError) as err:
6febd1c1 1468 self.report_error('postprocessing: %s' % str(err))
8222d8de 1469 return
cd58dc3e 1470 self.record_download_archive(info_dict)
8222d8de
JMF
1471
1472 def download(self, url_list):
1473 """Download a given list of URLs."""
acd69589 1474 outtmpl = self.params.get('outtmpl', DEFAULT_OUTTMPL)
0c75c3fa 1475 if (len(url_list) > 1 and
8fb3ac36
PH
1476 '%' not in outtmpl and
1477 self.params.get('max_downloads') != 1):
acd69589 1478 raise SameFileError(outtmpl)
8222d8de
JMF
1479
1480 for url in url_list:
1481 try:
5f6a1245 1482 # It also downloads the videos
63e0be34 1483 res = self.extract_info(url)
8222d8de 1484 except UnavailableVideoError:
6febd1c1 1485 self.report_error('unable to download video')
8222d8de 1486 except MaxDownloadsReached:
6febd1c1 1487 self.to_screen('[info] Maximum number of downloaded files reached.')
8222d8de 1488 raise
63e0be34
PH
1489 else:
1490 if self.params.get('dump_single_json', False):
1491 self.to_stdout(json.dumps(res))
8222d8de
JMF
1492
1493 return self._download_retcode
1494
1dcc4c0c 1495 def download_with_info_file(self, info_filename):
31bd3925
JMF
1496 with contextlib.closing(fileinput.FileInput(
1497 [info_filename], mode='r',
1498 openhook=fileinput.hook_encoded('utf-8'))) as f:
1499 # FileInput doesn't have a read method, we can't call json.load
cb202fd2 1500 info = self.filter_requested_info(json.loads('\n'.join(f)))
d4943898
JMF
1501 try:
1502 self.process_ie_result(info, download=True)
1503 except DownloadError:
1504 webpage_url = info.get('webpage_url')
1505 if webpage_url is not None:
6febd1c1 1506 self.report_warning('The info failed to download, trying with "%s"' % webpage_url)
d4943898
JMF
1507 return self.download([webpage_url])
1508 else:
1509 raise
1510 return self._download_retcode
1dcc4c0c 1511
cb202fd2
S
1512 @staticmethod
1513 def filter_requested_info(info_dict):
1514 return dict(
1515 (k, v) for k, v in info_dict.items()
1516 if k not in ['requested_formats', 'requested_subtitles'])
1517
8222d8de
JMF
1518 def post_process(self, filename, ie_info):
1519 """Run all the postprocessors on the given file."""
1520 info = dict(ie_info)
1521 info['filepath'] = filename
6350728b
JMF
1522 pps_chain = []
1523 if ie_info.get('__postprocessors') is not None:
1524 pps_chain.extend(ie_info['__postprocessors'])
1525 pps_chain.extend(self._pps)
1526 for pp in pps_chain:
8222d8de 1527 try:
592e97e8 1528 files_to_delete, info = pp.run(info)
8222d8de 1529 except PostProcessingError as e:
bbcbf4d4 1530 self.report_error(e.msg)
592e97e8
JMF
1531 if files_to_delete and not self.params.get('keepvideo', False):
1532 for old_filename in files_to_delete:
f3ff1a36 1533 self.to_screen('Deleting original file %s (pass -k to keep)' % old_filename)
592e97e8
JMF
1534 try:
1535 os.remove(encodeFilename(old_filename))
1536 except (IOError, OSError):
1537 self.report_warning('Unable to remove downloaded original file')
c1c9a79c 1538
5db07df6
PH
1539 def _make_archive_id(self, info_dict):
1540 # Future-proof against any change in case
1541 # and backwards compatibility with prior versions
d31209a1 1542 extractor = info_dict.get('extractor_key')
7012b23c
PH
1543 if extractor is None:
1544 if 'id' in info_dict:
1545 extractor = info_dict.get('ie_key') # key in a playlist
1546 if extractor is None:
5db07df6 1547 return None # Incomplete video information
6febd1c1 1548 return extractor.lower() + ' ' + info_dict['id']
5db07df6
PH
1549
1550 def in_download_archive(self, info_dict):
1551 fn = self.params.get('download_archive')
1552 if fn is None:
1553 return False
1554
1555 vid_id = self._make_archive_id(info_dict)
1556 if vid_id is None:
7012b23c 1557 return False # Incomplete video information
5db07df6 1558
c1c9a79c
PH
1559 try:
1560 with locked_file(fn, 'r', encoding='utf-8') as archive_file:
1561 for line in archive_file:
1562 if line.strip() == vid_id:
1563 return True
1564 except IOError as ioe:
1565 if ioe.errno != errno.ENOENT:
1566 raise
1567 return False
1568
1569 def record_download_archive(self, info_dict):
1570 fn = self.params.get('download_archive')
1571 if fn is None:
1572 return
5db07df6
PH
1573 vid_id = self._make_archive_id(info_dict)
1574 assert vid_id
c1c9a79c 1575 with locked_file(fn, 'a', encoding='utf-8') as archive_file:
6febd1c1 1576 archive_file.write(vid_id + '\n')
dd82ffea 1577
8c51aa65 1578 @staticmethod
8abeeb94 1579 def format_resolution(format, default='unknown'):
fb04e403
PH
1580 if format.get('vcodec') == 'none':
1581 return 'audio only'
f49d89ee
PH
1582 if format.get('resolution') is not None:
1583 return format['resolution']
8c51aa65
JMF
1584 if format.get('height') is not None:
1585 if format.get('width') is not None:
6febd1c1 1586 res = '%sx%s' % (format['width'], format['height'])
8c51aa65 1587 else:
6febd1c1 1588 res = '%sp' % format['height']
f49d89ee 1589 elif format.get('width') is not None:
6febd1c1 1590 res = '?x%d' % format['width']
8c51aa65 1591 else:
8abeeb94 1592 res = default
8c51aa65
JMF
1593 return res
1594
c57f7757
PH
1595 def _format_note(self, fdict):
1596 res = ''
1597 if fdict.get('ext') in ['f4f', 'f4m']:
1598 res += '(unsupported) '
1599 if fdict.get('format_note') is not None:
1600 res += fdict['format_note'] + ' '
1601 if fdict.get('tbr') is not None:
1602 res += '%4dk ' % fdict['tbr']
1603 if fdict.get('container') is not None:
1604 if res:
1605 res += ', '
1606 res += '%s container' % fdict['container']
1607 if (fdict.get('vcodec') is not None and
1608 fdict.get('vcodec') != 'none'):
1609 if res:
1610 res += ', '
1611 res += fdict['vcodec']
91c7271a 1612 if fdict.get('vbr') is not None:
c57f7757
PH
1613 res += '@'
1614 elif fdict.get('vbr') is not None and fdict.get('abr') is not None:
1615 res += 'video@'
1616 if fdict.get('vbr') is not None:
1617 res += '%4dk' % fdict['vbr']
fbb21cf5
PH
1618 if fdict.get('fps') is not None:
1619 res += ', %sfps' % fdict['fps']
c57f7757
PH
1620 if fdict.get('acodec') is not None:
1621 if res:
1622 res += ', '
1623 if fdict['acodec'] == 'none':
1624 res += 'video only'
1625 else:
1626 res += '%-5s' % fdict['acodec']
1627 elif fdict.get('abr') is not None:
1628 if res:
1629 res += ', '
1630 res += 'audio'
1631 if fdict.get('abr') is not None:
1632 res += '@%3dk' % fdict['abr']
1633 if fdict.get('asr') is not None:
1634 res += ' (%5dHz)' % fdict['asr']
1635 if fdict.get('filesize') is not None:
1636 if res:
1637 res += ', '
1638 res += format_bytes(fdict['filesize'])
9732d77e
PH
1639 elif fdict.get('filesize_approx') is not None:
1640 if res:
1641 res += ', '
1642 res += '~' + format_bytes(fdict['filesize_approx'])
c57f7757 1643 return res
91c7271a 1644
c57f7757 1645 def list_formats(self, info_dict):
94badb25 1646 formats = info_dict.get('formats', [info_dict])
b81a359e
PH
1647 table = [
1648 [f['format_id'], f['ext'], self.format_resolution(f), self._format_note(f)]
1649 for f in formats
e65566a9 1650 if f.get('preference') is None or f['preference'] >= -1000]
94badb25 1651 if len(formats) > 1:
b81a359e 1652 table[-1][-1] += (' ' if table[-1][-1] else '') + '(best)'
57dd9a8f 1653
b81a359e 1654 header_line = ['format code', 'extension', 'resolution', 'note']
cfb56d1a 1655 self.to_screen(
b81a359e
PH
1656 '[info] Available formats for %s:\n%s' %
1657 (info_dict['id'], render_table(header_line, table)))
cfb56d1a
PH
1658
1659 def list_thumbnails(self, info_dict):
1660 thumbnails = info_dict.get('thumbnails')
1661 if not thumbnails:
1662 tn_url = info_dict.get('thumbnail')
1663 if tn_url:
1664 thumbnails = [{'id': '0', 'url': tn_url}]
1665 else:
1666 self.to_screen(
1667 '[info] No thumbnails present for %s' % info_dict['id'])
1668 return
1669
1670 self.to_screen(
1671 '[info] Thumbnails for %s:' % info_dict['id'])
1672 self.to_screen(render_table(
1673 ['ID', 'width', 'height', 'URL'],
1674 [[t['id'], t.get('width', 'unknown'), t.get('height', 'unknown'), t['url']] for t in thumbnails]))
dca08720 1675
360e1ca5 1676 def list_subtitles(self, video_id, subtitles, name='subtitles'):
a504ced0 1677 if not subtitles:
360e1ca5 1678 self.to_screen('%s has no %s' % (video_id, name))
a504ced0 1679 return
a504ced0 1680 self.to_screen(
edab9dbf
JMF
1681 'Available %s for %s:' % (name, video_id))
1682 self.to_screen(render_table(
1683 ['Language', 'formats'],
1684 [[lang, ', '.join(f['ext'] for f in reversed(formats))]
1685 for lang, formats in subtitles.items()]))
a504ced0 1686
dca08720
PH
1687 def urlopen(self, req):
1688 """ Start an HTTP download """
37419b4f 1689
d05cfe06
S
1690 # According to RFC 3986, URLs can not contain non-ASCII characters, however this is not
1691 # always respected by websites, some tend to give out URLs with non percent-encoded
1692 # non-ASCII characters (see telemb.py, ard.py [#3412])
37419b4f 1693 # urllib chokes on URLs with non-ASCII characters (see http://bugs.python.org/issue3991)
d05cfe06
S
1694 # To work around aforementioned issue we will replace request's original URL with
1695 # percent-encoded one
8f9312c3 1696 req_is_string = isinstance(req, compat_basestring)
68b09730 1697 url = req if req_is_string else req.get_full_url()
d05cfe06 1698 url_escaped = escape_url(url)
37419b4f
S
1699
1700 # Substitute URL if any change after escaping
1701 if url != url_escaped:
68b09730 1702 if req_is_string:
37419b4f
S
1703 req = url_escaped
1704 else:
1705 req = compat_urllib_request.Request(
1706 url_escaped, data=req.data, headers=req.headers,
1707 origin_req_host=req.origin_req_host, unverifiable=req.unverifiable)
1708
19a41fc6 1709 return self._opener.open(req, timeout=self._socket_timeout)
dca08720
PH
1710
1711 def print_debug_header(self):
1712 if not self.params.get('verbose'):
1713 return
62fec3b2 1714
4192b51c
PH
1715 if type('') is not compat_str:
1716 # Python 2.6 on SLES11 SP1 (https://github.com/rg3/youtube-dl/issues/3326)
1717 self.report_warning(
1718 'Your Python is broken! Update to a newer and supported version')
1719
c6afed48
PH
1720 stdout_encoding = getattr(
1721 sys.stdout, 'encoding', 'missing (%s)' % type(sys.stdout).__name__)
b0472057 1722 encoding_str = (
734f90bb
PH
1723 '[debug] Encodings: locale %s, fs %s, out %s, pref %s\n' % (
1724 locale.getpreferredencoding(),
1725 sys.getfilesystemencoding(),
c6afed48 1726 stdout_encoding,
b0472057 1727 self.get_encoding()))
4192b51c 1728 write_string(encoding_str, encoding=None)
734f90bb
PH
1729
1730 self._write_string('[debug] youtube-dl version ' + __version__ + '\n')
dca08720
PH
1731 try:
1732 sp = subprocess.Popen(
1733 ['git', 'rev-parse', '--short', 'HEAD'],
1734 stdout=subprocess.PIPE, stderr=subprocess.PIPE,
1735 cwd=os.path.dirname(os.path.abspath(__file__)))
1736 out, err = sp.communicate()
1737 out = out.decode().strip()
1738 if re.match('[0-9a-f]+', out):
734f90bb 1739 self._write_string('[debug] Git HEAD: ' + out + '\n')
70a1165b 1740 except Exception:
dca08720
PH
1741 try:
1742 sys.exc_clear()
70a1165b 1743 except Exception:
dca08720 1744 pass
d28b5171
PH
1745 self._write_string('[debug] Python version %s - %s\n' % (
1746 platform.python_version(), platform_name()))
1747
73fac4e9 1748 exe_versions = FFmpegPostProcessor.get_versions(self)
4c83c967 1749 exe_versions['rtmpdump'] = rtmpdump_version()
d28b5171
PH
1750 exe_str = ', '.join(
1751 '%s %s' % (exe, v)
1752 for exe, v in sorted(exe_versions.items())
1753 if v
1754 )
1755 if not exe_str:
1756 exe_str = 'none'
1757 self._write_string('[debug] exe versions: %s\n' % exe_str)
dca08720
PH
1758
1759 proxy_map = {}
1760 for handler in self._opener.handlers:
1761 if hasattr(handler, 'proxies'):
1762 proxy_map.update(handler.proxies)
734f90bb 1763 self._write_string('[debug] Proxy map: ' + compat_str(proxy_map) + '\n')
dca08720 1764
58b1f00d
PH
1765 if self.params.get('call_home', False):
1766 ipaddr = self.urlopen('https://yt-dl.org/ip').read().decode('utf-8')
1767 self._write_string('[debug] Public IP address: %s\n' % ipaddr)
1768 latest_version = self.urlopen(
1769 'https://yt-dl.org/latest/version').read().decode('utf-8')
1770 if version_tuple(latest_version) > version_tuple(__version__):
1771 self.report_warning(
1772 'You are using an outdated version (newest version: %s)! '
1773 'See https://yt-dl.org/update if you need help updating.' %
1774 latest_version)
1775
e344693b 1776 def _setup_opener(self):
6ad14cab 1777 timeout_val = self.params.get('socket_timeout')
19a41fc6 1778 self._socket_timeout = 600 if timeout_val is None else float(timeout_val)
6ad14cab 1779
dca08720
PH
1780 opts_cookiefile = self.params.get('cookiefile')
1781 opts_proxy = self.params.get('proxy')
1782
1783 if opts_cookiefile is None:
1784 self.cookiejar = compat_cookiejar.CookieJar()
1785 else:
1786 self.cookiejar = compat_cookiejar.MozillaCookieJar(
1787 opts_cookiefile)
1788 if os.access(opts_cookiefile, os.R_OK):
1789 self.cookiejar.load()
1790
1791 cookie_processor = compat_urllib_request.HTTPCookieProcessor(
1792 self.cookiejar)
1793 if opts_proxy is not None:
1794 if opts_proxy == '':
1795 proxies = {}
1796 else:
1797 proxies = {'http': opts_proxy, 'https': opts_proxy}
1798 else:
1799 proxies = compat_urllib_request.getproxies()
1800 # Set HTTPS proxy to HTTP one if given (https://github.com/rg3/youtube-dl/issues/805)
1801 if 'http' in proxies and 'https' not in proxies:
1802 proxies['https'] = proxies['http']
91410c9b 1803 proxy_handler = PerRequestProxyHandler(proxies)
a0ddb8a2
PH
1804
1805 debuglevel = 1 if self.params.get('debug_printtraffic') else 0
be4a824d
PH
1806 https_handler = make_HTTPS_handler(self.params, debuglevel=debuglevel)
1807 ydlh = YoutubeDLHandler(self.params, debuglevel=debuglevel)
dca08720 1808 opener = compat_urllib_request.build_opener(
2461f79d
PH
1809 proxy_handler, https_handler, cookie_processor, ydlh)
1810
dca08720
PH
1811 # Delete the default user-agent header, which would otherwise apply in
1812 # cases where our custom HTTP handler doesn't come into play
1813 # (See https://github.com/rg3/youtube-dl/issues/1309 for details)
1814 opener.addheaders = []
1815 self._opener = opener
62fec3b2
PH
1816
1817 def encode(self, s):
1818 if isinstance(s, bytes):
1819 return s # Already encoded
1820
1821 try:
1822 return s.encode(self.get_encoding())
1823 except UnicodeEncodeError as err:
1824 err.reason = err.reason + '. Check your system encoding configuration or use the --encoding option.'
1825 raise
1826
1827 def get_encoding(self):
1828 encoding = self.params.get('encoding')
1829 if encoding is None:
1830 encoding = preferredencoding()
1831 return encoding
ec82d85a
PH
1832
1833 def _write_thumbnails(self, info_dict, filename):
1834 if self.params.get('writethumbnail', False):
1835 thumbnails = info_dict.get('thumbnails')
1836 if thumbnails:
1837 thumbnails = [thumbnails[-1]]
1838 elif self.params.get('write_all_thumbnails', False):
1839 thumbnails = info_dict.get('thumbnails')
1840 else:
1841 return
1842
1843 if not thumbnails:
1844 # No thumbnails present, so return immediately
1845 return
1846
1847 for t in thumbnails:
1848 thumb_ext = determine_ext(t['url'], 'jpg')
1849 suffix = '_%s' % t['id'] if len(thumbnails) > 1 else ''
1850 thumb_display_id = '%s ' % t['id'] if len(thumbnails) > 1 else ''
1851 thumb_filename = os.path.splitext(filename)[0] + suffix + '.' + thumb_ext
1852
1853 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(thumb_filename)):
1854 self.to_screen('[%s] %s: Thumbnail %sis already present' %
1855 (info_dict['extractor'], info_dict['id'], thumb_display_id))
1856 else:
1857 self.to_screen('[%s] %s: Downloading thumbnail %s...' %
1858 (info_dict['extractor'], info_dict['id'], thumb_display_id))
1859 try:
1860 uf = self.urlopen(t['url'])
1861 with open(thumb_filename, 'wb') as thumbf:
1862 shutil.copyfileobj(uf, thumbf)
1863 self.to_screen('[%s] %s: Writing thumbnail %sto: %s' %
1864 (info_dict['extractor'], info_dict['id'], thumb_display_id, thumb_filename))
1865 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
1866 self.report_warning('Unable to download thumbnail "%s": %s' %
1867 (t['url'], compat_str(err)))