| import logging |
| from logging.handlers import MemoryHandler |
| from collections import deque |
|
|
| logs = None |
| formatter = logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s") |
|
|
|
|
| def get_logs(): |
| return "\n".join([formatter.format(x) for x in logs]) |
|
|
|
|
| def setup_logger(verbose: bool = False, capacity: int = 300): |
| global logs |
| if logs: |
| return |
|
|
| |
| logger = logging.getLogger() |
| logger.setLevel(logging.DEBUG if verbose else logging.INFO) |
|
|
| stream_handler = logging.StreamHandler() |
| stream_handler.setFormatter(logging.Formatter("%(message)s")) |
| logger.addHandler(stream_handler) |
|
|
| |
| logs = deque(maxlen=capacity) |
| memory_handler = MemoryHandler(capacity, flushLevel=logging.INFO) |
| memory_handler.buffer = logs |
| memory_handler.setFormatter(formatter) |
| logger.addHandler(memory_handler) |
|
|
