2022-11-30 00:09:23 -08:00
|
|
|
from prompt_toolkit import prompt
|
|
|
|
from prompt_toolkit.completion import Completer, Completion
|
|
|
|
|
|
|
|
|
|
|
|
class BasePrompt(Completer):
|
|
|
|
|
|
|
|
def __init__(self, manager=None, parent=None):
|
|
|
|
super(BasePrompt, self).__init__()
|
|
|
|
|
2022-12-07 23:41:49 -08:00
|
|
|
if (not manager and not parent): # pragma: no cover
|
2022-11-30 00:09:23 -08:00
|
|
|
raise RuntimeError("Must define either a database manager or a parent object.")
|
|
|
|
|
|
|
|
self._prompt = ''
|
|
|
|
self._values = []
|
|
|
|
self._parent = parent
|
|
|
|
self._manager = manager
|
2022-12-10 10:14:06 -08:00
|
|
|
self._commands = {'help': self.help}
|
|
|
|
|
|
|
|
@property
|
|
|
|
def commands(self):
|
|
|
|
return self._commands
|
2022-11-30 00:09:23 -08:00
|
|
|
|
|
|
|
@property
|
|
|
|
def usage(self):
|
|
|
|
return self.__class__.__name__
|
|
|
|
|
|
|
|
@property
|
|
|
|
def help_text(self):
|
|
|
|
return self.__doc__
|
|
|
|
|
|
|
|
@property
|
|
|
|
def manager(self):
|
|
|
|
if self._manager:
|
|
|
|
return self._manager
|
|
|
|
elif self._parent:
|
|
|
|
return self._parent.manager
|
|
|
|
|
|
|
|
@property
|
|
|
|
def parent(self):
|
|
|
|
return self._parent
|
|
|
|
|
|
|
|
@property
|
|
|
|
def prompt(self):
|
|
|
|
return self._prompt
|
|
|
|
|
|
|
|
@property
|
|
|
|
def values(self):
|
2022-12-10 10:14:06 -08:00
|
|
|
return [k for k in self.commands.keys() if not k.startswith('_')]
|
2022-11-30 00:09:23 -08:00
|
|
|
|
|
|
|
def get_completions(self, document, complete_event):
|
|
|
|
word = document.get_word_before_cursor()
|
|
|
|
found = False
|
|
|
|
for value in self.values:
|
|
|
|
if word in value:
|
|
|
|
found = True
|
|
|
|
yield Completion(value, start_position=-len(word))
|
|
|
|
if not found:
|
|
|
|
try:
|
|
|
|
for result in self.default_completer(document, complete_event):
|
|
|
|
yield result
|
|
|
|
except NotImplementedError:
|
|
|
|
pass
|
|
|
|
|
|
|
|
def start(self, cmd=''):
|
|
|
|
while True:
|
|
|
|
if not cmd:
|
2022-12-10 10:14:06 -08:00
|
|
|
cmd = prompt(f'{self.prompt} ', completer=self, complete_while_typing=True)
|
2022-11-30 00:09:23 -08:00
|
|
|
if not cmd:
|
|
|
|
return
|
|
|
|
cmd, *parts = cmd.split()
|
|
|
|
if not self.process(cmd, *parts):
|
|
|
|
return
|
|
|
|
cmd = ''
|
|
|
|
|
2022-12-10 10:14:06 -08:00
|
|
|
def help(self, parts):
|
|
|
|
if not parts:
|
|
|
|
print(self.__doc__)
|
|
|
|
else:
|
|
|
|
print(getattr(self, parts[0]).__doc__)
|
|
|
|
return True
|
|
|
|
|
2022-11-30 00:09:23 -08:00
|
|
|
def default_completer(self, document, complete_event):
|
|
|
|
raise NotImplementedError()
|
|
|
|
|
|
|
|
def process(self, cmd, *parts):
|
|
|
|
raise NotImplementedError()
|