Log youtube-dl warnings and errors in bot log

Signed-off-by: lzzy12 <jhashivam2020@gmail.com>
This commit is contained in:
lzzy12 2020-03-31 11:15:09 +05:30
parent 85d4005c16
commit e2c5c8beaf

View File

@ -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