| 459 | |
| 460 | |
| 461 | class Logger: |
| 462 | def __init__(self, name, log_level=logging.INFO, log_file=None, log_mode="both", disable_formatter=False): |
| 463 | self.logger = logging.getLogger(name) |
| 464 | self.logger.setLevel(log_level) |
| 465 | |
| 466 | self.formatter = logging.Formatter('%(asctime)s [%(levelname)s] %(message)s', datefmt='%Y-%m-%d %H:%M:%S') |
| 467 | |
| 468 | # Log to console |
| 469 | if log_mode == "both" or log_mode == "terminal": |
| 470 | console_handler = logging.StreamHandler() |
| 471 | if not disable_formatter: |
| 472 | console_handler.setFormatter(self.formatter) |
| 473 | self.logger.addHandler(console_handler) |
| 474 | |
| 475 | # Log to file |
| 476 | if log_file is not None: |
| 477 | if log_mode == "both" or log_mode == "file": |
| 478 | file_handler = logging.FileHandler(log_file, mode='w') |
| 479 | if not disable_formatter: |
| 480 | file_handler.setFormatter(self.formatter) |
| 481 | self.logger.addHandler(file_handler) |
| 482 | |
| 483 | def add_file_handler(self, file_name): |
| 484 | file_handler = logging.FileHandler(file_name, mode='w') |
| 485 | file_handler.setFormatter(self.formatter) |
| 486 | self.logger.addHandler(file_handler) |
| 487 | |
| 488 | def debug(self, message): |
| 489 | self.logger.debug(message) |
| 490 | |
| 491 | def info(self, message): |
| 492 | self.logger.info(message) |
| 493 | |
| 494 | def warning(self, message): |
| 495 | self.logger.warning(message) |
| 496 | |
| 497 | def error(self, message): |
| 498 | self.logger.error(message) |
| 499 | |
| 500 | def critical(self, message): |
| 501 | self.logger.critical(message) |
no outgoing calls
no test coverage detected