From 8c58bb558adceb8ff32c7a3e1a88d2a13371dbfa Mon Sep 17 00:00:00 2001
From: kotfu <kotfu@kotfu.net>
Date: Sun, 7 Jan 2018 18:30:34 -0700
Subject: Properly set docstring so it contains help message

---
 cmd2.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

(limited to 'cmd2.py')

diff --git a/cmd2.py b/cmd2.py
index d327b828..c284a66f 100755
--- a/cmd2.py
+++ b/cmd2.py
@@ -258,8 +258,16 @@ def with_argument_parser(argparser):
                     temp_arglist.append(strip_quotes(arg))
                 lexed_arglist = temp_arglist
             opts = argparser.parse_args(lexed_arglist)
-
             func(instance, arg, opts)
+
+        funcdoc = func.__doc__
+        if funcdoc:
+            funcdoc += '\n'
+        else:
+            # if it's None, make it an empty string
+            funcdoc = ''
+        
+        cmd_wrapper.__doc__ = '{}{}'.format(funcdoc, argparser.format_help())
         return cmd_wrapper
     return arg_decorator
 
-- 
cgit v1.2.1