Files
FastAnime/viu_media/cli/utils/exception.py
2025-08-18 01:08:27 +03:00

25 lines
586 B
Python

import sys
from rich.traceback import install as rich_install
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 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