57 lines
1.4 KiB
Python
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()
|