Auto-generated patch by python-black
[integration/test.git] / csit / scripts / data_generate.py
index b0725210888f6d88d1dec40f3a898e3f5a70ed36..c876fb425808aca233c4ea71e2aa4a4bf051417d 100644 (file)
@@ -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