goated

convert_recipes.py

@@ -14,18 +14,19 @@ def generic_constructor(loader, tag_suffix, node):
 
 yaml.SafeLoader.add_multi_constructor('!', generic_constructor)
 
-def parse_recipes(directory):
+def parse_recipes(directories):
 
     recipes = []
     
-    for filename in os.listdir(directory):
-        if filename.endswith(".yml"):
-            with open(os.path.join(directory, filename), 'r') as f:
-                data = yaml.safe_load(f)
-                if data:
-                    for item in data:
-                        if item.get('type') == 'reaction':
-                            recipes.append(item)
+    for directory in directories:
+        for filename in os.listdir(directory):
+            if filename.endswith(".yml"):
+                with open(os.path.join(directory, filename), 'r') as f:
+                    data = yaml.safe_load(f)
+                    if data:
+                        for item in data:
+                            if item.get('type') == 'reaction':
+                                recipes.append(item)
     return recipes
 
 
@@ -67,7 +68,7 @@ def process_recipes(raw_recipes):
     return processed
 
 if __name__ == "__main__":
-    raw = parse_recipes('./mixingrecipes')
+    raw = parse_recipes(['./mixingrecipes', './Prototypes/_Goobstation/Recipes/Reactions/'])
     processed = process_recipes(raw)
     
     with open('recipes.json', 'w') as f: