logger.py

import datetime
import logging
import pathlib
import sys

root_logger = logging.getLogger("rov-vision")


def init_logger():
    root_logger.setLevel(logging.DEBUG)

    # Send all log messages to a file
    logs_path = (pathlib.Path(__file__).parent / "logs").resolve()

    filename = datetime.datetime.now().strftime(str(logs_path) + "/%Y-%m-%d_%H%M%S.log")
    file_handler = logging.FileHandler(filename)
    file_handler.setLevel(logging.DEBUG)
    file_formatter = logging.Formatter("{asctime} [{levelname}] [{name}] {message}", style="{")
    file_handler.setFormatter(file_formatter)
    root_logger.addHandler(file_handler)

    # Send INFO and WARNING log messages to stdout
    simple_formatter = logging.Formatter("[{levelname}] {message}", style="{")

    stdout_handler = logging.StreamHandler(sys.stdout)
    stdout_handler.setLevel(logging.INFO)
    stdout_handler.setFormatter(simple_formatter)
    # Messages of severity above warning go to stderr instead
    stdout_handler.addFilter(lambda record: record.levelno <= logging.WARNING)
    root_logger.addHandler(stdout_handler)

    # Send ERROR and CRITICAL messages to stderr
    stderr_handler = logging.StreamHandler(sys.stderr)
    stderr_handler.setLevel(logging.ERROR)
    stderr_handler.setFormatter(simple_formatter)
    root_logger.addHandler(stderr_handler)


init_logger()