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