37 lines
1.0 KiB
Python
37 lines
1.0 KiB
Python
import logging, os
|
|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
|
|
# Configure the logger
|
|
|
|
def setup_logger(name: str, log_file: str = None, level: int = logging.INFO) -> logging.Logger:
|
|
"""Set up a logger with the specified name and level."""
|
|
if log_file is None:
|
|
log_file = f"{name}.log"
|
|
|
|
# Create a logger
|
|
logger = logging.getLogger(name)
|
|
logger.setLevel(level)
|
|
|
|
# Create file handler
|
|
file_handler = logging.FileHandler(log_file)
|
|
file_handler.setLevel(level)
|
|
|
|
# Create console handler
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setLevel(level)
|
|
|
|
# Create a formatter and set it for both handlers
|
|
formatter = logging.Formatter('%(asctime)s - %(name)s - [%(levelname)s] %(message)s',
|
|
datefmt='%Y-%m-%d %H:%M:%S')
|
|
file_handler.setFormatter(formatter)
|
|
console_handler.setFormatter(formatter)
|
|
|
|
|
|
# Add handlers to the logger
|
|
logger.addHandler(file_handler)
|
|
logger.addHandler(console_handler)
|
|
|
|
return logger
|
|
|