From 8f1101df131a4d3e03b377738507d88b745831c0 Mon Sep 17 00:00:00 2001
From: "Yiting.Li" <liyiting@huawei.com>
Date: Tue, 22 Dec 2015 17:11:12 -0800
Subject:     Upload the contribution of vstf as bottleneck network framework.

    End to End Performance test

    JIRA:BOTTLENECK-29

Change-Id: Ib2c553c8b60d6cda9e7a7b52b737c9139f706ebd
Signed-off-by: Yiting.Li <liyiting@huawei.com>
---
 vstf/vstf/common/vstfcli.py | 62 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)
 create mode 100755 vstf/vstf/common/vstfcli.py

(limited to 'vstf/vstf/common/vstfcli.py')

diff --git a/vstf/vstf/common/vstfcli.py b/vstf/vstf/common/vstfcli.py
new file mode 100755
index 00000000..9dc99779
--- /dev/null
+++ b/vstf/vstf/common/vstfcli.py
@@ -0,0 +1,62 @@
+import argparse
+import sys
+
+
+class VstfHelpFormatter(argparse.HelpFormatter):
+    def start_section(self, heading):
+        # Title-case the headings
+        heading = '%s%s' % (heading[0].upper(), heading[1:])
+        super(VstfHelpFormatter, self).start_section(heading)
+
+
+class VstfParser(argparse.ArgumentParser):
+    def __init__(self,
+                 prog='vstf',
+                 description="",
+                 epilog='',
+                 add_help=True,
+                 formatter_class=VstfHelpFormatter):
+
+        super(VstfParser, self).__init__(
+            prog=prog,
+            description=description,
+            epilog=epilog,
+            add_help=add_help,
+            formatter_class=formatter_class)
+        self.subcommands = {}
+
+    def _find_actions(self, subparsers, actions_module):
+        for attr in (a for a in dir(actions_module) if a.startswith('do_')):
+            command = attr[3:].replace('_', '-')
+            callback = getattr(actions_module, attr)
+            desc = callback.__doc__ or ''
+            action_help = desc.strip()
+            arguments = getattr(callback, 'arguments', [])
+            subparser = subparsers.add_parser(command,
+                                              help=action_help,
+                                              description=desc,
+                                              add_help=False,
+                                              formatter_class=VstfHelpFormatter)
+            subparser.add_argument('-h', '--help',
+                                   action='help',
+                                   help=argparse.SUPPRESS)
+            self.subcommands[command] = subparser
+            for (args, kwargs) in arguments:
+                subparser.add_argument(*args, **kwargs)
+            subparser.set_defaults(func=callback)
+
+    def set_subcommand_parser(self, target, metavar="<subcommand>"):
+        subparsers = self.add_subparsers(metavar=metavar)
+        self._find_actions(subparsers, target)
+        return subparsers
+
+    def set_parser_to_subcommand(self, subparser, target):
+        self._find_actions(subparser, target)
+
+
+if __name__ == "__main__":
+    from vstf.common import test_func
+    parser = VstfParser(prog="vstf", description="test parser")
+    parser.set_subcommand_parser(test_func)
+    args = parser.parse_args(sys.argv[1:])
+    args.func(args)
-- 
cgit 1.2.3-korg