tyshell: convenience command shortcuts
[saartuer.git] / tyshell
diff --git a/tyshell b/tyshell
index 9350e125272cd4f2ff7e046808ed0364d96ec2ff..0b9b94361fde70972e65c48a897716e7ee8f9316 100755 (executable)
--- a/tyshell
+++ b/tyshell
@@ -20,7 +20,7 @@ atexit.register(print, "Bye")
 
 # available commands
 def helpcmd(c):
 
 # available commands
 def helpcmd(c):
-       print("Available commands: %s" % ", ".join(sorted(commands.keys())))
+       print("Available commands: %s" % ", ".join(sorted(longcommands.keys())))
 
 def extcmd(cmd):
        def run(c):
 
 def extcmd(cmd):
        def run(c):
@@ -53,6 +53,39 @@ commands = {
        'buzz': sendcmd(tuerSock, 'buzz'),
 }
 
        'buzz': sendcmd(tuerSock, 'buzz'),
 }
 
+# command convenience shortcuts
+def filterCommonPrefix (strings, length):
+       # ignores duplicates in the string list "strings"
+       toremove=[]
+       for a in strings:
+               for b in strings:
+                       if a != b:
+                               if a[:length] == b[:length]:
+                                       toremove.append(a)
+                                       toremove.append(b)
+       ret = list(strings) # copy
+       for x in toremove:
+               try:
+                       ret.remove(x)
+               except ValueError:
+                       pass
+       return ret
+
+def shortcutify (dic):
+       maxlen = 0
+       for x in dic.keys():
+               if len(x) > maxlen:
+                       maxlen = len(x)
+       for i in range(maxlen):
+               shortable = filterCommonPrefix (dic.keys(), i)
+               for x in shortable:
+                       dic[x[:i]] = dic[x]
+       return dic # only for convenience, as dic is passed by reference
+
+longcommands = commands.copy()
+shortcutify (commands)
+
+
 # input loop
 print("Welcome to tyshell. Use help to see what you can do.")
 while True:
 # input loop
 print("Welcome to tyshell. Use help to see what you can do.")
 while True: