mirror of
https://github.com/Benexl/FastAnime.git
synced 2025-12-25 12:24:52 -08:00
31 lines
725 B
Python
31 lines
725 B
Python
import os
|
|
import sys
|
|
|
|
from rich.traceback import install as rich_install
|
|
|
|
from ...core.constants import PROJECT_NAME
|
|
|
|
|
|
def custom_exception_hook(exc_type, exc_value, exc_traceback):
|
|
print(f"{exc_type.__name__}: {exc_value}")
|
|
|
|
|
|
default_exception_hook = sys.excepthook
|
|
|
|
|
|
def setup_exceptions_handler(
|
|
trace: bool | None,
|
|
dev: bool | None,
|
|
rich_traceback: bool | None,
|
|
rich_traceback_theme: str,
|
|
):
|
|
if dev:
|
|
# auto set env
|
|
os.environ[f"{PROJECT_NAME}_DEBUG"] = "1"
|
|
if trace or dev:
|
|
sys.excepthook = default_exception_hook
|
|
if rich_traceback:
|
|
rich_install(show_locals=True, theme=rich_traceback_theme)
|
|
else:
|
|
sys.excepthook = custom_exception_hook
|