diff options
Diffstat (limited to 'src/Logging.py')
| -rw-r--r-- | src/Logging.py | 23 |
1 files changed, 13 insertions, 10 deletions
diff --git a/src/Logging.py b/src/Logging.py index f82770fc..fe7f3f91 100644 --- a/src/Logging.py +++ b/src/Logging.py @@ -16,7 +16,7 @@ class BitBotFormatter(logging.Formatter): return utils.iso8601_format(datetime_obj) class Log(object): - def __init__(self, level: str, location: str): + def __init__(self, to_file: bool, level: str, location: str): logging.addLevelName(LEVELS["trace"], "TRACE") self.logger = logging.getLogger(__name__) @@ -35,16 +35,19 @@ class Log(object): stdout_handler.setFormatter(formatter) self.logger.addHandler(stdout_handler) - trace_handler = logging.handlers.TimedRotatingFileHandler( - os.path.join(location, "trace.log"), when="midnight", backupCount=5) - trace_handler.setLevel(LEVELS["trace"]) - trace_handler.setFormatter(formatter) - self.logger.addHandler(trace_handler) + if to_file: + trace_path = os.path.join(location, "trace.log") + trace_handler = logging.handlers.TimedRotatingFileHandler( + trace_path, when="midnight", backupCount=5) + trace_handler.setLevel(LEVELS["trace"]) + trace_handler.setFormatter(formatter) + self.logger.addHandler(trace_handler) - warn_handler = logging.FileHandler(os.path.join(location, "warn.log")) - warn_handler.setLevel(LEVELS["warn"]) - warn_handler.setFormatter(formatter) - self.logger.addHandler(warn_handler) + warn_path = os.path.join(location, "warn.log") + warn_handler = logging.FileHandler(warn_path) + warn_handler.setLevel(LEVELS["warn"]) + warn_handler.setFormatter(formatter) + self.logger.addHandler(warn_handler) def trace(self, message: str, params: typing.List, **kwargs): self._log(message, params, LEVELS["trace"], kwargs) |