]> jfr.im git - yt-dlp.git/blame - youtube_dl/YoutubeDL.py
[wat] Remove unused import
[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
9d2ecdbc 7import datetime
c1c9a79c 8import errno
8222d8de 9import io
8694c600 10import json
62fec3b2 11import locale
8222d8de 12import os
dca08720 13import platform
8222d8de
JMF
14import re
15import shutil
dca08720 16import subprocess
8222d8de
JMF
17import socket
18import sys
19import time
20import traceback
21
1e5b9a95
PH
22if os.name == 'nt':
23 import ctypes
24
ce02ed60 25from .utils import (
dca08720 26 compat_cookiejar,
ce02ed60 27 compat_http_client,
ce02ed60
PH
28 compat_str,
29 compat_urllib_error,
30 compat_urllib_request,
31 ContentTooShortError,
32 date_from_str,
33 DateRange,
acd69589 34 DEFAULT_OUTTMPL,
ce02ed60
PH
35 determine_ext,
36 DownloadError,
37 encodeFilename,
38 ExtractorError,
02dbf93f 39 format_bytes,
525ef922 40 formatSeconds,
1c088fa8 41 get_term_width,
ce02ed60 42 locked_file,
dca08720 43 make_HTTPS_handler,
ce02ed60 44 MaxDownloadsReached,
b7ab0590 45 PagedList,
ce02ed60 46 PostProcessingError,
dca08720 47 platform_name,
ce02ed60
PH
48 preferredencoding,
49 SameFileError,
50 sanitize_filename,
51 subtitles_filename,
52 takewhile_inclusive,
53 UnavailableVideoError,
29eb5174 54 url_basename,
ce02ed60
PH
55 write_json_file,
56 write_string,
dca08720 57 YoutubeDLHandler,
6350728b 58 prepend_extension,
ce02ed60 59)
023fa8c4 60from .extractor import get_info_extractor, gen_extractors
3bc2ddcc 61from .downloader import get_suitable_downloader
56327689 62from .postprocessor import FFmpegMergerPP
dca08720 63from .version import __version__
8222d8de
JMF
64
65
66class YoutubeDL(object):
67 """YoutubeDL class.
68
69 YoutubeDL objects are the ones responsible of downloading the
70 actual video file and writing it to disk if the user has requested
71 it, among some other tasks. In most cases there should be one per
72 program. As, given a video URL, the downloader doesn't know how to
73 extract all the needed information, task that InfoExtractors do, it
74 has to pass the URL to one of them.
75
76 For this, YoutubeDL objects have a method that allows
77 InfoExtractors to be registered in a given order. When it is passed
78 a URL, the YoutubeDL object handles it to the first InfoExtractor it
79 finds that reports being able to handle it. The InfoExtractor extracts
80 all the information about the video or videos the URL refers to, and
81 YoutubeDL process the extracted information, possibly using a File
82 Downloader to download the video.
83
84 YoutubeDL objects accept a lot of parameters. In order not to saturate
85 the object constructor with arguments, it receives a dictionary of
86 options instead. These options are available through the params
87 attribute for the InfoExtractors to use. The YoutubeDL also
88 registers itself as the downloader in charge for the InfoExtractors
89 that are added to it, so this is a "mutual registration".
90
91 Available options:
92
93 username: Username for authentication purposes.
94 password: Password for authentication purposes.
c6c19746 95 videopassword: Password for acces a video.
8222d8de
JMF
96 usenetrc: Use netrc for authentication instead.
97 verbose: Print additional info to stdout.
98 quiet: Do not print messages to stdout.
ad8915b7 99 no_warnings: Do not print out anything for warnings.
8222d8de
JMF
100 forceurl: Force printing final URL.
101 forcetitle: Force printing title.
102 forceid: Force printing ID.
103 forcethumbnail: Force printing thumbnail URL.
104 forcedescription: Force printing description.
105 forcefilename: Force printing final filename.
525ef922 106 forceduration: Force printing duration.
8694c600 107 forcejson: Force printing info_dict as JSON.
8222d8de
JMF
108 simulate: Do not download the video files.
109 format: Video format code.
110 format_limit: Highest quality format to try.
111 outtmpl: Template for output names.
112 restrictfilenames: Do not allow "&" and spaces in file names
113 ignoreerrors: Do not stop on download errors.
114 nooverwrites: Prevent overwriting files.
115 playliststart: Playlist item to start at.
116 playlistend: Playlist item to end at.
117 matchtitle: Download only matching titles.
118 rejecttitle: Reject downloads for matching titles.
8bf9319e 119 logger: Log messages to a logging.Logger instance.
8222d8de
JMF
120 logtostderr: Log messages to stderr instead of stdout.
121 writedescription: Write the video description to a .description file
122 writeinfojson: Write the video description to a .info.json file
1fb07d10 123 writeannotations: Write the video annotations to a .annotations.xml file
8222d8de
JMF
124 writethumbnail: Write the thumbnail image to a file
125 writesubtitles: Write the video subtitles to a file
b004821f 126 writeautomaticsub: Write the automatic subtitles to a file
8222d8de 127 allsubtitles: Downloads all the subtitles of the video
0b7f3118 128 (requires writesubtitles or writeautomaticsub)
8222d8de 129 listsubtitles: Lists all available subtitles for the video
b98a6b2f 130 subtitlesformat: Subtitle format [srt/sbv/vtt] (default=srt)
aa6a10c4 131 subtitleslangs: List of languages of the subtitles to download
8222d8de
JMF
132 keepvideo: Keep the video file after post-processing
133 daterange: A DateRange object, download only if the upload_date is in the range.
134 skip_download: Skip the actual download of the video file
c35f9e72 135 cachedir: Location of the cache files in the filesystem.
c3c88a26 136 None to disable filesystem cache.
47192f92 137 noplaylist: Download single video instead of a playlist if in doubt.
8dbe9899
PH
138 age_limit: An integer representing the user's age in years.
139 Unsuitable videos for the given age are skipped.
5fe18bdb
PH
140 min_views: An integer representing the minimum view count the video
141 must have in order to not be skipped.
142 Videos without view count information are always
143 downloaded. None for no limit.
144 max_views: An integer representing the maximum view count.
145 Videos that are more popular than that are not
146 downloaded.
147 Videos without view count information are always
148 downloaded. None for no limit.
149 download_archive: File name of a file where all downloads are recorded.
c1c9a79c
PH
150 Videos already present in the file are not downloaded
151 again.
dca08720 152 cookiefile: File name where cookies should be read from and dumped to.
a1ee09e8 153 nocheckcertificate:Do not verify SSL certificates
7e8c0af0
PH
154 prefer_insecure: Use HTTP instead of HTTPS to retrieve information.
155 At the moment, this is only supported by YouTube.
a1ee09e8 156 proxy: URL of the proxy server to use
e344693b 157 socket_timeout: Time to wait for unresponsive hosts, in seconds
0783b09b
PH
158 bidi_workaround: Work around buggy terminals without bidirectional text
159 support, using fridibi
a0ddb8a2 160 debug_printtraffic:Print out sent and received HTTP traffic
7b0817e8 161 include_ads: Download ads as well
04b4d394
PH
162 default_search: Prepend this string if an input url is not valid.
163 'auto' for elaborate guessing
62fec3b2 164 encoding: Use this encoding instead of the system-specified.
e8ee972c 165 extract_flat: Do not resolve URLs, return the immediate result.
fe7e0c98 166
8222d8de
JMF
167 The following parameters are not used by YoutubeDL itself, they are used by
168 the FileDownloader:
169 nopart, updatetime, buffersize, ratelimit, min_filesize, max_filesize, test,
170 noresizebuffer, retries, continuedl, noprogress, consoletitle
76b1bd67
JMF
171
172 The following options are used by the post processors:
173 prefer_ffmpeg: If True, use ffmpeg instead of avconv if both are available,
174 otherwise prefer avconv.
8222d8de
JMF
175 """
176
177 params = None
178 _ies = []
179 _pps = []
180 _download_retcode = None
181 _num_downloads = None
182 _screen_file = None
183
a3fb4675 184 def __init__(self, params=None):
8222d8de 185 """Create a FileDownloader object with the given options."""
e9f9a10f
JMF
186 if params is None:
187 params = {}
8222d8de 188 self._ies = []
56c73665 189 self._ies_instances = {}
8222d8de 190 self._pps = []
933605d7 191 self._progress_hooks = []
8222d8de
JMF
192 self._download_retcode = 0
193 self._num_downloads = 0
194 self._screen_file = [sys.stdout, sys.stderr][params.get('logtostderr', False)]
0783b09b 195 self._err_file = sys.stderr
e9f9a10f 196 self.params = params
34308b30 197
0783b09b 198 if params.get('bidi_workaround', False):
1c088fa8
PH
199 try:
200 import pty
201 master, slave = pty.openpty()
202 width = get_term_width()
203 if width is None:
204 width_args = []
205 else:
206 width_args = ['-w', str(width)]
5d681e96 207 sp_kwargs = dict(
1c088fa8
PH
208 stdin=subprocess.PIPE,
209 stdout=slave,
210 stderr=self._err_file)
5d681e96
PH
211 try:
212 self._output_process = subprocess.Popen(
213 ['bidiv'] + width_args, **sp_kwargs
214 )
215 except OSError:
5d681e96
PH
216 self._output_process = subprocess.Popen(
217 ['fribidi', '-c', 'UTF-8'] + width_args, **sp_kwargs)
218 self._output_channel = os.fdopen(master, 'rb')
1c088fa8
PH
219 except OSError as ose:
220 if ose.errno == 2:
6febd1c1 221 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
222 else:
223 raise
0783b09b 224
34308b30
PH
225 if (sys.version_info >= (3,) and sys.platform != 'win32' and
226 sys.getfilesystemencoding() in ['ascii', 'ANSI_X3.4-1968']
227 and not params['restrictfilenames']):
228 # On Python 3, the Unicode filesystem API will throw errors (#1474)
229 self.report_warning(
6febd1c1
PH
230 'Assuming --restrict-filenames since file system encoding '
231 'cannot encode all charactes. '
232 'Set the LC_ALL environment variable to fix this.')
4a98cdbf 233 self.params['restrictfilenames'] = True
34308b30 234
a3927cf7 235 if '%(stitle)s' in self.params.get('outtmpl', ''):
6febd1c1 236 self.report_warning('%(stitle)s is deprecated. Use the %(title)s and the --restrict-filenames flag(which also secures %(uploader)s et al) instead.')
8222d8de 237
dca08720
PH
238 self._setup_opener()
239
8222d8de
JMF
240 def add_info_extractor(self, ie):
241 """Add an InfoExtractor object to the end of the list."""
242 self._ies.append(ie)
56c73665 243 self._ies_instances[ie.ie_key()] = ie
8222d8de
JMF
244 ie.set_downloader(self)
245
56c73665
JMF
246 def get_info_extractor(self, ie_key):
247 """
248 Get an instance of an IE with name ie_key, it will try to get one from
249 the _ies list, if there's no instance it will create a new one and add
250 it to the extractor list.
251 """
252 ie = self._ies_instances.get(ie_key)
253 if ie is None:
254 ie = get_info_extractor(ie_key)()
255 self.add_info_extractor(ie)
256 return ie
257
023fa8c4
JMF
258 def add_default_info_extractors(self):
259 """
260 Add the InfoExtractors returned by gen_extractors to the end of the list
261 """
262 for ie in gen_extractors():
263 self.add_info_extractor(ie)
264
8222d8de
JMF
265 def add_post_processor(self, pp):
266 """Add a PostProcessor object to the end of the chain."""
267 self._pps.append(pp)
268 pp.set_downloader(self)
269
933605d7
JMF
270 def add_progress_hook(self, ph):
271 """Add the progress hook (currently only for the file downloader)"""
272 self._progress_hooks.append(ph)
8ab470f1 273
1c088fa8 274 def _bidi_workaround(self, message):
5d681e96 275 if not hasattr(self, '_output_channel'):
1c088fa8
PH
276 return message
277
5d681e96 278 assert hasattr(self, '_output_process')
11b85ce6 279 assert isinstance(message, compat_str)
6febd1c1
PH
280 line_count = message.count('\n') + 1
281 self._output_process.stdin.write((message + '\n').encode('utf-8'))
5d681e96 282 self._output_process.stdin.flush()
6febd1c1 283 res = ''.join(self._output_channel.readline().decode('utf-8')
1c088fa8 284 for _ in range(line_count))
6febd1c1 285 return res[:-len('\n')]
1c088fa8 286
8222d8de 287 def to_screen(self, message, skip_eol=False):
0783b09b
PH
288 """Print message to stdout if not in quiet mode."""
289 return self.to_stdout(message, skip_eol, check_quiet=True)
290
734f90bb 291 def _write_string(self, s, out=None):
b58ddb32 292 write_string(s, out=out, encoding=self.params.get('encoding'))
734f90bb 293
0783b09b 294 def to_stdout(self, message, skip_eol=False, check_quiet=False):
8222d8de 295 """Print message to stdout if not in quiet mode."""
8bf9319e 296 if self.params.get('logger'):
43afe285 297 self.params['logger'].debug(message)
0783b09b 298 elif not check_quiet or not self.params.get('quiet', False):
1c088fa8 299 message = self._bidi_workaround(message)
6febd1c1 300 terminator = ['\n', ''][skip_eol]
8222d8de 301 output = message + terminator
1c088fa8 302
734f90bb 303 self._write_string(output, self._screen_file)
8222d8de
JMF
304
305 def to_stderr(self, message):
306 """Print message to stderr."""
11b85ce6 307 assert isinstance(message, compat_str)
8bf9319e 308 if self.params.get('logger'):
43afe285
IB
309 self.params['logger'].error(message)
310 else:
1c088fa8 311 message = self._bidi_workaround(message)
6febd1c1 312 output = message + '\n'
734f90bb 313 self._write_string(output, self._err_file)
8222d8de 314
1e5b9a95
PH
315 def to_console_title(self, message):
316 if not self.params.get('consoletitle', False):
317 return
318 if os.name == 'nt' and ctypes.windll.kernel32.GetConsoleWindow():
319 # c_wchar_p() might not be necessary if `message` is
320 # already of type unicode()
321 ctypes.windll.kernel32.SetConsoleTitleW(ctypes.c_wchar_p(message))
322 elif 'TERM' in os.environ:
734f90bb 323 self._write_string('\033]0;%s\007' % message, self._screen_file)
1e5b9a95 324
bdde425c
PH
325 def save_console_title(self):
326 if not self.params.get('consoletitle', False):
327 return
328 if 'TERM' in os.environ:
efd6c574 329 # Save the title on stack
734f90bb 330 self._write_string('\033[22;0t', self._screen_file)
bdde425c
PH
331
332 def restore_console_title(self):
333 if not self.params.get('consoletitle', False):
334 return
335 if 'TERM' in os.environ:
efd6c574 336 # Restore the title from stack
734f90bb 337 self._write_string('\033[23;0t', self._screen_file)
bdde425c
PH
338
339 def __enter__(self):
340 self.save_console_title()
341 return self
342
343 def __exit__(self, *args):
344 self.restore_console_title()
f89197d7 345
dca08720
PH
346 if self.params.get('cookiefile') is not None:
347 self.cookiejar.save()
bdde425c 348
8222d8de
JMF
349 def trouble(self, message=None, tb=None):
350 """Determine action to take when a download problem appears.
351
352 Depending on if the downloader has been configured to ignore
353 download errors or not, this method may throw an exception or
354 not when errors are found, after printing the message.
355
356 tb, if given, is additional traceback information.
357 """
358 if message is not None:
359 self.to_stderr(message)
360 if self.params.get('verbose'):
361 if tb is None:
362 if sys.exc_info()[0]: # if .trouble has been called from an except block
6febd1c1 363 tb = ''
8222d8de 364 if hasattr(sys.exc_info()[1], 'exc_info') and sys.exc_info()[1].exc_info[0]:
6febd1c1 365 tb += ''.join(traceback.format_exception(*sys.exc_info()[1].exc_info))
8222d8de
JMF
366 tb += compat_str(traceback.format_exc())
367 else:
368 tb_data = traceback.format_list(traceback.extract_stack())
6febd1c1 369 tb = ''.join(tb_data)
8222d8de
JMF
370 self.to_stderr(tb)
371 if not self.params.get('ignoreerrors', False):
372 if sys.exc_info()[0] and hasattr(sys.exc_info()[1], 'exc_info') and sys.exc_info()[1].exc_info[0]:
373 exc_info = sys.exc_info()[1].exc_info
374 else:
375 exc_info = sys.exc_info()
376 raise DownloadError(message, exc_info)
377 self._download_retcode = 1
378
379 def report_warning(self, message):
380 '''
381 Print the message to stderr, it will be prefixed with 'WARNING:'
382 If stderr is a tty file the 'WARNING:' will be colored
383 '''
6d07ce01
JMF
384 if self.params.get('logger') is not None:
385 self.params['logger'].warning(message)
8222d8de 386 else:
ad8915b7
PH
387 if self.params.get('no_warnings'):
388 return
6d07ce01
JMF
389 if self._err_file.isatty() and os.name != 'nt':
390 _msg_header = '\033[0;33mWARNING:\033[0m'
391 else:
392 _msg_header = 'WARNING:'
393 warning_message = '%s %s' % (_msg_header, message)
394 self.to_stderr(warning_message)
8222d8de
JMF
395
396 def report_error(self, message, tb=None):
397 '''
398 Do the same as trouble, but prefixes the message with 'ERROR:', colored
399 in red if stderr is a tty file.
400 '''
0783b09b 401 if self._err_file.isatty() and os.name != 'nt':
6febd1c1 402 _msg_header = '\033[0;31mERROR:\033[0m'
8222d8de 403 else:
6febd1c1
PH
404 _msg_header = 'ERROR:'
405 error_message = '%s %s' % (_msg_header, message)
8222d8de
JMF
406 self.trouble(error_message, tb)
407
8222d8de
JMF
408 def report_file_already_downloaded(self, file_name):
409 """Report file has already been fully downloaded."""
410 try:
6febd1c1 411 self.to_screen('[download] %s has already been downloaded' % file_name)
ce02ed60 412 except UnicodeEncodeError:
6febd1c1 413 self.to_screen('[download] The file has already been downloaded')
8222d8de 414
8222d8de
JMF
415 def prepare_filename(self, info_dict):
416 """Generate the output filename."""
417 try:
418 template_dict = dict(info_dict)
419
420 template_dict['epoch'] = int(time.time())
421 autonumber_size = self.params.get('autonumber_size')
422 if autonumber_size is None:
423 autonumber_size = 5
6febd1c1 424 autonumber_templ = '%0' + str(autonumber_size) + 'd'
8222d8de 425 template_dict['autonumber'] = autonumber_templ % self._num_downloads
702665c0 426 if template_dict.get('playlist_index') is not None:
6febd1c1 427 template_dict['playlist_index'] = '%05d' % template_dict['playlist_index']
17b75c0d
PH
428 if template_dict.get('resolution') is None:
429 if template_dict.get('width') and template_dict.get('height'):
430 template_dict['resolution'] = '%dx%d' % (template_dict['width'], template_dict['height'])
431 elif template_dict.get('height'):
805ef3c6 432 template_dict['resolution'] = '%sp' % template_dict['height']
17b75c0d 433 elif template_dict.get('width'):
805ef3c6 434 template_dict['resolution'] = '?x%d' % template_dict['width']
8222d8de 435
586a91b6 436 sanitize = lambda k, v: sanitize_filename(
45598aab 437 compat_str(v),
8222d8de 438 restricted=self.params.get('restrictfilenames'),
6febd1c1 439 is_id=(k == 'id'))
586a91b6 440 template_dict = dict((k, sanitize(k, v))
45598aab
PH
441 for k, v in template_dict.items()
442 if v is not None)
6febd1c1 443 template_dict = collections.defaultdict(lambda: 'NA', template_dict)
8222d8de 444
acd69589
PH
445 outtmpl = self.params.get('outtmpl', DEFAULT_OUTTMPL)
446 tmpl = os.path.expanduser(outtmpl)
586a91b6 447 filename = tmpl % template_dict
8222d8de 448 return filename
8222d8de 449 except ValueError as err:
6febd1c1 450 self.report_error('Error in output template: ' + str(err) + ' (encoding: ' + repr(preferredencoding()) + ')')
8222d8de
JMF
451 return None
452
453 def _match_entry(self, info_dict):
454 """ Returns None iff the file should be downloaded """
455
6febd1c1 456 video_title = info_dict.get('title', info_dict.get('id', 'video'))
7012b23c
PH
457 if 'title' in info_dict:
458 # This can happen when we're just evaluating the playlist
459 title = info_dict['title']
460 matchtitle = self.params.get('matchtitle', False)
461 if matchtitle:
462 if not re.search(matchtitle, title, re.IGNORECASE):
6febd1c1 463 return '"' + title + '" title did not match pattern "' + matchtitle + '"'
7012b23c
PH
464 rejecttitle = self.params.get('rejecttitle', False)
465 if rejecttitle:
466 if re.search(rejecttitle, title, re.IGNORECASE):
6febd1c1 467 return '"' + title + '" title matched reject pattern "' + rejecttitle + '"'
8222d8de
JMF
468 date = info_dict.get('upload_date', None)
469 if date is not None:
470 dateRange = self.params.get('daterange', DateRange())
471 if date not in dateRange:
6febd1c1 472 return '%s upload date is not in range %s' % (date_from_str(date).isoformat(), dateRange)
5fe18bdb
PH
473 view_count = info_dict.get('view_count', None)
474 if view_count is not None:
475 min_views = self.params.get('min_views')
476 if min_views is not None and view_count < min_views:
6febd1c1 477 return 'Skipping %s, because it has not reached minimum view count (%d/%d)' % (video_title, view_count, min_views)
5fe18bdb
PH
478 max_views = self.params.get('max_views')
479 if max_views is not None and view_count > max_views:
6febd1c1 480 return 'Skipping %s, because it has exceeded the maximum view count (%d/%d)' % (video_title, view_count, max_views)
8dbe9899
PH
481 age_limit = self.params.get('age_limit')
482 if age_limit is not None:
be843678
PH
483 actual_age_limit = info_dict.get('age_limit')
484 if actual_age_limit is None:
485 actual_age_limit = 0
486 if age_limit < actual_age_limit:
6febd1c1 487 return 'Skipping "' + title + '" because it is age restricted'
c1c9a79c 488 if self.in_download_archive(info_dict):
6febd1c1 489 return '%s has already been recorded in archive' % video_title
8222d8de 490 return None
fe7e0c98 491
b6c45014
JMF
492 @staticmethod
493 def add_extra_info(info_dict, extra_info):
494 '''Set the keys from extra_info in info dict if they are missing'''
495 for key, value in extra_info.items():
496 info_dict.setdefault(key, value)
497
7fc3fa05
PH
498 def extract_info(self, url, download=True, ie_key=None, extra_info={},
499 process=True):
8222d8de
JMF
500 '''
501 Returns a list with a dictionary for each video we find.
502 If 'download', also downloads the videos.
503 extra_info is a dict containing the extra values to add to each result
504 '''
fe7e0c98 505
8222d8de 506 if ie_key:
56c73665 507 ies = [self.get_info_extractor(ie_key)]
8222d8de
JMF
508 else:
509 ies = self._ies
510
511 for ie in ies:
512 if not ie.suitable(url):
513 continue
514
515 if not ie.working():
6febd1c1
PH
516 self.report_warning('The program functionality for this site has been marked as broken, '
517 'and will probably not work.')
8222d8de
JMF
518
519 try:
520 ie_result = ie.extract(url)
521 if ie_result is None: # Finished already (backwards compatibility; listformats and friends should be moved here)
522 break
523 if isinstance(ie_result, list):
524 # Backwards compatibility: old IE result format
8222d8de
JMF
525 ie_result = {
526 '_type': 'compat_list',
527 'entries': ie_result,
528 }
ea38e55f 529 self.add_default_extra_info(ie_result, ie, url)
7fc3fa05
PH
530 if process:
531 return self.process_ie_result(ie_result, download, extra_info)
532 else:
533 return ie_result
8222d8de
JMF
534 except ExtractorError as de: # An error we somewhat expected
535 self.report_error(compat_str(de), de.format_traceback())
536 break
d3e5bbf4
PH
537 except MaxDownloadsReached:
538 raise
8222d8de
JMF
539 except Exception as e:
540 if self.params.get('ignoreerrors', False):
541 self.report_error(compat_str(e), tb=compat_str(traceback.format_exc()))
542 break
543 else:
544 raise
545 else:
1a489545 546 self.report_error('no suitable InfoExtractor for URL %s' % url)
fe7e0c98 547
ea38e55f
PH
548 def add_default_extra_info(self, ie_result, ie, url):
549 self.add_extra_info(ie_result, {
550 'extractor': ie.IE_NAME,
551 'webpage_url': url,
552 'webpage_url_basename': url_basename(url),
553 'extractor_key': ie.ie_key(),
554 })
555
8222d8de
JMF
556 def process_ie_result(self, ie_result, download=True, extra_info={}):
557 """
558 Take the result of the ie(may be modified) and resolve all unresolved
559 references (URLs, playlist items).
560
561 It will also download the videos if 'download'.
562 Returns the resolved ie_result.
563 """
564
e8ee972c
PH
565 result_type = ie_result.get('_type', 'video')
566
567 if self.params.get('extract_flat', False):
568 if result_type in ('url', 'url_transparent'):
569 return ie_result
570
8222d8de 571 if result_type == 'video':
b6c45014 572 self.add_extra_info(ie_result, extra_info)
feee2ecf 573 return self.process_video_result(ie_result, download=download)
8222d8de
JMF
574 elif result_type == 'url':
575 # We have to add extra_info to the results because it may be
576 # contained in a playlist
577 return self.extract_info(ie_result['url'],
578 download,
579 ie_key=ie_result.get('ie_key'),
580 extra_info=extra_info)
7fc3fa05
PH
581 elif result_type == 'url_transparent':
582 # Use the information from the embedding page
583 info = self.extract_info(
584 ie_result['url'], ie_key=ie_result.get('ie_key'),
585 extra_info=extra_info, download=False, process=False)
586
587 def make_result(embedded_info):
588 new_result = ie_result.copy()
589 for f in ('_type', 'url', 'ext', 'player_url', 'formats',
1538eff6 590 'entries', 'ie_key', 'duration',
ef4fd848
PH
591 'subtitles', 'annotations', 'format',
592 'thumbnail', 'thumbnails'):
7fc3fa05
PH
593 if f in new_result:
594 del new_result[f]
595 if f in embedded_info:
596 new_result[f] = embedded_info[f]
597 return new_result
598 new_result = make_result(info)
599
600 assert new_result.get('_type') != 'url_transparent'
601 if new_result.get('_type') == 'compat_list':
602 new_result['entries'] = [
603 make_result(e) for e in new_result['entries']]
604
605 return self.process_ie_result(
606 new_result, download=download, extra_info=extra_info)
8222d8de
JMF
607 elif result_type == 'playlist':
608 # We process each entry in the playlist
609 playlist = ie_result.get('title', None) or ie_result.get('id', None)
6febd1c1 610 self.to_screen('[download] Downloading playlist: %s' % playlist)
8222d8de
JMF
611
612 playlist_results = []
613
8222d8de 614 playliststart = self.params.get('playliststart', 1) - 1
a19fd00c
PH
615 playlistend = self.params.get('playlistend', None)
616 # For backwards compatibility, interpret -1 as whole list
8222d8de 617 if playlistend == -1:
a19fd00c 618 playlistend = None
8222d8de 619
b7ab0590
PH
620 if isinstance(ie_result['entries'], list):
621 n_all_entries = len(ie_result['entries'])
622 entries = ie_result['entries'][playliststart:playlistend]
623 n_entries = len(entries)
624 self.to_screen(
625 "[%s] playlist %s: Collected %d video ids (downloading %d of them)" %
626 (ie_result['extractor'], playlist, n_all_entries, n_entries))
627 else:
628 assert isinstance(ie_result['entries'], PagedList)
629 entries = ie_result['entries'].getslice(
630 playliststart, playlistend)
631 n_entries = len(entries)
632 self.to_screen(
633 "[%s] playlist %s: Downloading %d videos" %
634 (ie_result['extractor'], playlist, n_entries))
8222d8de 635
fe7e0c98 636 for i, entry in enumerate(entries, 1):
6febd1c1 637 self.to_screen('[download] Downloading video #%s of %s' % (i, n_entries))
8222d8de 638 extra = {
fe7e0c98
JMF
639 'playlist': playlist,
640 'playlist_index': i + playliststart,
b6c45014 641 'extractor': ie_result['extractor'],
9103bbc5 642 'webpage_url': ie_result['webpage_url'],
29eb5174 643 'webpage_url_basename': url_basename(ie_result['webpage_url']),
be97abc2 644 'extractor_key': ie_result['extractor_key'],
fe7e0c98 645 }
7012b23c
PH
646
647 reason = self._match_entry(entry)
648 if reason is not None:
6febd1c1 649 self.to_screen('[download] ' + reason)
7012b23c
PH
650 continue
651
8222d8de
JMF
652 entry_result = self.process_ie_result(entry,
653 download=download,
654 extra_info=extra)
655 playlist_results.append(entry_result)
656 ie_result['entries'] = playlist_results
657 return ie_result
658 elif result_type == 'compat_list':
659 def _fixup(r):
b6c45014 660 self.add_extra_info(r,
9103bbc5
JMF
661 {
662 'extractor': ie_result['extractor'],
663 'webpage_url': ie_result['webpage_url'],
29eb5174 664 'webpage_url_basename': url_basename(ie_result['webpage_url']),
be97abc2 665 'extractor_key': ie_result['extractor_key'],
9103bbc5 666 })
8222d8de
JMF
667 return r
668 ie_result['entries'] = [
b6c45014 669 self.process_ie_result(_fixup(r), download, extra_info)
8222d8de
JMF
670 for r in ie_result['entries']
671 ]
672 return ie_result
673 else:
674 raise Exception('Invalid result type: %s' % result_type)
675
a9c58ad9
JMF
676 def select_format(self, format_spec, available_formats):
677 if format_spec == 'best' or format_spec is None:
678 return available_formats[-1]
679 elif format_spec == 'worst':
680 return available_formats[0]
ba7678f9
PH
681 elif format_spec == 'bestaudio':
682 audio_formats = [
683 f for f in available_formats
684 if f.get('vcodec') == 'none']
685 if audio_formats:
686 return audio_formats[-1]
687 elif format_spec == 'worstaudio':
688 audio_formats = [
689 f for f in available_formats
690 if f.get('vcodec') == 'none']
691 if audio_formats:
692 return audio_formats[0]
bc6d5978
JMF
693 elif format_spec == 'bestvideo':
694 video_formats = [
695 f for f in available_formats
696 if f.get('acodec') == 'none']
697 if video_formats:
698 return video_formats[-1]
699 elif format_spec == 'worstvideo':
700 video_formats = [
701 f for f in available_formats
702 if f.get('acodec') == 'none']
703 if video_formats:
704 return video_formats[0]
a9c58ad9 705 else:
6febd1c1 706 extensions = ['mp4', 'flv', 'webm', '3gp']
49e86983
JMF
707 if format_spec in extensions:
708 filter_f = lambda f: f['ext'] == format_spec
709 else:
710 filter_f = lambda f: f['format_id'] == format_spec
fe7e0c98 711 matches = list(filter(filter_f, available_formats))
a9c58ad9
JMF
712 if matches:
713 return matches[-1]
714 return None
715
dd82ffea
JMF
716 def process_video_result(self, info_dict, download=True):
717 assert info_dict.get('_type', 'video') == 'video'
718
bec1fad2
PH
719 if 'id' not in info_dict:
720 raise ExtractorError('Missing "id" field in extractor result')
721 if 'title' not in info_dict:
722 raise ExtractorError('Missing "title" field in extractor result')
723
dd82ffea
JMF
724 if 'playlist' not in info_dict:
725 # It isn't part of a playlist
726 info_dict['playlist'] = None
727 info_dict['playlist_index'] = None
728
d5519808
PH
729 thumbnails = info_dict.get('thumbnails')
730 if thumbnails:
be6d7229
PH
731 thumbnails.sort(key=lambda t: (
732 t.get('width'), t.get('height'), t.get('url')))
d5519808
PH
733 for t in thumbnails:
734 if 'width' in t and 'height' in t:
735 t['resolution'] = '%dx%d' % (t['width'], t['height'])
736
737 if thumbnails and 'thumbnail' not in info_dict:
738 info_dict['thumbnail'] = thumbnails[-1]['url']
739
c9ae7b95 740 if 'display_id' not in info_dict and 'id' in info_dict:
0afef30b
PH
741 info_dict['display_id'] = info_dict['id']
742
955c4514 743 if info_dict.get('upload_date') is None and info_dict.get('timestamp') is not None:
9d2ecdbc 744 upload_date = datetime.datetime.utcfromtimestamp(
955c4514 745 info_dict['timestamp'])
9d2ecdbc
PH
746 info_dict['upload_date'] = upload_date.strftime('%Y%m%d')
747
6ff000b8 748 # This extractors handle format selection themselves
6febd1c1 749 if info_dict['extractor'] in ['Youku']:
12893efe
JMF
750 if download:
751 self.process_info(info_dict)
6ff000b8
JMF
752 return info_dict
753
dd82ffea
JMF
754 # We now pick which formats have to be downloaded
755 if info_dict.get('formats') is None:
756 # There's only one format available
757 formats = [info_dict]
758 else:
759 formats = info_dict['formats']
760
db95dc13
PH
761 if not formats:
762 raise ExtractorError('No video formats found!')
763
dd82ffea 764 # We check that all the formats have the format and format_id fields
db95dc13 765 for i, format in enumerate(formats):
bec1fad2
PH
766 if 'url' not in format:
767 raise ExtractorError('Missing "url" key in result (index %d)' % i)
768
dd82ffea 769 if format.get('format_id') is None:
8016c922 770 format['format_id'] = compat_str(i)
8c51aa65 771 if format.get('format') is None:
6febd1c1 772 format['format'] = '{id} - {res}{note}'.format(
8c51aa65
JMF
773 id=format['format_id'],
774 res=self.format_resolution(format),
6febd1c1 775 note=' ({0})'.format(format['format_note']) if format.get('format_note') is not None else '',
8c51aa65 776 )
c1002e96
PH
777 # Automatically determine file extension if missing
778 if 'ext' not in format:
cce929ea 779 format['ext'] = determine_ext(format['url']).lower()
dd82ffea 780
99e206d5
JMF
781 format_limit = self.params.get('format_limit', None)
782 if format_limit:
f4d96df0
PH
783 formats = list(takewhile_inclusive(
784 lambda f: f['format_id'] != format_limit, formats
785 ))
4bcc7bd1
PH
786
787 # TODO Central sorting goes here
99e206d5 788
f89197d7 789 if formats[0] is not info_dict:
b3d9ef88
JMF
790 # only set the 'formats' fields if the original info_dict list them
791 # otherwise we end up with a circular reference, the first (and unique)
f89197d7 792 # element in the 'formats' field in info_dict is info_dict itself,
b3d9ef88
JMF
793 # wich can't be exported to json
794 info_dict['formats'] = formats
bfaae0a7 795 if self.params.get('listformats', None):
796 self.list_formats(info_dict)
797 return
798
de3ef3ed 799 req_format = self.params.get('format')
a9c58ad9
JMF
800 if req_format is None:
801 req_format = 'best'
dd82ffea 802 formats_to_download = []
dd82ffea 803 # The -1 is for supporting YoutubeIE
a9c58ad9 804 if req_format in ('-1', 'all'):
dd82ffea
JMF
805 formats_to_download = formats
806 else:
a9c5e5ca 807 # We can accept formats requested in the format: 34/5/best, we pick
416a5efc 808 # the first that is available, starting from left
dd82ffea
JMF
809 req_formats = req_format.split('/')
810 for rf in req_formats:
6350728b
JMF
811 if re.match(r'.+?\+.+?', rf) is not None:
812 # Two formats have been requested like '137+139'
813 format_1, format_2 = rf.split('+')
814 formats_info = (self.select_format(format_1, formats),
815 self.select_format(format_2, formats))
816 if all(formats_info):
a9c5e5ca
PH
817 selected_format = {
818 'requested_formats': formats_info,
819 'format': rf,
820 'ext': formats_info[0]['ext'],
821 }
6350728b
JMF
822 else:
823 selected_format = None
824 else:
825 selected_format = self.select_format(rf, formats)
a9c58ad9
JMF
826 if selected_format is not None:
827 formats_to_download = [selected_format]
dd82ffea
JMF
828 break
829 if not formats_to_download:
6febd1c1 830 raise ExtractorError('requested format not available',
78a3a9f8 831 expected=True)
dd82ffea
JMF
832
833 if download:
834 if len(formats_to_download) > 1:
6febd1c1 835 self.to_screen('[info] %s: downloading video in %s formats' % (info_dict['id'], len(formats_to_download)))
dd82ffea
JMF
836 for format in formats_to_download:
837 new_info = dict(info_dict)
838 new_info.update(format)
839 self.process_info(new_info)
840 # We update the info dict with the best quality format (backwards compatibility)
841 info_dict.update(formats_to_download[-1])
842 return info_dict
843
8222d8de
JMF
844 def process_info(self, info_dict):
845 """Process a single resolved IE result."""
846
847 assert info_dict.get('_type', 'video') == 'video'
fd288278
PH
848
849 max_downloads = self.params.get('max_downloads')
850 if max_downloads is not None:
851 if self._num_downloads >= int(max_downloads):
852 raise MaxDownloadsReached()
8222d8de
JMF
853
854 info_dict['fulltitle'] = info_dict['title']
855 if len(info_dict['title']) > 200:
6febd1c1 856 info_dict['title'] = info_dict['title'][:197] + '...'
8222d8de
JMF
857
858 # Keep for backwards compatibility
859 info_dict['stitle'] = info_dict['title']
860
11b85ce6 861 if 'format' not in info_dict:
8222d8de
JMF
862 info_dict['format'] = info_dict['ext']
863
864 reason = self._match_entry(info_dict)
865 if reason is not None:
6febd1c1 866 self.to_screen('[download] ' + reason)
8222d8de
JMF
867 return
868
fd288278 869 self._num_downloads += 1
8222d8de
JMF
870
871 filename = self.prepare_filename(info_dict)
872
873 # Forced printings
874 if self.params.get('forcetitle', False):
0783b09b 875 self.to_stdout(info_dict['fulltitle'])
8222d8de 876 if self.params.get('forceid', False):
0783b09b 877 self.to_stdout(info_dict['id'])
8222d8de 878 if self.params.get('forceurl', False):
edde6c56 879 # For RTMP URLs, also include the playpath
6febd1c1 880 self.to_stdout(info_dict['url'] + info_dict.get('play_path', ''))
216d71d0 881 if self.params.get('forcethumbnail', False) and info_dict.get('thumbnail') is not None:
0783b09b 882 self.to_stdout(info_dict['thumbnail'])
216d71d0 883 if self.params.get('forcedescription', False) and info_dict.get('description') is not None:
0783b09b 884 self.to_stdout(info_dict['description'])
8222d8de 885 if self.params.get('forcefilename', False) and filename is not None:
0783b09b 886 self.to_stdout(filename)
525ef922
PH
887 if self.params.get('forceduration', False) and info_dict.get('duration') is not None:
888 self.to_stdout(formatSeconds(info_dict['duration']))
8222d8de 889 if self.params.get('forceformat', False):
0783b09b 890 self.to_stdout(info_dict['format'])
9d153818 891 if self.params.get('forcejson', False):
a0d96c98 892 info_dict['_filename'] = filename
0783b09b 893 self.to_stdout(json.dumps(info_dict))
8222d8de
JMF
894
895 # Do nothing else if in simulate mode
896 if self.params.get('simulate', False):
897 return
898
899 if filename is None:
900 return
901
902 try:
903 dn = os.path.dirname(encodeFilename(filename))
d26e981d 904 if dn and not os.path.exists(dn):
8222d8de
JMF
905 os.makedirs(dn)
906 except (OSError, IOError) as err:
6febd1c1 907 self.report_error('unable to create directory ' + compat_str(err))
8222d8de
JMF
908 return
909
910 if self.params.get('writedescription', False):
6febd1c1 911 descfn = filename + '.description'
7b6fefc9 912 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(descfn)):
6febd1c1 913 self.to_screen('[info] Video description is already present')
7b6fefc9
PH
914 else:
915 try:
6febd1c1 916 self.to_screen('[info] Writing video description to: ' + descfn)
7b6fefc9
PH
917 with io.open(encodeFilename(descfn), 'w', encoding='utf-8') as descfile:
918 descfile.write(info_dict['description'])
919 except (KeyError, TypeError):
6febd1c1 920 self.report_warning('There\'s no description to write.')
7b6fefc9 921 except (OSError, IOError):
6febd1c1 922 self.report_error('Cannot write description file ' + descfn)
7b6fefc9 923 return
8222d8de 924
1fb07d10 925 if self.params.get('writeannotations', False):
6febd1c1 926 annofn = filename + '.annotations.xml'
7b6fefc9 927 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(annofn)):
6febd1c1 928 self.to_screen('[info] Video annotations are already present')
7b6fefc9
PH
929 else:
930 try:
6febd1c1 931 self.to_screen('[info] Writing video annotations to: ' + annofn)
7b6fefc9
PH
932 with io.open(encodeFilename(annofn), 'w', encoding='utf-8') as annofile:
933 annofile.write(info_dict['annotations'])
934 except (KeyError, TypeError):
6febd1c1 935 self.report_warning('There are no annotations to write.')
7b6fefc9 936 except (OSError, IOError):
6febd1c1 937 self.report_error('Cannot write annotations file: ' + annofn)
7b6fefc9 938 return
1fb07d10 939
c4a91be7 940 subtitles_are_requested = any([self.params.get('writesubtitles', False),
0b7f3118 941 self.params.get('writeautomaticsub')])
c4a91be7 942
fe7e0c98 943 if subtitles_are_requested and 'subtitles' in info_dict and info_dict['subtitles']:
8222d8de
JMF
944 # subtitles download errors are already managed as troubles in relevant IE
945 # that way it will silently go on when used with unsupporting IE
8222d8de 946 subtitles = info_dict['subtitles']
ca715127 947 sub_format = self.params.get('subtitlesformat', 'srt')
5d51a883
JMF
948 for sub_lang in subtitles.keys():
949 sub = subtitles[sub_lang]
6804038d
JMF
950 if sub is None:
951 continue
8222d8de 952 try:
d4051a8e 953 sub_filename = subtitles_filename(filename, sub_lang, sub_format)
7b6fefc9 954 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(sub_filename)):
6febd1c1 955 self.to_screen('[info] Video subtitle %s.%s is already_present' % (sub_lang, sub_format))
7b6fefc9 956 else:
6febd1c1 957 self.to_screen('[info] Writing video subtitles to: ' + sub_filename)
7b6fefc9
PH
958 with io.open(encodeFilename(sub_filename), 'w', encoding='utf-8') as subfile:
959 subfile.write(sub)
8222d8de 960 except (OSError, IOError):
e4db1951 961 self.report_error('Cannot write subtitles file ' + sub_filename)
8222d8de
JMF
962 return
963
8222d8de 964 if self.params.get('writeinfojson', False):
6febd1c1 965 infofn = os.path.splitext(filename)[0] + '.info.json'
7b6fefc9 966 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(infofn)):
6febd1c1 967 self.to_screen('[info] Video description metadata is already present')
7b6fefc9 968 else:
6febd1c1 969 self.to_screen('[info] Writing video description metadata as JSON to: ' + infofn)
7b6fefc9 970 try:
1538eff6 971 write_json_file(info_dict, encodeFilename(infofn))
7b6fefc9 972 except (OSError, IOError):
6febd1c1 973 self.report_error('Cannot write metadata to JSON file ' + infofn)
7b6fefc9 974 return
8222d8de
JMF
975
976 if self.params.get('writethumbnail', False):
d8269e1d 977 if info_dict.get('thumbnail') is not None:
6febd1c1
PH
978 thumb_format = determine_ext(info_dict['thumbnail'], 'jpg')
979 thumb_filename = os.path.splitext(filename)[0] + '.' + thumb_format
0a9ce268 980 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(thumb_filename)):
6febd1c1 981 self.to_screen('[%s] %s: Thumbnail is already present' %
7b6fefc9
PH
982 (info_dict['extractor'], info_dict['id']))
983 else:
6febd1c1 984 self.to_screen('[%s] %s: Downloading thumbnail ...' %
7b6fefc9
PH
985 (info_dict['extractor'], info_dict['id']))
986 try:
e9c092f1 987 uf = self.urlopen(info_dict['thumbnail'])
7b6fefc9
PH
988 with open(thumb_filename, 'wb') as thumbf:
989 shutil.copyfileobj(uf, thumbf)
6febd1c1 990 self.to_screen('[%s] %s: Writing thumbnail to: %s' %
7b6fefc9
PH
991 (info_dict['extractor'], info_dict['id'], thumb_filename))
992 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
6febd1c1 993 self.report_warning('Unable to download thumbnail "%s": %s' %
7b6fefc9 994 (info_dict['thumbnail'], compat_str(err)))
8222d8de
JMF
995
996 if not self.params.get('skip_download', False):
997 if self.params.get('nooverwrites', False) and os.path.exists(encodeFilename(filename)):
998 success = True
999 else:
1000 try:
6350728b
JMF
1001 def dl(name, info):
1002 fd = get_suitable_downloader(info)(self, self.params)
1003 for ph in self._progress_hooks:
1004 fd.add_progress_hook(ph)
8d5797b0
PH
1005 if self.params.get('verbose'):
1006 self.to_stdout('[debug] Invoking downloader on %r' % info.get('url'))
6350728b
JMF
1007 return fd.download(name, info)
1008 if info_dict.get('requested_formats') is not None:
1009 downloaded = []
1010 success = True
b7f81164 1011 merger = FFmpegMergerPP(self, not self.params.get('keepvideo'))
58c3c7ae
JMF
1012 if not merger._get_executable():
1013 postprocessors = []
1014 self.report_warning('You have requested multiple '
1015 'formats but ffmpeg or avconv are not installed.'
1016 ' The formats won\'t be merged')
1017 else:
1018 postprocessors = [merger]
6350728b
JMF
1019 for f in info_dict['requested_formats']:
1020 new_info = dict(info_dict)
1021 new_info.update(f)
1022 fname = self.prepare_filename(new_info)
1023 fname = prepend_extension(fname, 'f%s' % f['format_id'])
1024 downloaded.append(fname)
1025 partial_success = dl(fname, new_info)
1026 success = success and partial_success
58c3c7ae 1027 info_dict['__postprocessors'] = postprocessors
6350728b
JMF
1028 info_dict['__files_to_merge'] = downloaded
1029 else:
1030 # Just a single file
1031 success = dl(filename, info_dict)
8222d8de 1032 except (compat_urllib_error.URLError, compat_http_client.HTTPException, socket.error) as err:
6febd1c1 1033 self.report_error('unable to download video data: %s' % str(err))
8222d8de 1034 return
c40c6aaa
JMF
1035 except (OSError, IOError) as err:
1036 raise UnavailableVideoError(err)
8222d8de 1037 except (ContentTooShortError, ) as err:
6febd1c1 1038 self.report_error('content too short (expected %s bytes and served %s)' % (err.expected, err.downloaded))
8222d8de
JMF
1039 return
1040
1041 if success:
1042 try:
1043 self.post_process(filename, info_dict)
1044 except (PostProcessingError) as err:
6febd1c1 1045 self.report_error('postprocessing: %s' % str(err))
8222d8de
JMF
1046 return
1047
c1c9a79c
PH
1048 self.record_download_archive(info_dict)
1049
8222d8de
JMF
1050 def download(self, url_list):
1051 """Download a given list of URLs."""
acd69589 1052 outtmpl = self.params.get('outtmpl', DEFAULT_OUTTMPL)
0c75c3fa 1053 if (len(url_list) > 1 and
acd69589 1054 '%' not in outtmpl
0c75c3fa 1055 and self.params.get('max_downloads') != 1):
acd69589 1056 raise SameFileError(outtmpl)
8222d8de
JMF
1057
1058 for url in url_list:
1059 try:
1060 #It also downloads the videos
dca08720 1061 self.extract_info(url)
8222d8de 1062 except UnavailableVideoError:
6febd1c1 1063 self.report_error('unable to download video')
8222d8de 1064 except MaxDownloadsReached:
6febd1c1 1065 self.to_screen('[info] Maximum number of downloaded files reached.')
8222d8de
JMF
1066 raise
1067
1068 return self._download_retcode
1069
1dcc4c0c 1070 def download_with_info_file(self, info_filename):
395293a8 1071 with io.open(info_filename, 'r', encoding='utf-8') as f:
1dcc4c0c 1072 info = json.load(f)
d4943898
JMF
1073 try:
1074 self.process_ie_result(info, download=True)
1075 except DownloadError:
1076 webpage_url = info.get('webpage_url')
1077 if webpage_url is not None:
6febd1c1 1078 self.report_warning('The info failed to download, trying with "%s"' % webpage_url)
d4943898
JMF
1079 return self.download([webpage_url])
1080 else:
1081 raise
1082 return self._download_retcode
1dcc4c0c 1083
8222d8de
JMF
1084 def post_process(self, filename, ie_info):
1085 """Run all the postprocessors on the given file."""
1086 info = dict(ie_info)
1087 info['filepath'] = filename
1088 keep_video = None
6350728b
JMF
1089 pps_chain = []
1090 if ie_info.get('__postprocessors') is not None:
1091 pps_chain.extend(ie_info['__postprocessors'])
1092 pps_chain.extend(self._pps)
1093 for pp in pps_chain:
8222d8de 1094 try:
fe7e0c98 1095 keep_video_wish, new_info = pp.run(info)
8222d8de
JMF
1096 if keep_video_wish is not None:
1097 if keep_video_wish:
1098 keep_video = keep_video_wish
1099 elif keep_video is None:
1100 # No clear decision yet, let IE decide
1101 keep_video = keep_video_wish
1102 except PostProcessingError as e:
bbcbf4d4 1103 self.report_error(e.msg)
8222d8de
JMF
1104 if keep_video is False and not self.params.get('keepvideo', False):
1105 try:
6febd1c1 1106 self.to_screen('Deleting original file %s (pass -k to keep)' % filename)
8222d8de
JMF
1107 os.remove(encodeFilename(filename))
1108 except (IOError, OSError):
6febd1c1 1109 self.report_warning('Unable to remove downloaded video file')
c1c9a79c 1110
5db07df6
PH
1111 def _make_archive_id(self, info_dict):
1112 # Future-proof against any change in case
1113 # and backwards compatibility with prior versions
d31209a1 1114 extractor = info_dict.get('extractor_key')
7012b23c
PH
1115 if extractor is None:
1116 if 'id' in info_dict:
1117 extractor = info_dict.get('ie_key') # key in a playlist
1118 if extractor is None:
5db07df6 1119 return None # Incomplete video information
6febd1c1 1120 return extractor.lower() + ' ' + info_dict['id']
5db07df6
PH
1121
1122 def in_download_archive(self, info_dict):
1123 fn = self.params.get('download_archive')
1124 if fn is None:
1125 return False
1126
1127 vid_id = self._make_archive_id(info_dict)
1128 if vid_id is None:
7012b23c 1129 return False # Incomplete video information
5db07df6 1130
c1c9a79c
PH
1131 try:
1132 with locked_file(fn, 'r', encoding='utf-8') as archive_file:
1133 for line in archive_file:
1134 if line.strip() == vid_id:
1135 return True
1136 except IOError as ioe:
1137 if ioe.errno != errno.ENOENT:
1138 raise
1139 return False
1140
1141 def record_download_archive(self, info_dict):
1142 fn = self.params.get('download_archive')
1143 if fn is None:
1144 return
5db07df6
PH
1145 vid_id = self._make_archive_id(info_dict)
1146 assert vid_id
c1c9a79c 1147 with locked_file(fn, 'a', encoding='utf-8') as archive_file:
6febd1c1 1148 archive_file.write(vid_id + '\n')
dd82ffea 1149
8c51aa65 1150 @staticmethod
8abeeb94 1151 def format_resolution(format, default='unknown'):
fb04e403
PH
1152 if format.get('vcodec') == 'none':
1153 return 'audio only'
f49d89ee
PH
1154 if format.get('resolution') is not None:
1155 return format['resolution']
8c51aa65
JMF
1156 if format.get('height') is not None:
1157 if format.get('width') is not None:
6febd1c1 1158 res = '%sx%s' % (format['width'], format['height'])
8c51aa65 1159 else:
6febd1c1 1160 res = '%sp' % format['height']
f49d89ee 1161 elif format.get('width') is not None:
6febd1c1 1162 res = '?x%d' % format['width']
8c51aa65 1163 else:
8abeeb94 1164 res = default
8c51aa65
JMF
1165 return res
1166
c57f7757
PH
1167 def _format_note(self, fdict):
1168 res = ''
1169 if fdict.get('ext') in ['f4f', 'f4m']:
1170 res += '(unsupported) '
1171 if fdict.get('format_note') is not None:
1172 res += fdict['format_note'] + ' '
1173 if fdict.get('tbr') is not None:
1174 res += '%4dk ' % fdict['tbr']
1175 if fdict.get('container') is not None:
1176 if res:
1177 res += ', '
1178 res += '%s container' % fdict['container']
1179 if (fdict.get('vcodec') is not None and
1180 fdict.get('vcodec') != 'none'):
1181 if res:
1182 res += ', '
1183 res += fdict['vcodec']
91c7271a 1184 if fdict.get('vbr') is not None:
c57f7757
PH
1185 res += '@'
1186 elif fdict.get('vbr') is not None and fdict.get('abr') is not None:
1187 res += 'video@'
1188 if fdict.get('vbr') is not None:
1189 res += '%4dk' % fdict['vbr']
1190 if fdict.get('acodec') is not None:
1191 if res:
1192 res += ', '
1193 if fdict['acodec'] == 'none':
1194 res += 'video only'
1195 else:
1196 res += '%-5s' % fdict['acodec']
1197 elif fdict.get('abr') is not None:
1198 if res:
1199 res += ', '
1200 res += 'audio'
1201 if fdict.get('abr') is not None:
1202 res += '@%3dk' % fdict['abr']
1203 if fdict.get('asr') is not None:
1204 res += ' (%5dHz)' % fdict['asr']
1205 if fdict.get('filesize') is not None:
1206 if res:
1207 res += ', '
1208 res += format_bytes(fdict['filesize'])
9732d77e
PH
1209 elif fdict.get('filesize_approx') is not None:
1210 if res:
1211 res += ', '
1212 res += '~' + format_bytes(fdict['filesize_approx'])
c57f7757 1213 return res
91c7271a 1214
c57f7757 1215 def list_formats(self, info_dict):
02dbf93f 1216 def line(format, idlen=20):
6febd1c1 1217 return (('%-' + compat_str(idlen + 1) + 's%-10s%-12s%s') % (
8c51aa65
JMF
1218 format['format_id'],
1219 format['ext'],
8c51aa65 1220 self.format_resolution(format),
c57f7757 1221 self._format_note(format),
02dbf93f 1222 ))
57dd9a8f 1223
94badb25 1224 formats = info_dict.get('formats', [info_dict])
6febd1c1 1225 idlen = max(len('format code'),
02dbf93f
PH
1226 max(len(f['format_id']) for f in formats))
1227 formats_s = [line(f, idlen) for f in formats]
94badb25 1228 if len(formats) > 1:
c57f7757
PH
1229 formats_s[0] += (' ' if self._format_note(formats[0]) else '') + '(worst)'
1230 formats_s[-1] += (' ' if self._format_note(formats[-1]) else '') + '(best)'
57dd9a8f
PH
1231
1232 header_line = line({
6febd1c1
PH
1233 'format_id': 'format code', 'ext': 'extension',
1234 'resolution': 'resolution', 'format_note': 'note'}, idlen=idlen)
1235 self.to_screen('[info] Available formats for %s:\n%s\n%s' %
1236 (info_dict['id'], header_line, '\n'.join(formats_s)))
dca08720
PH
1237
1238 def urlopen(self, req):
1239 """ Start an HTTP download """
19a41fc6 1240 return self._opener.open(req, timeout=self._socket_timeout)
dca08720
PH
1241
1242 def print_debug_header(self):
1243 if not self.params.get('verbose'):
1244 return
62fec3b2 1245
4192b51c
PH
1246 if type('') is not compat_str:
1247 # Python 2.6 on SLES11 SP1 (https://github.com/rg3/youtube-dl/issues/3326)
1248 self.report_warning(
1249 'Your Python is broken! Update to a newer and supported version')
1250
b0472057 1251 encoding_str = (
734f90bb
PH
1252 '[debug] Encodings: locale %s, fs %s, out %s, pref %s\n' % (
1253 locale.getpreferredencoding(),
1254 sys.getfilesystemencoding(),
1255 sys.stdout.encoding,
b0472057 1256 self.get_encoding()))
4192b51c 1257 write_string(encoding_str, encoding=None)
734f90bb
PH
1258
1259 self._write_string('[debug] youtube-dl version ' + __version__ + '\n')
dca08720
PH
1260 try:
1261 sp = subprocess.Popen(
1262 ['git', 'rev-parse', '--short', 'HEAD'],
1263 stdout=subprocess.PIPE, stderr=subprocess.PIPE,
1264 cwd=os.path.dirname(os.path.abspath(__file__)))
1265 out, err = sp.communicate()
1266 out = out.decode().strip()
1267 if re.match('[0-9a-f]+', out):
734f90bb 1268 self._write_string('[debug] Git HEAD: ' + out + '\n')
dca08720
PH
1269 except:
1270 try:
1271 sys.exc_clear()
1272 except:
1273 pass
734f90bb 1274 self._write_string('[debug] Python version %s - %s' %
6febd1c1 1275 (platform.python_version(), platform_name()) + '\n')
dca08720
PH
1276
1277 proxy_map = {}
1278 for handler in self._opener.handlers:
1279 if hasattr(handler, 'proxies'):
1280 proxy_map.update(handler.proxies)
734f90bb 1281 self._write_string('[debug] Proxy map: ' + compat_str(proxy_map) + '\n')
dca08720 1282
e344693b 1283 def _setup_opener(self):
6ad14cab 1284 timeout_val = self.params.get('socket_timeout')
19a41fc6 1285 self._socket_timeout = 600 if timeout_val is None else float(timeout_val)
6ad14cab 1286
dca08720
PH
1287 opts_cookiefile = self.params.get('cookiefile')
1288 opts_proxy = self.params.get('proxy')
1289
1290 if opts_cookiefile is None:
1291 self.cookiejar = compat_cookiejar.CookieJar()
1292 else:
1293 self.cookiejar = compat_cookiejar.MozillaCookieJar(
1294 opts_cookiefile)
1295 if os.access(opts_cookiefile, os.R_OK):
1296 self.cookiejar.load()
1297
1298 cookie_processor = compat_urllib_request.HTTPCookieProcessor(
1299 self.cookiejar)
1300 if opts_proxy is not None:
1301 if opts_proxy == '':
1302 proxies = {}
1303 else:
1304 proxies = {'http': opts_proxy, 'https': opts_proxy}
1305 else:
1306 proxies = compat_urllib_request.getproxies()
1307 # Set HTTPS proxy to HTTP one if given (https://github.com/rg3/youtube-dl/issues/805)
1308 if 'http' in proxies and 'https' not in proxies:
1309 proxies['https'] = proxies['http']
1310 proxy_handler = compat_urllib_request.ProxyHandler(proxies)
a0ddb8a2
PH
1311
1312 debuglevel = 1 if self.params.get('debug_printtraffic') else 0
dca08720 1313 https_handler = make_HTTPS_handler(
a0ddb8a2
PH
1314 self.params.get('nocheckcertificate', False), debuglevel=debuglevel)
1315 ydlh = YoutubeDLHandler(debuglevel=debuglevel)
dca08720 1316 opener = compat_urllib_request.build_opener(
a0ddb8a2 1317 https_handler, proxy_handler, cookie_processor, ydlh)
dca08720
PH
1318 # Delete the default user-agent header, which would otherwise apply in
1319 # cases where our custom HTTP handler doesn't come into play
1320 # (See https://github.com/rg3/youtube-dl/issues/1309 for details)
1321 opener.addheaders = []
1322 self._opener = opener
62fec3b2
PH
1323
1324 def encode(self, s):
1325 if isinstance(s, bytes):
1326 return s # Already encoded
1327
1328 try:
1329 return s.encode(self.get_encoding())
1330 except UnicodeEncodeError as err:
1331 err.reason = err.reason + '. Check your system encoding configuration or use the --encoding option.'
1332 raise
1333
1334 def get_encoding(self):
1335 encoding = self.params.get('encoding')
1336 if encoding is None:
1337 encoding = preferredencoding()
1338 return encoding