1 files changed, 90 insertions, 0 deletions
diff --git a/cvp/docker/db/init_db.py b/cvp/docker/db/init_db.py
new file mode 100644
index 00000000..2aac6fc5
--- /dev/null
+++ b/cvp/docker/db/init_db.py
@@ -0,0 +1,90 @@
+##############################################################################
+# Copyright (c) 2016 HUAWEI TECHNOLOGIES CO.,LTD 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
+##############################################################################
+
+import requests
+import json
+import sys
+
+db_host_ip = sys.argv[1]
+testapi_port = sys.argv[2]
+
+target_url = 'http://{}:{}/api/v1'.format(db_host_ip, testapi_port)
+print(target_url)
+
+
+def get(url):
+    return requests.get(url).json()
+
+
+def post(url, data):
+    headers = {'Content-Type': 'application/json'}
+    res = requests.post(url, data=json.dumps(data), headers=headers)
+    print(res.text)
+
+
+def pod():
+    target = '{}/pods'.format(target_url)
+
+    with open('pods.json', 'r') as f:
+        pods = json.load(f)
+    for p in pods:
+        post(target, p)
+
+    add_pod('master', 'metal')
+    add_pod('virtual_136_2', 'virtual')
+
+
+def project():
+    target = '{}/projects'.format(target_url)
+    with open('projects.json', 'r') as f:
+        projects = json.load(f)
+    for p in projects:
+        post(target, p)
+
+
+def cases():
+    with open('cases.json', 'r') as f:
+        for line in f:
+            try:
+                cases = json.loads(line)
+                for c in cases["testcases"]:
+                    target = '{}/projects/{}/cases'.format(target_url,
+                                                           c['project_name'])
+                    print(target)
+                    post(target, c)
+            except:
+                print("useless data")
+    add_case("functest", "tempest_custom")
+
+
+def add_pod(name, mode):
+    data = {
+        "role": "",
+        "name": name,
+        "details": '',
+        "mode": mode,
+        "creation_date": "2017-2-23 11:23:03.765581"
+    }
+    pod_url = '{}/pods'.format(target_url)
+    post(pod_url, data)
+
+
+def add_case(project, case):
+    data = {
+        "project_name": project,
+        "name": case,
+    }
+    case_url = '{}/projects/{}/cases'.format(target_url, project)
+    post(case_url, data)
+
+
+if __name__ == '__main__':
+    pod()
+    project()
+    cases()