update setup

This commit is contained in:
Vladimir Mandic
2023-04-15 17:22:58 -04:00
parent 45dcc69484
commit dd25f2f515
11 changed files with 81 additions and 48 deletions
+13 -10
View File
@@ -1,7 +1,5 @@
import sys
import anyio
import starlette
import gradio
import logging
from rich import print # pylint: disable=redefined-builtin
from rich.console import Console
from rich.theme import Theme
@@ -13,13 +11,18 @@ console = Console(log_time=True, log_time_format='%H:%M:%S-%f', theme=Theme({
"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, suppress=[anyio, starlette, gradio])
traceback_install(console=console, extra_lines=1, width=console.width, word_wrap=False, indent_guides=False)
already_displayed = {}
def install():
def install(suppress=[]):
pretty_install(console=console)
traceback_install(console=console, extra_lines=1, width=console.width, word_wrap=False, indent_guides=False, suppress=[anyio, starlette, gradio])
traceback_install(console=console, extra_lines=1, width=console.width, word_wrap=False, indent_guides=False, suppress=suppress)
logging.basicConfig(level=logging.INFO, format='%(asctime)s | %(levelname)s | %(pathname)s | %(message)s')
for handler in logging.getLogger().handlers:
handler.setLevel(logging.INFO)
def print_error_explanation(message):
@@ -28,9 +31,9 @@ def print_error_explanation(message):
print(line, file=sys.stderr)
def display(e: Exception, task):
def display(e: Exception, task, suppress=[]):
print(f"{task or 'error'}: {type(e).__name__}", file=sys.stderr)
console.print_exception(show_locals=False, max_frames=2, extra_lines=1, suppress=[anyio, starlette, gradio], theme="ansi_dark", word_wrap=False, width=min([console.width, 200]))
console.print_exception(show_locals=False, max_frames=2, extra_lines=1, suppress=suppress, theme="ansi_dark", word_wrap=False, width=min([console.width, 200]))
def display_once(e: Exception, task):
@@ -47,5 +50,5 @@ def run(code, task):
display(e, task)
def exception():
console.print_exception(show_locals=False, max_frames=10, extra_lines=2, suppress=[anyio, starlette, gradio], theme="ansi_dark", word_wrap=False, width=min([console.width, 200]))
def exception(suppress=[]):
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]))