byml_to_yml: Add option to dump to JSON

Useful for web things.
zeldamods · Jan 21, 2019 · 21db57c · 21db57c
1 parent 6e1b5aa
commit 21db57c
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/byml/byml_to_yml.py b/byml/byml_to_yml.py
@@ -1,4 +1,5 @@
 import argparse
+import json
 import os
 import sys
 import yaml
@@ -9,6 +10,7 @@
 
 def main() -> None:
     parser = argparse.ArgumentParser(description='Converts a BYML file to YAML.')
+    parser.add_argument('-j', '--to-json', action='store_true', help='Convert to JSON (warning: one-way conversion; does not preserve type information)')
     parser.add_argument('byml', help='Path to a BYML file', nargs='?', default='-')
     parser.add_argument('yml', help='Path to destination YAML file', nargs='?', default='-')
     args = parser.parse_args()
@@ -30,7 +32,10 @@ def main() -> None:
             sys.exit(1)
         output = sys.stdout if args.yml == '-' else open(args.yml, 'w', encoding='utf-8')
         with output:
-            yaml.dump(root, output, Dumper=dumper, allow_unicode=True, encoding='utf-8')
+            if args.to_json:
+                json.dump(root, output, ensure_ascii=False)
+            else:
+                yaml.dump(root, output, Dumper=dumper, allow_unicode=True, encoding='utf-8')
 
 if __name__ == '__main__':
     main()