Merge pull request #98 from ManualForArchipelago/csv-helper

Helper function for loading a csv file.  People might find this useful for hooks

Author: silasary

src/Helpers.py

@@ -1,8 +1,9 @@
+import csv
 import os
 import pkgutil
 import json
 
-from BaseClasses import MultiWorld, Item, Location
+from BaseClasses import MultiWorld, Item
 from typing import Optional, List, TYPE_CHECKING
 from worlds.AutoWorld import World
 from .hooks.Helpers import before_is_category_enabled, before_is_item_enabled, before_is_location_enabled
@@ -24,6 +25,17 @@ def load_data_file(*args) -> dict:
 
     return filedata
 
+def load_data_csv(*args) -> list[dict]:
+    fname = os.path.join("data", *args)
+
+    try:
+        lines = pkgutil.get_data(__name__, fname).decode().splitlines()
+    except:
+        lines = []
+    filedata = list(csv.DictReader(lines))
+
+    return filedata
+
 def is_option_enabled(multiworld: MultiWorld, player: int, name: str) -> bool:
     return get_option_value(multiworld, player, name) > 0