Merge "Calculate a suitable column width for positional arguments"

1 files changed, 12 insertions, 0 deletions

diff --git a/keystoneclient/shell.py b/keystoneclient/shell.py
index 1305c89..61b93e9 100644
--- a/keystoneclient/shell.py
+++ b/keystoneclient/shell.py
@@ -438,6 +438,18 @@ class OpenStackIdentityShell(object):
 
 # I'm picky about my shell help.
 class OpenStackHelpFormatter(argparse.HelpFormatter):
+    INDENT_BEFORE_ARGUMENTS = 6
+    MAX_WIDTH_ARGUMENTS = 32
+
+    def add_arguments(self, actions):
+        for action in filter(lambda x: not x.option_strings, actions):
+            for choice in action.choices:
+                length = len(choice) + self.INDENT_BEFORE_ARGUMENTS
+                if(length > self._max_help_position and
+                   length <= self.MAX_WIDTH_ARGUMENTS):
+                    self._max_help_position = length
+        super(OpenStackHelpFormatter, self).add_arguments(actions)
+
     def start_section(self, heading):
         # Title-case the headings
         heading = '%s%s' % (heading[0].upper(), heading[1:])