1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
#!/usr/bin/env python3
import argparse
import json
import sys
def flatten(*path, obj):
if isinstance(obj, list):
for n, value in enumerate(obj):
yield from flatten(*path, n, obj=value)
elif isinstance(obj, dict):
for key, value in obj.items():
yield from flatten(*path, key, obj=value)
else:
yield path, obj
def path_str(*path):
return "".join(
f"[{part}]" if isinstance(part, int) else f".{part}" for part in path
)
def main():
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument("--sort-keys", "-s", action="store_true")
parser.add_argument("--json", "-j", action="store_true")
parser.add_argument("file", nargs="?")
args = parser.parse_args()
if args.file:
with open(args.file) as f:
data = json.load(f)
else:
data = json.load(sys.stdin)
flattened = flatten(obj=data)
if args.sort_keys:
flattened = sorted(flattened)
flattened_fmt = ((path_str(*path), value) for path, value in flattened)
if args.json:
print(json.dumps(dict(flattened_fmt), indent=True))
else:
print("\n".join(f"{path}:{value}" for path, value in flattened_fmt))
if __name__ == "__main__":
main()
|
