tools/gen_json.py

   1 #!/usr/bin/env python
   2
   3 import json
   4 import os
   5
   6
   7 def main(update=True):
   8     data = os.path.join(os.path.dirname(__file__), '..', 'data')
   9     deck_dir = os.path.join(data, 'location_decks')
  10
  11     for yaml_fn in os.listdir(deck_dir):
  12         basename, extension = os.path.splitext(yaml_fn)
  13         if extension != '.yaml':
  14             continue
  15         json_fn = basename + '.json'
  16
  17         yaml_path = os.path.join(deck_dir, yaml_fn)
  18         json_path = os.path.join(deck_dir, json_fn)
  19
  20         if not update and os.path.exists(json_path):
  21             continue
  22
  23         import yaml
  24         with open(yaml_path) as yaml_f:
  25             obj = yaml.safe_load(yaml_f)
  26         with open(json_path, 'w') as json_f:
  27             json.dump(obj, json_f, indent=2)
  28
  29
  30 if __name__ == '__main__':
  31     main()