initial public release

src/api-service/__app__/onefuzzlib/reports.py

@@ -0,0 +1,56 @@
+#!/usr/bin/env python
+#
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+import json
+import logging
+from typing import Optional
+
+from onefuzztypes.models import Report
+from pydantic import ValidationError
+
+from .azure.containers import get_blob
+
+
+def parse_report(content: str, metadata: Optional[str] = None) -> Optional[Report]:
+    if isinstance(content, bytes):
+        try:
+            content = content.decode()
+        except UnicodeDecodeError as err:
+            logging.error(
+                "unable to parse report (%s): unicode decode of report failed - %s",
+                metadata,
+                err,
+            )
+            return None
+
+    try:
+        data = json.loads(content)
+    except json.decoder.JSONDecodeError as err:
+        logging.error(
+            "unable to parse report (%s): json decoding failed - %s", metadata, err
+        )
+        return None
+
+    try:
+        entry = Report.parse_obj(data)
+    except ValidationError as err:
+        logging.error("unable to parse report (%s): %s", metadata, err)
+        return None
+
+    return entry
+
+
+def get_report(container: str, filename: str) -> Optional[Report]:
+    metadata = "/".join([container, filename])
+    if not filename.endswith(".json"):
+        logging.error("get_report invalid extension: %s", metadata)
+        return None
+
+    blob = get_blob(container, filename)
+    if blob is None:
+        logging.error("get_report invalid blob: %s", metadata)
+        return None
+
+    return parse_report(blob, metadata=metadata)