diff options
Diffstat (limited to 'yt_dlp')
-rw-r--r-- | yt_dlp/cookies.py | 23 | ||||
-rw-r--r-- | yt_dlp/utils/_utils.py | 30 |
2 files changed, 34 insertions, 19 deletions
diff --git a/yt_dlp/cookies.py b/yt_dlp/cookies.py index 53fe0ec2d..16f1918e6 100644 --- a/yt_dlp/cookies.py +++ b/yt_dlp/cookies.py @@ -41,30 +41,15 @@ from .utils import ( try_call, write_string, ) +from .utils._utils import _YDLLogger CHROMIUM_BASED_BROWSERS = {'brave', 'chrome', 'chromium', 'edge', 'opera', 'vivaldi'} SUPPORTED_BROWSERS = CHROMIUM_BASED_BROWSERS | {'firefox', 'safari'} -class YDLLogger: - def __init__(self, ydl=None): - self._ydl = ydl - - def debug(self, message): - if self._ydl: - self._ydl.write_debug(message) - - def info(self, message): - if self._ydl: - self._ydl.to_screen(f'[Cookies] {message}') - - def warning(self, message, only_once=False): - if self._ydl: - self._ydl.report_warning(message, only_once) - - def error(self, message): - if self._ydl: - self._ydl.report_error(message) +class YDLLogger(_YDLLogger): + def warning(self, message, only_once=False): # compat + return super().warning(message, once=only_once) class ProgressBar(MultilinePrinter): _DELAY, _timer = 0.1, 0 diff --git a/yt_dlp/utils/_utils.py b/yt_dlp/utils/_utils.py index 3023c33b2..4af955743 100644 --- a/yt_dlp/utils/_utils.py +++ b/yt_dlp/utils/_utils.py @@ -5994,3 +5994,33 @@ class FormatSorter: format['tbr'] = try_call(lambda: format['vbr'] + format['abr']) or None return tuple(self._calculate_field_preference(format, field) for field in self._order) + + +# XXX: Temporary +class _YDLLogger: + def __init__(self, ydl=None): + self._ydl = ydl + + def debug(self, message): + if self._ydl: + self._ydl.write_debug(message) + + def info(self, message): + if self._ydl: + self._ydl.to_screen(message) + + def warning(self, message, *, once=False): + if self._ydl: + self._ydl.report_warning(message, only_once=once) + + def error(self, message, *, is_error=True): + if self._ydl: + self._ydl.report_error(message, is_error=is_error) + + def stdout(self, message): + if self._ydl: + self._ydl.to_stdout(message) + + def stderr(self, message): + if self._ydl: + self._ydl.to_stderr(message) |