From 3c4ed5f7923190de621d065b496a5b8bdd025e5d Mon Sep 17 00:00:00 2001 From: mike Date: Sat, 22 Jun 2024 15:31:31 -0700 Subject: [PATCH] Add option to mute test logs --- domain/BackupRotator.py | 6 +++++- domain/Logger.py | 13 ++++++++----- main.py | 20 ++++++++++++++++++++ 3 files changed, 33 insertions(+), 6 deletions(-) diff --git a/domain/BackupRotator.py b/domain/BackupRotator.py index c83c171..8079c20 100755 --- a/domain/BackupRotator.py +++ b/domain/BackupRotator.py @@ -31,13 +31,17 @@ class BackupRotator: config_paths: [Path] = None, debug: bool = False, systemd: bool = False, - write_to_syslog: bool = False + write_to_syslog: bool = False, + do_test_logs: bool = True, ): + self.__do_test_logs = do_test_logs + self.__logger = Logger( name=type(self).__name__, debug=debug, systemd=systemd, write_to_syslog=write_to_syslog, + do_test_logs=do_test_logs, ) self.__config = Config( diff --git a/domain/Logger.py b/domain/Logger.py index 8e3fada..0d3fe45 100644 --- a/domain/Logger.py +++ b/domain/Logger.py @@ -12,13 +12,15 @@ class Logger: name: str, debug: bool = False, write_to_syslog: bool = False, - systemd: bool = False + systemd: bool = False, + do_test_logs: bool = True, ): self.__name = name self.__debug = debug self.__write_to_syslog = write_to_syslog self.__systemd = systemd + self.__do_test_logs = do_test_logs self._init_logger() @@ -73,10 +75,11 @@ class Logger: self.__logger.addHandler(handler) # This is annoying inside cron - self.debug("Test debug log") - self.info("Test info log") - self.warn("Test warn log") - self.error("Test error log") + if self.__do_test_logs: + self.debug("Test debug log") + self.info("Test info log") + self.warn("Test warn log") + self.error("Test error log") def debug(self, s): self.__logger.debug(s) diff --git a/main.py b/main.py index 4f9c723..a0e4dba 100755 --- a/main.py +++ b/main.py @@ -42,6 +42,25 @@ def main(): ) ) + parser.add_argument( + "--no-test-logs", + default=True, + dest="do_test_logs", + action="store_false", + help=( + "Pass if you do not want to see test logs for all log levels." + ) + ) + parser.add_argument( + "--test-logs", + default=True, + dest="do_test_logs", + action="store_true", + help=( + "Pass if you want to see test logs for all log levels." + ) + ) + parser.add_argument( "--config", "-c", dest="config_paths", @@ -68,6 +87,7 @@ def main(): debug=args.debug, systemd=args.systemd, write_to_syslog=args.write_to_syslog, + do_test_logs=args.do_test_logs, ) rotator.run( global_dry_run=args.global_dry_run