From abb21b78411793efcdf8832f6837201bfc4ce15c Mon Sep 17 00:00:00 2001 From: Ralf Jung Date: Thu, 10 Oct 2013 19:15:12 +0200 Subject: [PATCH] tyshell: add tab completion --- tyshell | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tyshell b/tyshell index 1ffccfe..1b90bd0 100755 --- a/tyshell +++ b/tyshell @@ -52,6 +52,12 @@ commands = { 'buzz': sendcmd(tuerSock, 'buzz'), } +def complete_command(cmd): + '''returns a list of commands (as strings) starting with cmd''' + return list(filter(lambda x: x.startswith(cmd), commands.keys())) +readline.set_completer(lambda cmd, num: (complete_command(cmd)+[None])[num]) # wrap complete_command for readline's weird completer API +readline.parse_and_bind("tab: complete") # run completion on tab + # input loop print("Welcome to tyshell. Use help to see what you can do.") while True: @@ -66,7 +72,7 @@ while True: if command[0] in commands: # needed in case a complete command is a prefix of another one cmdoptions = [command[0]] else: - cmdoptions = list(filter(lambda x: command[0].startswith(x), commands.keys())) + cmdoptions = complete_command(command[0]) # check how many we found if len(cmdoptions) == 0: # no commands fit prefix print("Command %s not found. Use help." % command[0]) -- 2.30.2