From e978976ae9341b6ae5242055dde98f1e43710945 Mon Sep 17 00:00:00 2001 From: Joao Mario Lago Date: Wed, 24 Jul 2024 17:01:58 -0300 Subject: [PATCH] logger: Allows for disabling loggin to buffer * Allows disable logger to avoid filling buffer when using as lib --- blueos_repository/logger.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/blueos_repository/logger.py b/blueos_repository/logger.py index d09dc84..0ff88fe 100644 --- a/blueos_repository/logger.py +++ b/blueos_repository/logger.py @@ -48,8 +48,17 @@ class Logger: Simple logger class to consolidate logs and generate a manifest.log """ + enabled: bool = True log_buffer: ManifestLog = ManifestLog() + @staticmethod + def disable() -> None: + Logger.enabled = False + + @staticmethod + def enable() -> None: + Logger.enabled = True + @staticmethod def start_docker_rate_limit(rate_limit: RateLimit) -> None: """ @@ -80,6 +89,9 @@ def _check_and_init_entry(entry: str) -> None: @staticmethod def log(entry: str, message: str, level: LogEntryStatus) -> None: + if not Logger.enabled: + return + Logger._check_and_init_entry(entry) Logger.log_buffer.extensions[entry].append(LogEntry(level, message))