X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=csit%2Fscripts%2Fdata_generate.py;h=c876fb425808aca233c4ea71e2aa4a4bf051417d;hb=1c106006ffb33c1c60e2f8abe61e88a7bd95a7bd;hp=b0725210888f6d88d1dec40f3a898e3f5a70ed36;hpb=e235045c2ad12d41416e2583609a0178c8bd9287;p=integration%2Ftest.git

diff --git a/csit/scripts/data_generate.py b/csit/scripts/data_generate.py
index b072521088..c876fb4258 100644
--- a/csit/scripts/data_generate.py
+++ b/csit/scripts/data_generate.py
@@ -9,55 +9,52 @@ def generate():
     BODY = {}
 
     ts = time.time()
-    formatted_ts = \
-        datetime.fromtimestamp(ts).strftime('%Y-%m-%dT%H:%M:%S.%fZ')
-    BODY['@timestamp'] = formatted_ts
+    formatted_ts = datetime.fromtimestamp(ts).strftime("%Y-%m-%dT%H:%M:%S.%fZ")
+    BODY["@timestamp"] = formatted_ts
 
     # Plots are obtained from csv files ( in archives directory in $WORKSPACE).
 
-    csv_files = glob.glob('archives/*.csv')
-    BODY['project'] = 'opendaylight'
-    BODY['subject'] = 'test'
+    csv_files = glob.glob("archives/*.csv")
+    BODY["project"] = "opendaylight"
+    BODY["subject"] = "test"
 
     # If there are no csv files, then it is a functional test.
     # Parse csv files and fill perfomance parameter values
 
     if len(csv_files) == 0:
-        BODY['test-type'] = 'functional'
+        BODY["test-type"] = "functional"
     else:
-        BODY['test-type'] = 'performance'
-        BODY['plots'] = {}
+        BODY["test-type"] = "performance"
+        BODY["plots"] = {}
         for f in csv_files:
-            key = (f.split('/')[-1])[:-4]
-            BODY['plots'][key] = {}
+            key = (f.split("/")[-1])[:-4]
+            BODY["plots"][key] = {}
             with open(f) as file:
                 lines = file.readlines()
-            props = lines[0].strip().split(',')
-            vals = lines[1].strip().split(',')
+            props = lines[0].strip().split(",")
+            vals = lines[1].strip().split(",")
             for i in range(len(props)):
-                BODY['plots'][key][props[i]] = float(vals[i])
+                BODY["plots"][key][props[i]] = float(vals[i])
 
     # Fill the required parameters whose values are obtained from environment.
 
-    BODY['jenkins-silo'] = os.environ['SILO']
-    BODY['test-name'] = os.environ['JOB_NAME']
-    BODY['test-run'] = int(os.environ['BUILD_NUMBER'])
+    BODY["jenkins-silo"] = os.environ["SILO"]
+    BODY["test-name"] = os.environ["JOB_NAME"]
+    BODY["test-run"] = int(os.environ["BUILD_NUMBER"])
 
     # Parsing robot log for stats on start-time, pass/fail tests and duration.
 
-    robot_log = os.environ['WORKSPACE'] + '/output.xml'
+    robot_log = os.environ["WORKSPACE"] + "/output.xml"
     tree = ET.parse(robot_log)
-    BODY['id'] = '{}-{}'.format(os.environ['JOB_NAME'],
-                                os.environ['BUILD_NUMBER'])
-    BODY['start-time'] = tree.getroot().attrib['generated']
-    BODY['pass-tests'] = int(tree.getroot().find('statistics')
-                             [0][1].get('pass'))
-    BODY['fail-tests'] = int(tree.getroot().find('statistics')
-                             [0][1].get('fail'))
-    endtime = tree.getroot().find('suite').find('status').get('endtime')
-    starttime = tree.getroot().find('suite').find('status').get('starttime')
-    elap_time = datetime.strptime(endtime, '%Y%m%d %H:%M:%S.%f') \
-        - datetime.strptime(starttime, '%Y%m%d %H:%M:%S.%f')
-    BODY['duration'] = str(elap_time)
+    BODY["id"] = "{}-{}".format(os.environ["JOB_NAME"], os.environ["BUILD_NUMBER"])
+    BODY["start-time"] = tree.getroot().attrib["generated"]
+    BODY["pass-tests"] = int(tree.getroot().find("statistics")[0][1].get("pass"))
+    BODY["fail-tests"] = int(tree.getroot().find("statistics")[0][1].get("fail"))
+    endtime = tree.getroot().find("suite").find("status").get("endtime")
+    starttime = tree.getroot().find("suite").find("status").get("starttime")
+    elap_time = datetime.strptime(endtime, "%Y%m%d %H:%M:%S.%f") - datetime.strptime(
+        starttime, "%Y%m%d %H:%M:%S.%f"
+    )
+    BODY["duration"] = str(elap_time)
 
     return BODY