19 lines
815 B
Python
19 lines
815 B
Python
|
import logging
|
||
|
|
from logging import Logger
|
||
|
|
from logging.handlers import RotatingFileHandler
|
||
|
|
import os
|
||
|
|
import datetime
|
||
|
|
from atlas_rag.evaluation.benchmark import BenchMarkConfig
|
||
|
|
|
||
|
|
def setup_logger(config:BenchMarkConfig, logger_name = "MyLogger") -> Logger:
|
||
|
|
date_time = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
|
||
|
|
log_file_path = f'./log/{config.dataset_name}_event{config.include_events}_concept{config.include_concept}_{date_time}.log'
|
||
|
|
logger = logging.getLogger(logger_name)
|
||
|
|
logger.setLevel(logging.INFO)
|
||
|
|
max_bytes = 50 * 1024 * 1024
|
||
|
|
if not os.path.exists(log_file_path):
|
||
|
|
os.makedirs(os.path.dirname(log_file_path), exist_ok=True)
|
||
|
|
handler = RotatingFileHandler(log_file_path, maxBytes=max_bytes, backupCount=5)
|
||
|
|
logger.addHandler(handler)
|
||
|
|
|
||
|
|
return logger
|