diff --git a/Services/ml_model/global_logger.py b/Services/ml_model/global_logger.py index 7941175..dc29ec3 100644 --- a/Services/ml_model/global_logger.py +++ b/Services/ml_model/global_logger.py @@ -9,17 +9,17 @@ def __init__(self, log_file=None): ch = logging.StreamHandler() ch.setLevel(logging.DEBUG) - # Create file handler if log_file is provided + # Create formatter + formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s - %(data)s') + ch.setFormatter(formatter) + self.logger.addHandler(ch) + + # Add file handler if specified if log_file: fh = logging.FileHandler(log_file) fh.setLevel(logging.DEBUG) - formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s - %(data)s') fh.setFormatter(formatter) self.logger.addHandler(fh) - - formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s - %(data)s') - ch.setFormatter(formatter) - self.logger.addHandler(ch) def info(self, message, extra=None): if extra is None: diff --git a/Services/ml_model/test_logger.py b/Services/ml_model/test_logger.py new file mode 100644 index 0000000..cf6ad4d --- /dev/null +++ b/Services/ml_model/test_logger.py @@ -0,0 +1,15 @@ +from global_logger import global_logger + +def test_logger(): + # Test different log levels + global_logger.info("Test info message") + global_logger.error("Test error message") + global_logger.debug("Test debug message") + + # Test with extra data + global_logger.info("Test with data", extra={"data": "test_value"}) + + print("Check the log file and console output for messages") + +if __name__ == "__main__": + test_logger()