mirror of
https://github.com/vladmandic/sdnext.git
synced 2026-01-27 15:02:48 +03:00
58 lines
2.0 KiB
Python
58 lines
2.0 KiB
Python
import logging
|
|
import warnings
|
|
from rich.console import Console
|
|
from rich.theme import Theme
|
|
from rich.pretty import install as pretty_install
|
|
from rich.traceback import install as traceback_install
|
|
from installer import log as installer_log
|
|
|
|
|
|
log = installer_log
|
|
console = Console(log_time=True, log_time_format='%H:%M:%S-%f', theme=Theme({
|
|
"traceback.border": "black",
|
|
"traceback.border.syntax_error": "black",
|
|
"inspect.value.border": "black",
|
|
}))
|
|
|
|
pretty_install(console=console)
|
|
traceback_install(console=console, extra_lines=1, width=console.width, word_wrap=False, indent_guides=False)
|
|
already_displayed = {}
|
|
|
|
|
|
def install(suppress=[]): # noqa: B006
|
|
warnings.filterwarnings("ignore", category=UserWarning)
|
|
pretty_install(console=console)
|
|
traceback_install(console=console, extra_lines=1, width=console.width, word_wrap=False, indent_guides=False, suppress=suppress)
|
|
logging.basicConfig(level=logging.ERROR, format='%(asctime)s | %(levelname)s | %(pathname)s | %(message)s')
|
|
# for handler in logging.getLogger().handlers:
|
|
# handler.setLevel(logging.INFO)
|
|
|
|
|
|
def print_error_explanation(message):
|
|
lines = message.strip().split("\n")
|
|
for line in lines:
|
|
log.error(line)
|
|
|
|
|
|
def display(e: Exception, task, suppress=[]): # noqa: B006
|
|
log.error(f"{task or 'error'}: {type(e).__name__}")
|
|
console.print_exception(show_locals=False, max_frames=5, extra_lines=1, suppress=suppress, theme="ansi_dark", word_wrap=False, width=min([console.width, 200]))
|
|
|
|
|
|
def display_once(e: Exception, task):
|
|
if task in already_displayed:
|
|
return
|
|
display(e, task)
|
|
already_displayed[task] = 1
|
|
|
|
|
|
def run(code, task):
|
|
try:
|
|
code()
|
|
except Exception as e:
|
|
display(e, task)
|
|
|
|
|
|
def exception(suppress=[]): # noqa: B006
|
|
console.print_exception(show_locals=False, max_frames=10, extra_lines=2, suppress=suppress, theme="ansi_dark", word_wrap=False, width=min([console.width, 200]))
|