File size: 832 Bytes
d87e8d0 |
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 29 30 31 32 |
import logging
import os
def get_logger(name: str):
logs_dir = "logs"
os.makedirs(logs_dir, exist_ok=True)
logger = logging.getLogger(name)
logger.setLevel(logging.DEBUG)
# File handler
file_handler = logging.FileHandler(os.path.join(logs_dir, "pipeline.log"))
file_handler.setLevel(logging.DEBUG)
# Console handler
console_handler = logging.StreamHandler()
console_handler.setLevel(logging.INFO)
# Formatter
formatter = logging.Formatter(
"[%(asctime)s] [%(levelname)s] - %(name)s - %(message)s", "%Y-%m-%d %H:%M:%S"
)
file_handler.setFormatter(formatter)
console_handler.setFormatter(formatter)
# Avoid duplicate handlers
if not logger.handlers:
logger.addHandler(file_handler)
logger.addHandler(console_handler)
return logger
|