diff options
Diffstat (limited to 'pywhoisd_dataloader.py')
-rwxr-xr-x | pywhoisd_dataloader.py | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/pywhoisd_dataloader.py b/pywhoisd_dataloader.py new file mode 100755 index 0000000..7b80ef5 --- /dev/null +++ b/pywhoisd_dataloader.py @@ -0,0 +1,54 @@ +#!/usr/bin/python3 +import signal, sys + +from lib.outputmanager import OutputManager +from lib.commandmanager import CommandManager +from lib.completionmanager import CompletionManager + +from lib.exceptions import CmdNotFoundException, CommandException + +class Manager: + def __init__(self): + self.loader = None + self.completer = CompletionManager() + + def start(self): + try: + signal.signal(signal.SIGINT, signal.default_int_handler) + while True: + try: + try: + line = input('% ') + except KeyboardInterrupt: + output_manager.line_break() + continue + + cmd_name = line.strip().split(' ') + if len(cmd_name) > 0 and len(cmd_name[0]) > 0: + cmd = cmd_manager.find(cmd_name[0]) + cmd.execute(self.loader, line[1:] if len(line) > 1 else []) + + except CommandException as err: + output_manager.error(str(err)).linebreak() + if err.print_usage: + output_manager.normal('Usage: {0}'.format(cmd.usage(line[0]))).line_break() + + except CmdNotFoundException as err: + output_manager.error('Error: {0}'.format(str(err))).linebreak() + + except EOFError: + output_manager.line_break() + exit(0) + except: + raise + +if __name__ == '__main__': + try: + manager = Manager() + cmd_manager = CommandManager() + output_manager = OutputManager() + manager.start() + + except: # TODO: Improve exceptions handling + print("Unexpected error:", sys.exc_info()[0]) + raise |