4 from __future__
import absolute_import
, unicode_literals
31 from string
import ascii_letters
35 compat_get_terminal_size
,
42 compat_tokenize_tokenize
,
44 compat_urllib_request
,
45 compat_urllib_request_DataHandler
,
46 windows_enable_vt_mode
,
48 from .cookies
import load_cookies
58 DOT_DESKTOP_LINK_TEMPLATE
,
59 DOT_URL_LINK_TEMPLATE
,
60 DOT_WEBLOC_LINK_TEMPLATE
,
88 PerRequestProxyHandler
,
93 process_communicate_or_kill
,
94 register_socks_protocols
,
109 supports_terminal_sequences
,
115 UnavailableVideoError
,
121 YoutubeDLCookieProcessor
,
123 YoutubeDLRedirectHandler
,
125 from .cache
import Cache
126 from .extractor
import (
127 gen_extractor_classes
,
130 _PLUGIN_CLASSES
as plugin_extractors
132 from .extractor
.openload
import PhantomJSwrapper
133 from .downloader
import (
135 get_suitable_downloader
,
136 shorten_protocol_name
138 from .downloader
.rtmp
import rtmpdump_version
139 from .postprocessor
import (
142 FFmpegFixupDurationPP
,
145 FFmpegFixupStretchedPP
,
146 FFmpegFixupTimestampPP
,
149 MoveFilesAfterDownloadPP
,
150 _PLUGIN_CLASSES
as plugin_postprocessors
152 from .update
import detect_variant
153 from .version
import __version__
155 if compat_os_name
== 'nt':
159 class YoutubeDL(object):
162 YoutubeDL objects are the ones responsible of downloading the
163 actual video file and writing it to disk if the user has requested
164 it, among some other tasks. In most cases there should be one per
165 program. As, given a video URL, the downloader doesn't know how to
166 extract all the needed information, task that InfoExtractors do, it
167 has to pass the URL to one of them.
169 For this, YoutubeDL objects have a method that allows
170 InfoExtractors to be registered in a given order. When it is passed
171 a URL, the YoutubeDL object handles it to the first InfoExtractor it
172 finds that reports being able to handle it. The InfoExtractor extracts
173 all the information about the video or videos the URL refers to, and
174 YoutubeDL process the extracted information, possibly using a File
175 Downloader to download the video.
177 YoutubeDL objects accept a lot of parameters. In order not to saturate
178 the object constructor with arguments, it receives a dictionary of
179 options instead. These options are available through the params
180 attribute for the InfoExtractors to use. The YoutubeDL also
181 registers itself as the downloader in charge for the InfoExtractors
182 that are added to it, so this is a "mutual registration".
186 username: Username for authentication purposes.
187 password: Password for authentication purposes.
188 videopassword: Password for accessing a video.
189 ap_mso: Adobe Pass multiple-system operator identifier.
190 ap_username: Multiple-system operator account username.
191 ap_password: Multiple-system operator account password.
192 usenetrc: Use netrc for authentication instead.
193 verbose: Print additional info to stdout.
194 quiet: Do not print messages to stdout.
195 no_warnings: Do not print out anything for warnings.
196 forceprint: A list of templates to force print
197 forceurl: Force printing final URL. (Deprecated)
198 forcetitle: Force printing title. (Deprecated)
199 forceid: Force printing ID. (Deprecated)
200 forcethumbnail: Force printing thumbnail URL. (Deprecated)
201 forcedescription: Force printing description. (Deprecated)
202 forcefilename: Force printing final filename. (Deprecated)
203 forceduration: Force printing duration. (Deprecated)
204 forcejson: Force printing info_dict as JSON.
205 dump_single_json: Force printing the info_dict of the whole playlist
206 (or video) as a single JSON line.
207 force_write_download_archive: Force writing download archive regardless
208 of 'skip_download' or 'simulate'.
209 simulate: Do not download the video files. If unset (or None),
210 simulate only if listsubtitles, listformats or list_thumbnails is used
211 format: Video format code. see "FORMAT SELECTION" for more details.
212 allow_unplayable_formats: Allow unplayable formats to be extracted and downloaded.
213 ignore_no_formats_error: Ignore "No video formats" error. Usefull for
214 extracting metadata even if the video is not actually
215 available for download (experimental)
216 format_sort: How to sort the video formats. see "Sorting Formats"
218 format_sort_force: Force the given format_sort. see "Sorting Formats"
220 allow_multiple_video_streams: Allow multiple video streams to be merged
222 allow_multiple_audio_streams: Allow multiple audio streams to be merged
224 check_formats Whether to test if the formats are downloadable.
225 Can be True (check all), False (check none)
226 or None (check only if requested by extractor)
227 paths: Dictionary of output paths. The allowed keys are 'home'
228 'temp' and the keys of OUTTMPL_TYPES (in utils.py)
229 outtmpl: Dictionary of templates for output names. Allowed keys
230 are 'default' and the keys of OUTTMPL_TYPES (in utils.py).
231 For compatibility with youtube-dl, a single string can also be used
232 outtmpl_na_placeholder: Placeholder for unavailable meta fields.
233 restrictfilenames: Do not allow "&" and spaces in file names
234 trim_file_name: Limit length of filename (extension excluded)
235 windowsfilenames: Force the filenames to be windows compatible
236 ignoreerrors: Do not stop on download/postprocessing errors.
237 Can be 'only_download' to ignore only download errors.
238 Default is 'only_download' for CLI, but False for API
239 skip_playlist_after_errors: Number of allowed failures until the rest of
240 the playlist is skipped
241 force_generic_extractor: Force downloader to use the generic extractor
242 overwrites: Overwrite all video and metadata files if True,
243 overwrite only non-video files if None
244 and don't overwrite any file if False
245 For compatibility with youtube-dl,
246 "nooverwrites" may also be used instead
247 playliststart: Playlist item to start at.
248 playlistend: Playlist item to end at.
249 playlist_items: Specific indices of playlist to download.
250 playlistreverse: Download playlist items in reverse order.
251 playlistrandom: Download playlist items in random order.
252 matchtitle: Download only matching titles.
253 rejecttitle: Reject downloads for matching titles.
254 logger: Log messages to a logging.Logger instance.
255 logtostderr: Log messages to stderr instead of stdout.
256 consoletitle: Display progress in console window's titlebar.
257 writedescription: Write the video description to a .description file
258 writeinfojson: Write the video description to a .info.json file
259 clean_infojson: Remove private fields from the infojson
260 getcomments: Extract video comments. This will not be written to disk
261 unless writeinfojson is also given
262 writeannotations: Write the video annotations to a .annotations.xml file
263 writethumbnail: Write the thumbnail image to a file
264 allow_playlist_files: Whether to write playlists' description, infojson etc
265 also to disk when using the 'write*' options
266 write_all_thumbnails: Write all thumbnail formats to files
267 writelink: Write an internet shortcut file, depending on the
268 current platform (.url/.webloc/.desktop)
269 writeurllink: Write a Windows internet shortcut file (.url)
270 writewebloclink: Write a macOS internet shortcut file (.webloc)
271 writedesktoplink: Write a Linux internet shortcut file (.desktop)
272 writesubtitles: Write the video subtitles to a file
273 writeautomaticsub: Write the automatically generated subtitles to a file
274 allsubtitles: Deprecated - Use subtitleslangs = ['all']
275 Downloads all the subtitles of the video
276 (requires writesubtitles or writeautomaticsub)
277 listsubtitles: Lists all available subtitles for the video
278 subtitlesformat: The format code for subtitles
279 subtitleslangs: List of languages of the subtitles to download (can be regex).
280 The list may contain "all" to refer to all the available
281 subtitles. The language can be prefixed with a "-" to
282 exclude it from the requested languages. Eg: ['all', '-live_chat']
283 keepvideo: Keep the video file after post-processing
284 daterange: A DateRange object, download only if the upload_date is in the range.
285 skip_download: Skip the actual download of the video file
286 cachedir: Location of the cache files in the filesystem.
287 False to disable filesystem cache.
288 noplaylist: Download single video instead of a playlist if in doubt.
289 age_limit: An integer representing the user's age in years.
290 Unsuitable videos for the given age are skipped.
291 min_views: An integer representing the minimum view count the video
292 must have in order to not be skipped.
293 Videos without view count information are always
294 downloaded. None for no limit.
295 max_views: An integer representing the maximum view count.
296 Videos that are more popular than that are not
298 Videos without view count information are always
299 downloaded. None for no limit.
300 download_archive: File name of a file where all downloads are recorded.
301 Videos already present in the file are not downloaded
303 break_on_existing: Stop the download process after attempting to download a
304 file that is in the archive.
305 break_on_reject: Stop the download process when encountering a video that
306 has been filtered out.
307 cookiefile: File name where cookies should be read from and dumped to
308 cookiesfrombrowser: A tuple containing the name of the browser and the profile
309 name/path from where cookies are loaded.
310 Eg: ('chrome', ) or (vivaldi, 'default')
311 nocheckcertificate:Do not verify SSL certificates
312 prefer_insecure: Use HTTP instead of HTTPS to retrieve information.
313 At the moment, this is only supported by YouTube.
314 proxy: URL of the proxy server to use
315 geo_verification_proxy: URL of the proxy to use for IP address verification
316 on geo-restricted sites.
317 socket_timeout: Time to wait for unresponsive hosts, in seconds
318 bidi_workaround: Work around buggy terminals without bidirectional text
319 support, using fridibi
320 debug_printtraffic:Print out sent and received HTTP traffic
321 include_ads: Download ads as well
322 default_search: Prepend this string if an input url is not valid.
323 'auto' for elaborate guessing
324 encoding: Use this encoding instead of the system-specified.
325 extract_flat: Do not resolve URLs, return the immediate result.
326 Pass in 'in_playlist' to only show this behavior for
328 postprocessors: A list of dictionaries, each with an entry
329 * key: The name of the postprocessor. See
330 yt_dlp/postprocessor/__init__.py for a list.
331 * when: When to run the postprocessor. Can be one of
332 pre_process|before_dl|post_process|after_move.
333 Assumed to be 'post_process' if not given
334 post_hooks: Deprecated - Register a custom postprocessor instead
335 A list of functions that get called as the final step
336 for each video file, after all postprocessors have been
337 called. The filename will be passed as the only argument.
338 progress_hooks: A list of functions that get called on download
339 progress, with a dictionary with the entries
340 * status: One of "downloading", "error", or "finished".
341 Check this first and ignore unknown values.
342 * info_dict: The extracted info_dict
344 If status is one of "downloading", or "finished", the
345 following properties may also be present:
346 * filename: The final filename (always present)
347 * tmpfilename: The filename we're currently writing to
348 * downloaded_bytes: Bytes on disk
349 * total_bytes: Size of the whole file, None if unknown
350 * total_bytes_estimate: Guess of the eventual file size,
352 * elapsed: The number of seconds since download started.
353 * eta: The estimated time in seconds, None if unknown
354 * speed: The download speed in bytes/second, None if
356 * fragment_index: The counter of the currently
357 downloaded video fragment.
358 * fragment_count: The number of fragments (= individual
359 files that will be merged)
361 Progress hooks are guaranteed to be called at least once
362 (with status "finished") if the download is successful.
363 postprocessor_hooks: A list of functions that get called on postprocessing
364 progress, with a dictionary with the entries
365 * status: One of "started", "processing", or "finished".
366 Check this first and ignore unknown values.
367 * postprocessor: Name of the postprocessor
368 * info_dict: The extracted info_dict
370 Progress hooks are guaranteed to be called at least twice
371 (with status "started" and "finished") if the processing is successful.
372 merge_output_format: Extension to use when merging formats.
373 final_ext: Expected final extension; used to detect when the file was
374 already downloaded and converted. "merge_output_format" is
375 replaced by this extension when given
376 fixup: Automatically correct known faults of the file.
378 - "never": do nothing
379 - "warn": only emit a warning
380 - "detect_or_warn": check whether we can do anything
381 about it, warn otherwise (default)
382 source_address: Client-side IP address to bind to.
383 call_home: Boolean, true iff we are allowed to contact the
384 yt-dlp servers for debugging. (BROKEN)
385 sleep_interval_requests: Number of seconds to sleep between requests
387 sleep_interval: Number of seconds to sleep before each download when
388 used alone or a lower bound of a range for randomized
389 sleep before each download (minimum possible number
390 of seconds to sleep) when used along with
392 max_sleep_interval:Upper bound of a range for randomized sleep before each
393 download (maximum possible number of seconds to sleep).
394 Must only be used along with sleep_interval.
395 Actual sleep time will be a random float from range
396 [sleep_interval; max_sleep_interval].
397 sleep_interval_subtitles: Number of seconds to sleep before each subtitle download
398 listformats: Print an overview of available video formats and exit.
399 list_thumbnails: Print a table of all thumbnails and exit.
400 match_filter: A function that gets called with the info_dict of
402 If it returns a message, the video is ignored.
403 If it returns None, the video is downloaded.
404 match_filter_func in utils.py is one example for this.
405 no_color: Do not emit color codes in output.
406 geo_bypass: Bypass geographic restriction via faking X-Forwarded-For
409 Two-letter ISO 3166-2 country code that will be used for
410 explicit geographic restriction bypassing via faking
411 X-Forwarded-For HTTP header
413 IP range in CIDR notation that will be used similarly to
416 The following options determine which downloader is picked:
417 external_downloader: A dictionary of protocol keys and the executable of the
418 external downloader to use for it. The allowed protocols
419 are default|http|ftp|m3u8|dash|rtsp|rtmp|mms.
420 Set the value to 'native' to use the native downloader
421 hls_prefer_native: Deprecated - Use external_downloader = {'m3u8': 'native'}
422 or {'m3u8': 'ffmpeg'} instead.
423 Use the native HLS downloader instead of ffmpeg/avconv
424 if True, otherwise use ffmpeg/avconv if False, otherwise
425 use downloader suggested by extractor if None.
426 compat_opts: Compatibility options. See "Differences in default behavior".
427 The following options do not work when used through the API:
428 filename, abort-on-error, multistreams, no-live-chat, format-sort
429 no-clean-infojson, no-playlist-metafiles, no-keep-subs.
430 Refer __init__.py for their implementation
431 progress_template: Dictionary of templates for progress outputs.
432 Allowed keys are 'download', 'postprocess',
433 'download-title' (console title) and 'postprocess-title'.
434 The template is mapped on a dictionary with keys 'progress' and 'info'
436 The following parameters are not used by YoutubeDL itself, they are used by
437 the downloader (see yt_dlp/downloader/common.py):
438 nopart, updatetime, buffersize, ratelimit, throttledratelimit, min_filesize,
439 max_filesize, test, noresizebuffer, retries, fragment_retries, continuedl,
440 noprogress, xattr_set_filesize, hls_use_mpegts, http_chunk_size,
441 external_downloader_args.
443 The following options are used by the post processors:
444 prefer_ffmpeg: If False, use avconv instead of ffmpeg if both are available,
445 otherwise prefer ffmpeg. (avconv support is deprecated)
446 ffmpeg_location: Location of the ffmpeg/avconv binary; either the path
447 to the binary or its containing directory.
448 postprocessor_args: A dictionary of postprocessor/executable keys (in lower case)
449 and a list of additional command-line arguments for the
450 postprocessor/executable. The dict can also have "PP+EXE" keys
451 which are used when the given exe is used by the given PP.
452 Use 'default' as the name for arguments to passed to all PP
453 For compatibility with youtube-dl, a single list of args
456 The following options are used by the extractors:
457 extractor_retries: Number of times to retry for known errors
458 dynamic_mpd: Whether to process dynamic DASH manifests (default: True)
459 hls_split_discontinuity: Split HLS playlists to different formats at
460 discontinuities such as ad breaks (default: False)
461 extractor_args: A dictionary of arguments to be passed to the extractors.
462 See "EXTRACTOR ARGUMENTS" for details.
463 Eg: {'youtube': {'skip': ['dash', 'hls']}}
464 youtube_include_dash_manifest: Deprecated - Use extractor_args instead.
465 If True (default), DASH manifests and related
466 data will be downloaded and processed by extractor.
467 You can reduce network I/O by disabling it if you don't
468 care about DASH. (only for youtube)
469 youtube_include_hls_manifest: Deprecated - Use extractor_args instead.
470 If True (default), HLS manifests and related
471 data will be downloaded and processed by extractor.
472 You can reduce network I/O by disabling it if you don't
473 care about HLS. (only for youtube)
476 _NUMERIC_FIELDS
= set((
477 'width', 'height', 'tbr', 'abr', 'asr', 'vbr', 'fps', 'filesize', 'filesize_approx',
478 'timestamp', 'release_timestamp',
479 'duration', 'view_count', 'like_count', 'dislike_count', 'repost_count',
480 'average_rating', 'comment_count', 'age_limit',
481 'start_time', 'end_time',
482 'chapter_number', 'season_number', 'episode_number',
483 'track_number', 'disc_number', 'release_year',
488 _pps
= {'pre_process': [], 'before_dl': [], 'after_move': [], 'post_process': []}
489 _printed_messages
= set()
490 _first_webpage_request
= True
491 _download_retcode
= None
492 _num_downloads
= None
494 _playlist_urls
= set()
497 def __init__(self
, params
=None, auto_init
=True):
498 """Create a FileDownloader object with the given options."""
502 self
._ies
_instances
= {}
503 self
._pps
= {'pre_process': [], 'before_dl': [], 'after_move': [], 'post_process': []}
504 self
._printed
_messages
= set()
505 self
._first
_webpage
_request
= True
506 self
._post
_hooks
= []
507 self
._progress
_hooks
= []
508 self
._postprocessor
_hooks
= []
509 self
._download
_retcode
= 0
510 self
._num
_downloads
= 0
511 self
._screen
_file
= [sys
.stdout
, sys
.stderr
][params
.get('logtostderr', False)]
512 self
._err
_file
= sys
.stderr
514 self
.cache
= Cache(self
)
516 windows_enable_vt_mode()
517 # FIXME: This will break if we ever print color to stdout
518 self
.params
['no_color'] = self
.params
.get('no_color') or not supports_terminal_sequences(self
._err
_file
)
520 if sys
.version_info
< (3, 6):
522 'Python version %d.%d is not supported! Please update to Python 3.6 or above' % sys
.version_info
[:2])
524 if self
.params
.get('allow_unplayable_formats'):
526 f
'You have asked for {self._color_text("unplayable formats", "blue")} to be listed/downloaded. '
527 'This is a developer option intended for debugging. \n'
528 ' If you experience any issues while using this option, '
529 f
'{self._color_text("DO NOT", "red")} open a bug report')
531 def check_deprecated(param
, option
, suggestion
):
532 if self
.params
.get(param
) is not None:
533 self
.report_warning('%s is deprecated. Use %s instead' % (option
, suggestion
))
537 if check_deprecated('cn_verification_proxy', '--cn-verification-proxy', '--geo-verification-proxy'):
538 if self
.params
.get('geo_verification_proxy') is None:
539 self
.params
['geo_verification_proxy'] = self
.params
['cn_verification_proxy']
541 check_deprecated('autonumber', '--auto-number', '-o "%(autonumber)s-%(title)s.%(ext)s"')
542 check_deprecated('usetitle', '--title', '-o "%(title)s-%(id)s.%(ext)s"')
543 check_deprecated('useid', '--id', '-o "%(id)s.%(ext)s"')
545 for msg
in self
.params
.get('warnings', []):
546 self
.report_warning(msg
)
548 if 'overwrites' not in self
.params
and self
.params
.get('nooverwrites') is not None:
549 # nooverwrites was unnecessarily changed to overwrites
550 # in 0c3d0f51778b153f65c21906031c2e091fcfb641
551 # This ensures compatibility with both keys
552 self
.params
['overwrites'] = not self
.params
['nooverwrites']
553 elif self
.params
.get('overwrites') is None:
554 self
.params
.pop('overwrites', None)
556 self
.params
['nooverwrites'] = not self
.params
['overwrites']
558 if params
.get('bidi_workaround', False):
561 master
, slave
= pty
.openpty()
562 width
= compat_get_terminal_size().columns
566 width_args
= ['-w', str(width
)]
568 stdin
=subprocess
.PIPE
,
570 stderr
=self
._err
_file
)
572 self
._output
_process
= subprocess
.Popen(
573 ['bidiv'] + width_args
, **sp_kwargs
576 self
._output
_process
= subprocess
.Popen(
577 ['fribidi', '-c', 'UTF-8'] + width_args
, **sp_kwargs
)
578 self
._output
_channel
= os
.fdopen(master
, 'rb')
579 except OSError as ose
:
580 if ose
.errno
== errno
.ENOENT
:
581 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.')
585 if (sys
.platform
!= 'win32'
586 and sys
.getfilesystemencoding() in ['ascii', 'ANSI_X3.4-1968']
587 and not params
.get('restrictfilenames', False)):
588 # Unicode filesystem API will throw errors (#1474, #13027)
590 'Assuming --restrict-filenames since file system encoding '
591 'cannot encode all characters. '
592 'Set the LC_ALL environment variable to fix this.')
593 self
.params
['restrictfilenames'] = True
595 self
.outtmpl_dict
= self
.parse_outtmpl()
597 # Creating format selector here allows us to catch syntax errors before the extraction
598 self
.format_selector
= (
599 None if self
.params
.get('format') is None
600 else self
.build_format_selector(self
.params
['format']))
605 self
.print_debug_header()
606 self
.add_default_info_extractors()
608 for pp_def_raw
in self
.params
.get('postprocessors', []):
609 pp_def
= dict(pp_def_raw
)
610 when
= pp_def
.pop('when', 'post_process')
611 pp_class
= get_postprocessor(pp_def
.pop('key'))
612 pp
= pp_class(self
, **compat_kwargs(pp_def
))
613 self
.add_post_processor(pp
, when
=when
)
615 for ph
in self
.params
.get('post_hooks', []):
616 self
.add_post_hook(ph
)
618 for ph
in self
.params
.get('progress_hooks', []):
619 self
.add_progress_hook(ph
)
621 register_socks_protocols()
623 def preload_download_archive(fn
):
624 """Preload the archive, if any is specified"""
627 self
.write_debug('Loading archive file %r\n' % fn
)
629 with locked_file(fn
, 'r', encoding
='utf-8') as archive_file
:
630 for line
in archive_file
:
631 self
.archive
.add(line
.strip())
632 except IOError as ioe
:
633 if ioe
.errno
!= errno
.ENOENT
:
639 preload_download_archive(self
.params
.get('download_archive'))
641 def warn_if_short_id(self
, argv
):
642 # short YouTube ID starting with dash?
644 i
for i
, a
in enumerate(argv
)
645 if re
.match(r
'^-[0-9A-Za-z_-]{10}$', a
)]
649 + [a
for i
, a
in enumerate(argv
) if i
not in idxs
]
650 + ['--'] + [argv
[i
] for i
in idxs
]
653 'Long argument string detected. '
654 'Use -- to separate parameters and URLs, like this:\n%s\n' %
655 args_to_str(correct_argv
))
657 def add_info_extractor(self
, ie
):
658 """Add an InfoExtractor object to the end of the list."""
660 self
._ies
[ie_key
] = ie
661 if not isinstance(ie
, type):
662 self
._ies
_instances
[ie_key
] = ie
663 ie
.set_downloader(self
)
665 def _get_info_extractor_class(self
, ie_key
):
666 ie
= self
._ies
.get(ie_key
)
668 ie
= get_info_extractor(ie_key
)
669 self
.add_info_extractor(ie
)
672 def get_info_extractor(self
, ie_key
):
674 Get an instance of an IE with name ie_key, it will try to get one from
675 the _ies list, if there's no instance it will create a new one and add
676 it to the extractor list.
678 ie
= self
._ies
_instances
.get(ie_key
)
680 ie
= get_info_extractor(ie_key
)()
681 self
.add_info_extractor(ie
)
684 def add_default_info_extractors(self
):
686 Add the InfoExtractors returned by gen_extractors to the end of the list
688 for ie
in gen_extractor_classes():
689 self
.add_info_extractor(ie
)
691 def add_post_processor(self
, pp
, when
='post_process'):
692 """Add a PostProcessor object to the end of the chain."""
693 self
._pps
[when
].append(pp
)
694 pp
.set_downloader(self
)
696 def add_post_hook(self
, ph
):
697 """Add the post hook"""
698 self
._post
_hooks
.append(ph
)
700 def add_progress_hook(self
, ph
):
701 """Add the download progress hook"""
702 self
._progress
_hooks
.append(ph
)
704 def add_postprocessor_hook(self
, ph
):
705 """Add the postprocessing progress hook"""
706 self
._postprocessor
_hooks
.append(ph
)
708 def _bidi_workaround(self
, message
):
709 if not hasattr(self
, '_output_channel'):
712 assert hasattr(self
, '_output_process')
713 assert isinstance(message
, compat_str
)
714 line_count
= message
.count('\n') + 1
715 self
._output
_process
.stdin
.write((message
+ '\n').encode('utf-8'))
716 self
._output
_process
.stdin
.flush()
717 res
= ''.join(self
._output
_channel
.readline().decode('utf-8')
718 for _
in range(line_count
))
719 return res
[:-len('\n')]
721 def _write_string(self
, message
, out
=None, only_once
=False):
723 if message
in self
._printed
_messages
:
725 self
._printed
_messages
.add(message
)
726 write_string(message
, out
=out
, encoding
=self
.params
.get('encoding'))
728 def to_stdout(self
, message
, skip_eol
=False, quiet
=False):
729 """Print message to stdout"""
730 if self
.params
.get('logger'):
731 self
.params
['logger'].debug(message
)
732 elif not quiet
or self
.params
.get('verbose'):
734 '%s%s' % (self
._bidi
_workaround
(message
), ('' if skip_eol
else '\n')),
735 self
._err
_file
if quiet
else self
._screen
_file
)
737 def to_stderr(self
, message
, only_once
=False):
738 """Print message to stderr"""
739 assert isinstance(message
, compat_str
)
740 if self
.params
.get('logger'):
741 self
.params
['logger'].error(message
)
743 self
._write
_string
('%s\n' % self
._bidi
_workaround
(message
), self
._err
_file
, only_once
=only_once
)
745 def to_console_title(self
, message
):
746 if not self
.params
.get('consoletitle', False):
748 if compat_os_name
== 'nt':
749 if ctypes
.windll
.kernel32
.GetConsoleWindow():
750 # c_wchar_p() might not be necessary if `message` is
751 # already of type unicode()
752 ctypes
.windll
.kernel32
.SetConsoleTitleW(ctypes
.c_wchar_p(message
))
753 elif 'TERM' in os
.environ
:
754 self
._write
_string
('\033]0;%s\007' % message
, self
._screen
_file
)
756 def save_console_title(self
):
757 if not self
.params
.get('consoletitle', False):
759 if self
.params
.get('simulate'):
761 if compat_os_name
!= 'nt' and 'TERM' in os
.environ
:
762 # Save the title on stack
763 self
._write
_string
('\033[22;0t', self
._screen
_file
)
765 def restore_console_title(self
):
766 if not self
.params
.get('consoletitle', False):
768 if self
.params
.get('simulate'):
770 if compat_os_name
!= 'nt' and 'TERM' in os
.environ
:
771 # Restore the title from stack
772 self
._write
_string
('\033[23;0t', self
._screen
_file
)
775 self
.save_console_title()
778 def __exit__(self
, *args
):
779 self
.restore_console_title()
781 if self
.params
.get('cookiefile') is not None:
782 self
.cookiejar
.save(ignore_discard
=True, ignore_expires
=True)
784 def trouble(self
, message
=None, tb
=None):
785 """Determine action to take when a download problem appears.
787 Depending on if the downloader has been configured to ignore
788 download errors or not, this method may throw an exception or
789 not when errors are found, after printing the message.
791 tb, if given, is additional traceback information.
793 if message
is not None:
794 self
.to_stderr(message
)
795 if self
.params
.get('verbose'):
797 if sys
.exc_info()[0]: # if .trouble has been called from an except block
799 if hasattr(sys
.exc_info()[1], 'exc_info') and sys
.exc_info()[1].exc_info
[0]:
800 tb
+= ''.join(traceback
.format_exception(*sys
.exc_info()[1].exc_info
))
801 tb
+= encode_compat_str(traceback
.format_exc())
803 tb_data
= traceback
.format_list(traceback
.extract_stack())
804 tb
= ''.join(tb_data
)
807 if not self
.params
.get('ignoreerrors'):
808 if sys
.exc_info()[0] and hasattr(sys
.exc_info()[1], 'exc_info') and sys
.exc_info()[1].exc_info
[0]:
809 exc_info
= sys
.exc_info()[1].exc_info
811 exc_info
= sys
.exc_info()
812 raise DownloadError(message
, exc_info
)
813 self
._download
_retcode
= 1
815 def to_screen(self
, message
, skip_eol
=False):
816 """Print message to stdout if not in quiet mode"""
818 message
, skip_eol
, quiet
=self
.params
.get('quiet', False))
820 def _color_text(self
, text
, color
):
821 if self
.params
.get('no_color'):
823 return f
'{TERMINAL_SEQUENCES[color.upper()]}{text}{TERMINAL_SEQUENCES["RESET_STYLE"]}'
825 def report_warning(self
, message
, only_once
=False):
827 Print the message to stderr, it will be prefixed with 'WARNING:'
828 If stderr is a tty file the 'WARNING:' will be colored
830 if self
.params
.get('logger') is not None:
831 self
.params
['logger'].warning(message
)
833 if self
.params
.get('no_warnings'):
835 self
.to_stderr(f
'{self._color_text("WARNING:", "yellow")} {message}', only_once
)
837 def report_error(self
, message
, tb
=None):
839 Do the same as trouble, but prefixes the message with 'ERROR:', colored
840 in red if stderr is a tty file.
842 self
.trouble(f
'{self._color_text("ERROR:", "red")} {message}', tb
)
844 def write_debug(self
, message
, only_once
=False):
845 '''Log debug message or Print message to stderr'''
846 if not self
.params
.get('verbose', False):
848 message
= '[debug] %s' % message
849 if self
.params
.get('logger'):
850 self
.params
['logger'].debug(message
)
852 self
.to_stderr(message
, only_once
)
854 def report_file_already_downloaded(self
, file_name
):
855 """Report file has already been fully downloaded."""
857 self
.to_screen('[download] %s has already been downloaded' % file_name
)
858 except UnicodeEncodeError:
859 self
.to_screen('[download] The file has already been downloaded')
861 def report_file_delete(self
, file_name
):
862 """Report that existing file will be deleted."""
864 self
.to_screen('Deleting existing file %s' % file_name
)
865 except UnicodeEncodeError:
866 self
.to_screen('Deleting existing file')
868 def raise_no_formats(self
, info
, forced
=False):
869 has_drm
= info
.get('__has_drm')
870 msg
= 'This video is DRM protected' if has_drm
else 'No video formats found!'
871 expected
= self
.params
.get('ignore_no_formats_error')
872 if forced
or not expected
:
873 raise ExtractorError(msg
, video_id
=info
['id'], ie
=info
['extractor'],
874 expected
=has_drm
or expected
)
876 self
.report_warning(msg
)
878 def parse_outtmpl(self
):
879 outtmpl_dict
= self
.params
.get('outtmpl', {})
880 if not isinstance(outtmpl_dict
, dict):
881 outtmpl_dict
= {'default': outtmpl_dict}
882 outtmpl_dict
.update({
883 k
: v
for k
, v
in DEFAULT_OUTTMPL
.items()
884 if outtmpl_dict
.get(k
) is None})
885 for key
, val
in outtmpl_dict
.items():
886 if isinstance(val
, bytes):
888 'Parameter outtmpl is bytes, but should be a unicode string. '
889 'Put from __future__ import unicode_literals at the top of your code file or consider switching to Python 3.x.')
892 def get_output_path(self
, dir_type
='', filename
=None):
893 paths
= self
.params
.get('paths', {})
894 assert isinstance(paths
, dict)
896 expand_path(paths
.get('home', '').strip()),
897 expand_path(paths
.get(dir_type
, '').strip()) if dir_type
else '',
900 # Temporary fix for #4787
901 # 'Treat' all problem characters by passing filename through preferredencoding
902 # to workaround encoding issues with subprocess on python2 @ Windows
903 if sys
.version_info
< (3, 0) and sys
.platform
== 'win32':
904 path
= encodeFilename(path
, True).decode(preferredencoding())
905 return sanitize_path(path
, force
=self
.params
.get('windowsfilenames'))
908 def _outtmpl_expandpath(outtmpl
):
909 # expand_path translates '%%' into '%' and '$$' into '$'
910 # correspondingly that is not what we want since we need to keep
911 # '%%' intact for template dict substitution step. Working around
912 # with boundary-alike separator hack.
913 sep
= ''.join([random
.choice(ascii_letters
) for _
in range(32)])
914 outtmpl
= outtmpl
.replace('%%', '%{0}%'.format(sep
)).replace('$$', '${0}$'.format(sep
))
916 # outtmpl should be expand_path'ed before template dict substitution
917 # because meta fields may contain env variables we don't want to
918 # be expanded. For example, for outtmpl "%(title)s.%(ext)s" and
919 # title "Hello $PATH", we don't want `$PATH` to be expanded.
920 return expand_path(outtmpl
).replace(sep
, '')
923 def escape_outtmpl(outtmpl
):
924 ''' Escape any remaining strings like %s, %abc% etc. '''
926 STR_FORMAT_RE_TMPL
.format('', '(?![%(\0])'),
927 lambda mobj
: ('' if mobj
.group('has_key') else '%') + mobj
.group(0),
931 def validate_outtmpl(cls
, outtmpl
):
932 ''' @return None or Exception object '''
934 STR_FORMAT_RE_TMPL
.format('[^)]*', '[ljqBU]'),
935 lambda mobj
: f
'{mobj.group(0)[:-1]}s',
936 cls
._outtmpl
_expandpath
(outtmpl
))
938 cls
.escape_outtmpl(outtmpl
) % collections
.defaultdict(int)
940 except ValueError as err
:
943 def prepare_outtmpl(self
, outtmpl
, info_dict
, sanitize
=None):
944 """ Make the outtmpl and info_dict suitable for substitution: ydl.escape_outtmpl(outtmpl) % info_dict """
945 info_dict
.setdefault('epoch', int(time
.time())) # keep epoch consistent once set
947 info_dict
= dict(info_dict
) # Do not sanitize so as not to consume LazyList
948 for key
in ('__original_infodict', '__postprocessors'):
949 info_dict
.pop(key
, None)
950 info_dict
['duration_string'] = ( # %(duration>%H-%M-%S)s is wrong if duration > 24hrs
951 formatSeconds(info_dict
['duration'], '-' if sanitize
else ':')
952 if info_dict
.get('duration', None) is not None
954 info_dict
['autonumber'] = self
.params
.get('autonumber_start', 1) - 1 + self
._num
_downloads
955 if info_dict
.get('resolution') is None:
956 info_dict
['resolution'] = self
.format_resolution(info_dict
, default
=None)
958 # For fields playlist_index, playlist_autonumber and autonumber convert all occurrences
959 # of %(field)s to %(field)0Nd for backward compatibility
960 field_size_compat_map
= {
961 'playlist_index': len(str(info_dict
.get('_last_playlist_index') or '')),
962 'playlist_autonumber': len(str(info_dict
.get('n_entries') or '')),
963 'autonumber': self
.params
.get('autonumber_size') or 5,
967 EXTERNAL_FORMAT_RE
= re
.compile(STR_FORMAT_RE_TMPL
.format('[^)]*', f
'[{STR_FORMAT_TYPES}ljqBU]'))
972 # Field is of the form key1.key2...
973 # where keys (except first) can be string, int or slice
974 FIELD_RE
= r
'\w*(?:\.(?:\w+|{num}|{num}?(?::{num}?){{1,2}}))*'.format(num
=r
'(?:-?\d+)')
975 MATH_FIELD_RE
= r
'''{field}|{num}'''.format(field
=FIELD_RE
, num
=r
'-?\d+(?:.\d+)?')
976 MATH_OPERATORS_RE
= r
'(?:%s)' % '|'.join(map(re
.escape
, MATH_FUNCTIONS
.keys()))
977 INTERNAL_FORMAT_RE
= re
.compile(r
'''(?x)
980 (?P<maths>(?:{math_op}{math_field})*)
981 (?:>(?P<strf_format>.+?))?
982 (?P<alternate>(?<!\\),[^|)]+)?
983 (?:\|(?P<default>.*?))?
984 $'''.format(field
=FIELD_RE
, math_op
=MATH_OPERATORS_RE
, math_field
=MATH_FIELD_RE
))
986 def _traverse_infodict(k
):
990 return traverse_obj(info_dict
, k
, is_user_input
=True, traverse_string
=True)
992 def get_value(mdict
):
994 value
= _traverse_infodict(mdict
['fields'])
997 value
= float_or_none(value
)
998 if value
is not None:
1001 offset_key
= mdict
['maths']
1003 value
= float_or_none(value
)
1007 MATH_FIELD_RE
if operator
else MATH_OPERATORS_RE
,
1008 offset_key
).group(0)
1009 offset_key
= offset_key
[len(item
):]
1010 if operator
is None:
1011 operator
= MATH_FUNCTIONS
[item
]
1013 item
, multiplier
= (item
[1:], -1) if item
[0] == '-' else (item
, 1)
1014 offset
= float_or_none(item
)
1016 offset
= float_or_none(_traverse_infodict(item
))
1018 value
= operator(value
, multiplier
* offset
)
1019 except (TypeError, ZeroDivisionError):
1022 # Datetime formatting
1023 if mdict
['strf_format']:
1024 value
= strftime_or_none(value
, mdict
['strf_format'].replace('\\,', ','))
1028 na
= self
.params
.get('outtmpl_na_placeholder', 'NA')
1030 def _dumpjson_default(obj
):
1031 if isinstance(obj
, (set, LazyList
)):
1033 raise TypeError(f
'Object of type {type(obj).__name__} is not JSON serializable')
1035 def create_key(outer_mobj
):
1036 if not outer_mobj
.group('has_key'):
1037 return f
'%{outer_mobj.group(0)}'
1038 key
= outer_mobj
.group('key')
1039 mobj
= re
.match(INTERNAL_FORMAT_RE
, key
)
1040 initial_field
= mobj
.group('fields').split('.')[-1] if mobj
else ''
1041 value
, default
= None, na
1043 mobj
= mobj
.groupdict()
1044 default
= mobj
['default'] if mobj
['default'] is not None else default
1045 value
= get_value(mobj
)
1046 if value
is None and mobj
['alternate']:
1047 mobj
= re
.match(INTERNAL_FORMAT_RE
, mobj
['alternate'][1:])
1051 fmt
= outer_mobj
.group('format')
1052 if fmt
== 's' and value
is not None and key
in field_size_compat_map
.keys():
1053 fmt
= '0{:d}d'.format(field_size_compat_map
[key
])
1055 value
= default
if value
is None else value
1057 str_fmt
= f
'{fmt[:-1]}s'
1058 if fmt
[-1] == 'l': # list
1059 delim
= '\n' if '#' in (outer_mobj
.group('conversion') or '') else ', '
1060 value
, fmt
= delim
.join(variadic(value
)), str_fmt
1061 elif fmt
[-1] == 'j': # json
1062 value
, fmt
= json
.dumps(value
, default
=_dumpjson_default
), str_fmt
1063 elif fmt
[-1] == 'q': # quoted
1064 value
, fmt
= compat_shlex_quote(str(value
)), str_fmt
1065 elif fmt
[-1] == 'B': # bytes
1066 value
= f
'%{str_fmt}'.encode('utf-8') % str(value
).encode('utf-8')
1067 value
, fmt
= value
.decode('utf-8', 'ignore'), 's'
1068 elif fmt
[-1] == 'U': # unicode normalized
1069 opts
= outer_mobj
.group('conversion') or ''
1070 value
, fmt
= unicodedata
.normalize(
1071 # "+" = compatibility equivalence, "#" = NFD
1072 'NF%s%s' % ('K' if '+' in opts
else '', 'D' if '#' in opts
else 'C'),
1074 elif fmt
[-1] == 'c':
1076 value
= str(value
)[0]
1079 elif fmt
[-1] not in 'rs': # numeric
1080 value
= float_or_none(value
)
1082 value
, fmt
= default
, 's'
1086 # If value is an object, sanitize might convert it to a string
1087 # So we convert it to repr first
1088 value
, fmt
= repr(value
), str_fmt
1089 if fmt
[-1] in 'csr':
1090 value
= sanitize(initial_field
, value
)
1092 key
= '%s\0%s' % (key
.replace('%', '%\0'), outer_mobj
.group('format'))
1093 TMPL_DICT
[key
] = value
1094 return '{prefix}%({key}){fmt}'.format(key
=key
, fmt
=fmt
, prefix
=outer_mobj
.group('prefix'))
1096 return EXTERNAL_FORMAT_RE
.sub(create_key
, outtmpl
), TMPL_DICT
1098 def evaluate_outtmpl(self
, outtmpl
, info_dict
, *args
, **kwargs
):
1099 outtmpl
, info_dict
= self
.prepare_outtmpl(outtmpl
, info_dict
, *args
, **kwargs
)
1100 return self
.escape_outtmpl(outtmpl
) % info_dict
1102 def _prepare_filename(self
, info_dict
, tmpl_type
='default'):
1104 sanitize
= lambda k
, v
: sanitize_filename(
1106 restricted
=self
.params
.get('restrictfilenames'),
1107 is_id
=(k
== 'id' or k
.endswith('_id')))
1108 outtmpl
= self
.outtmpl_dict
.get(tmpl_type
, self
.outtmpl_dict
['default'])
1109 outtmpl
, template_dict
= self
.prepare_outtmpl(outtmpl
, info_dict
, sanitize
)
1110 outtmpl
= self
.escape_outtmpl(self
._outtmpl
_expandpath
(outtmpl
))
1111 filename
= outtmpl
% template_dict
1113 force_ext
= OUTTMPL_TYPES
.get(tmpl_type
)
1114 if filename
and force_ext
is not None:
1115 filename
= replace_extension(filename
, force_ext
, info_dict
.get('ext'))
1117 # https://github.com/blackjack4494/youtube-dlc/issues/85
1118 trim_file_name
= self
.params
.get('trim_file_name', False)
1120 fn_groups
= filename
.rsplit('.')
1123 if len(fn_groups
) > 2:
1124 sub_ext
= fn_groups
[-2]
1125 filename
= '.'.join(filter(None, [fn_groups
[0][:trim_file_name
], sub_ext
, ext
]))
1128 except ValueError as err
:
1129 self
.report_error('Error in output template: ' + str(err
) + ' (encoding: ' + repr(preferredencoding()) + ')')
1132 def prepare_filename(self
, info_dict
, dir_type
='', warn
=False):
1133 """Generate the output filename."""
1135 filename
= self
._prepare
_filename
(info_dict
, dir_type
or 'default')
1136 if not filename
and dir_type
not in ('', 'temp'):
1140 if not self
.params
.get('paths'):
1142 elif filename
== '-':
1143 self
.report_warning('--paths is ignored when an outputting to stdout', only_once
=True)
1144 elif os
.path
.isabs(filename
):
1145 self
.report_warning('--paths is ignored since an absolute path is given in output template', only_once
=True)
1146 if filename
== '-' or not filename
:
1149 return self
.get_output_path(dir_type
, filename
)
1151 def _match_entry(self
, info_dict
, incomplete
=False, silent
=False):
1152 """ Returns None if the file should be downloaded """
1154 video_title
= info_dict
.get('title', info_dict
.get('id', 'video'))
1157 if 'title' in info_dict
:
1158 # This can happen when we're just evaluating the playlist
1159 title
= info_dict
['title']
1160 matchtitle
= self
.params
.get('matchtitle', False)
1162 if not re
.search(matchtitle
, title
, re
.IGNORECASE
):
1163 return '"' + title
+ '" title did not match pattern "' + matchtitle
+ '"'
1164 rejecttitle
= self
.params
.get('rejecttitle', False)
1166 if re
.search(rejecttitle
, title
, re
.IGNORECASE
):
1167 return '"' + title
+ '" title matched reject pattern "' + rejecttitle
+ '"'
1168 date
= info_dict
.get('upload_date')
1169 if date
is not None:
1170 dateRange
= self
.params
.get('daterange', DateRange())
1171 if date
not in dateRange
:
1172 return '%s upload date is not in range %s' % (date_from_str(date
).isoformat(), dateRange
)
1173 view_count
= info_dict
.get('view_count')
1174 if view_count
is not None:
1175 min_views
= self
.params
.get('min_views')
1176 if min_views
is not None and view_count
< min_views
:
1177 return 'Skipping %s, because it has not reached minimum view count (%d/%d)' % (video_title
, view_count
, min_views
)
1178 max_views
= self
.params
.get('max_views')
1179 if max_views
is not None and view_count
> max_views
:
1180 return 'Skipping %s, because it has exceeded the maximum view count (%d/%d)' % (video_title
, view_count
, max_views
)
1181 if age_restricted(info_dict
.get('age_limit'), self
.params
.get('age_limit')):
1182 return 'Skipping "%s" because it is age restricted' % video_title
1184 match_filter
= self
.params
.get('match_filter')
1185 if match_filter
is not None:
1187 ret
= match_filter(info_dict
, incomplete
=incomplete
)
1189 # For backward compatibility
1190 ret
= None if incomplete
else match_filter(info_dict
)
1195 if self
.in_download_archive(info_dict
):
1196 reason
= '%s has already been recorded in the archive' % video_title
1197 break_opt
, break_err
= 'break_on_existing', ExistingVideoReached
1199 reason
= check_filter()
1200 break_opt
, break_err
= 'break_on_reject', RejectedVideoReached
1201 if reason
is not None:
1203 self
.to_screen('[download] ' + reason
)
1204 if self
.params
.get(break_opt
, False):
1209 def add_extra_info(info_dict
, extra_info
):
1210 '''Set the keys from extra_info in info dict if they are missing'''
1211 for key
, value
in extra_info
.items():
1212 info_dict
.setdefault(key
, value
)
1214 def extract_info(self
, url
, download
=True, ie_key
=None, extra_info
=None,
1215 process
=True, force_generic_extractor
=False):
1217 Return a list with a dictionary for each video extracted.
1220 url -- URL to extract
1223 download -- whether to download videos during extraction
1224 ie_key -- extractor key hint
1225 extra_info -- dictionary containing the extra values to add to each result
1226 process -- whether to resolve all unresolved references (URLs, playlist items),
1227 must be True for download to work.
1228 force_generic_extractor -- force using the generic extractor
1231 if extra_info
is None:
1234 if not ie_key
and force_generic_extractor
:
1238 ies
= {ie_key: self._get_info_extractor_class(ie_key)}
1242 for ie_key
, ie
in ies
.items():
1243 if not ie
.suitable(url
):
1246 if not ie
.working():
1247 self
.report_warning('The program functionality for this site has been marked as broken, '
1248 'and will probably not work.')
1250 temp_id
= ie
.get_temp_id(url
)
1251 if temp_id
is not None and self
.in_download_archive({'id': temp_id, 'ie_key': ie_key}
):
1252 self
.to_screen("[%s] %s: has already been recorded in archive" % (
1255 return self
.__extract
_info
(url
, self
.get_info_extractor(ie_key
), download
, extra_info
, process
)
1257 self
.report_error('no suitable InfoExtractor for URL %s' % url
)
1259 def __handle_extraction_exceptions(func
):
1260 @functools.wraps(func
)
1261 def wrapper(self
, *args
, **kwargs
):
1263 return func(self
, *args
, **kwargs
)
1264 except GeoRestrictedError
as e
:
1267 msg
+= '\nThis video is available in %s.' % ', '.join(
1268 map(ISO3166Utils
.short2full
, e
.countries
))
1269 msg
+= '\nYou might want to use a VPN or a proxy server (with --proxy) to workaround.'
1270 self
.report_error(msg
)
1271 except ExtractorError
as e
: # An error we somewhat expected
1272 self
.report_error(compat_str(e
), e
.format_traceback())
1273 except ThrottledDownload
:
1274 self
.to_stderr('\r')
1275 self
.report_warning('The download speed is below throttle limit. Re-extracting data')
1276 return wrapper(self
, *args
, **kwargs
)
1277 except (MaxDownloadsReached
, ExistingVideoReached
, RejectedVideoReached
, LazyList
.IndexError):
1279 except Exception as e
:
1280 if self
.params
.get('ignoreerrors'):
1281 self
.report_error(error_to_compat_str(e
), tb
=encode_compat_str(traceback
.format_exc()))
1286 @__handle_extraction_exceptions
1287 def __extract_info(self
, url
, ie
, download
, extra_info
, process
):
1288 ie_result
= ie
.extract(url
)
1289 if ie_result
is None: # Finished already (backwards compatibility; listformats and friends should be moved here)
1291 if isinstance(ie_result
, list):
1292 # Backwards compatibility: old IE result format
1294 '_type': 'compat_list',
1295 'entries': ie_result
,
1297 if extra_info
.get('original_url'):
1298 ie_result
.setdefault('original_url', extra_info
['original_url'])
1299 self
.add_default_extra_info(ie_result
, ie
, url
)
1301 return self
.process_ie_result(ie_result
, download
, extra_info
)
1305 def add_default_extra_info(self
, ie_result
, ie
, url
):
1307 self
.add_extra_info(ie_result
, {
1309 'original_url': url
,
1310 'webpage_url_basename': url_basename(url
),
1313 self
.add_extra_info(ie_result
, {
1314 'extractor': ie
.IE_NAME
,
1315 'extractor_key': ie
.ie_key(),
1318 def process_ie_result(self
, ie_result
, download
=True, extra_info
=None):
1320 Take the result of the ie(may be modified) and resolve all unresolved
1321 references (URLs, playlist items).
1323 It will also download the videos if 'download'.
1324 Returns the resolved ie_result.
1326 if extra_info
is None:
1328 result_type
= ie_result
.get('_type', 'video')
1330 if result_type
in ('url', 'url_transparent'):
1331 ie_result
['url'] = sanitize_url(ie_result
['url'])
1332 if ie_result
.get('original_url'):
1333 extra_info
.setdefault('original_url', ie_result
['original_url'])
1335 extract_flat
= self
.params
.get('extract_flat', False)
1336 if ((extract_flat
== 'in_playlist' and 'playlist' in extra_info
)
1337 or extract_flat
is True):
1338 info_copy
= ie_result
.copy()
1339 ie
= try_get(ie_result
.get('ie_key'), self
.get_info_extractor
)
1340 if ie
and not ie_result
.get('id'):
1341 info_copy
['id'] = ie
.get_temp_id(ie_result
['url'])
1342 self
.add_default_extra_info(info_copy
, ie
, ie_result
['url'])
1343 self
.add_extra_info(info_copy
, extra_info
)
1344 self
.__forced
_printings
(info_copy
, self
.prepare_filename(info_copy
), incomplete
=True)
1345 if self
.params
.get('force_write_download_archive', False):
1346 self
.record_download_archive(info_copy
)
1349 if result_type
== 'video':
1350 self
.add_extra_info(ie_result
, extra_info
)
1351 ie_result
= self
.process_video_result(ie_result
, download
=download
)
1352 additional_urls
= (ie_result
or {}).get('additional_urls')
1354 # TODO: Improve MetadataParserPP to allow setting a list
1355 if isinstance(additional_urls
, compat_str
):
1356 additional_urls
= [additional_urls
]
1358 '[info] %s: %d additional URL(s) requested' % (ie_result
['id'], len(additional_urls
)))
1359 self
.write_debug('Additional URLs: "%s"' % '", "'.join(additional_urls
))
1360 ie_result
['additional_entries'] = [
1362 url
, download
, extra_info
,
1363 force_generic_extractor
=self
.params
.get('force_generic_extractor'))
1364 for url
in additional_urls
1367 elif result_type
== 'url':
1368 # We have to add extra_info to the results because it may be
1369 # contained in a playlist
1370 return self
.extract_info(
1371 ie_result
['url'], download
,
1372 ie_key
=ie_result
.get('ie_key'),
1373 extra_info
=extra_info
)
1374 elif result_type
== 'url_transparent':
1375 # Use the information from the embedding page
1376 info
= self
.extract_info(
1377 ie_result
['url'], ie_key
=ie_result
.get('ie_key'),
1378 extra_info
=extra_info
, download
=False, process
=False)
1380 # extract_info may return None when ignoreerrors is enabled and
1381 # extraction failed with an error, don't crash and return early
1386 force_properties
= dict(
1387 (k
, v
) for k
, v
in ie_result
.items() if v
is not None)
1388 for f
in ('_type', 'url', 'id', 'extractor', 'extractor_key', 'ie_key'):
1389 if f
in force_properties
:
1390 del force_properties
[f
]
1391 new_result
= info
.copy()
1392 new_result
.update(force_properties
)
1394 # Extracted info may not be a video result (i.e.
1395 # info.get('_type', 'video') != video) but rather an url or
1396 # url_transparent. In such cases outer metadata (from ie_result)
1397 # should be propagated to inner one (info). For this to happen
1398 # _type of info should be overridden with url_transparent. This
1399 # fixes issue from https://github.com/ytdl-org/youtube-dl/pull/11163.
1400 if new_result
.get('_type') == 'url':
1401 new_result
['_type'] = 'url_transparent'
1403 return self
.process_ie_result(
1404 new_result
, download
=download
, extra_info
=extra_info
)
1405 elif result_type
in ('playlist', 'multi_video'):
1406 # Protect from infinite recursion due to recursively nested playlists
1407 # (see https://github.com/ytdl-org/youtube-dl/issues/27833)
1408 webpage_url
= ie_result
['webpage_url']
1409 if webpage_url
in self
._playlist
_urls
:
1411 '[download] Skipping already downloaded playlist: %s'
1412 % ie_result
.get('title') or ie_result
.get('id'))
1415 self
._playlist
_level
+= 1
1416 self
._playlist
_urls
.add(webpage_url
)
1417 self
._sanitize
_thumbnails
(ie_result
)
1419 return self
.__process
_playlist
(ie_result
, download
)
1421 self
._playlist
_level
-= 1
1422 if not self
._playlist
_level
:
1423 self
._playlist
_urls
.clear()
1424 elif result_type
== 'compat_list':
1425 self
.report_warning(
1426 'Extractor %s returned a compat_list result. '
1427 'It needs to be updated.' % ie_result
.get('extractor'))
1430 self
.add_extra_info(r
, {
1431 'extractor': ie_result
['extractor'],
1432 'webpage_url': ie_result
['webpage_url'],
1433 'webpage_url_basename': url_basename(ie_result
['webpage_url']),
1434 'extractor_key': ie_result
['extractor_key'],
1437 ie_result
['entries'] = [
1438 self
.process_ie_result(_fixup(r
), download
, extra_info
)
1439 for r
in ie_result
['entries']
1443 raise Exception('Invalid result type: %s' % result_type
)
1445 def _ensure_dir_exists(self
, path
):
1446 return make_dir(path
, self
.report_error
)
1448 def __process_playlist(self
, ie_result
, download
):
1449 # We process each entry in the playlist
1450 playlist
= ie_result
.get('title') or ie_result
.get('id')
1451 self
.to_screen('[download] Downloading playlist: %s' % playlist
)
1453 if 'entries' not in ie_result
:
1454 raise EntryNotInPlaylist()
1455 incomplete_entries
= bool(ie_result
.get('requested_entries'))
1456 if incomplete_entries
:
1457 def fill_missing_entries(entries
, indexes
):
1458 ret
= [None] * max(*indexes
)
1459 for i
, entry
in zip(indexes
, entries
):
1462 ie_result
['entries'] = fill_missing_entries(ie_result
['entries'], ie_result
['requested_entries'])
1464 playlist_results
= []
1466 playliststart
= self
.params
.get('playliststart', 1)
1467 playlistend
= self
.params
.get('playlistend')
1468 # For backwards compatibility, interpret -1 as whole list
1469 if playlistend
== -1:
1472 playlistitems_str
= self
.params
.get('playlist_items')
1473 playlistitems
= None
1474 if playlistitems_str
is not None:
1475 def iter_playlistitems(format
):
1476 for string_segment
in format
.split(','):
1477 if '-' in string_segment
:
1478 start
, end
= string_segment
.split('-')
1479 for item
in range(int(start
), int(end
) + 1):
1482 yield int(string_segment
)
1483 playlistitems
= orderedSet(iter_playlistitems(playlistitems_str
))
1485 ie_entries
= ie_result
['entries']
1487 'Downloading %d videos' if not isinstance(ie_entries
, list)
1488 else 'Collected %d videos; downloading %%d of them' % len(ie_entries
))
1490 if isinstance(ie_entries
, list):
1492 return ie_entries
[i
- 1]
1494 if not isinstance(ie_entries
, PagedList
):
1495 ie_entries
= LazyList(ie_entries
)
1498 return YoutubeDL
.__handle
_extraction
_exceptions
(
1499 lambda self
, i
: ie_entries
[i
- 1]
1503 items
= playlistitems
if playlistitems
is not None else itertools
.count(playliststart
)
1507 if playlistitems
is None and playlistend
is not None and playlistend
< i
:
1511 entry
= get_entry(i
)
1513 raise EntryNotInPlaylist()
1514 except (IndexError, EntryNotInPlaylist
):
1515 if incomplete_entries
:
1516 raise EntryNotInPlaylist()
1517 elif not playlistitems
:
1519 entries
.append(entry
)
1521 if entry
is not None:
1522 self
._match
_entry
(entry
, incomplete
=True, silent
=True)
1523 except (ExistingVideoReached
, RejectedVideoReached
):
1525 ie_result
['entries'] = entries
1527 # Save playlist_index before re-ordering
1529 ((playlistitems
[i
- 1] if playlistitems
else i
+ playliststart
- 1), entry
)
1530 for i
, entry
in enumerate(entries
, 1)
1531 if entry
is not None]
1532 n_entries
= len(entries
)
1534 if not playlistitems
and (playliststart
or playlistend
):
1535 playlistitems
= list(range(playliststart
, playliststart
+ n_entries
))
1536 ie_result
['requested_entries'] = playlistitems
1538 if self
.params
.get('allow_playlist_files', True):
1540 'playlist': playlist
,
1541 'playlist_id': ie_result
.get('id'),
1542 'playlist_title': ie_result
.get('title'),
1543 'playlist_uploader': ie_result
.get('uploader'),
1544 'playlist_uploader_id': ie_result
.get('uploader_id'),
1545 'playlist_index': 0,
1547 ie_copy
.update(dict(ie_result
))
1549 if self
._write
_info
_json
('playlist', ie_result
,
1550 self
.prepare_filename(ie_copy
, 'pl_infojson')) is None:
1552 if self
._write
_description
('playlist', ie_result
,
1553 self
.prepare_filename(ie_copy
, 'pl_description')) is None:
1555 # TODO: This should be passed to ThumbnailsConvertor if necessary
1556 self
._write
_thumbnails
('playlist', ie_copy
, self
.prepare_filename(ie_copy
, 'pl_thumbnail'))
1558 if self
.params
.get('playlistreverse', False):
1559 entries
= entries
[::-1]
1560 if self
.params
.get('playlistrandom', False):
1561 random
.shuffle(entries
)
1563 x_forwarded_for
= ie_result
.get('__x_forwarded_for_ip')
1565 self
.to_screen('[%s] playlist %s: %s' % (ie_result
['extractor'], playlist
, msg
% n_entries
))
1567 max_failures
= self
.params
.get('skip_playlist_after_errors') or float('inf')
1568 for i
, entry_tuple
in enumerate(entries
, 1):
1569 playlist_index
, entry
= entry_tuple
1570 if 'playlist-index' in self
.params
.get('compat_opts', []):
1571 playlist_index
= playlistitems
[i
- 1] if playlistitems
else i
+ playliststart
- 1
1572 self
.to_screen('[download] Downloading video %s of %s' % (i
, n_entries
))
1573 # This __x_forwarded_for_ip thing is a bit ugly but requires
1576 entry
['__x_forwarded_for_ip'] = x_forwarded_for
1578 'n_entries': n_entries
,
1579 '_last_playlist_index': max(playlistitems
) if playlistitems
else (playlistend
or n_entries
),
1580 'playlist_index': playlist_index
,
1581 'playlist_autonumber': i
,
1582 'playlist': playlist
,
1583 'playlist_id': ie_result
.get('id'),
1584 'playlist_title': ie_result
.get('title'),
1585 'playlist_uploader': ie_result
.get('uploader'),
1586 'playlist_uploader_id': ie_result
.get('uploader_id'),
1587 'extractor': ie_result
['extractor'],
1588 'webpage_url': ie_result
['webpage_url'],
1589 'webpage_url_basename': url_basename(ie_result
['webpage_url']),
1590 'extractor_key': ie_result
['extractor_key'],
1593 if self
._match
_entry
(entry
, incomplete
=True) is not None:
1596 entry_result
= self
.__process
_iterable
_entry
(entry
, download
, extra
)
1597 if not entry_result
:
1599 if failures
>= max_failures
:
1601 'Skipping the remaining entries in playlist "%s" since %d items failed extraction' % (playlist
, failures
))
1603 # TODO: skip failed (empty) entries?
1604 playlist_results
.append(entry_result
)
1605 ie_result
['entries'] = playlist_results
1606 self
.to_screen('[download] Finished downloading playlist: %s' % playlist
)
1609 @__handle_extraction_exceptions
1610 def __process_iterable_entry(self
, entry
, download
, extra_info
):
1611 return self
.process_ie_result(
1612 entry
, download
=download
, extra_info
=extra_info
)
1614 def _build_format_filter(self
, filter_spec
):
1615 " Returns a function to filter the formats according to the filter_spec "
1625 operator_rex
= re
.compile(r
'''(?x)\s*
1626 (?P<key>width|height|tbr|abr|vbr|asr|filesize|filesize_approx|fps)\s*
1627 (?P<op>%s)(?P<none_inclusive>\s*\?)?\s*
1628 (?P<value>[0-9.]+(?:[kKmMgGtTpPeEzZyY]i?[Bb]?)?)\s*
1629 ''' % '|'.join(map(re
.escape
, OPERATORS
.keys())))
1630 m
= operator_rex
.fullmatch(filter_spec
)
1633 comparison_value
= int(m
.group('value'))
1635 comparison_value
= parse_filesize(m
.group('value'))
1636 if comparison_value
is None:
1637 comparison_value
= parse_filesize(m
.group('value') + 'B')
1638 if comparison_value
is None:
1640 'Invalid value %r in format specification %r' % (
1641 m
.group('value'), filter_spec
))
1642 op
= OPERATORS
[m
.group('op')]
1647 '^=': lambda attr
, value
: attr
.startswith(value
),
1648 '$=': lambda attr
, value
: attr
.endswith(value
),
1649 '*=': lambda attr
, value
: value
in attr
,
1651 str_operator_rex
= re
.compile(r
'''(?x)\s*
1652 (?P<key>[a-zA-Z0-9._-]+)\s*
1653 (?P<negation>!\s*)?(?P<op>%s)(?P<none_inclusive>\s*\?)?\s*
1654 (?P<value>[a-zA-Z0-9._-]+)\s*
1655 ''' % '|'.join(map(re
.escape
, STR_OPERATORS
.keys())))
1656 m
= str_operator_rex
.fullmatch(filter_spec
)
1658 comparison_value
= m
.group('value')
1659 str_op
= STR_OPERATORS
[m
.group('op')]
1660 if m
.group('negation'):
1661 op
= lambda attr
, value
: not str_op(attr
, value
)
1666 raise SyntaxError('Invalid filter specification %r' % filter_spec
)
1669 actual_value
= f
.get(m
.group('key'))
1670 if actual_value
is None:
1671 return m
.group('none_inclusive')
1672 return op(actual_value
, comparison_value
)
1675 def _default_format_spec(self
, info_dict
, download
=True):
1678 merger
= FFmpegMergerPP(self
)
1679 return merger
.available
and merger
.can_merge()
1682 not self
.params
.get('simulate')
1686 or info_dict
.get('is_live', False)
1687 or self
.outtmpl_dict
['default'] == '-'))
1690 or self
.params
.get('allow_multiple_audio_streams', False)
1691 or 'format-spec' in self
.params
.get('compat_opts', []))
1694 'best/bestvideo+bestaudio' if prefer_best
1695 else 'bestvideo*+bestaudio/best' if not compat
1696 else 'bestvideo+bestaudio/best')
1698 def build_format_selector(self
, format_spec
):
1699 def syntax_error(note
, start
):
1701 'Invalid format specification: '
1702 '{0}\n\t{1}\n\t{2}^'.format(note
, format_spec
, ' ' * start
[1]))
1703 return SyntaxError(message
)
1705 PICKFIRST
= 'PICKFIRST'
1709 FormatSelector
= collections
.namedtuple('FormatSelector', ['type', 'selector', 'filters'])
1711 allow_multiple_streams
= {'audio': self
.params
.get('allow_multiple_audio_streams', False),
1712 'video': self
.params
.get('allow_multiple_video_streams', False)}
1714 check_formats
= self
.params
.get('check_formats')
1716 def _parse_filter(tokens
):
1718 for type, string
, start
, _
, _
in tokens
:
1719 if type == tokenize
.OP
and string
== ']':
1720 return ''.join(filter_parts
)
1722 filter_parts
.append(string
)
1724 def _remove_unused_ops(tokens
):
1725 # Remove operators that we don't use and join them with the surrounding strings
1726 # for example: 'mp4' '-' 'baseline' '-' '16x9' is converted to 'mp4-baseline-16x9'
1727 ALLOWED_OPS
= ('/', '+', ',', '(', ')')
1728 last_string
, last_start
, last_end
, last_line
= None, None, None, None
1729 for type, string
, start
, end
, line
in tokens
:
1730 if type == tokenize
.OP
and string
== '[':
1732 yield tokenize
.NAME
, last_string
, last_start
, last_end
, last_line
1734 yield type, string
, start
, end
, line
1735 # everything inside brackets will be handled by _parse_filter
1736 for type, string
, start
, end
, line
in tokens
:
1737 yield type, string
, start
, end
, line
1738 if type == tokenize
.OP
and string
== ']':
1740 elif type == tokenize
.OP
and string
in ALLOWED_OPS
:
1742 yield tokenize
.NAME
, last_string
, last_start
, last_end
, last_line
1744 yield type, string
, start
, end
, line
1745 elif type in [tokenize
.NAME
, tokenize
.NUMBER
, tokenize
.OP
]:
1747 last_string
= string
1751 last_string
+= string
1753 yield tokenize
.NAME
, last_string
, last_start
, last_end
, last_line
1755 def _parse_format_selection(tokens
, inside_merge
=False, inside_choice
=False, inside_group
=False):
1757 current_selector
= None
1758 for type, string
, start
, _
, _
in tokens
:
1759 # ENCODING is only defined in python 3.x
1760 if type == getattr(tokenize
, 'ENCODING', None):
1762 elif type in [tokenize
.NAME
, tokenize
.NUMBER
]:
1763 current_selector
= FormatSelector(SINGLE
, string
, [])
1764 elif type == tokenize
.OP
:
1766 if not inside_group
:
1767 # ')' will be handled by the parentheses group
1768 tokens
.restore_last_token()
1770 elif inside_merge
and string
in ['/', ',']:
1771 tokens
.restore_last_token()
1773 elif inside_choice
and string
== ',':
1774 tokens
.restore_last_token()
1777 if not current_selector
:
1778 raise syntax_error('"," must follow a format selector', start
)
1779 selectors
.append(current_selector
)
1780 current_selector
= None
1782 if not current_selector
:
1783 raise syntax_error('"/" must follow a format selector', start
)
1784 first_choice
= current_selector
1785 second_choice
= _parse_format_selection(tokens
, inside_choice
=True)
1786 current_selector
= FormatSelector(PICKFIRST
, (first_choice
, second_choice
), [])
1788 if not current_selector
:
1789 current_selector
= FormatSelector(SINGLE
, 'best', [])
1790 format_filter
= _parse_filter(tokens
)
1791 current_selector
.filters
.append(format_filter
)
1793 if current_selector
:
1794 raise syntax_error('Unexpected "("', start
)
1795 group
= _parse_format_selection(tokens
, inside_group
=True)
1796 current_selector
= FormatSelector(GROUP
, group
, [])
1798 if not current_selector
:
1799 raise syntax_error('Unexpected "+"', start
)
1800 selector_1
= current_selector
1801 selector_2
= _parse_format_selection(tokens
, inside_merge
=True)
1803 raise syntax_error('Expected a selector', start
)
1804 current_selector
= FormatSelector(MERGE
, (selector_1
, selector_2
), [])
1806 raise syntax_error('Operator not recognized: "{0}"'.format(string
), start
)
1807 elif type == tokenize
.ENDMARKER
:
1809 if current_selector
:
1810 selectors
.append(current_selector
)
1813 def _merge(formats_pair
):
1814 format_1
, format_2
= formats_pair
1817 formats_info
.extend(format_1
.get('requested_formats', (format_1
,)))
1818 formats_info
.extend(format_2
.get('requested_formats', (format_2
,)))
1820 if not allow_multiple_streams
['video'] or not allow_multiple_streams
['audio']:
1821 get_no_more
= {'video': False, 'audio': False}
1822 for (i
, fmt_info
) in enumerate(formats_info
):
1823 if fmt_info
.get('acodec') == fmt_info
.get('vcodec') == 'none':
1826 for aud_vid
in ['audio', 'video']:
1827 if not allow_multiple_streams
[aud_vid
] and fmt_info
.get(aud_vid
[0] + 'codec') != 'none':
1828 if get_no_more
[aud_vid
]:
1831 get_no_more
[aud_vid
] = True
1833 if len(formats_info
) == 1:
1834 return formats_info
[0]
1836 video_fmts
= [fmt_info
for fmt_info
in formats_info
if fmt_info
.get('vcodec') != 'none']
1837 audio_fmts
= [fmt_info
for fmt_info
in formats_info
if fmt_info
.get('acodec') != 'none']
1839 the_only_video
= video_fmts
[0] if len(video_fmts
) == 1 else None
1840 the_only_audio
= audio_fmts
[0] if len(audio_fmts
) == 1 else None
1842 output_ext
= self
.params
.get('merge_output_format')
1845 output_ext
= the_only_video
['ext']
1846 elif the_only_audio
and not video_fmts
:
1847 output_ext
= the_only_audio
['ext']
1851 filtered
= lambda *keys
: filter(None, (traverse_obj(fmt
, *keys
) for fmt
in formats_info
))
1854 'requested_formats': formats_info
,
1855 'format': '+'.join(filtered('format')),
1856 'format_id': '+'.join(filtered('format_id')),
1858 'protocol': '+'.join(map(determine_protocol
, formats_info
)),
1859 'language': '+'.join(orderedSet(filtered('language'))),
1860 'format_note': '+'.join(orderedSet(filtered('format_note'))),
1861 'filesize_approx': sum(filtered('filesize', 'filesize_approx')),
1862 'tbr': sum(filtered('tbr', 'vbr', 'abr')),
1867 'width': the_only_video
.get('width'),
1868 'height': the_only_video
.get('height'),
1869 'resolution': the_only_video
.get('resolution') or self
.format_resolution(the_only_video
),
1870 'fps': the_only_video
.get('fps'),
1871 'vcodec': the_only_video
.get('vcodec'),
1872 'vbr': the_only_video
.get('vbr'),
1873 'stretched_ratio': the_only_video
.get('stretched_ratio'),
1878 'acodec': the_only_audio
.get('acodec'),
1879 'abr': the_only_audio
.get('abr'),
1880 'asr': the_only_audio
.get('asr'),
1885 def _check_formats(formats
):
1886 if not check_formats
:
1890 self
.to_screen('[info] Testing format %s' % f
['format_id'])
1891 temp_file
= tempfile
.NamedTemporaryFile(
1892 suffix
='.tmp', delete
=False,
1893 dir=self
.get_output_path('temp') or None)
1896 success
, _
= self
.dl(temp_file
.name
, f
, test
=True)
1897 except (DownloadError
, IOError, OSError, ValueError) + network_exceptions
:
1900 if os
.path
.exists(temp_file
.name
):
1902 os
.remove(temp_file
.name
)
1904 self
.report_warning('Unable to delete temporary file "%s"' % temp_file
.name
)
1908 self
.to_screen('[info] Unable to download format %s. Skipping...' % f
['format_id'])
1910 def _build_selector_function(selector
):
1911 if isinstance(selector
, list): # ,
1912 fs
= [_build_selector_function(s
) for s
in selector
]
1914 def selector_function(ctx
):
1917 return selector_function
1919 elif selector
.type == GROUP
: # ()
1920 selector_function
= _build_selector_function(selector
.selector
)
1922 elif selector
.type == PICKFIRST
: # /
1923 fs
= [_build_selector_function(s
) for s
in selector
.selector
]
1925 def selector_function(ctx
):
1927 picked_formats
= list(f(ctx
))
1929 return picked_formats
1932 elif selector
.type == MERGE
: # +
1933 selector_1
, selector_2
= map(_build_selector_function
, selector
.selector
)
1935 def selector_function(ctx
):
1936 for pair
in itertools
.product(
1937 selector_1(copy
.deepcopy(ctx
)), selector_2(copy
.deepcopy(ctx
))):
1940 elif selector
.type == SINGLE
: # atom
1941 format_spec
= selector
.selector
or 'best'
1943 # TODO: Add allvideo, allaudio etc by generalizing the code with best/worst selector
1944 if format_spec
== 'all':
1945 def selector_function(ctx
):
1946 yield from _check_formats(ctx
['formats'])
1947 elif format_spec
== 'mergeall':
1948 def selector_function(ctx
):
1949 formats
= list(_check_formats(ctx
['formats']))
1952 merged_format
= formats
[-1]
1953 for f
in formats
[-2::-1]:
1954 merged_format
= _merge((merged_format
, f
))
1958 format_fallback
, format_reverse
, format_idx
= False, True, 1
1960 r
'(?P<bw>best|worst|b|w)(?P<type>video|audio|v|a)?(?P<mod>\*)?(?:\.(?P<n>[1-9]\d*))?$',
1962 if mobj
is not None:
1963 format_idx
= int_or_none(mobj
.group('n'), default
=1)
1964 format_reverse
= mobj
.group('bw')[0] == 'b'
1965 format_type
= (mobj
.group('type') or [None])[0]
1966 not_format_type
= {'v': 'a', 'a': 'v'}
.get(format_type
)
1967 format_modified
= mobj
.group('mod') is not None
1969 format_fallback
= not format_type
and not format_modified
# for b, w
1971 (lambda f
: f
.get('%scodec' % format_type
) != 'none')
1972 if format_type
and format_modified
# bv*, ba*, wv*, wa*
1973 else (lambda f
: f
.get('%scodec' % not_format_type
) == 'none')
1974 if format_type
# bv, ba, wv, wa
1975 else (lambda f
: f
.get('vcodec') != 'none' and f
.get('acodec') != 'none')
1976 if not format_modified
# b, w
1977 else lambda f
: True) # b*, w*
1978 filter_f
= lambda f
: _filter_f(f
) and (
1979 f
.get('vcodec') != 'none' or f
.get('acodec') != 'none')
1981 if format_spec
in ('m4a', 'mp3', 'ogg', 'aac'): # audio extension
1982 filter_f
= lambda f
: f
.get('ext') == format_spec
and f
.get('acodec') != 'none'
1983 elif format_spec
in ('mp4', 'flv', 'webm', '3gp'): # video extension
1984 filter_f
= lambda f
: f
.get('ext') == format_spec
and f
.get('acodec') != 'none' and f
.get('vcodec') != 'none'
1985 elif format_spec
in ('mhtml', ): # storyboards extension
1986 filter_f
= lambda f
: f
.get('ext') == format_spec
and f
.get('acodec') == 'none' and f
.get('vcodec') == 'none'
1988 filter_f
= lambda f
: f
.get('format_id') == format_spec
# id
1990 def selector_function(ctx
):
1991 formats
= list(ctx
['formats'])
1992 matches
= list(filter(filter_f
, formats
)) if filter_f
is not None else formats
1993 if format_fallback
and ctx
['incomplete_formats'] and not matches
:
1994 # for extractors with incomplete formats (audio only (soundcloud)
1995 # or video only (imgur)) best/worst will fallback to
1996 # best/worst {video,audio}-only format
1998 matches
= LazyList(_check_formats(matches
[::-1 if format_reverse
else 1]))
2000 yield matches
[format_idx
- 1]
2004 filters
= [self
._build
_format
_filter
(f
) for f
in selector
.filters
]
2006 def final_selector(ctx
):
2007 ctx_copy
= copy
.deepcopy(ctx
)
2008 for _filter
in filters
:
2009 ctx_copy
['formats'] = list(filter(_filter
, ctx_copy
['formats']))
2010 return selector_function(ctx_copy
)
2011 return final_selector
2013 stream
= io
.BytesIO(format_spec
.encode('utf-8'))
2015 tokens
= list(_remove_unused_ops(compat_tokenize_tokenize(stream
.readline
)))
2016 except tokenize
.TokenError
:
2017 raise syntax_error('Missing closing/opening brackets or parenthesis', (0, len(format_spec
)))
2019 class TokenIterator(object):
2020 def __init__(self
, tokens
):
2021 self
.tokens
= tokens
2028 if self
.counter
>= len(self
.tokens
):
2029 raise StopIteration()
2030 value
= self
.tokens
[self
.counter
]
2036 def restore_last_token(self
):
2039 parsed_selector
= _parse_format_selection(iter(TokenIterator(tokens
)))
2040 return _build_selector_function(parsed_selector
)
2042 def _calc_headers(self
, info_dict
):
2043 res
= std_headers
.copy()
2045 add_headers
= info_dict
.get('http_headers')
2047 res
.update(add_headers
)
2049 cookies
= self
._calc
_cookies
(info_dict
)
2051 res
['Cookie'] = cookies
2053 if 'X-Forwarded-For' not in res
:
2054 x_forwarded_for_ip
= info_dict
.get('__x_forwarded_for_ip')
2055 if x_forwarded_for_ip
:
2056 res
['X-Forwarded-For'] = x_forwarded_for_ip
2060 def _calc_cookies(self
, info_dict
):
2061 pr
= sanitized_Request(info_dict
['url'])
2062 self
.cookiejar
.add_cookie_header(pr
)
2063 return pr
.get_header('Cookie')
2065 def _sanitize_thumbnails(self
, info_dict
):
2066 thumbnails
= info_dict
.get('thumbnails')
2067 if thumbnails
is None:
2068 thumbnail
= info_dict
.get('thumbnail')
2070 info_dict
['thumbnails'] = thumbnails
= [{'url': thumbnail}
]
2072 thumbnails
.sort(key
=lambda t
: (
2073 t
.get('preference') if t
.get('preference') is not None else -1,
2074 t
.get('width') if t
.get('width') is not None else -1,
2075 t
.get('height') if t
.get('height') is not None else -1,
2076 t
.get('id') if t
.get('id') is not None else '',
2079 def thumbnail_tester():
2080 if self
.params
.get('check_formats'):
2082 to_screen
= lambda msg
: self
.to_screen(f
'[info] {msg}')
2085 to_screen
= self
.write_debug
2087 def test_thumbnail(t
):
2088 if not test_all
and not t
.get('_test_url'):
2090 to_screen('Testing thumbnail %s' % t
['id'])
2092 self
.urlopen(HEADRequest(t
['url']))
2093 except network_exceptions
as err
:
2094 to_screen('Unable to connect to thumbnail %s URL "%s" - %s. Skipping...' % (
2095 t
['id'], t
['url'], error_to_compat_str(err
)))
2099 return test_thumbnail
2101 for i
, t
in enumerate(thumbnails
):
2102 if t
.get('id') is None:
2104 if t
.get('width') and t
.get('height'):
2105 t
['resolution'] = '%dx%d' % (t
['width'], t
['height'])
2106 t
['url'] = sanitize_url(t
['url'])
2108 if self
.params
.get('check_formats') is not False:
2109 info_dict
['thumbnails'] = LazyList(filter(thumbnail_tester(), thumbnails
[::-1])).reverse()
2111 info_dict
['thumbnails'] = thumbnails
2113 def process_video_result(self
, info_dict
, download
=True):
2114 assert info_dict
.get('_type', 'video') == 'video'
2116 if 'id' not in info_dict
:
2117 raise ExtractorError('Missing "id" field in extractor result')
2118 if 'title' not in info_dict
:
2119 raise ExtractorError('Missing "title" field in extractor result',
2120 video_id
=info_dict
['id'], ie
=info_dict
['extractor'])
2122 def report_force_conversion(field
, field_not
, conversion
):
2123 self
.report_warning(
2124 '"%s" field is not %s - forcing %s conversion, there is an error in extractor'
2125 % (field
, field_not
, conversion
))
2127 def sanitize_string_field(info
, string_field
):
2128 field
= info
.get(string_field
)
2129 if field
is None or isinstance(field
, compat_str
):
2131 report_force_conversion(string_field
, 'a string', 'string')
2132 info
[string_field
] = compat_str(field
)
2134 def sanitize_numeric_fields(info
):
2135 for numeric_field
in self
._NUMERIC
_FIELDS
:
2136 field
= info
.get(numeric_field
)
2137 if field
is None or isinstance(field
, compat_numeric_types
):
2139 report_force_conversion(numeric_field
, 'numeric', 'int')
2140 info
[numeric_field
] = int_or_none(field
)
2142 sanitize_string_field(info_dict
, 'id')
2143 sanitize_numeric_fields(info_dict
)
2145 if 'playlist' not in info_dict
:
2146 # It isn't part of a playlist
2147 info_dict
['playlist'] = None
2148 info_dict
['playlist_index'] = None
2150 self
._sanitize
_thumbnails
(info_dict
)
2152 thumbnail
= info_dict
.get('thumbnail')
2153 thumbnails
= info_dict
.get('thumbnails')
2155 info_dict
['thumbnail'] = sanitize_url(thumbnail
)
2157 info_dict
['thumbnail'] = thumbnails
[-1]['url']
2159 if info_dict
.get('display_id') is None and 'id' in info_dict
:
2160 info_dict
['display_id'] = info_dict
['id']
2162 for ts_key
, date_key
in (
2163 ('timestamp', 'upload_date'),
2164 ('release_timestamp', 'release_date'),
2166 if info_dict
.get(date_key
) is None and info_dict
.get(ts_key
) is not None:
2167 # Working around out-of-range timestamp values (e.g. negative ones on Windows,
2168 # see http://bugs.python.org/issue1646728)
2170 upload_date
= datetime
.datetime
.utcfromtimestamp(info_dict
[ts_key
])
2171 info_dict
[date_key
] = upload_date
.strftime('%Y%m%d')
2172 except (ValueError, OverflowError, OSError):
2175 live_keys
= ('is_live', 'was_live')
2176 live_status
= info_dict
.get('live_status')
2177 if live_status
is None:
2178 for key
in live_keys
:
2179 if info_dict
.get(key
) is False:
2181 if info_dict
.get(key
):
2184 if all(info_dict
.get(key
) is False for key
in live_keys
):
2185 live_status
= 'not_live'
2187 info_dict
['live_status'] = live_status
2188 for key
in live_keys
:
2189 if info_dict
.get(key
) is None:
2190 info_dict
[key
] = (live_status
== key
)
2192 # Auto generate title fields corresponding to the *_number fields when missing
2193 # in order to always have clean titles. This is very common for TV series.
2194 for field
in ('chapter', 'season', 'episode'):
2195 if info_dict
.get('%s_number' % field
) is not None and not info_dict
.get(field
):
2196 info_dict
[field
] = '%s %d' % (field
.capitalize(), info_dict
['%s_number' % field
])
2198 for cc_kind
in ('subtitles', 'automatic_captions'):
2199 cc
= info_dict
.get(cc_kind
)
2201 for _
, subtitle
in cc
.items():
2202 for subtitle_format
in subtitle
:
2203 if subtitle_format
.get('url'):
2204 subtitle_format
['url'] = sanitize_url(subtitle_format
['url'])
2205 if subtitle_format
.get('ext') is None:
2206 subtitle_format
['ext'] = determine_ext(subtitle_format
['url']).lower()
2208 automatic_captions
= info_dict
.get('automatic_captions')
2209 subtitles
= info_dict
.get('subtitles')
2211 info_dict
['requested_subtitles'] = self
.process_subtitles(
2212 info_dict
['id'], subtitles
, automatic_captions
)
2214 # We now pick which formats have to be downloaded
2215 if info_dict
.get('formats') is None:
2216 # There's only one format available
2217 formats
= [info_dict
]
2219 formats
= info_dict
['formats']
2221 info_dict
['__has_drm'] = any(f
.get('has_drm') for f
in formats
)
2222 if not self
.params
.get('allow_unplayable_formats'):
2223 formats
= [f
for f
in formats
if not f
.get('has_drm')]
2226 self
.raise_no_formats(info_dict
)
2228 def is_wellformed(f
):
2231 self
.report_warning(
2232 '"url" field is missing or empty - skipping format, '
2233 'there is an error in extractor')
2235 if isinstance(url
, bytes):
2236 sanitize_string_field(f
, 'url')
2239 # Filter out malformed formats for better extraction robustness
2240 formats
= list(filter(is_wellformed
, formats
))
2244 # We check that all the formats have the format and format_id fields
2245 for i
, format
in enumerate(formats
):
2246 sanitize_string_field(format
, 'format_id')
2247 sanitize_numeric_fields(format
)
2248 format
['url'] = sanitize_url(format
['url'])
2249 if not format
.get('format_id'):
2250 format
['format_id'] = compat_str(i
)
2252 # Sanitize format_id from characters used in format selector expression
2253 format
['format_id'] = re
.sub(r
'[\s,/+\[\]()]', '_', format
['format_id'])
2254 format_id
= format
['format_id']
2255 if format_id
not in formats_dict
:
2256 formats_dict
[format_id
] = []
2257 formats_dict
[format_id
].append(format
)
2259 # Make sure all formats have unique format_id
2260 for format_id
, ambiguous_formats
in formats_dict
.items():
2261 if len(ambiguous_formats
) > 1:
2262 for i
, format
in enumerate(ambiguous_formats
):
2263 format
['format_id'] = '%s-%d' % (format_id
, i
)
2265 for i
, format
in enumerate(formats
):
2266 if format
.get('format') is None:
2267 format
['format'] = '{id} - {res}{note}'.format(
2268 id=format
['format_id'],
2269 res
=self
.format_resolution(format
),
2270 note
=format_field(format
, 'format_note', ' (%s)'),
2272 # Automatically determine file extension if missing
2273 if format
.get('ext') is None:
2274 format
['ext'] = determine_ext(format
['url']).lower()
2275 # Automatically determine protocol if missing (useful for format
2276 # selection purposes)
2277 if format
.get('protocol') is None:
2278 format
['protocol'] = determine_protocol(format
)
2279 # Add HTTP headers, so that external programs can use them from the
2281 full_format_info
= info_dict
.copy()
2282 full_format_info
.update(format
)
2283 format
['http_headers'] = self
._calc
_headers
(full_format_info
)
2284 # Remove private housekeeping stuff
2285 if '__x_forwarded_for_ip' in info_dict
:
2286 del info_dict
['__x_forwarded_for_ip']
2288 # TODO Central sorting goes here
2290 if not formats
or formats
[0] is not info_dict
:
2291 # only set the 'formats' fields if the original info_dict list them
2292 # otherwise we end up with a circular reference, the first (and unique)
2293 # element in the 'formats' field in info_dict is info_dict itself,
2294 # which can't be exported to json
2295 info_dict
['formats'] = formats
2297 info_dict
, _
= self
.pre_process(info_dict
)
2299 if self
.params
.get('list_thumbnails'):
2300 self
.list_thumbnails(info_dict
)
2301 if self
.params
.get('listformats'):
2302 if not info_dict
.get('formats') and not info_dict
.get('url'):
2303 self
.to_screen('%s has no formats' % info_dict
['id'])
2305 self
.list_formats(info_dict
)
2306 if self
.params
.get('listsubtitles'):
2307 if 'automatic_captions' in info_dict
:
2308 self
.list_subtitles(
2309 info_dict
['id'], automatic_captions
, 'automatic captions')
2310 self
.list_subtitles(info_dict
['id'], subtitles
, 'subtitles')
2311 list_only
= self
.params
.get('simulate') is None and (
2312 self
.params
.get('list_thumbnails') or self
.params
.get('listformats') or self
.params
.get('listsubtitles'))
2314 # Without this printing, -F --print-json will not work
2315 self
.__forced
_printings
(info_dict
, self
.prepare_filename(info_dict
), incomplete
=True)
2318 format_selector
= self
.format_selector
2319 if format_selector
is None:
2320 req_format
= self
._default
_format
_spec
(info_dict
, download
=download
)
2321 self
.write_debug('Default format spec: %s' % req_format
)
2322 format_selector
= self
.build_format_selector(req_format
)
2324 # While in format selection we may need to have an access to the original
2325 # format set in order to calculate some metrics or do some processing.
2326 # For now we need to be able to guess whether original formats provided
2327 # by extractor are incomplete or not (i.e. whether extractor provides only
2328 # video-only or audio-only formats) for proper formats selection for
2329 # extractors with such incomplete formats (see
2330 # https://github.com/ytdl-org/youtube-dl/pull/5556).
2331 # Since formats may be filtered during format selection and may not match
2332 # the original formats the results may be incorrect. Thus original formats
2333 # or pre-calculated metrics should be passed to format selection routines
2335 # We will pass a context object containing all necessary additional data
2336 # instead of just formats.
2337 # This fixes incorrect format selection issue (see
2338 # https://github.com/ytdl-org/youtube-dl/issues/10083).
2339 incomplete_formats
= (
2340 # All formats are video-only or
2341 all(f
.get('vcodec') != 'none' and f
.get('acodec') == 'none' for f
in formats
)
2342 # all formats are audio-only
2343 or all(f
.get('vcodec') == 'none' and f
.get('acodec') != 'none' for f
in formats
))
2347 'incomplete_formats': incomplete_formats
,
2350 formats_to_download
= list(format_selector(ctx
))
2351 if not formats_to_download
:
2352 if not self
.params
.get('ignore_no_formats_error'):
2353 raise ExtractorError('Requested format is not available', expected
=True,
2354 video_id
=info_dict
['id'], ie
=info_dict
['extractor'])
2356 self
.report_warning('Requested format is not available')
2357 # Process what we can, even without any available formats.
2358 self
.process_info(dict(info_dict
))
2361 '[info] %s: Downloading %d format(s): %s' % (
2362 info_dict
['id'], len(formats_to_download
),
2363 ", ".join([f
['format_id'] for f
in formats_to_download
])))
2364 for fmt
in formats_to_download
:
2365 new_info
= dict(info_dict
)
2366 # Save a reference to the original info_dict so that it can be modified in process_info if needed
2367 new_info
['__original_infodict'] = info_dict
2368 new_info
.update(fmt
)
2369 self
.process_info(new_info
)
2370 # We update the info dict with the best quality format (backwards compatibility)
2371 if formats_to_download
:
2372 info_dict
.update(formats_to_download
[-1])
2375 def process_subtitles(self
, video_id
, normal_subtitles
, automatic_captions
):
2376 """Select the requested subtitles and their format"""
2378 if normal_subtitles
and self
.params
.get('writesubtitles'):
2379 available_subs
.update(normal_subtitles
)
2380 if automatic_captions
and self
.params
.get('writeautomaticsub'):
2381 for lang
, cap_info
in automatic_captions
.items():
2382 if lang
not in available_subs
:
2383 available_subs
[lang
] = cap_info
2385 if (not self
.params
.get('writesubtitles') and not
2386 self
.params
.get('writeautomaticsub') or not
2390 all_sub_langs
= available_subs
.keys()
2391 if self
.params
.get('allsubtitles', False):
2392 requested_langs
= all_sub_langs
2393 elif self
.params
.get('subtitleslangs', False):
2394 # A list is used so that the order of languages will be the same as
2395 # given in subtitleslangs. See https://github.com/yt-dlp/yt-dlp/issues/1041
2396 requested_langs
= []
2397 for lang_re
in self
.params
.get('subtitleslangs'):
2398 if lang_re
== 'all':
2399 requested_langs
.extend(all_sub_langs
)
2401 discard
= lang_re
[0] == '-'
2403 lang_re
= lang_re
[1:]
2404 current_langs
= filter(re
.compile(lang_re
+ '$').match
, all_sub_langs
)
2406 for lang
in current_langs
:
2407 while lang
in requested_langs
:
2408 requested_langs
.remove(lang
)
2410 requested_langs
.extend(current_langs
)
2411 requested_langs
= orderedSet(requested_langs
)
2412 elif 'en' in available_subs
:
2413 requested_langs
= ['en']
2415 requested_langs
= [list(all_sub_langs
)[0]]
2417 self
.write_debug('Downloading subtitles: %s' % ', '.join(requested_langs
))
2419 formats_query
= self
.params
.get('subtitlesformat', 'best')
2420 formats_preference
= formats_query
.split('/') if formats_query
else []
2422 for lang
in requested_langs
:
2423 formats
= available_subs
.get(lang
)
2425 self
.report_warning('%s subtitles not available for %s' % (lang
, video_id
))
2427 for ext
in formats_preference
:
2431 matches
= list(filter(lambda f
: f
['ext'] == ext
, formats
))
2437 self
.report_warning(
2438 'No subtitle format found matching "%s" for language %s, '
2439 'using %s' % (formats_query
, lang
, f
['ext']))
2443 def __forced_printings(self
, info_dict
, filename
, incomplete
):
2444 def print_mandatory(field
, actual_field
=None):
2445 if actual_field
is None:
2446 actual_field
= field
2447 if (self
.params
.get('force%s' % field
, False)
2448 and (not incomplete
or info_dict
.get(actual_field
) is not None)):
2449 self
.to_stdout(info_dict
[actual_field
])
2451 def print_optional(field
):
2452 if (self
.params
.get('force%s' % field
, False)
2453 and info_dict
.get(field
) is not None):
2454 self
.to_stdout(info_dict
[field
])
2456 info_dict
= info_dict
.copy()
2457 if filename
is not None:
2458 info_dict
['filename'] = filename
2459 if info_dict
.get('requested_formats') is not None:
2460 # For RTMP URLs, also include the playpath
2461 info_dict
['urls'] = '\n'.join(f
['url'] + f
.get('play_path', '') for f
in info_dict
['requested_formats'])
2462 elif 'url' in info_dict
:
2463 info_dict
['urls'] = info_dict
['url'] + info_dict
.get('play_path', '')
2465 if self
.params
.get('forceprint') or self
.params
.get('forcejson'):
2466 self
.post_extract(info_dict
)
2467 for tmpl
in self
.params
.get('forceprint', []):
2468 mobj
= re
.match(r
'\w+(=?)$', tmpl
)
2469 if mobj
and mobj
.group(1):
2470 tmpl
= f
'{tmpl[:-1]} = %({tmpl[:-1]})s'
2472 tmpl
= '%({})s'.format(tmpl
)
2473 self
.to_stdout(self
.evaluate_outtmpl(tmpl
, info_dict
))
2475 print_mandatory('title')
2476 print_mandatory('id')
2477 print_mandatory('url', 'urls')
2478 print_optional('thumbnail')
2479 print_optional('description')
2480 print_optional('filename')
2481 if self
.params
.get('forceduration') and info_dict
.get('duration') is not None:
2482 self
.to_stdout(formatSeconds(info_dict
['duration']))
2483 print_mandatory('format')
2485 if self
.params
.get('forcejson'):
2486 self
.to_stdout(json
.dumps(self
.sanitize_info(info_dict
)))
2488 def dl(self
, name
, info
, subtitle
=False, test
=False):
2489 if not info
.get('url'):
2490 self
.raise_no_formats(info
, True)
2493 verbose
= self
.params
.get('verbose')
2496 'quiet': self
.params
.get('quiet') or not verbose
,
2498 'noprogress': not verbose
,
2500 'skip_unavailable_fragments': False,
2501 'keep_fragments': False,
2503 '_no_ytdl_file': True,
2506 params
= self
.params
2507 fd
= get_suitable_downloader(info
, params
, to_stdout
=(name
== '-'))(self
, params
)
2509 for ph
in self
._progress
_hooks
:
2510 fd
.add_progress_hook(ph
)
2511 urls
= '", "'.join([f
['url'] for f
in info
.get('requested_formats', [])] or [info
['url']])
2512 self
.write_debug('Invoking downloader on "%s"' % urls
)
2513 new_info
= dict(info
)
2514 if new_info
.get('http_headers') is None:
2515 new_info
['http_headers'] = self
._calc
_headers
(new_info
)
2516 return fd
.download(name
, new_info
, subtitle
)
2518 def process_info(self
, info_dict
):
2519 """Process a single resolved IE result."""
2521 assert info_dict
.get('_type', 'video') == 'video'
2523 max_downloads
= self
.params
.get('max_downloads')
2524 if max_downloads
is not None:
2525 if self
._num
_downloads
>= int(max_downloads
):
2526 raise MaxDownloadsReached()
2528 # TODO: backward compatibility, to be removed
2529 info_dict
['fulltitle'] = info_dict
['title']
2531 if 'format' not in info_dict
and 'ext' in info_dict
:
2532 info_dict
['format'] = info_dict
['ext']
2534 if self
._match
_entry
(info_dict
) is not None:
2537 self
.post_extract(info_dict
)
2538 self
._num
_downloads
+= 1
2540 # info_dict['_filename'] needs to be set for backward compatibility
2541 info_dict
['_filename'] = full_filename
= self
.prepare_filename(info_dict
, warn
=True)
2542 temp_filename
= self
.prepare_filename(info_dict
, 'temp')
2546 self
.__forced
_printings
(info_dict
, full_filename
, incomplete
=('format' not in info_dict
))
2548 if self
.params
.get('simulate'):
2549 if self
.params
.get('force_write_download_archive', False):
2550 self
.record_download_archive(info_dict
)
2551 # Do nothing else if in simulate mode
2554 if full_filename
is None:
2556 if not self
._ensure
_dir
_exists
(encodeFilename(full_filename
)):
2558 if not self
._ensure
_dir
_exists
(encodeFilename(temp_filename
)):
2561 if self
._write
_description
('video', info_dict
,
2562 self
.prepare_filename(info_dict
, 'description')) is None:
2565 sub_files
= self
._write
_subtitles
(info_dict
, temp_filename
)
2566 if sub_files
is None:
2568 files_to_move
.update(dict(sub_files
))
2570 thumb_files
= self
._write
_thumbnails
(
2571 'video', info_dict
, temp_filename
, self
.prepare_filename(info_dict
, 'thumbnail'))
2572 if thumb_files
is None:
2574 files_to_move
.update(dict(thumb_files
))
2576 infofn
= self
.prepare_filename(info_dict
, 'infojson')
2577 _infojson_written
= self
._write
_info
_json
('video', info_dict
, infofn
)
2578 if _infojson_written
:
2579 info_dict
['__infojson_filename'] = infofn
2580 elif _infojson_written
is None:
2583 # Note: Annotations are deprecated
2585 if self
.params
.get('writeannotations', False):
2586 annofn
= self
.prepare_filename(info_dict
, 'annotation')
2588 if not self
._ensure
_dir
_exists
(encodeFilename(annofn
)):
2590 if not self
.params
.get('overwrites', True) and os
.path
.exists(encodeFilename(annofn
)):
2591 self
.to_screen('[info] Video annotations are already present')
2592 elif not info_dict
.get('annotations'):
2593 self
.report_warning('There are no annotations to write.')
2596 self
.to_screen('[info] Writing video annotations to: ' + annofn
)
2597 with io
.open(encodeFilename(annofn
), 'w', encoding
='utf-8') as annofile
:
2598 annofile
.write(info_dict
['annotations'])
2599 except (KeyError, TypeError):
2600 self
.report_warning('There are no annotations to write.')
2601 except (OSError, IOError):
2602 self
.report_error('Cannot write annotations file: ' + annofn
)
2605 # Write internet shortcut files
2606 url_link
= webloc_link
= desktop_link
= False
2607 if self
.params
.get('writelink', False):
2608 if sys
.platform
== "darwin": # macOS.
2610 elif sys
.platform
.startswith("linux"):
2612 else: # if sys.platform in ['win32', 'cygwin']:
2614 if self
.params
.get('writeurllink', False):
2616 if self
.params
.get('writewebloclink', False):
2618 if self
.params
.get('writedesktoplink', False):
2621 if url_link
or webloc_link
or desktop_link
:
2622 if 'webpage_url' not in info_dict
:
2623 self
.report_error('Cannot write internet shortcut file because the "webpage_url" field is missing in the media information')
2625 ascii_url
= iri_to_uri(info_dict
['webpage_url'])
2627 def _write_link_file(extension
, template
, newline
, embed_filename
):
2628 linkfn
= replace_extension(full_filename
, extension
, info_dict
.get('ext'))
2629 if self
.params
.get('overwrites', True) and os
.path
.exists(encodeFilename(linkfn
)):
2630 self
.to_screen('[info] Internet shortcut is already present')
2633 self
.to_screen('[info] Writing internet shortcut to: ' + linkfn
)
2634 with io
.open(encodeFilename(to_high_limit_path(linkfn
)), 'w', encoding
='utf-8', newline
=newline
) as linkfile
:
2635 template_vars
= {'url': ascii_url}
2637 template_vars
['filename'] = linkfn
[:-(len(extension
) + 1)]
2638 linkfile
.write(template
% template_vars
)
2639 except (OSError, IOError):
2640 self
.report_error('Cannot write internet shortcut ' + linkfn
)
2645 if not _write_link_file('url', DOT_URL_LINK_TEMPLATE
, '\r\n', embed_filename
=False):
2648 if not _write_link_file('webloc', DOT_WEBLOC_LINK_TEMPLATE
, '\n', embed_filename
=False):
2651 if not _write_link_file('desktop', DOT_DESKTOP_LINK_TEMPLATE
, '\n', embed_filename
=True):
2655 info_dict
, files_to_move
= self
.pre_process(info_dict
, 'before_dl', files_to_move
)
2656 except PostProcessingError
as err
:
2657 self
.report_error('Preprocessing: %s' % str(err
))
2660 must_record_download_archive
= False
2661 if self
.params
.get('skip_download', False):
2662 info_dict
['filepath'] = temp_filename
2663 info_dict
['__finaldir'] = os
.path
.dirname(os
.path
.abspath(encodeFilename(full_filename
)))
2664 info_dict
['__files_to_move'] = files_to_move
2665 info_dict
= self
.run_pp(MoveFilesAfterDownloadPP(self
, False), info_dict
)
2668 info_dict
.setdefault('__postprocessors', [])
2671 def existing_file(*filepaths
):
2672 ext
= info_dict
.get('ext')
2673 final_ext
= self
.params
.get('final_ext', ext
)
2675 for file in orderedSet(filepaths
):
2676 if final_ext
!= ext
:
2677 converted
= replace_extension(file, final_ext
, ext
)
2678 if os
.path
.exists(encodeFilename(converted
)):
2679 existing_files
.append(converted
)
2680 if os
.path
.exists(encodeFilename(file)):
2681 existing_files
.append(file)
2683 if not existing_files
or self
.params
.get('overwrites', False):
2684 for file in orderedSet(existing_files
):
2685 self
.report_file_delete(file)
2686 os
.remove(encodeFilename(file))
2689 info_dict
['ext'] = os
.path
.splitext(existing_files
[0])[1][1:]
2690 return existing_files
[0]
2693 if info_dict
.get('requested_formats') is not None:
2695 def compatible_formats(formats
):
2696 # TODO: some formats actually allow this (mkv, webm, ogg, mp4), but not all of them.
2697 video_formats
= [format
for format
in formats
if format
.get('vcodec') != 'none']
2698 audio_formats
= [format
for format
in formats
if format
.get('acodec') != 'none']
2699 if len(video_formats
) > 2 or len(audio_formats
) > 2:
2703 exts
= set(format
.get('ext') for format
in formats
)
2705 set(('mp3', 'mp4', 'm4a', 'm4p', 'm4b', 'm4r', 'm4v', 'ismv', 'isma')),
2708 for ext_sets
in COMPATIBLE_EXTS
:
2709 if ext_sets
.issuperset(exts
):
2711 # TODO: Check acodec/vcodec
2714 requested_formats
= info_dict
['requested_formats']
2715 old_ext
= info_dict
['ext']
2716 if self
.params
.get('merge_output_format') is None:
2717 if not compatible_formats(requested_formats
):
2718 info_dict
['ext'] = 'mkv'
2719 self
.report_warning(
2720 'Requested formats are incompatible for merge and will be merged into mkv')
2721 if (info_dict
['ext'] == 'webm'
2722 and info_dict
.get('thumbnails')
2723 # check with type instead of pp_key, __name__, or isinstance
2724 # since we dont want any custom PPs to trigger this
2725 and any(type(pp
) == EmbedThumbnailPP
for pp
in self
._pps
['post_process'])):
2726 info_dict
['ext'] = 'mkv'
2727 self
.report_warning(
2728 'webm doesn\'t support embedding a thumbnail, mkv will be used')
2729 new_ext
= info_dict
['ext']
2731 def correct_ext(filename
, ext
=new_ext
):
2734 filename_real_ext
= os
.path
.splitext(filename
)[1][1:]
2736 os
.path
.splitext(filename
)[0]
2737 if filename_real_ext
in (old_ext
, new_ext
)
2739 return '%s.%s' % (filename_wo_ext
, ext
)
2741 # Ensure filename always has a correct extension for successful merge
2742 full_filename
= correct_ext(full_filename
)
2743 temp_filename
= correct_ext(temp_filename
)
2744 dl_filename
= existing_file(full_filename
, temp_filename
)
2745 info_dict
['__real_download'] = False
2747 if dl_filename
is not None:
2748 self
.report_file_already_downloaded(dl_filename
)
2749 elif get_suitable_downloader(info_dict
, self
.params
, to_stdout
=temp_filename
== '-'):
2750 info_dict
['url'] = '\n'.join(f
['url'] for f
in requested_formats
)
2751 success
, real_download
= self
.dl(temp_filename
, info_dict
)
2752 info_dict
['__real_download'] = real_download
2755 merger
= FFmpegMergerPP(self
)
2756 if self
.params
.get('allow_unplayable_formats'):
2757 self
.report_warning(
2758 'You have requested merging of multiple formats '
2759 'while also allowing unplayable formats to be downloaded. '
2760 'The formats won\'t be merged to prevent data corruption.')
2761 elif not merger
.available
:
2762 self
.report_warning(
2763 'You have requested merging of multiple formats but ffmpeg is not installed. '
2764 'The formats won\'t be merged.')
2766 if temp_filename
== '-':
2767 reason
= ('using a downloader other than ffmpeg' if FFmpegFD
.can_merge_formats(info_dict
)
2768 else 'but the formats are incompatible for simultaneous download' if merger
.available
2769 else 'but ffmpeg is not installed')
2770 self
.report_warning(
2771 f
'You have requested downloading multiple formats to stdout {reason}. '
2772 'The formats will be streamed one after the other')
2773 fname
= temp_filename
2774 for f
in requested_formats
:
2775 new_info
= dict(info_dict
)
2776 del new_info
['requested_formats']
2778 if temp_filename
!= '-':
2779 fname
= prepend_extension(
2780 correct_ext(temp_filename
, new_info
['ext']),
2781 'f%s' % f
['format_id'], new_info
['ext'])
2782 if not self
._ensure
_dir
_exists
(fname
):
2784 f
['filepath'] = fname
2785 downloaded
.append(fname
)
2786 partial_success
, real_download
= self
.dl(fname
, new_info
)
2787 info_dict
['__real_download'] = info_dict
['__real_download'] or real_download
2788 success
= success
and partial_success
2789 if merger
.available
and not self
.params
.get('allow_unplayable_formats'):
2790 info_dict
['__postprocessors'].append(merger
)
2791 info_dict
['__files_to_merge'] = downloaded
2792 # Even if there were no downloads, it is being merged only now
2793 info_dict
['__real_download'] = True
2795 for file in downloaded
:
2796 files_to_move
[file] = None
2798 # Just a single file
2799 dl_filename
= existing_file(full_filename
, temp_filename
)
2800 if dl_filename
is None or dl_filename
== temp_filename
:
2801 # dl_filename == temp_filename could mean that the file was partially downloaded with --no-part.
2802 # So we should try to resume the download
2803 success
, real_download
= self
.dl(temp_filename
, info_dict
)
2804 info_dict
['__real_download'] = real_download
2806 self
.report_file_already_downloaded(dl_filename
)
2808 dl_filename
= dl_filename
or temp_filename
2809 info_dict
['__finaldir'] = os
.path
.dirname(os
.path
.abspath(encodeFilename(full_filename
)))
2811 except network_exceptions
as err
:
2812 self
.report_error('unable to download video data: %s' % error_to_compat_str(err
))
2814 except (OSError, IOError) as err
:
2815 raise UnavailableVideoError(err
)
2816 except (ContentTooShortError
, ) as err
:
2817 self
.report_error('content too short (expected %s bytes and served %s)' % (err
.expected
, err
.downloaded
))
2820 if success
and full_filename
!= '-':
2824 fixup_policy
= self
.params
.get('fixup')
2825 vid
= info_dict
['id']
2827 if fixup_policy
in ('ignore', 'never'):
2829 elif fixup_policy
== 'warn':
2831 elif fixup_policy
!= 'force':
2832 assert fixup_policy
in ('detect_or_warn', None)
2833 if not info_dict
.get('__real_download'):
2836 def ffmpeg_fixup(cndn
, msg
, cls
):
2840 self
.report_warning(f
'{vid}: {msg}')
2844 info_dict
['__postprocessors'].append(pp
)
2846 self
.report_warning(f
'{vid}: {msg}. Install ffmpeg to fix this automatically')
2848 stretched_ratio
= info_dict
.get('stretched_ratio')
2850 stretched_ratio
not in (1, None),
2851 f
'Non-uniform pixel ratio {stretched_ratio}',
2852 FFmpegFixupStretchedPP
)
2855 (info_dict
.get('requested_formats') is None
2856 and info_dict
.get('container') == 'm4a_dash'
2857 and info_dict
.get('ext') == 'm4a'),
2858 'writing DASH m4a. Only some players support this container',
2861 downloader
= (get_suitable_downloader(info_dict
, self
.params
).__name
__
2862 if 'protocol' in info_dict
else None)
2863 ffmpeg_fixup(info_dict
.get('requested_formats') is None and downloader
== 'HlsFD',
2864 'malformed AAC bitstream detected', FFmpegFixupM3u8PP
)
2865 ffmpeg_fixup(downloader
== 'WebSocketFragmentFD', 'malformed timestamps detected', FFmpegFixupTimestampPP
)
2866 ffmpeg_fixup(downloader
== 'WebSocketFragmentFD', 'malformed duration detected', FFmpegFixupDurationPP
)
2870 info_dict
= self
.post_process(dl_filename
, info_dict
, files_to_move
)
2871 except PostProcessingError
as err
:
2872 self
.report_error('Postprocessing: %s' % str(err
))
2875 for ph
in self
._post
_hooks
:
2876 ph(info_dict
['filepath'])
2877 except Exception as err
:
2878 self
.report_error('post hooks: %s' % str(err
))
2880 must_record_download_archive
= True
2882 if must_record_download_archive
or self
.params
.get('force_write_download_archive', False):
2883 self
.record_download_archive(info_dict
)
2884 max_downloads
= self
.params
.get('max_downloads')
2885 if max_downloads
is not None and self
._num
_downloads
>= int(max_downloads
):
2886 raise MaxDownloadsReached()
2888 def download(self
, url_list
):
2889 """Download a given list of URLs."""
2890 outtmpl
= self
.outtmpl_dict
['default']
2891 if (len(url_list
) > 1
2893 and '%' not in outtmpl
2894 and self
.params
.get('max_downloads') != 1):
2895 raise SameFileError(outtmpl
)
2897 for url
in url_list
:
2899 # It also downloads the videos
2900 res
= self
.extract_info(
2901 url
, force_generic_extractor
=self
.params
.get('force_generic_extractor', False))
2902 except UnavailableVideoError
:
2903 self
.report_error('unable to download video')
2904 except MaxDownloadsReached
:
2905 self
.to_screen('[info] Maximum number of downloads reached')
2907 except ExistingVideoReached
:
2908 self
.to_screen('[info] Encountered a video that is already in the archive, stopping due to --break-on-existing')
2910 except RejectedVideoReached
:
2911 self
.to_screen('[info] Encountered a video that did not match filter, stopping due to --break-on-reject')
2914 if self
.params
.get('dump_single_json', False):
2915 self
.post_extract(res
)
2916 self
.to_stdout(json
.dumps(self
.sanitize_info(res
)))
2918 return self
._download
_retcode
2920 def download_with_info_file(self
, info_filename
):
2921 with contextlib
.closing(fileinput
.FileInput(
2922 [info_filename
], mode
='r',
2923 openhook
=fileinput
.hook_encoded('utf-8'))) as f
:
2924 # FileInput doesn't have a read method, we can't call json.load
2925 info
= self
.sanitize_info(json
.loads('\n'.join(f
)), self
.params
.get('clean_infojson', True))
2927 self
.process_ie_result(info
, download
=True)
2928 except (DownloadError
, EntryNotInPlaylist
, ThrottledDownload
):
2929 webpage_url
= info
.get('webpage_url')
2930 if webpage_url
is not None:
2931 self
.report_warning('The info failed to download, trying with "%s"' % webpage_url
)
2932 return self
.download([webpage_url
])
2935 return self
._download
_retcode
2938 def sanitize_info(info_dict
, remove_private_keys
=False):
2939 ''' Sanitize the infodict for converting to json '''
2940 if info_dict
is None:
2942 info_dict
.setdefault('epoch', int(time
.time()))
2943 remove_keys
= {'__original_infodict'}
# Always remove this since this may contain a copy of the entire dict
2944 keep_keys
= ['_type'], # Always keep this to facilitate load-info-json
2945 if remove_private_keys
:
2947 'requested_formats', 'requested_subtitles', 'requested_entries',
2948 'filepath', 'entries', 'original_url', 'playlist_autonumber',
2950 empty_values
= (None, {}, [], set(), tuple())
2951 reject
= lambda k
, v
: k
not in keep_keys
and (
2952 k
.startswith('_') or k
in remove_keys
or v
in empty_values
)
2954 reject
= lambda k
, v
: k
in remove_keys
2955 filter_fn
= lambda obj
: (
2956 list(map(filter_fn
, obj
)) if isinstance(obj
, (LazyList
, list, tuple, set))
2957 else obj
if not isinstance(obj
, dict)
2958 else dict((k
, filter_fn(v
)) for k
, v
in obj
.items() if not reject(k
, v
)))
2959 return filter_fn(info_dict
)
2962 def filter_requested_info(info_dict
, actually_filter
=True):
2963 ''' Alias of sanitize_info for backward compatibility '''
2964 return YoutubeDL
.sanitize_info(info_dict
, actually_filter
)
2966 def run_pp(self
, pp
, infodict
):
2967 files_to_delete
= []
2968 if '__files_to_move' not in infodict
:
2969 infodict
['__files_to_move'] = {}
2971 files_to_delete
, infodict
= pp
.run(infodict
)
2972 except PostProcessingError
as e
:
2973 # Must be True and not 'only_download'
2974 if self
.params
.get('ignoreerrors') is True:
2975 self
.report_error(e
)
2979 if not files_to_delete
:
2981 if self
.params
.get('keepvideo', False):
2982 for f
in files_to_delete
:
2983 infodict
['__files_to_move'].setdefault(f
, '')
2985 for old_filename
in set(files_to_delete
):
2986 self
.to_screen('Deleting original file %s (pass -k to keep)' % old_filename
)
2988 os
.remove(encodeFilename(old_filename
))
2989 except (IOError, OSError):
2990 self
.report_warning('Unable to remove downloaded original file')
2991 if old_filename
in infodict
['__files_to_move']:
2992 del infodict
['__files_to_move'][old_filename
]
2996 def post_extract(info_dict
):
2997 def actual_post_extract(info_dict
):
2998 if info_dict
.get('_type') in ('playlist', 'multi_video'):
2999 for video_dict
in info_dict
.get('entries', {}):
3000 actual_post_extract(video_dict
or {})
3003 post_extractor
= info_dict
.get('__post_extractor') or (lambda: {})
3004 extra
= post_extractor().items()
3005 info_dict
.update(extra
)
3006 info_dict
.pop('__post_extractor', None)
3008 original_infodict
= info_dict
.get('__original_infodict') or {}
3009 original_infodict
.update(extra
)
3010 original_infodict
.pop('__post_extractor', None)
3012 actual_post_extract(info_dict
or {})
3014 def pre_process(self
, ie_info
, key
='pre_process', files_to_move
=None):
3015 info
= dict(ie_info
)
3016 info
['__files_to_move'] = files_to_move
or {}
3017 for pp
in self
._pps
[key
]:
3018 info
= self
.run_pp(pp
, info
)
3019 return info
, info
.pop('__files_to_move', None)
3021 def post_process(self
, filename
, ie_info
, files_to_move
=None):
3022 """Run all the postprocessors on the given file."""
3023 info
= dict(ie_info
)
3024 info
['filepath'] = filename
3025 info
['__files_to_move'] = files_to_move
or {}
3027 for pp
in ie_info
.get('__postprocessors', []) + self
._pps
['post_process']:
3028 info
= self
.run_pp(pp
, info
)
3029 info
= self
.run_pp(MoveFilesAfterDownloadPP(self
), info
)
3030 del info
['__files_to_move']
3031 for pp
in self
._pps
['after_move']:
3032 info
= self
.run_pp(pp
, info
)
3035 def _make_archive_id(self
, info_dict
):
3036 video_id
= info_dict
.get('id')
3039 # Future-proof against any change in case
3040 # and backwards compatibility with prior versions
3041 extractor
= info_dict
.get('extractor_key') or info_dict
.get('ie_key') # key in a playlist
3042 if extractor
is None:
3043 url
= str_or_none(info_dict
.get('url'))
3046 # Try to find matching extractor for the URL and take its ie_key
3047 for ie_key
, ie
in self
._ies
.items():
3048 if ie
.suitable(url
):
3053 return '%s %s' % (extractor
.lower(), video_id
)
3055 def in_download_archive(self
, info_dict
):
3056 fn
= self
.params
.get('download_archive')
3060 vid_id
= self
._make
_archive
_id
(info_dict
)
3062 return False # Incomplete video information
3064 return vid_id
in self
.archive
3066 def record_download_archive(self
, info_dict
):
3067 fn
= self
.params
.get('download_archive')
3070 vid_id
= self
._make
_archive
_id
(info_dict
)
3072 with locked_file(fn
, 'a', encoding
='utf-8') as archive_file
:
3073 archive_file
.write(vid_id
+ '\n')
3074 self
.archive
.add(vid_id
)
3077 def format_resolution(format
, default
='unknown'):
3078 is_images
= format
.get('vcodec') == 'none' and format
.get('acodec') == 'none'
3079 if format
.get('vcodec') == 'none' and format
.get('acodec') != 'none':
3081 if format
.get('resolution') is not None:
3082 return format
['resolution']
3083 if format
.get('width') and format
.get('height'):
3084 res
= '%dx%d' % (format
['width'], format
['height'])
3085 elif format
.get('height'):
3086 res
= '%sp' % format
['height']
3087 elif format
.get('width'):
3088 res
= '%dx?' % format
['width']
3093 return f
'{res} images' if is_images
else res
3095 def _format_note(self
, fdict
):
3097 if fdict
.get('ext') in ['f4f', 'f4m']:
3098 res
+= '(unsupported) '
3099 if fdict
.get('language'):
3102 res
+= '[%s] ' % fdict
['language']
3103 if fdict
.get('format_note') is not None:
3104 res
+= fdict
['format_note'] + ' '
3105 if fdict
.get('tbr') is not None:
3106 res
+= '%4dk ' % fdict
['tbr']
3107 if fdict
.get('container') is not None:
3110 res
+= '%s container' % fdict
['container']
3111 if (fdict
.get('vcodec') is not None
3112 and fdict
.get('vcodec') != 'none'):
3115 res
+= fdict
['vcodec']
3116 if fdict
.get('vbr') is not None:
3118 elif fdict
.get('vbr') is not None and fdict
.get('abr') is not None:
3120 if fdict
.get('vbr') is not None:
3121 res
+= '%4dk' % fdict
['vbr']
3122 if fdict
.get('fps') is not None:
3125 res
+= '%sfps' % fdict
['fps']
3126 if fdict
.get('acodec') is not None:
3129 if fdict
['acodec'] == 'none':
3132 res
+= '%-5s' % fdict
['acodec']
3133 elif fdict
.get('abr') is not None:
3137 if fdict
.get('abr') is not None:
3138 res
+= '@%3dk' % fdict
['abr']
3139 if fdict
.get('asr') is not None:
3140 res
+= ' (%5dHz)' % fdict
['asr']
3141 if fdict
.get('filesize') is not None:
3144 res
+= format_bytes(fdict
['filesize'])
3145 elif fdict
.get('filesize_approx') is not None:
3148 res
+= '~' + format_bytes(fdict
['filesize_approx'])
3151 def list_formats(self
, info_dict
):
3152 formats
= info_dict
.get('formats', [info_dict
])
3154 'list-formats' not in self
.params
.get('compat_opts', [])
3155 and self
.params
.get('listformats_table', True) is not False)
3159 format_field(f
, 'format_id'),
3160 format_field(f
, 'ext'),
3161 self
.format_resolution(f
),
3162 format_field(f
, 'fps', '%d'),
3164 format_field(f
, 'filesize', ' %s', func
=format_bytes
) + format_field(f
, 'filesize_approx', '~%s', func
=format_bytes
),
3165 format_field(f
, 'tbr', '%4dk'),
3166 shorten_protocol_name(f
.get('protocol', '').replace("native", "n")),
3168 format_field(f
, 'vcodec', default
='unknown').replace('none', ''),
3169 format_field(f
, 'vbr', '%4dk'),
3170 format_field(f
, 'acodec', default
='unknown').replace('none', ''),
3171 format_field(f
, 'abr', '%3dk'),
3172 format_field(f
, 'asr', '%5dHz'),
3173 ', '.join(filter(None, (
3174 'UNSUPPORTED' if f
.get('ext') in ('f4f', 'f4m') else '',
3175 format_field(f
, 'language', '[%s]'),
3176 format_field(f
, 'format_note'),
3177 format_field(f
, 'container', ignore
=(None, f
.get('ext'))),
3179 ] for f
in formats
if f
.get('preference') is None or f
['preference'] >= -1000]
3180 header_line
= ['ID', 'EXT', 'RESOLUTION', 'FPS', '|', ' FILESIZE', ' TBR', 'PROTO',
3181 '|', 'VCODEC', ' VBR', 'ACODEC', ' ABR', ' ASR', 'MORE INFO']
3185 format_field(f
, 'format_id'),
3186 format_field(f
, 'ext'),
3187 self
.format_resolution(f
),
3188 self
._format
_note
(f
)]
3190 if f
.get('preference') is None or f
['preference'] >= -1000]
3191 header_line
= ['format code', 'extension', 'resolution', 'note']
3194 '[info] Available formats for %s:' % info_dict
['id'])
3195 self
.to_stdout(render_table(
3196 header_line
, table
, delim
=new_format
, extraGap
=(0 if new_format
else 1), hideEmpty
=new_format
))
3198 def list_thumbnails(self
, info_dict
):
3199 thumbnails
= list(info_dict
.get('thumbnails'))
3201 self
.to_screen('[info] No thumbnails present for %s' % info_dict
['id'])
3205 '[info] Thumbnails for %s:' % info_dict
['id'])
3206 self
.to_stdout(render_table(
3207 ['ID', 'width', 'height', 'URL'],
3208 [[t
['id'], t
.get('width', 'unknown'), t
.get('height', 'unknown'), t
['url']] for t
in thumbnails
]))
3210 def list_subtitles(self
, video_id
, subtitles
, name
='subtitles'):
3212 self
.to_screen('%s has no %s' % (video_id
, name
))
3215 'Available %s for %s:' % (name
, video_id
))
3217 def _row(lang
, formats
):
3218 exts
, names
= zip(*((f
['ext'], f
.get('name') or 'unknown') for f
in reversed(formats
)))
3219 if len(set(names
)) == 1:
3220 names
= [] if names
[0] == 'unknown' else names
[:1]
3221 return [lang
, ', '.join(names
), ', '.join(exts
)]
3223 self
.to_stdout(render_table(
3224 ['Language', 'Name', 'Formats'],
3225 [_row(lang
, formats
) for lang
, formats
in subtitles
.items()],
3228 def urlopen(self
, req
):
3229 """ Start an HTTP download """
3230 if isinstance(req
, compat_basestring
):
3231 req
= sanitized_Request(req
)
3232 return self
._opener
.open(req
, timeout
=self
._socket
_timeout
)
3234 def print_debug_header(self
):
3235 if not self
.params
.get('verbose'):
3238 stdout_encoding
= getattr(
3239 sys
.stdout
, 'encoding', 'missing (%s)' % type(sys
.stdout
).__name
__)
3241 '[debug] Encodings: locale %s, fs %s, out %s, pref %s\n' % (
3242 locale
.getpreferredencoding(),
3243 sys
.getfilesystemencoding(),
3245 self
.get_encoding()))
3246 write_string(encoding_str
, encoding
=None)
3248 source
= detect_variant()
3249 self
._write
_string
('[debug] yt-dlp version %s%s\n' % (__version__
, '' if source
== 'unknown' else f
' ({source})'))
3251 self
._write
_string
('[debug] Lazy loading extractors enabled\n')
3252 if plugin_extractors
or plugin_postprocessors
:
3253 self
._write
_string
('[debug] Plugins: %s\n' % [
3254 '%s%s' % (klass
.__name
__, '' if klass
.__name
__ == name
else f
' as {name}')
3255 for name
, klass
in itertools
.chain(plugin_extractors
.items(), plugin_postprocessors
.items())])
3256 if self
.params
.get('compat_opts'):
3258 '[debug] Compatibility options: %s\n' % ', '.join(self
.params
.get('compat_opts')))
3260 sp
= subprocess
.Popen(
3261 ['git', 'rev-parse', '--short', 'HEAD'],
3262 stdout
=subprocess
.PIPE
, stderr
=subprocess
.PIPE
,
3263 cwd
=os
.path
.dirname(os
.path
.abspath(__file__
)))
3264 out
, err
= process_communicate_or_kill(sp
)
3265 out
= out
.decode().strip()
3266 if re
.match('[0-9a-f]+', out
):
3267 self
._write
_string
('[debug] Git HEAD: %s\n' % out
)
3274 def python_implementation():
3275 impl_name
= platform
.python_implementation()
3276 if impl_name
== 'PyPy' and hasattr(sys
, 'pypy_version_info'):
3277 return impl_name
+ ' version %d.%d.%d' % sys
.pypy_version_info
[:3]
3280 self
._write
_string
('[debug] Python version %s (%s %s) - %s\n' % (
3281 platform
.python_version(),
3282 python_implementation(),
3283 platform
.architecture()[0],
3286 exe_versions
= FFmpegPostProcessor
.get_versions(self
)
3287 exe_versions
['rtmpdump'] = rtmpdump_version()
3288 exe_versions
['phantomjs'] = PhantomJSwrapper
._version
()
3289 exe_str
= ', '.join(
3290 f
'{exe} {v}' for exe
, v
in sorted(exe_versions
.items()) if v
3292 self
._write
_string
('[debug] exe versions: %s\n' % exe_str
)
3294 from .downloader
.websocket
import has_websockets
3295 from .postprocessor
.embedthumbnail
import has_mutagen
3296 from .cookies
import SQLITE_AVAILABLE
, KEYRING_AVAILABLE
3298 lib_str
= ', '.join(sorted(filter(None, (
3299 compat_pycrypto_AES
and compat_pycrypto_AES
.__name
__.split('.')[0],
3300 has_websockets
and 'websockets',
3301 has_mutagen
and 'mutagen',
3302 SQLITE_AVAILABLE
and 'sqlite',
3303 KEYRING_AVAILABLE
and 'keyring',
3305 self
._write
_string
('[debug] Optional libraries: %s\n' % lib_str
)
3306 self
._write
_string
('[debug] ANSI escape support: stdout = %s, stderr = %s\n' % (
3307 supports_terminal_sequences(self
._screen
_file
),
3308 supports_terminal_sequences(self
._err
_file
)))
3311 for handler
in self
._opener
.handlers
:
3312 if hasattr(handler
, 'proxies'):
3313 proxy_map
.update(handler
.proxies
)
3314 self
._write
_string
('[debug] Proxy map: ' + compat_str(proxy_map
) + '\n')
3316 if self
.params
.get('call_home', False):
3317 ipaddr
= self
.urlopen('https://yt-dl.org/ip').read().decode('utf-8')
3318 self
._write
_string
('[debug] Public IP address: %s\n' % ipaddr
)
3320 latest_version
= self
.urlopen(
3321 'https://yt-dl.org/latest/version').read().decode('utf-8')
3322 if version_tuple(latest_version
) > version_tuple(__version__
):
3323 self
.report_warning(
3324 'You are using an outdated version (newest version: %s)! '
3325 'See https://yt-dl.org/update if you need help updating.' %
3328 def _setup_opener(self
):
3329 timeout_val
= self
.params
.get('socket_timeout')
3330 self
._socket
_timeout
= 600 if timeout_val
is None else float(timeout_val
)
3332 opts_cookiesfrombrowser
= self
.params
.get('cookiesfrombrowser')
3333 opts_cookiefile
= self
.params
.get('cookiefile')
3334 opts_proxy
= self
.params
.get('proxy')
3336 self
.cookiejar
= load_cookies(opts_cookiefile
, opts_cookiesfrombrowser
, self
)
3338 cookie_processor
= YoutubeDLCookieProcessor(self
.cookiejar
)
3339 if opts_proxy
is not None:
3340 if opts_proxy
== '':
3343 proxies
= {'http': opts_proxy, 'https': opts_proxy}
3345 proxies
= compat_urllib_request
.getproxies()
3346 # Set HTTPS proxy to HTTP one if given (https://github.com/ytdl-org/youtube-dl/issues/805)
3347 if 'http' in proxies
and 'https' not in proxies
:
3348 proxies
['https'] = proxies
['http']
3349 proxy_handler
= PerRequestProxyHandler(proxies
)
3351 debuglevel
= 1 if self
.params
.get('debug_printtraffic') else 0
3352 https_handler
= make_HTTPS_handler(self
.params
, debuglevel
=debuglevel
)
3353 ydlh
= YoutubeDLHandler(self
.params
, debuglevel
=debuglevel
)
3354 redirect_handler
= YoutubeDLRedirectHandler()
3355 data_handler
= compat_urllib_request_DataHandler()
3357 # When passing our own FileHandler instance, build_opener won't add the
3358 # default FileHandler and allows us to disable the file protocol, which
3359 # can be used for malicious purposes (see
3360 # https://github.com/ytdl-org/youtube-dl/issues/8227)
3361 file_handler
= compat_urllib_request
.FileHandler()
3363 def file_open(*args
, **kwargs
):
3364 raise compat_urllib_error
.URLError('file:// scheme is explicitly disabled in yt-dlp for security reasons')
3365 file_handler
.file_open
= file_open
3367 opener
= compat_urllib_request
.build_opener(
3368 proxy_handler
, https_handler
, cookie_processor
, ydlh
, redirect_handler
, data_handler
, file_handler
)
3370 # Delete the default user-agent header, which would otherwise apply in
3371 # cases where our custom HTTP handler doesn't come into play
3372 # (See https://github.com/ytdl-org/youtube-dl/issues/1309 for details)
3373 opener
.addheaders
= []
3374 self
._opener
= opener
3376 def encode(self
, s
):
3377 if isinstance(s
, bytes):
3378 return s
# Already encoded
3381 return s
.encode(self
.get_encoding())
3382 except UnicodeEncodeError as err
:
3383 err
.reason
= err
.reason
+ '. Check your system encoding configuration or use the --encoding option.'
3386 def get_encoding(self
):
3387 encoding
= self
.params
.get('encoding')
3388 if encoding
is None:
3389 encoding
= preferredencoding()
3392 def _write_info_json(self
, label
, ie_result
, infofn
):
3393 ''' Write infojson and returns True = written, False = skip, None = error '''
3394 if not self
.params
.get('writeinfojson'):
3397 self
.write_debug(f
'Skipping writing {label} infojson')
3399 elif not self
._ensure
_dir
_exists
(infofn
):
3401 elif not self
.params
.get('overwrites', True) and os
.path
.exists(infofn
):
3402 self
.to_screen(f
'[info] {label.title()} metadata is already present')
3404 self
.to_screen(f
'[info] Writing {label} metadata as JSON to: {infofn}')
3406 write_json_file(self
.sanitize_info(ie_result
, self
.params
.get('clean_infojson', True)), infofn
)
3407 except (OSError, IOError):
3408 self
.report_error(f
'Cannot write {label} metadata to JSON file {infofn}')
3412 def _write_description(self
, label
, ie_result
, descfn
):
3413 ''' Write description and returns True = written, False = skip, None = error '''
3414 if not self
.params
.get('writedescription'):
3417 self
.write_debug(f
'Skipping writing {label} description')
3419 elif not self
._ensure
_dir
_exists
(descfn
):
3421 elif not self
.params
.get('overwrites', True) and os
.path
.exists(descfn
):
3422 self
.to_screen(f
'[info] {label.title()} description is already present')
3423 elif ie_result
.get('description') is None:
3424 self
.report_warning(f
'There\'s no {label} description to write')
3428 self
.to_screen(f
'[info] Writing {label} description to: {descfn}')
3429 with io
.open(encodeFilename(descfn
), 'w', encoding
='utf-8') as descfile
:
3430 descfile
.write(ie_result
['description'])
3431 except (OSError, IOError):
3432 self
.report_error(f
'Cannot write {label} description file {descfn}')
3436 def _write_subtitles(self
, info_dict
, filename
):
3437 ''' Write subtitles to file and return list of (sub_filename, final_sub_filename); or None if error'''
3439 subtitles
= info_dict
.get('requested_subtitles')
3440 if not subtitles
or not (self
.params
.get('writesubtitles') or self
.params
.get('writeautomaticsub')):
3441 # subtitles download errors are already managed as troubles in relevant IE
3442 # that way it will silently go on when used with unsupporting IE
3445 sub_filename_base
= self
.prepare_filename(info_dict
, 'subtitle')
3446 if not sub_filename_base
:
3447 self
.to_screen('[info] Skipping writing video subtitles')
3449 for sub_lang
, sub_info
in subtitles
.items():
3450 sub_format
= sub_info
['ext']
3451 sub_filename
= subtitles_filename(filename
, sub_lang
, sub_format
, info_dict
.get('ext'))
3452 sub_filename_final
= subtitles_filename(sub_filename_base
, sub_lang
, sub_format
, info_dict
.get('ext'))
3453 if not self
.params
.get('overwrites', True) and os
.path
.exists(sub_filename
):
3454 self
.to_screen(f
'[info] Video subtitle {sub_lang}.{sub_format} is already present')
3455 sub_info
['filepath'] = sub_filename
3456 ret
.append((sub_filename
, sub_filename_final
))
3459 self
.to_screen(f
'[info] Writing video subtitles to: {sub_filename}')
3460 if sub_info
.get('data') is not None:
3462 # Use newline='' to prevent conversion of newline characters
3463 # See https://github.com/ytdl-org/youtube-dl/issues/10268
3464 with io
.open(sub_filename
, 'w', encoding
='utf-8', newline
='') as subfile
:
3465 subfile
.write(sub_info
['data'])
3466 sub_info
['filepath'] = sub_filename
3467 ret
.append((sub_filename
, sub_filename_final
))
3469 except (OSError, IOError):
3470 self
.report_error(f
'Cannot write video subtitles file {sub_filename}')
3474 sub_copy
= sub_info
.copy()
3475 sub_copy
.setdefault('http_headers', info_dict
.get('http_headers'))
3476 self
.dl(sub_filename
, sub_copy
, subtitle
=True)
3477 sub_info
['filepath'] = sub_filename
3478 ret
.append((sub_filename
, sub_filename_final
))
3479 except (ExtractorError
, IOError, OSError, ValueError) + network_exceptions
as err
:
3480 self
.report_warning(f
'Unable to download video subtitles for {sub_lang!r}: {err}')
3484 def _write_thumbnails(self
, label
, info_dict
, filename
, thumb_filename_base
=None):
3485 ''' Write thumbnails to file and return list of (thumb_filename, final_thumb_filename) '''
3486 write_all
= self
.params
.get('write_all_thumbnails', False)
3487 thumbnails
, ret
= [], []
3488 if write_all
or self
.params
.get('writethumbnail', False):
3489 thumbnails
= info_dict
.get('thumbnails') or []
3490 multiple
= write_all
and len(thumbnails
) > 1
3492 if thumb_filename_base
is None:
3493 thumb_filename_base
= filename
3494 if thumbnails
and not thumb_filename_base
:
3495 self
.write_debug(f
'Skipping writing {label} thumbnail')
3498 for t
in thumbnails
[::-1]:
3499 thumb_ext
= (f
'{t["id"]}.' if multiple
else '') + determine_ext(t
['url'], 'jpg')
3500 thumb_display_id
= f
'{label} thumbnail' + (f
' {t["id"]}' if multiple
else '')
3501 thumb_filename
= replace_extension(filename
, thumb_ext
, info_dict
.get('ext'))
3502 thumb_filename_final
= replace_extension(thumb_filename_base
, thumb_ext
, info_dict
.get('ext'))
3504 if not self
.params
.get('overwrites', True) and os
.path
.exists(thumb_filename
):
3505 ret
.append((thumb_filename
, thumb_filename_final
))
3506 t
['filepath'] = thumb_filename
3507 self
.to_screen(f
'[info] {thumb_display_id.title()} is already present')
3509 self
.to_screen(f
'[info] Downloading {thumb_display_id} ...')
3511 uf
= self
.urlopen(t
['url'])
3512 self
.to_screen(f
'[info] Writing {thumb_display_id} to: {thumb_filename}')
3513 with open(encodeFilename(thumb_filename
), 'wb') as thumbf
:
3514 shutil
.copyfileobj(uf
, thumbf
)
3515 ret
.append((thumb_filename
, thumb_filename_final
))
3516 t
['filepath'] = thumb_filename
3517 except network_exceptions
as err
:
3518 self
.report_warning(f
'Unable to download {thumb_display_id}: {err}')
3519 if ret
and not write_all
: