log.py

import os
import logging
from datetime import datetime


class LogHandler():
    def __init__(self,
                 run_mode: str,
                 model_name: str,
                 dataset_name: str,
                 ) -> None:
        self.run_mode = run_mode
        self.model_name = model_name
        self.dataset_name = dataset_name
        self.log_direction = model_name + "-" + dataset_name + "/" + run_mode


    def getLogger(self) -> logging.Logger:
        if not os.path.exists(self.log_direction):
            os.makedirs(self.log_direction)
        log_file_name = f"{self.log_direction}/log-{self.run_mode}-{self.model_name}-{self.dataset_name}-{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}.txt"
        logging.basicConfig(
            filename=log_file_name,
            filemode='a',
            level=logging.INFO,
            format='%(asctime)s - %(levelname)s : %(message)s',
        )
        logger = logging.getLogger()
        logger.info(f"run-mode: {self.run_mode}")
        return logger