Remove only :return:

kratos47mhs · Nov 20, 2019 · 8b639b5 · 8b639b5
1 parent 6339962
commit 8b639b5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/scripts/rest_client.py b/scripts/rest_client.py
@@ -10,6 +10,7 @@
 import argparse
 import inspect
 import json
+import re
 import logging
 import sys
 from pathlib import Path
@@ -238,7 +239,8 @@ def print_commands():
     print("Possible commands:\n")
     for x, y in inspect.getmembers(client):
         if not x.startswith('_'):
-            print(f"{x}""\n        "f"{getattr(client, x).__doc__.splitlines()[0]}""\n")
+            doc = re.sub(':return:.*', '', getattr(client, x).__doc__, flags=re.MULTILINE).rstrip()
+            print(f"{x}\n\t{doc}\n")
 
 
 def main(args):