| """ Logging helpers | |
| Hacked together by / Copyright 2020 Ross Wightman | |
| """ | |
| import logging | |
| import logging.handlers | |
| class FormatterNoInfo(logging.Formatter): | |
| def __init__(self, fmt='%(levelname)s: %(message)s'): | |
| logging.Formatter.__init__(self, fmt) | |
| def format(self, record): | |
| if record.levelno == logging.INFO: | |
| return str(record.getMessage()) | |
| return logging.Formatter.format(self, record) | |
| def setup_default_logging(default_level=logging.INFO, log_path=''): | |
| console_handler = logging.StreamHandler() | |
| console_handler.setFormatter(FormatterNoInfo()) | |
| logging.root.addHandler(console_handler) | |
| logging.root.setLevel(default_level) | |
| if log_path: | |
| file_handler = logging.handlers.RotatingFileHandler(log_path, maxBytes=(1024 ** 2 * 2), backupCount=3) | |
| file_formatter = logging.Formatter("%(asctime)s - %(name)20s: [%(levelname)8s] - %(message)s") | |
| file_handler.setFormatter(file_formatter) | |
| logging.root.addHandler(file_handler) | |