Upstream/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-12-13 14:37:43 +01:00
Code Issues Packages Projects Releases Wiki Activity

[cookies] Throttle progress-bar

Browse Source 
Closes #3710
This commit is contained in:
pukkandan 2022-05-12 10:54:49 +05:30
parent 8dcce6a89c
commit 2e4585da92
No known key found for this signature in database
GPG Key ID: 7EEE9E1E817D0A39
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
yt_dlp/cookies.py
View File

@ -7,6 +7,7 @@
import subprocess import subprocess
import sys import sys
import tempfile import tempfile
import time
from datetime import datetime, timedelta, timezone from datetime import datetime, timedelta, timezone
from enum import Enum, auto from enum import Enum, auto
from hashlib import pbkdf2_hmac from hashlib import pbkdf2_hmac
@ -49,6 +50,14 @@ def error(self, message):
if self._ydl: if self._ydl:
self._ydl.report_error(message) self._ydl.report_error(message)
class ProgressBar(MultilinePrinter):
_DELAY, _timer = 0.1, 0
def print(self, message):
if time.time() - self._timer > self._DELAY:
self.print_at_line(f'[Cookies] {message}', 0)
self._timer = time.time()
def progress_bar(self): def progress_bar(self):
"""Return a context manager with a print method. (Optional)""" """Return a context manager with a print method. (Optional)"""
# Do not print to files/pipes, loggers, or when --no-progress is used # Do not print to files/pipes, loggers, or when --no-progress is used
@ -60,10 +69,7 @@ def progress_bar(self):
return return
except BaseException: except BaseException:
return return
return self.ProgressBar(file, preserve_output=False)
printer = MultilinePrinter(file, preserve_output=False)
printer.print = lambda message: printer.print_at_line(f'[Cookies] {message}', 0)
return printer
def _create_progress_bar(logger): def _create_progress_bar(logger):