mirrors/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-30 12:01:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

[update] Set executable bit-mask

Browse source 
Closes #4621
This commit is contained in:
pukkandan 2022-08-14 19:03:58 +05:30
parent 8f84770acd
commit a6125983ab
No known key found for this signature in database
GPG key ID: 7EEE9E1E817D0A39
1 changed files with 23 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
yt_dlp/update.py
View file

@ -9,7 +9,7 @@
from zipimport import zipimporter from zipimport import zipimporter
from .compat import functools # isort: split from .compat import functools # isort: split
from .compat import compat_realpath from .compat import compat_realpath, compat_shlex_quote
from .utils import ( from .utils import (
Popen, Popen,
cached_method, cached_method,
@ -229,24 +229,32 @@ def update(self):
except OSError: except OSError:
return self._report_permission_error(new_filename) return self._report_permission_error(new_filename)
try: if old_filename:
if old_filename: try:
os.rename(self.filename, old_filename) os.rename(self.filename, old_filename)
except OSError: except OSError:
return self._report_error('Unable to move current version') return self._report_error('Unable to move current version')
try:
if old_filename:
os.rename(new_filename, self.filename)
except OSError:
self._report_error('Unable to overwrite current version')
return os.rename(old_filename, self.filename)
if detect_variant() not in ('win32_exe', 'py2exe'): try:
if old_filename: os.rename(new_filename, self.filename)
os.remove(old_filename) except OSError:
else: self._report_error('Unable to overwrite current version')
return os.rename(old_filename, self.filename)
if detect_variant() in ('win32_exe', 'py2exe'):
atexit.register(Popen, f'ping 127.0.0.1 -n 5 -w 1000 & del /F "{old_filename}"', atexit.register(Popen, f'ping 127.0.0.1 -n 5 -w 1000 & del /F "{old_filename}"',
shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL) shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
elif old_filename:
try:
os.remove(old_filename)
except OSError:
self._report_error('Unable to remove the old version')
try:
os.chmod(self.filename, 0o777)
except OSError:
return self._report_error(
f'Unable to set permissions. Run: sudo chmod a+rx {compat_shlex_quote(self.filename)}')
self.ydl.to_screen(f'Updated yt-dlp to version {self.new_version}') self.ydl.to_screen(f'Updated yt-dlp to version {self.new_version}')
return True return True