aswi2021jmsd-gitlab4/src/utils/logger.py @ 8f05fcc9

import inspect
import logging
import sys
from logging import handlers
import os
from pathlib import Path

from src.config.configuration import Configuration, LOG_LEVEL_MAPPING
from src.constants import LOG_NAME, LOG_DIR, LOG_FILENAME, LOG_FORMAT
from src.utils.file_anchor import FileAnchor


def configure_logging(config: Configuration):
    root_path = FileAnchor(config.root_dir, LOG_DIR).shortest_relative_path()
    if root_path is None:
        print(f"Critical error, '{config.root_dir}' does not exist in '{os.getcwd()}'.")
        exit(1)

    if not os.path.exists(root_path):
        os.makedirs(FileAnchor(config.root_dir, LOG_DIR).shortest_relative_path())

    handler = logging.handlers.TimedRotatingFileHandler(
        os.path.join(FileAnchor(config.root_dir, LOG_DIR).shortest_relative_path(), LOG_FILENAME),
        when='H', interval=1)
    formatter = logging.Formatter(LOG_FORMAT)
    handler.setFormatter(formatter)

    # set log level based on config file
    app_logger = logging.getLogger(LOG_NAME)
    app_logger.setLevel(LOG_LEVEL_MAPPING.get(config.log_level, logging.DEBUG))

    app_logger.addHandler(handler)

    log = logging.getLogger('werkzeug')
    log.disabled = True

    if 'pytest' in sys.modules:
        logging.disable(logging.CRITICAL)


class Logger:

    @staticmethod
    def get_names():
        stack = inspect.stack()

        file_name = inspect.getmodule(inspect.stack()[2][0]).__file__
        class_name = ""
        if "self" in stack[2][0].f_locals:
            class_name = stack[2][0].f_locals["self"].__class__.__name__ + "."
        function_name = stack[2][0].f_code.co_name

        return file_name, class_name, function_name

    @staticmethod
    def debug(message: str):
        names = Logger.get_names()
        app_logger = logging.getLogger(LOG_NAME)
        app_logger.debug(f'{Path(names[0]).name}[{names[1]}{names[2]}()]: {message}')

    @staticmethod
    def info(message: str):
        names = Logger.get_names()
        app_logger = logging.getLogger(LOG_NAME)
        app_logger.info(f'{Path(names[0]).name}[{names[1]}{names[2]}()]: {message}')

    @staticmethod
    def warning(message: str):
        names = Logger.get_names()
        app_logger = logging.getLogger(LOG_NAME)
        app_logger.warning(f'{Path(names[0]).name}[{names[1]}{names[2]}()]: {message}')

    @staticmethod
    def error(message: str):
        names = Logger.get_names()
        app_logger = logging.getLogger(LOG_NAME)
        app_logger.error(f'{Path(names[0]).name}[{names[1]}.{names[2]}()]: {message}')

    @staticmethod
    def critical(message: str):
        names = Logger.get_names()
        app_logger = logging.getLogger(LOG_NAME)
        app_logger.critical(f'{Path(names[0]).name}[{names[1]}.{names[2]}()]: {message}')