From 768bad65409c069fdc2b7bf94d07a2ad34e0a00d Mon Sep 17 00:00:00 2001 From: Ian Weller Date: Fri, 18 Dec 2009 10:45:25 -0600 Subject: [PATCH] Fix subcommand creation --- src/mw/cli.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/mw/cli.py b/src/mw/cli.py index 7dc49c5..e1d96c0 100644 --- a/src/mw/cli.py +++ b/src/mw/cli.py @@ -36,6 +36,10 @@ class CLI(object): cmd = clazz() self.commands[cmd.name] = cmd self.shortcuts[cmd.name] = cmd.shortcuts + self.all_commands = self.commands + for command in self.shortcuts: + for shortcut in self.shortcuts[command]: + self.all_commands[shortcut] = self.commands[command] def usage(self): print 'usage: %s [subcommand]' % self.me @@ -53,7 +57,7 @@ class CLI(object): def main(self): # determine what the subcommand is if len(sys.argv) > 1: - if sys.argv[1] in self.commands.keys(): + if sys.argv[1] in self.all_commands: the_command = sys.argv[1] # SWEET ACTION elif sys.argv[1] in ['--help', '-h']: self.usage() -- 2.30.2