Skip to content

Commit

Permalink
✨ Add logger.mute() (#77)
Browse files Browse the repository at this point in the history
* ✨ Add silence() context manager

* ♻️ Rename to mute
  • Loading branch information
falexwolf authored Apr 19, 2024
1 parent fd56232 commit fc78dfc
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions lamin_utils/_logger.py
Original file line number Diff line number Diff line change
Expand Up @@ -226,3 +226,20 @@ def set_verbosity(logger, verbosity: int):


RootLogger.set_verbosity = set_verbosity # type: ignore


def mute(logger):
"""Context manager to mute logger."""

class Muted:
def __enter__(self):
self.original_verbosity = logger._verbosity
logger.set_verbosity(0)

def __exit__(self, exc_type, exc_val, exc_tb):
logger.set_verbosity(self.original_verbosity)

return Muted()


RootLogger.mute = mute # type: ignore

0 comments on commit fc78dfc

Please sign in to comment.