logger_config.py

import logging
import os
from logging.handlers import RotatingFileHandler

import appdirs

import constants


def get_log_dir():
    return appdirs.user_log_dir(
        constants.APPLICATION_NAME, appauthor=constants.APPLICATION_AUTHOR
    )


def get_log_file(log_dir=get_log_dir()) -> str:
    return os.path.join(log_dir, constants.LOG_FILENAME)


def setup_logger():
    log_dir = get_log_dir()
    if os.path.isdir(log_dir):
        pass
    else:
        os.makedirs(log_dir)

    # Create logger
    logger = logging.getLogger(constants.APPLICATION_NAME)
    logger.setLevel(logging.INFO)

    # Create formatters
    file_formatter = logging.Formatter(
        "%(asctime)s - %(name)s - %(levelname)s - %(filename)s:%(lineno)d - %(message)s"
    )
    console_formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s")

    # File handler (rotating log files)
    file_handler = RotatingFileHandler(
        get_log_file(log_dir),
        maxBytes=1024 * 1024,  # 1MB
        backupCount=5,
    )
    file_handler.setLevel(logging.DEBUG)
    file_handler.setFormatter(file_formatter)

    # Console handler
    console_handler = logging.StreamHandler()
    console_handler.setLevel(logging.INFO)
    console_handler.setFormatter(console_formatter)

    # Add handlers to logger
    logger.addHandler(file_handler)
    logger.addHandler(console_handler)

    return logger


# Create and configure logger
logger = setup_logger()