0ptr/nullptr/command_line.py

98 lines
2.3 KiB
Python
Raw Normal View History

import shlex
import inspect
import sys
import importlib
2023-06-10 17:39:32 +00:00
import logging
def func_supports_argcount(f, cnt):
argspec = inspect.getargspec(f)
posargs = 0 if argspec.args is None else len(argspec.args)
if argspec.args[0] == 'self':
posargs = posargs - 1
defargs = 0 if argspec.defaults is None else len(argspec.defaults)
varargs = 0 if argspec.varargs is None else len(argspec.varargs)
minargs = posargs - defargs
maxargs = posargs
if cnt < minargs:
return False
if cnt > maxargs:
return varargs > 0
return True
class CommandLine:
def __init__(self):
self.reloading = False
self.stopping = False
def ask_multichoice(self, choices, prompt):
choice = None
while choice not in choices:
choice = input(prompt).lower()
return choice
def stop(self):
self.stopping = True
def prompt(self):
2023-06-09 11:19:47 +00:00
return '> '
def handle_not_found(self, c, args):
print(f'command not found; {c}')
def handle_error(self, cmd, args, e):
logging.error(e, exc_info=type(e).__name__ not in ['ApiError','CommandError', 'CentralCommandError'])
def handle_empty(self):
pass
2023-06-10 17:39:32 +00:00
def after_cmd(self):
pass
def handle_eof(self):
print('EOF')
def do_quit(self):
print('byebye!')
self.stopping = True
def do_reload(self):
self.reloading = True
def handle_cmd(self, c):
if c == '':
return self.handle_empty()
args = shlex.split(c)
cmd = args.pop(0)
str_handler = f'do_{cmd}'
if not hasattr(self, str_handler):
try:
self.handle_not_found(cmd, args)
return
except Exception as e:
self.handle_error(cmd, args, e)
handler = getattr(self, str_handler)
if not func_supports_argcount(handler, len(args)):
expect_args = ', '.join(inspect.getargspec(handler).args[1:])
print('expected args: ' + expect_args)
return
try:
handler(*args)
except Exception as e:
self.handle_error(cmd, args, e)
2023-06-10 17:39:32 +00:00
self.after_cmd()
def run(self):
while not self.stopping and not self.reloading:
p = self.prompt()
try:
c = input(p)
except EOFError:
self.handle_eof()
break
try:
self.handle_cmd(c)
except Exception as e:
logging.error(e, exc_info=True)