python_examples/cli/key_menu.py

57 lines
1.4 KiB
Python

from prompt_toolkit.key_binding import KeyBindings, merge_key_bindings
from prompt_toolkit.key_binding.bindings.basic import load_basic_bindings
from prompt_toolkit.shortcuts import prompt, PromptSession
from rich.table import Table
from rich.console import Console
keybindings = load_basic_bindings()
@keybindings.add('c-q')
def exit_(event):
""" Pressing 'c-q' will exit the application. """
exit(0)
@keybindings.add('c-m')
def open_menu(event):
""" Pressing 'c-m' will open the menu. """
show_keybindings_menu()
event.app.exit()
@keybindings.add('enter')
def process_input(event):
""" Pressing 'enter' will process the input. """
print(dir(event))
event.app.exit(result=event.current_buffer.text)
bindings_info = [
{"Key": "Ctrl-Q", "Action": "Exit the application"},
{"Key": "Ctrl-M", "Action": "Open the menu"},
]
console = Console()
def show_keybindings_menu():
table = Table(title="Keybindings")
table.add_column("Key", justify="right", style="cyan", no_wrap=True)
table.add_column("Action", style="magenta")
for binding in bindings_info:
table.add_row(binding["Key"], binding["Action"])
console.print(table)
def main_menu():
session = PromptSession("> ", key_bindings=keybindings)
while True:
show_keybindings_menu()
user_input = session.prompt()
console.print(f"You entered: {user_input}")
if __name__ == "__main__":
main_menu()