Improve logging

consumer.py

@@ -10,27 +10,11 @@ from functools import partial
 from threading import Thread, active_count
 
 from src.access import Access
+from src.util.helpers import MainLogger
 from src.docx_converter.docx_solver import DocxBook
 from src.epub_converter.epub_solver import EpubBook
 
 
-def configure_file_logger(name: str, filename: str = "logs/converter.log",
-                          filemode: str = "w+", logging_level: int = logging.INFO) -> logging.Logger:
-    logger = logging.getLogger(name)
-
-    folder_path = os.path.dirname(os.path.abspath(__file__))
-    file_path = os.path.join(folder_path, filename)
-
-    file_handler = logging.FileHandler(file_path, mode=filemode)
-    logger.addHandler(file_handler)
-
-    file_format = logging.Formatter(fmt="%(asctime)s - %(levelname)s - %(message)s "
-                                        "[%(filename)s:%(lineno)d in %(funcName)s]")
-    file_handler.setFormatter(file_format)
-    logger.setLevel(logging_level)
-    return logger
-
-
 def local_convert_book(book_type: [DocxBook, EpubBook], book_id: int, logger: logging.Logger, params: dict):
     logger.info(f"Start processing book-{book_id}.")
     try:
@@ -88,7 +72,8 @@ def callback(ch, method, properties, body: bytes, logger: logging.Logger, libre_
 
 
 def server_run():
-    logger = configure_file_logger("consumer")
+    logger = MainLogger(name="consumer")
+    logger_object = logger.configure_main_logger()
     channel = None
     try:
         folder_path = os.path.dirname(os.path.abspath(__file__))
@@ -112,17 +97,17 @@ def server_run():
     except TypeError as exc:
         print("TypeError: problem with config, " + str(exc))
     except ValueError as exc:
-        logger.log(logging.ERROR,
+        logger_object.log(logging.ERROR,
                    f"Queue {conf_param['queue']} is not declared.")
         raise exc
     locker = Event()
     locker.set()
     channel.basic_consume(queue=conf_param["queue"],
                           auto_ack=True,
-                          on_message_callback=partial(callback, logger=logger, libre_locker=locker))
-    logger.info("Connection has been established.")
+                          on_message_callback=partial(callback, logger=logger_object, libre_locker=locker))
+    logger_object.info("Connection has been established.")
     print("Waiting for messages...")
-    logger.info("Waiting for messages...")
+    logger_object.info("Waiting for messages...")
 
     channel.start_consuming()