summaryrefslogtreecommitdiff
path: root/etc/json-format.py
diff options
context:
space:
mode:
Diffstat (limited to 'etc/json-format.py')
-rwxr-xr-xetc/json-format.py85
1 files changed, 85 insertions, 0 deletions
diff --git a/etc/json-format.py b/etc/json-format.py
new file mode 100755
index 0000000..827205a
--- /dev/null
+++ b/etc/json-format.py
@@ -0,0 +1,85 @@
+#!/usr/bin/env python3
+
+# Copyright (C) 2021-2022 Klaus Aehlig. All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met:
+#
+# 1. Redistributions of source code must retain the above copyright
+# notice, this list of conditions and the following disclaimer.
+#
+# 2. Redistributions in binary form must reproduce the above copyright
+# notice, this list of conditions and the following disclaimer in the
+# documentation and/or other materials provided with the distribution.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+# HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+import json
+import sys
+
+from typing import Any, Dict, List, cast
+
+
+def is_simple(entry: Any) -> bool:
+ if isinstance(entry, list):
+ return len(cast(List[Any], entry)) == 0
+ if isinstance(entry, dict):
+ return len(cast(Dict[str, Any], entry)) == 0
+ return True
+
+
+def is_short(entry: Any, indent: int) -> bool:
+ return len(json.dumps(entry)) + indent < 80
+
+
+def hdumps(entry: Any, *, _current_indent: int = 0) -> str:
+ if entry:
+ if is_short(entry, _current_indent):
+ return json.dumps(entry)
+ if isinstance(entry, list):
+ result: str = "[ " + hdumps(cast(Any, entry[0]),
+ _current_indent=_current_indent + 2)
+ entries: List[Any] = entry[1:] # explicit type hint
+ for x in entries:
+ result += "\n" + " " * _current_indent + ", "
+ result += hdumps(x, _current_indent=_current_indent + 2)
+ result += "\n" + " " * _current_indent + "]"
+ return result
+ if isinstance(entry, dict):
+ result: str = "{ "
+ is_first: bool = True
+ keys = cast(Dict[str, Any], entry).keys() # explicit type hint
+ for k in keys:
+ key_entry: Any = entry[k] # explicit type hint
+ if not is_first:
+ result += "\n" + " " * _current_indent + ", "
+ result += json.dumps(k) + ":"
+ if is_simple(key_entry):
+ result += " " + json.dumps(key_entry)
+ elif is_short(key_entry,
+ _current_indent + len(json.dumps(k)) + 4):
+ result += " " + json.dumps(key_entry)
+ else:
+ result += "\n" + " " * _current_indent + " "
+ result += hdumps(key_entry,
+ _current_indent=_current_indent + 2)
+ is_first = False
+ result += "\n" + " " * _current_indent + "}"
+ return result
+ return json.dumps(entry)
+
+
+if __name__ == "__main__":
+ data = json.load(sys.stdin)
+ print(hdumps(data))