Fixed issue where categorization is skipped when there's a help_<command> function provided.

In verbose help, added check for argparse usage block (starting with 'usage: '), to skip that block and move to the next comment block
Added unit tests for new categorization code
Updated example to demonstrate skipping of argparse usage statement

1 files changed, 8 insertions, 1 deletions

diff --git a/examples/help_categories.py b/examples/help_categories.py
index 2a88edba..8a33e62c 100755
--- a/examples/help_categories.py
+++ b/examples/help_categories.py
@@ -4,7 +4,8 @@
 A sample application for tagging categories on commands.
 """
 
-from cmd2 import Cmd, categorize, __version__
+from cmd2 import Cmd, categorize, __version__, with_argparser
+import argparse
 
 
 class HelpCategories(Cmd):
@@ -51,6 +52,12 @@ class HelpCategories(Cmd):
         """Redeploy command"""
         self.poutput('Redeploy')
 
+    restart_parser = argparse.ArgumentParser(formatter_class=argparse.RawTextHelpFormatter)
+    restart_parser.add_argument('when', default='now',
+                                choices=['now', 'later', 'sometime', 'whenever'],
+                                help='Specify when to restart')
+
+    @with_argparser(restart_parser)
     def do_restart(self, _):
         """Restart command"""
         self.poutput('Restart')