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._log_render import LogRender
from rich.console import Console from rich.console import Console
from rich.style import Style
from rich.text import Text from rich.text import Text
from rich.theme import Theme
from .utils import PathLike, to_path 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 = logging.getLogger(name)
logger.setLevel(level) 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 # 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. # 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 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) 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) self._log_render = LogRender(show_level=True, show_time=False, show_path=False)
def emit(self, record: logging.LogRecord) -> None: def emit(self, record: logging.LogRecord) -> None: