diff options
author | Joe Server <jserver@broadway.com> | 2013-03-08 10:55:03 -0500 |
---|---|---|
committer | Joe Server <jserver@broadway.com> | 2013-03-08 10:55:03 -0500 |
commit | fb030f15bd8cd1ebded9bc313ae8239e66a28eb5 (patch) | |
tree | 5df991c0cbc71d01c91338e973e0e61ba326b438 | |
parent | 1f34d45e325f3e3e965b22dcf43820e3d8c4f1e6 (diff) | |
download | cliff-fb030f15bd8cd1ebded9bc313ae8239e66a28eb5.tar.gz |
make converting underscores optional in CommandManager
-rw-r--r-- | cliff/commandmanager.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/cliff/commandmanager.py b/cliff/commandmanager.py index 135714a..754f793 100644 --- a/cliff/commandmanager.py +++ b/cliff/commandmanager.py @@ -27,16 +27,20 @@ class CommandManager(object): :param namespace: String containing the setuptools entrypoint namespace for the plugins to be loaded. For example, ``'cliff.formatter.list'``. + :param convert_underscores: Whether cliff should convert underscores to + to spaces in entry_point commands. """ - def __init__(self, namespace): + def __init__(self, namespace, convert_underscores=True): self.commands = {} self.namespace = namespace + self.convert_underscores = convert_underscores self._load_commands() def _load_commands(self): for ep in pkg_resources.iter_entry_points(self.namespace): LOG.debug('found command %r', ep.name) - self.commands[ep.name.replace('_', ' ')] = ep + cmd_name = ep.name.replace('_', ' ') if self.convert_underscores else ep.name + self.commands[cmd_name] = ep return def __iter__(self): |