add metrics preparing for grafana

UnitTestBot · SBOne-Kenobi · Nov 9, 2022 · Oct 24, 2022 · Nov 2, 2022 · Nov 2, 2022
commit cd98c8c3427e6a8238db8a08f9d34788418b4d97
diff --git a/monitoring/insert_metadata.py b/monitoring/insert_metadata.py
@@ -5,24 +5,12 @@
 from collections import OrderedDict
 from datetime import datetime
 from os import environ
-from os.path import exists
 from platform import uname
 from time import time
 from typing import Optional, List
 
 from monitoring_settings import JSON_VERSION
-
-
-def load(json_file: str) -> Optional[any]:
-    """
-    Try load object from json file
-    :param json_file: path to json file
-    :return: object from given json file or None
-    """
-    if exists(json_file):
-        with open(json_file, "r") as f:
-            return json.load(f)
-    return None
+from utils import load
 
 
 def try_get_output(args: str) -> Optional[str]:

diff --git a/monitoring/prepare_metrics.py b/monitoring/prepare_metrics.py
@@ -0,0 +1,91 @@
+import argparse
+import json
+from typing import List
+
+from utils import load
+
+
+def build_metric_struct(name: str, value: any, labels: dict) -> dict:
+    return {
+        "metric": name,
+        "labels": labels,
+        "value": value
+    }
+
+
+def build_metrics_from_data(data: dict, labels: dict) -> List[dict]:
+    result = []
+    fuzzing_ratio = data["parameters"]["fuzzing_ratio"]
+    new_labels = {
+        **labels,
+        "fuzzing_ratio": fuzzing_ratio
+    }
+    metrics = data["metrics"]
+    for metric in metrics:
+        result.append(build_metric_struct(metric, metrics[metric], new_labels))
+    return result
+
+
+def build_metrics_from_data_array(metrics: List[dict], labels: dict) -> List[dict]:
+    result = []
+    for metric in metrics:
+        result.extend(build_metrics_from_data(metric, labels))
+    return result
+
+
+def build_metrics_from_target(target: dict) -> List[dict]:
+    result = []
+    project = target["target"]
+
+    result.extend(build_metrics_from_data_array(
+        target["summarised"],
+        {
+            "project": project
+        }
+    ))
+
+    for class_item in target["by_class"]:
+        class_name = class_item["class_name"]
+        result.extend(build_metrics_from_data_array(
+            class_item["data"],
+            {
+                "project": project,
+                "class": class_name
+            }
+        ))
+
+    return result
+
+
+def build_metrics_from_targets(targets: List[dict]) -> List[dict]:
+    metrics = []
+    for target in targets:
+        metrics.extend(build_metrics_from_target(target))
+    return metrics
+
+
+def get_args():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        '--stats_file', required=True,
+        help='files with statistics after insertion metadata', type=str
+    )
+    parser.add_argument(
+        '--output_file', required=True,
+        help='output file', type=str
+    )
+
+    args = parser.parse_args()
+    return args
+
+
+def main():
+    args = get_args()
+    stats = load(args.stats_file)
+    metrics = build_metrics_from_targets(stats["targets"])
+    with open(args.output_file, "w") as f:
+        json.dump(metrics, f, indent=4)
+
+
+if __name__ == "__main__":
+    main()
diff --git a/monitoring/utils.py b/monitoring/utils.py
@@ -0,0 +1,15 @@
+import json
+from os.path import exists
+from typing import Optional
+
+
+def load(json_file: str) -> Optional[any]:
+    """
+    Try load object from json file
+    :param json_file: path to json file
+    :return: object from given json file or None
+    """
+    if exists(json_file):
+        with open(json_file, "r") as f:
+            return json.load(f)
+    return None