Add option to mute test logs

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(

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")
+		if self.__do_test_logs:
-		self.info("Test info log")
+			self.debug("Test debug log")
-		self.warn("Test warn log")
+			self.info("Test info log")
-		self.error("Test error log")
+			self.warn("Test warn log")
+			self.error("Test error log")
 	
 	def debug(self, s):
 		self.__logger.debug(s)

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