X-Git-Url: https://git.ctpug.org.za/?a=blobdiff_plain;f=tools%2Fgen_json.py;h=e3468cd530ea1be35970a5e4670c907bb5c7ff33;hb=b07be6d16b82960992e90989d2fc773490785889;hp=6ae4b514c3302f29d4dddedc728153b355954193;hpb=44b9016415981987c6c5fe9e637a860b67a786a7;p=naja.git

diff --git a/tools/gen_json.py b/tools/gen_json.py
index 6ae4b51..e3468cd 100755
--- a/tools/gen_json.py
+++ b/tools/gen_json.py
@@ -3,22 +3,30 @@
 import json
 import os
 
-import yaml
 
-
-def main():
+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
 
+        import yaml
         with open(yaml_path) as yaml_f:
             obj = yaml.safe_load(yaml_f)
         with open(json_path, 'w') as json_f: