1 files changed, 88 insertions, 0 deletions
diff --git a/client/nfvbench_client.py b/client/nfvbench_client.py
new file mode 100644
index 0000000..3973b9c
--- /dev/null
+++ b/client/nfvbench_client.py
@@ -0,0 +1,88 @@
+#!/usr/bin/env python
+# Copyright 2017 Cisco Systems, Inc.  All rights reserved.
+#
+#    Licensed under the Apache License, Version 2.0 (the "License"); you may
+#    not use this file except in compliance with the License. You may obtain
+#    a copy of the License at
+#
+#         http://www.apache.org/licenses/LICENSE-2.0
+#
+#    Unless required by applicable law or agreed to in writing, software
+#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+#    License for the specific language governing permissions and limitations
+#    under the License.
+#
+
+#
+# This is an example of python application controling a nfvbench server
+# using the nfvbench client API.
+# The nfvbench server must run in background using the --server option.
+# Since HTML pages are not required, the path to pass to --server can be any directory on the host.
+#
+import argparse
+import json
+import sys
+
+from client import NfvbenchClient
+
+
+#
+# At the CLI, the user can either:
+# - pass an nfvbench configuration as a string (-c <config>)
+# - pass an nfvbench configuration as a file name containing the
+#   configuration (-f <config_file_path>)
+# - or pass a test config (-e <config>) that will be echoed back by the server as is
+#
+def main():
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('-f', '--file', dest='file',
+                        action='store',
+                        help='NFVbench config file to execute (json format)',
+                        metavar='<config_file_path>')
+    parser.add_argument('-c', '--config', dest='config',
+                        action='store',
+                        help='NFVbench config to execute (json format)',
+                        metavar='<config>')
+    parser.add_argument('-e', '--echo', dest='echo',
+                        action='store',
+                        help='NFVbench config to echo (json format)',
+                        metavar='<config>')
+    parser.add_argument('-t', '--timeout', dest='timeout',
+                        default=900,
+                        action='store',
+                        help='time (seconds) to wait for NFVbench result',
+                        metavar='<config>')
+    parser.add_argument('--use-socketio', dest='use_socketio',
+                        action='store_true',
+                        help='NFVbench config to echo (json format)')
+    parser.add_argument('url', help='nfvbench server url (e.g. http://10.0.0.1:5000)')
+    opts = parser.parse_args()
+
+    if not opts.file and not opts.config and not opts.echo:
+        print('at least one of -f or -c or -e required')
+        sys.exit(-1)
+
+    nfvbench = NfvbenchClient(opts.url, opts.use_socketio)
+    # convert JSON into a dict
+    try:
+        timeout = int(opts.timeout)
+        if opts.file:
+            with open(opts.file) as fd:
+                config = json.loads(fd.read())
+                result = nfvbench.run_config(config, timeout=timeout)
+        elif opts.config:
+            config = json.loads(opts.config)
+            result = nfvbench.run_config(config, timeout=timeout)
+        elif opts.echo:
+            config = json.loads(opts.echo)
+            result = nfvbench.echo_config(config, timeout=timeout)
+        print('Result:', result)
+    except ValueError as ex:
+        print('Input configuration is invalid: ' + str(ex))
+        print()
+
+
+if __name__ == "__main__":
+    main()