Added ensure_ascii + style=<pretty|compact|minified> args to data.json.write()

Author: adamlui

remove-json-keys/src/remove_json_keys/lib/data/json.py

@@ -27,7 +27,19 @@ def remove_keys(json_path, keys):
         files_processed_cnt += 1
     return keys_removed, keys_skipped, files_processed_cnt
 
-def write(file_path, data, encoding='utf-8'):
+def write(file_path, data, encoding='utf-8', ensure_ascii=False, style='pretty'):
     Path(file_path).parent.mkdir(parents=True, exist_ok=True)
     with open(file_path, 'w', encoding=encoding) as file:
-        json.dump(data, file, indent=2, ensure_ascii=False)
+        if style == 'pretty': # single key/val spans multi-lines
+            json.dump(data, file, indent=2, ensure_ascii=ensure_ascii)
+        elif style == 'compact': # single key/val per line
+            file.write('{\n')
+            items = list(data.items())
+            for i, (key, val) in enumerate(items):
+                line_end = ',' if i < len(items) - 1 else ''
+                inner = json.dumps(val, ensure_ascii=ensure_ascii)
+                inner = '{ ' + inner[1:-1] + ' }' # pad braces
+                file.write(f'  "{key}": {inner}{line_end}\n')
+            file.write('}\n')
+        else: # minified to single line
+            json.dump(data, file, separators=(',', ':'), ensure_ascii=ensure_ascii)

translate-messages/src/translate_messages/lib/data/json.py

@@ -7,7 +7,19 @@ def read(file_path, encoding='utf-8'):
     with open(file_path, 'r', encoding=encoding) as file:
         return json5.load(file)
 
-def write(file_path, data, encoding='utf-8'):
+def write(file_path, data, encoding='utf-8', ensure_ascii=False, style='pretty'):
     Path(file_path).parent.mkdir(parents=True, exist_ok=True)
     with open(file_path, 'w', encoding=encoding) as file:
-        json.dump(data, file, indent=2, ensure_ascii=False)
+        if style == 'pretty': # single key/val spans multi-lines
+            json.dump(data, file, indent=2, ensure_ascii=ensure_ascii)
+        elif style == 'compact': # single key/val per line
+            file.write('{\n')
+            items = list(data.items())
+            for i, (key, val) in enumerate(items):
+                line_end = ',' if i < len(items) - 1 else ''
+                inner = json.dumps(val, ensure_ascii=ensure_ascii)
+                inner = '{ ' + inner[1:-1] + ' }' # pad braces
+                file.write(f'  "{key}": {inner}{line_end}\n')
+            file.write('}\n')
+        else: # minified to single line
+            json.dump(data, file, separators=(',', ':'), ensure_ascii=ensure_ascii)