| 8 | |
| 9 | |
| 10 | class MyLogger: |
| 11 | def __init__(self): |
| 12 | self.logger = logging.getLogger("BERTopic") |
| 13 | |
| 14 | def configure(self, level): |
| 15 | self.set_level(level) |
| 16 | self._add_handler() |
| 17 | self.logger.propagate = False |
| 18 | |
| 19 | def info(self, message): |
| 20 | self.logger.info(f"{message}") |
| 21 | |
| 22 | def warning(self, message): |
| 23 | self.logger.warning(f"WARNING: {message}") |
| 24 | |
| 25 | def set_level(self, level): |
| 26 | levels = ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"] |
| 27 | if level in levels: |
| 28 | self.logger.setLevel(level) |
| 29 | |
| 30 | def _add_handler(self): |
| 31 | sh = logging.StreamHandler() |
| 32 | sh.setFormatter(logging.Formatter("%(asctime)s - %(name)s - %(message)s")) |
| 33 | self.logger.addHandler(sh) |
| 34 | |
| 35 | # Remove duplicate handlers |
| 36 | if len(self.logger.handlers) > 1: |
| 37 | self.logger.handlers = [self.logger.handlers[0]] |
| 38 | |
| 39 | |
| 40 | def check_documents_type(documents): |
no outgoing calls