36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
|
from prompt_toolkit import PromptSession
|
||
|
|
from prompt_toolkit.completion import NestedCompleter
|
||
|
|
from prompt_toolkit.completion import FuzzyWordCompleter
|
||
|
|
from site_tools.cli import app
|
||
|
|
from enum import EnumMeta
|
||
|
|
from inspect import signature
|
||
|
|
from rich import print
|
||
|
|
|
||
|
|
|
||
|
|
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}")
|