summary refs log tree commit diff
path: root/dot_local/bin/jflatten.py
blob: bacc754e8323aa73f3bb100b7d5d64420819bd98 (plain)
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()