33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
from enum import EnumMeta
|
|
from inspect import signature
|
|
|
|
from prompt_toolkit import PromptSession
|
|
from prompt_toolkit.completion import FuzzyWordCompleter, NestedCompleter
|
|
from rich import print
|
|
|
|
from site_tools.cli import app
|
|
|
|
|
|
def dmsh():
|
|
session = PromptSession()
|
|
|
|
def cmd2dict(cmd):
|
|
sig = signature(cmd)
|
|
if not sig.parameters:
|
|
return None
|
|
cmds = {}
|
|
for k, v in list(sig.parameters.items()):
|
|
print(v, dir(v))
|
|
if v.annotation.__class__ == EnumMeta:
|
|
cmds[k] = FuzzyWordCompleter([e.value for e in v.annotation])
|
|
else:
|
|
cmds[k] = None
|
|
return cmds
|
|
commands = dict(site=dict((c.callback.__name__, cmd2dict(c.callback)) for c in app.registered_commands))
|
|
print(commands)
|
|
completer = NestedCompleter.from_nested_dict(commands)
|
|
text = session.prompt("DM> ", completer=completer, complete_while_typing=True, enable_history_search=False)
|
|
words = text.split()
|
|
|
|
print(f"You said {words}")
|