basic noita parsing

polylan_submitter/noita/services/objectives.py

@@ -0,0 +1,40 @@
+from noita.models import LogfileSubmission, Objectiv
+from noita.services.decode import parse_log, resolve
+
+
+from collections import Counter
+
+
+def parse_objectives_from_logfile(logfile: LogfileSubmission) -> Counter:
+    """Parse a log file, and output a count for each ID."""
+    file_data = logfile.file.read().decode()
+
+    ids = []
+    for entry in parse_log(file_data):
+        idx, _seed = resolve(entry["hash"], entry["ts"])
+
+        if idx:
+            ids.append(idx)
+
+    return Counter(ids)
+
+
+def parse_objectives_and_store(logfile: LogfileSubmission) -> None:
+    """Parse a logfile and store output."""
+
+    if not logfile.user:
+        return
+
+    counter = parse_objectives_from_logfile(logfile)
+
+    for idx, count in counter.items():
+        if idx in {"-", "DEBUG"}:
+            continue
+
+        obj, created = Objectiv.objects.get_or_create(
+            objectiv_id=idx,
+            user=logfile.user,
+        )
+
+        obj.count += count
+        obj.save(update_fields=["count"])