about summary refs log tree commit diff stats
path: root/archive/2025/summer/bsc_gerg/src/logger.py
blob: 6ba0b84ed4eadd145dcaca77c3ec332019f8f186 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import logging
import os
import sys

from dotenv import load_dotenv

load_dotenv()
debug = os.getenv("DEBUG") == "true"

def simple_custom_logger(name):
    custom_logger = logging.getLogger(name)
    if debug: custom_logger.setLevel(logging.DEBUG)

    console_handler = logging.StreamHandler(sys.stdout)

    # Set the logging level for the console handler
    console_handler.setLevel(logging.DEBUG)

    # Create a formatter and set it for the console handler
    formatter = logging.Formatter('%(asctime)s [%(name)s] %(levelname)s: %(message)s')
    console_handler.setFormatter(formatter)

    # Add the console handler to the logger
    custom_logger.addHandler(console_handler)

    return custom_logger

logger = simple_custom_logger("TAS")