Pre-generate JSON, so we don't need to require pyyaml

[naja.git] / tools / gen_json.py

diff --git a/tools/gen_json.py b/tools/gen_json.py
new file mode 100755 (executable)
index 0000000..6ae4b51
--- /dev/null
+++ b/tools/gen_json.py
@@ -0,0 +1,29 @@
+#!/usr/bin/env python
+
+import json
+import os
+
+import yaml
+
+
+def main():
+    data = os.path.join(os.path.dirname(__file__), '..', 'data')
+    deck_dir = os.path.join(data, 'location_decks')
+
+    for yaml_fn in os.listdir(deck_dir):
+        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)
+
+        with open(yaml_path) as yaml_f:
+            obj = yaml.safe_load(yaml_f)
+        with open(json_path, 'w') as json_f:
+            json.dump(obj, json_f, indent=2)
+
+
+if __name__ == '__main__':
+    main()