mirrors/yt-dlp-deezer
1
0
Fork 0
mirror of https://github.com/LucBerge/yt-dlp.git synced 2025-03-17 19:57:52 +03:00
Code Issues Projects Releases Packages Wiki Activity

[update] Self-restart after update

Browse source
This commit is contained in:
pukkandan 2022-06-21 16:48:32 +05:30
parent 57e0f077a6
commit 8372be7469
No known key found for this signature in database
GPG key ID: 7EEE9E1E817D0A39
2 changed files with 34 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

16
yt_dlp/__init__.py
View file

@ -26,7 +26,7 @@ from .postprocessor import (
MetadataFromFieldPP,
MetadataParserPP,
)
from .update import run_update
from .update import Updater
from .utils import (
NO_DEFAULT,
POSTPROCESS_WHEN,
@ -879,17 +879,23 @@ def _real_main(argv=None):
return
with YoutubeDL(ydl_opts) as ydl:
pre_process = opts.update_self or opts.rm_cachedir
actual_use = all_urls or opts.load_info_filename
if opts.rm_cachedir:
ydl.cache.remove()
if opts.update_self and run_update(ydl) and actual_use:
# If updater returns True, exit. Required for windows
return 100, 'ERROR: The program must exit for the update to complete'
updater = Updater(ydl)
if opts.update_self and updater.update() and actual_use:
if updater.cmd:
return updater.restart()
# This code is reachable only for zip variant in py < 3.10
# It makes sense to exit here, but the old behavior is to continue
ydl.report_warning('Restart yt-dlp to use the updated version')
# return 100, 'ERROR: The program must exit for the update to complete'
if not actual_use:
if opts.update_self or opts.rm_cachedir:
if pre_process:
return ydl._download_retcode
ydl.warn_if_short_id(sys.argv[1:] if argv is None else argv)