From e2c5c8beaf71bd9f7e7c63354a5f427d6ee4bd97 Mon Sep 17 00:00:00 2001 From: lzzy12 Date: Tue, 31 Mar 2020 11:15:09 +0530 Subject: [PATCH] Log youtube-dl warnings and errors in bot log Signed-off-by: lzzy12 --- .../download_utils/youtube_dl_download_helper.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/bot/helper/mirror_utils/download_utils/youtube_dl_download_helper.py b/bot/helper/mirror_utils/download_utils/youtube_dl_download_helper.py index 8d25aca..70d8e79 100644 --- a/bot/helper/mirror_utils/download_utils/youtube_dl_download_helper.py +++ b/bot/helper/mirror_utils/download_utils/youtube_dl_download_helper.py @@ -4,6 +4,21 @@ from youtube_dl import YoutubeDL import threading from bot import LOGGER, download_dict_lock, download_dict, DOWNLOAD_DIR from ..status_utils.youtube_dl_download_status import YoutubeDLDownloadStatus +import logging + + +class MyLogger: + def __init__(self, obj): + self.obj = obj + + def debug(self, msg): + logging.debug(msg) + + def warning(self, msg): + logging.warning(msg) + + def error(self, msg): + logging.error(msg) class YoutubeDLHelper(DownloadHelper): @@ -15,6 +30,7 @@ class YoutubeDLHelper(DownloadHelper): self.__gid = "" self.opts = { 'progress_hooks': [self.__onDownloadProgress], + 'logger': MyLogger(self), 'usenetrc': True } self.__download_speed = 0