codeking/sp-hydra-veil-cli
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Implement a custom global exception hook

Browse source
This commit is contained in:
codeking 2025-02-22 21:44:58 +01:00
parent bed16c2a8f
commit 60a262ef9c
1 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
cli/__main__.py
View file

@ -1,5 +1,5 @@
from core.Constants import Constants from core.Constants import Constants
from core.Errors import MissingSubscriptionError, InvalidSubscriptionError from core.Errors import MissingSubscriptionError, InvalidSubscriptionError, UnknownConnectionTypeError
from core.controllers.ApplicationController import ApplicationController from core.controllers.ApplicationController import ApplicationController
from core.controllers.ApplicationVersionController import ApplicationVersionController from core.controllers.ApplicationVersionController import ApplicationVersionController
from core.controllers.ClientController import ClientController from core.controllers.ClientController import ClientController
@ -25,6 +25,7 @@ from typing import Optional, Union
import argparse import argparse
import pprint import pprint
import re import re
import sys
if __name__ == '__main__': if __name__ == '__main__':
@ -80,6 +81,16 @@ if __name__ == '__main__':
return candidate return candidate
def __handle_exception(identifier, message, traceback):
if issubclass(identifier, UnknownConnectionTypeError):
print('Please specify the desired connection method and try again.\n')
else:
sys.__excepthook__(identifier, message, traceback)
sys.excepthook = __handle_exception
pristine_parser = argparse.ArgumentParser(add_help=False) pristine_parser = argparse.ArgumentParser(add_help=False)
pristine_parser.add_argument('--pristine', '-p', action='store_true') pristine_parser.add_argument('--pristine', '-p', action='store_true')