refactor: dedicated tester

This commit is contained in:
Dobin
2024-03-16 14:38:50 +00:00
parent 2e491272c9
commit da567af27e
4 changed files with 110 additions and 77 deletions
+4 -4
View File
@@ -52,16 +52,16 @@ def writelog():
for line in log_messages:
f.write(line + "\n")
def setup_logging():
def setup_logging(level = logging.INFO):
root_logger = logging.getLogger()
root_logger.setLevel(logging.INFO)
root_logger.setLevel(level)
ch = logging.StreamHandler()
ch.setLevel(logging.INFO)
ch.setLevel(level)
ch.setFormatter(CustomFormatter())
list_handler = ListHandler(log_messages)
list_handler.setLevel(logging.DEBUG)
list_handler.setLevel(level)
list_handler.setFormatter(CustomFormatter())
root_logger.addHandler(ch)