Rooba
/
rich
mirror of https://github.com/Textualize/rich.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

using _log_render for traceback printing

This commit is contained in:
landi 2020-09-24 20:17:59 +02:00
parent d388b83954
commit f0d74ace14
1 changed files with 22 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
rich/logging.py
View File

@ -9,6 +9,7 @@ from ._log_render import LogRender
from .console import Console
from .highlighter import Highlighter, ReprHighlighter
from .text import Text
from .traceback import Traceback
class RichHandler(Handler):
@ -91,6 +92,16 @@ class RichHandler(Handler):
level = Text()
level.append(record.levelname, log_style)
traceback = None
if self.handle_tracebacks and record.exc_info:
traceback = Traceback(
width=self.tracebacks_width,
extra_lines=self.tracebacks_extra_lines,
theme=self.tracebacks_theme,
word_wrap=self.tracebacks_word_wrap,
)
message = record.getMessage()
use_markup = (
getattr(record, "markup") if hasattr(record, "markup") else self.markup
)
@ -104,26 +115,18 @@ class RichHandler(Handler):
if self.KEYWORDS:
message_text.highlight_words(self.KEYWORDS, "logging.keyword")
if self.handle_tracebacks and record.exc_info:
self.console.print_exception(
width=self.tracebacks_width,
extra_lines=self.tracebacks_extra_lines,
theme=self.tracebacks_theme,
word_wrap=self.tracebacks_word_wrap,
)
else:
self.console.print(
self._log_render(
self.console,
[message_text],
log_time=log_time,
time_format=time_format,
level=level,
path=path,
line_no=record.lineno,
link_path=record.pathname if self.enable_link_path else None,
)
self.console.print(
self._log_render(
self.console,
[message_text] if not traceback else [message_text, traceback],
log_time=log_time,
time_format=time_format,
level=level,
path=path,
line_no=record.lineno,
link_path=record.pathname if self.enable_link_path else None,
)
)
if __name__ == "__main__": # pragma: no cover