From 42118052bda60d04655a933bafb4c926e9e7e641 Mon Sep 17 00:00:00 2001
From: Dineshkarthik <dineshkarthik.r@gmail.com>
Date: Thu, 31 Dec 2020 14:27:41 +0100
Subject: [PATCH] refactor: improve logging

---
 media_downloader.py | 10 ++++------
 utils/log.py        | 15 +++++++++++++++
 2 files changed, 19 insertions(+), 6 deletions(-)
 create mode 100644 utils/log.py

diff --git a/media_downloader.py b/media_downloader.py
index 079f800..9a0c87a 100644
--- a/media_downloader.py
+++ b/media_downloader.py
@@ -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(
diff --git a/utils/log.py b/utils/log.py
new file mode 100644
index 0000000..baefb13
--- /dev/null
+++ b/utils/log.py
@@ -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