From a6892e258aaa7dbaf515a1043991492a0ae22727 Mon Sep 17 00:00:00 2001
From: Anders Roxell <anders.roxell@linaro.org>
Date: Wed, 20 Dec 2023 12:01:44 +0100
Subject: [PATCH] squad-download-attachments: save to a csv file

Signed-off-by: Anders Roxell <anders.roxell@linaro.org>
---
 README.md                  |  3 ++-
 squad-download-attachments | 37 +++++++++++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 7cf695d..25fd179 100644
--- a/README.md
+++ b/README.md
@@ -306,13 +306,14 @@ They will be stored in a directory 'stored_attachments/<environment>'_'<testrun_
 
 ```
 ./squad-download-attachments --help
-usage: squad-download-attachments [-h] --group GROUP --project PROJECT --build BUILD_ID
+usage: squad-download-attachments [-h] --group GROUP --project PROJECT --build BUILD_ID [--csv]
 
 options:
   -h, --help            show this help message and exit
   --group GROUP         The name of the SQUAD group.
   --project PROJECT     The name of the SQUAD project.
   --build BUILD         SQUAD build id.
+  --csv                 Create csv files.
 ```
 
 ### `read-skipfile-results`: Read results from
diff --git a/squad-download-attachments b/squad-download-attachments
index d3acd5f..38abe46 100755
--- a/squad-download-attachments
+++ b/squad-download-attachments
@@ -6,10 +6,12 @@
 # SPDX-License-Identifier: MIT
 
 import argparse
+import csv
 import json
 import logging
 from os import chdir
 from pathlib import Path
+import statistics
 import sys
 from squad_client.core.api import SquadApi
 from squad_client.core.models import ALL, Squad, TestRun, Environment
@@ -47,6 +49,14 @@ def arg_parser():
         help="squad build",
     )
 
+    parser.add_argument(
+        "--csv",
+        default=False,
+        action="store_true",
+        required=False,
+        help="generate csv files",
+    )
+
     return parser
 
 
@@ -109,6 +119,33 @@ def run():
                 with open(file_write, mode="w") as write_file:
                     write_file.write(pjson)
 
+                if not args.csv:
+                    continue
+
+                with open(file_write.replace(".json", ".csv"), mode="w") as csv_file:
+                    csv_writer = csv.writer(csv_file)
+                    dict_json = json.loads(pjson)
+                    if not dict_json["results"]:
+                        continue
+                    headers = ["median", "mean", "standard diviation", "name", "iteration", "name_iteration", "raw data..."]
+                    csv_writer.writerow(headers)
+                    for key in dict_json["results"]["_ResultData"]:
+                       iterations = 0
+                       for k in dict_json["results"]["_ResultData"][key]:
+                            csv_data = []
+                            float_arr = []
+                            for number in k["Values"]:
+                                float_arr.append(float(number))
+                            csv_data.append(statistics.median(float_arr))
+                            csv_data.append(statistics.mean(float_arr))
+                            csv_data.append(statistics.stdev(float_arr))
+                            csv_data.append(key)
+                            iterations = iterations + 1
+                            csv_data.append(f"iteration_{iterations}")
+                            csv_data.append(f"{key}_iterattion_{iterations}")
+                            csv_data.extend(k['Values'])
+                            csv_writer.writerow(csv_data)
+
 
 if __name__ == "__main__":
     sys.exit(run())