From 1a8231af3b3ebbc5932834900e516dbd3f193faa Mon Sep 17 00:00:00 2001
From: aerinon <randall.rupper@gmail.com>
Date: Thu, 20 Jun 2024 16:20:44 -0600
Subject: [PATCH] feat: minor stats compilation utility

---
 test/stats/SpoilerStats.py | 52 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 test/stats/SpoilerStats.py

diff --git a/test/stats/SpoilerStats.py b/test/stats/SpoilerStats.py
new file mode 100644
index 00000000..8e166c60
--- /dev/null
+++ b/test/stats/SpoilerStats.py
@@ -0,0 +1,52 @@
+import os
+import csv
+import collections
+
+
+def process_file(file_path):
+    items_at_locations = collections.defaultdict(list)
+    with open(file_path, 'r') as file:
+        location_section = False
+        for line in file:
+            line = line.strip()
+            if 'Locations:' in line:
+                location_section = True
+            elif 'Shops:' in line:
+                location_section = False
+            elif location_section and line:
+                location, item = line.split(":")
+                if 'Bottle' in item:
+                    item = 'Bottle'
+                items_at_locations[location].append(item.strip())
+    return items_at_locations
+
+
+def process_directory(directory_path):
+    all_items_at_locations = collections.defaultdict(list)
+    for file_name in os.listdir(directory_path):
+        file_path = os.path.join(directory_path, file_name)
+        items_at_locations = process_file(file_path)
+        for location, items in items_at_locations.items():
+            all_items_at_locations[location].extend(items)
+    return all_items_at_locations
+
+
+def write_to_csv(items_at_locations, csv_file_path):
+    # Get a list of all unique items across all locations
+    all_items = set(item for items in items_at_locations.values() for item in items)
+
+    with open(csv_file_path, 'w', newline='') as file:
+        writer = csv.writer(file)
+        # Write the header row
+        writer.writerow(['Location'] + sorted(list(all_items)))
+        # Write a row for each location
+        for location in sorted(items_at_locations.keys()):
+            item_counts = collections.Counter(items_at_locations[location])
+            # Write a column for each item
+            row = [location] + [item_counts.get(item, 0) for item in sorted(all_items)]
+            writer.writerow(row)
+
+
+if __name__ == '__main__':
+    items_at_locations = process_directory(os.path.join('..', '..', 'analysis2'))
+    write_to_csv(items_at_locations, os.path.join('..', '..', 'analysis2', 'output.csv'))
\ No newline at end of file