diff --git a/cli/__main__.py b/cli/__main__.py index bbdfc18..294e082 100644 --- a/cli/__main__.py +++ b/cli/__main__.py @@ -1,5 +1,5 @@ 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.ApplicationVersionController import ApplicationVersionController from core.controllers.ClientController import ClientController @@ -25,6 +25,7 @@ from typing import Optional, Union import argparse import pprint import re +import sys if __name__ == '__main__': @@ -80,6 +81,16 @@ if __name__ == '__main__': 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.add_argument('--pristine', '-p', action='store_true')