feat: logger atlevel method

src/ape/logging.py

@@ -3,6 +3,7 @@
 import sys
 import traceback
 from collections.abc import Callable, Sequence
+from contextlib import contextmanager
 from enum import IntEnum
 from pathlib import Path
 from typing import IO, Any, Optional, Union
@@ -185,6 +186,13 @@ def set_level(self, level: Union[str, int, LogLevel]):
  for _logger in self._extra_loggers.values():
  _logger.setLevel(level)
 
+ @contextmanager
+ def at_level(self, level: Union[str, int, LogLevel]):
+ initial_level = self.level
+ self.set_level(level)
+ yield
+ self.set_level(initial_level)
+
  def log_error(self, err: Exception):
  """
  Avoids logging empty messages.

tests/functional/test_logging.py

@@ -107,3 +107,12 @@ def cmd(cli_ctx):
 def test_set_level(level):
  logger.set_level(level)
  assert logger.level == LogLevel.INFO.value
+
+
+def test_at_level():
+ level_to_set = next(lvl for lvl in LogLevel if lvl != logger.level)
+ initial_level = logger.level
+ with logger.at_level(level_to_set):
+ assert logger.level == level_to_set
+
+ assert logger.level == initial_level