This commit is contained in:
parent
c8057e47d5
commit
dbd5fdb0f6
|
@ -10,6 +10,8 @@ from process.funcs import setup_file_logger
|
||||||
|
|
||||||
|
|
||||||
class ytdl_logger(object):
|
class ytdl_logger(object):
|
||||||
|
errors = []
|
||||||
|
|
||||||
def __init__(self, logger):
|
def __init__(self, logger):
|
||||||
self.logger = logger
|
self.logger = logger
|
||||||
|
|
||||||
|
@ -24,6 +26,7 @@ class ytdl_logger(object):
|
||||||
|
|
||||||
def error(self, msg):
|
def error(self, msg):
|
||||||
self.logger.error(msg)
|
self.logger.error(msg)
|
||||||
|
self.errors.append(msg)
|
||||||
|
|
||||||
|
|
||||||
def is_manager_lock_locked(lock) -> bool:
|
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)))
|
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)
|
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'])
|
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
|
output_dict = {'downloaded_video_id': None, 'blacklist_video_id': None, 'video_error_logger_msg': [], 'status_msg': [], 'logger_msg': []} # empty object
|
||||||
start_time = time.time()
|
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}'
|
m = f'{video["id"]} {video["title"]} -> Failed to download, error code: {error_code}'
|
||||||
output_dict['status_msg'].append(m)
|
output_dict['status_msg'].append(m)
|
||||||
output_dict['video_error_logger_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:
|
except Exception as e:
|
||||||
output_dict['video_error_logger_msg'].append(f"{video['id']} {video['title']} -> {e}")
|
output_dict['video_error_logger_msg'].append(f"{video['id']} {video['title']} -> {e}")
|
||||||
if locked:
|
if locked:
|
||||||
|
|
Reference in New Issue