1 files changed, 56 insertions, 0 deletions
diff --git a/python/mozperftest/mozperftest/metrics/notebook/transforms/single_json.py b/python/mozperftest/mozperftest/metrics/notebook/transforms/single_json.py
new file mode 100644
index 0000000000..375615fb23
--- /dev/null
+++ b/python/mozperftest/mozperftest/metrics/notebook/transforms/single_json.py
@@ -0,0 +1,56 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+from mozperftest.metrics.notebook.utilities import flat
+
+
+class SingleJsonRetriever:
+    """Transforms perfherder data into the standardized data format."""
+
+    entry_number = 0
+
+    def transform(self, data):
+        self.entry_number += 1
+
+        # flat(data, ()) returns a dict that have one key per dictionary path
+        # in the original data.
+        return [
+            {
+                "data": [{"value": i, "xaxis": self.entry_number} for i in v],
+                "subtest": k,
+            }
+            for k, v in flat(data, ()).items()
+        ]
+
+    def merge(self, sde):
+        grouped_data = {}
+        for entry in sde:
+            subtest = entry["subtest"]
+            data = grouped_data.get(subtest, [])
+            data.extend(entry["data"])
+            grouped_data.update({subtest: data})
+
+        merged_data = [{"data": v, "subtest": k} for k, v in grouped_data.items()]
+
+        self.entry_number = 0
+        return merged_data
+
+    def summary(self, suite):
+        """Summarize a suite of perfherder data into a single value.
+
+        Returning None means that there's no summary. Otherwise, an integer
+        or float must be returned.
+
+        Only available in the Perfherder layer.
+        """
+        return None
+
+    def subtest_summary(self, subtest):
+        """Summarize a set of replicates for a given subtest.
+
+        By default, it returns a None so we fall back to using the
+        average of the replicates which is the default.
+
+        Only available in the Perfherder layer.
+        """
+        return None