added global log level setter

visualDust · Oct 8, 2024 · e2c8c49 · e2c8c49
1 parent 1628dd2
commit e2c8c49
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/neetbox/logging/_logger.py b/neetbox/logging/_logger.py
@@ -101,9 +101,25 @@ def log_level(self, level: Union[LogLevel, str]):
                 "ERROR": LogLevel.ERROR,
             }[level]
         if type(level) is int:
-            assert level >= 0 and level <= 3
+            assert level >= 0 and level <= 3, "log level must be in [0, 3]"
             level = LogLevel(level)
         self._log_level = level
+
+    @classmethod
+    def set_global_log_level(cls, level: Union[LogLevel, str]):
+        if type(level) is str:
+            level = {
+                "ALL": LogLevel.ALL,
+                "DEBUG": LogLevel.DEBUG,
+                "INFO": LogLevel.INFO,
+                "WARNING": LogLevel.WARNING,
+                "ERROR": LogLevel.ERROR,
+            }[level]
+        if type(level) is int:
+            assert level >= 0 and level <= 3, "log level must be in [0, 3]"
+            level = LogLevel(level)
+        for logger in cls._IDENTITY2LOGGER.values():
+            logger.log_level = level
 
     def writer(self, name: str):
         def _add_private_writer(name, writer_func: Callable):