depau/youtube-dl
1
0
Fork 0
mirror of https://github.com/ytdl-org/youtube-dl.git synced 2024-06-28 22:59:34 +00:00
Code Issues Releases Wiki Activity

[YoutubeDL] restore macos terminal window title for --console-title

Browse source
This commit is contained in:
kikuyan 2021-06-28 11:39:31 +09:00
parent 379f52a495
commit 5d67355543
1 changed files with 31 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
youtube_dl/YoutubeDL.py
View file

@ -32,6 +32,7 @@ from .compat import (
compat_basestring,
compat_cookiejar,
compat_get_terminal_size,
compat_getenv,
compat_http_client,
compat_kwargs,
compat_numeric_types,
@ -53,6 +54,7 @@ from .utils import (
determine_protocol,
DownloadError,
encode_compat_str,
encodeArgument,
encodeFilename,
error_to_compat_str,
expand_path,
@ -355,6 +357,7 @@ class YoutubeDL(object):
self._num_downloads = 0
self._screen_file = [sys.stdout, sys.stderr][params.get('logtostderr', False)]
self._err_file = sys.stderr
self._apple_terminal = None
self.params = {
# Default parameters
'nocheckcertificate': False,
@ -546,8 +549,29 @@ class YoutubeDL(object):
if self.params.get('simulate', False):
return
if compat_os_name != 'nt' and 'TERM' in os.environ:
# Save the title on stack
self._write_string('\033[22;0t', self._screen_file)
if not (sys.platform == 'darwin' and compat_getenv('TERM_PROGRAM') == 'Apple_Terminal'):
# Save the title on stack
self._write_string('\033[22;0t', self._screen_file)
else:
# macOS Terminal app
window_title = None
try:
ttyname = os.ttyname(self._screen_file.fileno())
opts = ['-e', 'tell app "Terminal" to get custom title of item 1 of (every window whose tty is "%s")' % ttyname]
cmd = ([encodeFilename('osascript', True)]
+ [encodeArgument(o) for o in opts])
p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
stdout, stderr = p.communicate()
if p.returncode == 0:
window_title = stdout.decode('utf-8').strip()
except EnvironmentError:
pass
if window_title is None or window_title and preferredencoding() != 'UTF-8':
self.report_warning('Window title may not be restored correctly')
if window_title is None:
self._apple_terminal = '' # reset on restore anyway
else:
self._apple_terminal = window_title
def restore_console_title(self):
if not self.params.get('consoletitle', False):
@ -555,8 +579,11 @@ class YoutubeDL(object):
if self.params.get('simulate', False):
return
if compat_os_name != 'nt' and 'TERM' in os.environ:
# Restore the title from stack
self._write_string('\033[23;0t', self._screen_file)
if self._apple_terminal is None:
# Restore the title from stack
self._write_string('\033[23;0t', self._screen_file)
else:
self._write_string('\033]0;%s\007' % self._apple_terminal, self._screen_file)
def __enter__(self):
self.save_console_title()