Merge "Cli list option for suite"

1 files changed, 52 insertions, 0 deletions

diff --git a/cli/commands/suite.py b/cli/commands/suite.py
new file mode 100644
index 00000000..757f11a1
--- /dev/null
+++ b/cli/commands/suite.py
@@ -0,0 +1,52 @@
+##############################################################################
+# Copyright (c) 2016 ZTE Corp and others.
+#
+# All rights reserved. This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+# http://www.apache.org/licenses/LICENSE-2.0
+##############################################################################
+
+from prettytable import PrettyTable
+import os
+import click
+
+
+class Suite:
+
+    def __init__(self):
+        self.path = os.path.join(os.path.dirname(__file__), '..', '..', 'benchmarks/suite')
+
+    def list(self):
+        table = PrettyTable(["Name"])
+        table.align = 'l'
+        suites = os.listdir(self.path)
+        for suite in suites:
+            table.add_row([suite])
+        click.echo(table)
+
+    def run(self):
+        print("Run a suite")
+
+
+@click.group()
+def cli():
+    pass
+
+
+@cli.group()
+@click.pass_context
+def suite(ctx):
+    pass
+
+_suite = Suite()
+
+
+@suite.command("list", help="Lists all the available suites")
+def list():
+    _suite.list()
+
+
+@suite.command("run", help="Execute one complete suite")
+def execute():
+    _suite.run()