X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tools%2Fgen_json.py;h=e3468cd530ea1be35970a5e4670c907bb5c7ff33;hb=c85e7a3b924b6aab6a5995d84956081cb1552fbc;hp=bec7ffc2b1228627e02912ad8b00cdb3cb153ee2;hpb=25495fec3f11a25cdd632a19a352a10cee105d02;p=naja.git

diff --git a/tools/gen_json.py b/tools/gen_json.py
index bec7ffc..e3468cd 100755
--- a/tools/gen_json.py
+++ b/tools/gen_json.py
@@ -6,16 +6,22 @@ import os
 
 def main(update=True):
     data = os.path.join(os.path.dirname(__file__), '..', 'data')
+
     deck_dir = os.path.join(data, 'location_decks')
+    convert_to_json(deck_dir, update)
+    puzzle_dir = os.path.join(deck_dir, 'puzzles')
+    convert_to_json(puzzle_dir, update)
+
 
-    for yaml_fn in os.listdir(deck_dir):
+def convert_to_json(directory, update=True):
+    for yaml_fn in os.listdir(directory):
         basename, extension = os.path.splitext(yaml_fn)
         if extension != '.yaml':
             continue
         json_fn = basename + '.json'
 
-        yaml_path = os.path.join(deck_dir, yaml_fn)
-        json_path = os.path.join(deck_dir, json_fn)
+        yaml_path = os.path.join(directory, yaml_fn)
+        json_path = os.path.join(directory, json_fn)
 
         if not update and os.path.exists(json_path):
             continue