github/supysonic
1
0
Fork 0
mirror of https://github.com/spl0k/supysonic.git synced 2024-12-23 01:16:18 +00:00
Code Issues Releases Wiki Activity

CLI: autocomplete the first arg of commands

Browse Source
This commit is contained in:
spl0k 2013-09-07 16:47:39 +02:00
parent 012546b155
commit 4d140bdd05
1 changed files with 20 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
cli.py
View File

@ -21,6 +21,26 @@ class CLI(cmd.Cmd):
def postloop(self): def postloop(self):
print print
def completedefault(self, text, line, begidx, endidx):
# First check if we have subparsers for the command
parsers = getattr(self, line.split()[0] + '_subparsers', None)
if not parsers:
return []
# Count how many 'words' (minus one) to begidx (there must be a better way to do that)
whitespace = False
num_words = 0
for i in xrange(len(line.split()[0]), begidx):
ws = line[i].isspace()
if ws != whitespace:
if not ws:
num_words = num_words + 1
whitespace = ws
if num_words == 0:
return [ a for a in parsers.choices.keys() if a.startswith(text) ]
return []
folder_parser = CLIParser(prog = 'folder', add_help = False) folder_parser = CLIParser(prog = 'folder', add_help = False)
folder_subparsers = folder_parser.add_subparsers(dest = 'action') folder_subparsers = folder_parser.add_subparsers(dest = 'action')
folder_subparsers.add_parser('list', help = 'Lists folders', add_help = False) folder_subparsers.add_parser('list', help = 'Lists folders', add_help = False)