first commit

rasta_exp/utils.py

@@ -0,0 +1,49 @@
+from pathlib import Path
+from typing import Any, Dict
+import hashlib
+
+
+def parse_report(report_file: Path) -> Dict[str, Any]:
+    """Parse a report file."""
+    report: dict[str, Any] = {}
+    with report_file.open("r", errors="replace") as file:
+        lines = file.readlines()
+    for line in lines:
+        k, v = map(lambda x: x.strip(), line.split(":"))
+        if k in {"time", "kernel-cpu-time", "user-cpu-time"}:
+            report[k] = float(v)
+        else:
+            report[k] = int(v)
+    # TODO: normalize the mem values to Bytes (and not KB)
+    #     Warning: page-size is already in B
+    for k in ["max-rss-mem", "avg-rss-mem", "avg-total-mem"]:
+        if k in report:
+            report[k] *= 1024  # TODO: 1000 or 1024 (ie KiB or KB?)
+
+    if "exit-status" in report:
+        report["timeout"] = report["exit-status"] == 124
+    return report
+
+
+def sha256_sum(path: Path, chunk_size=4096) -> str:
+    """Compute the sha256 of an apk."""
+    hash = hashlib.sha256()
+    with path.open("rb") as f:
+        for chunk in iter(lambda: f.read(chunk_size), b""):
+            hash.update(chunk)
+    return hash.hexdigest()
+
+
+def removesuffix(string: str, suffix: str) -> str:
+    """`str.removesuffix` for python < 3.9"""
+    if string.endswith(suffix):
+        return string[: -len(suffix)]
+    return string
+
+
+def removeprefix(string: str, prefix: str) -> str:
+    """`str.removeprefix` for python < 3.9"""
+    if string.startswith(prefix):
+        return string[len(prefix):]
+    else:
+        return string