Skip to content

Commit

Permalink
feat: logger atlevel method
Browse files Browse the repository at this point in the history
  • Loading branch information
antazoey committed Sep 19, 2024
1 parent a2e2dd4 commit 1d312c0
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions src/ape/logging.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit 1d312c0

Please sign in to comment.