Move rich's console to log.py

This commit is contained in:
MinchinWeb 2021-07-08 21:33:22 -06:00
commit a52922bfb5
3 changed files with 8 additions and 10 deletions

View file

@ -1,12 +1,15 @@
import logging
from collections import defaultdict
from rich.console import Console
from rich.logging import RichHandler
__all__ = [
'init'
]
console = Console()
class LimitFilter(logging.Filter):
"""
@ -100,7 +103,7 @@ logging.setLoggerClass(FatalLogger)
logging.getLogger().__class__ = FatalLogger
def init(level=None, fatal='', handler=RichHandler(), name=None,
def init(level=None, fatal='', handler=RichHandler(console=console), name=None,
logs_dedup_min_level=None):
FatalLogger.warnings_fatal = fatal.startswith('warning')
FatalLogger.errors_fatal = bool(fatal)