cyberes
/
automated-youtube-dl
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

c

This commit is contained in:
Cyberes 2023-01-21 17:05:45 -07:00
parent c8057e47d5
commit dbd5fdb0f6
No known key found for this signature in database
GPG Key ID: 194A1C358AACFC39
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
process/threads.py
View File

@ -10,6 +10,8 @@ from process.funcs import setup_file_logger
class ytdl_logger(object):
errors = []
def __init__(self, logger):
self.logger = logger
@ -24,6 +26,7 @@ class ytdl_logger(object):
def error(self, msg):
self.logger.error(msg)
self.errors.append(msg)
def is_manager_lock_locked(lock) -> bool:
@ -74,7 +77,8 @@ def download_video(args) -> dict:
desc_with = int(np.round(os.get_terminal_size()[0] * (1 / 4)))
bar = tqdm(total=100, position=(offset if locked else None), desc=video['title'].ljust(desc_with)[:desc_with], bar_format='{l_bar}{bar}| {n_fmt}/{total_fmt} [{elapsed}<{remaining}{postfix}]', leave=False)
kwargs['ydl_opts']['logger'] = ytdl_logger(setup_file_logger(video['id'], kwargs['output_dir'] / f"{video['id']}.log"))
ylogger = ytdl_logger(setup_file_logger(video['id'], kwargs['output_dir'] / f"{video['id']}.log"))
kwargs['ydl_opts']['logger'] = ylogger
yt_dlp = ydl.YDL(kwargs['ydl_opts'])
output_dict = {'downloaded_video_id': None, 'blacklist_video_id': None, 'video_error_logger_msg': [], 'status_msg': [], 'logger_msg': []} # empty object
start_time = time.time()
@ -89,6 +93,7 @@ def download_video(args) -> dict:
m = f'{video["id"]} {video["title"]} -> Failed to download, error code: {error_code}'
output_dict['status_msg'].append(m)
output_dict['video_error_logger_msg'].append(m)
output_dict['video_error_logger_msg'] = output_dict['video_error_logger_msg'] + ylogger.errors
except Exception as e:
output_dict['video_error_logger_msg'].append(f"{video['id']} {video['title']} -> {e}")
if locked: