Dineshkarthik/telegram_media_downloader
1
1
Fork 0
mirror of https://github.com/Dineshkarthik/telegram_media_downloader.git synced 2024-09-20 07:16:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: improve logging

Browse source
This commit is contained in:
Dineshkarthik 2020-12-31 14:27:41 +01:00
parent 1f0c45cd2d
commit 42118052bd
No known key found for this signature in database
GPG key ID: E256EF0EECD6F201
2 changed files with 19 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
media_downloader.py
View file

@ -9,12 +9,13 @@ import pyrogram
import yaml
from utils.file_management import get_next_name, manage_duplicate_file
from utils.log import LogFilter
from utils.meta import print_meta
logging.basicConfig(level=logging.INFO)
logging.getLogger("pyrogram.session.session.send").setLevel(
logging.CRITICAL + 1
)
logging.getLogger("pyrogram.session.session").addFilter(LogFilter())
logging.getLogger("pyrogram.client").addFilter(LogFilter())
logger = logging.getLogger("media_downloader")
THIS_DIR = os.path.dirname(os.path.abspath(__file__))
@ -171,9 +172,6 @@ async def download_media(
_media, _type
)
if _can_download(_type, file_formats, file_format):
logger.info(
"Message[%d]: downloading...", message.message_id
)
if _is_exist(file_name):
file_name = get_next_name(file_name)
download_path = await client.download_media(

15
utils/log.py Normal file
View file

@ -0,0 +1,15 @@
"""Util module to handle logs."""
import logging
class LogFilter(logging.Filter):
"""
Custom Log Filter.
Ignore logs from specific functions.
"""
# pylint: disable = W0221
def filter(self, log_record):
if log_record.funcName == "send" or log_record.funcName == "get_file":
return False
return True