mirrors/pferd
1
0
Fork 0
mirror of https://github.com/Garmelon/PFERD.git synced 2023-12-21 10:23:01 +01:00
Code Issues Packages Projects Releases Wiki Activity

Color warning differently

Browse Source
This commit is contained in:
I-Al-Istannen 2020-05-08 19:48:21 +02:00
parent 2c48ab66d4
commit 2d20d2934c
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
PFERD/logging.py
View File

@ -7,7 +7,9 @@ from typing import Optional
from rich._log_render import LogRender
from rich.console import Console
from rich.style import Style
from rich.text import Text
from rich.theme import Theme
from .utils import PathLike, to_path
@ -23,7 +25,7 @@ def enable_logging(name: str = "PFERD", level: int = logging.INFO) -> None:
logger = logging.getLogger(name)
logger.setLevel(level)
logger.addHandler(RichLoggingHandler())
logger.addHandler(RichLoggingHandler(level=level))
# This should be logged by our own handler, and not the root logger's
# default handler, so we don't pass it on to the root logger.
@ -41,9 +43,11 @@ class RichLoggingHandler(logging.Handler):
A logging handler that uses rich for highlighting
"""
def __init__(self, level: int = logging.NOTSET, console: Optional[Console] = None) -> None:
def __init__(self, level: int) -> None:
super().__init__(level=level)
self.console = Console() if console is None else console
self.console = Console(theme=Theme({
"logging.level.warning": Style(color="yellow")
}))
self._log_render = LogRender(show_level=True, show_time=False, show_path=False)
def emit(self, record: logging.LogRecord) -> None: