accounting/bench
2
0
Fork 0
mirror of https://github.com/frappe/bench.git synced 2024-11-15 01:34:05 +00:00
Code Issues Releases Activity

fix: Setup exception handler for CommandFailedError only

Browse Source 
* Match only certain module lib paths
This commit is contained in:
Gavin D'souza 2023-05-10 15:40:11 +05:30
parent 8175239434
commit 94a1b8c3fb
No known key found for this signature in database
GPG Key ID: 3A7BF4D4340DE6F7
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
bench/cli.py
View File

@ -28,6 +28,8 @@ from bench.utils import (
get_cmd_from_sysargv, get_cmd_from_sysargv,
) )
from bench.utils.bench import get_env_cmd from bench.utils.bench import get_env_cmd
from importlib.util import find_spec
# these variables are used to show dynamic outputs on the terminal # these variables are used to show dynamic outputs on the terminal
dynamic_feed = False dynamic_feed = False
@ -38,6 +40,7 @@ bench.LOG_BUFFER = []
change_uid_msg = "You should not run this command as root" change_uid_msg = "You should not run this command as root"
src = os.path.dirname(__file__) src = os.path.dirname(__file__)
SKIP_MODULE_TRACEBACK = ("click",)
@contextmanager @contextmanager
@ -246,13 +249,22 @@ def setup_clear_cache():
def setup_exception_handler(): def setup_exception_handler():
from traceback import format_exception from traceback import format_exception
from bench.exceptions import CommandFailedError
def handle_exception(exc_type, exc_info, tb): def handle_exception(exc_type, exc_info, tb):
print("".join(generate_exc(exc_type, exc_info, tb))) if exc_type == CommandFailedError:
print("".join(generate_exc(exc_type, exc_info, tb)))
else:
sys.__excepthook__(exc_type, exc_info, tb)
def generate_exc(exc_type, exc_info, tb): def generate_exc(exc_type, exc_info, tb):
for t in format_exception(exc_type, exc_info, tb): TB_SKIP = [
if "/click/" not in t: os.path.dirname(find_spec(module).origin) for module in SKIP_MODULE_TRACEBACK
yield t ]
for tb_line in format_exception(exc_type, exc_info, tb):
for skip_module in TB_SKIP:
if skip_module not in tb_line:
yield tb_line
sys.excepthook = handle_exception sys.excepthook = handle_exception