From 22761864cd537b8d6f0118608283c72c777c0343 Mon Sep 17 00:00:00 2001
From: Pauli Virtanen <pauli.t.virtanen@jyu.fi>
Date: Wed, 14 Sep 2022 16:07:39 +0300
Subject: [PATCH] Add with_log_level

---
 usadelndsoc/__init__.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/usadelndsoc/__init__.py b/usadelndsoc/__init__.py
index 34bc4e1..84d4a11 100644
--- a/usadelndsoc/__init__.py
+++ b/usadelndsoc/__init__.py
@@ -1,6 +1,7 @@
 # -*- coding:utf-8; eval: (blacken-mode) -*-
 
 import logging
+import contextlib
 
 
 logger = logging.getLogger(__name__)
@@ -35,5 +36,15 @@ def _init_logging():
     logging.config.dictConfig(cfg)
 
 
+@contextlib.contextmanager
+def with_log_level(level):
+    old_level = int(logger.level)
+    logger.setLevel(level)
+    try:
+        yield
+    finally:
+        logger.setLevel(old_level)
+
+
 _init_logging()
-del _init_logging, logging
+del _init_logging, logging, contextlib
-- 
GitLab