From 3c4ed5f7923190de621d065b496a5b8bdd025e5d Mon Sep 17 00:00:00 2001
From: mike <mike@thinkmike>
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