Pepe_1 / utils /logger.py
nileshhanotia's picture
Update utils/logger.py
b144f8e verified
raw
history blame contribute delete
515 Bytes
import logging
import os
def setup_logger(name):
# Get log level from environment or default to INFO
log_level = os.getenv('LOG_LEVEL', 'INFO')
logger = logging.getLogger(name)
logger.setLevel(log_level)
if not logger.handlers:
handler = logging.StreamHandler()
formatter = logging.Formatter(
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
handler.setFormatter(formatter)
logger.addHandler(handler)
return logger