xref: /aosp_15_r20/external/bazelbuild-rules_python/sphinxdocs/private/inventory_builder.py (revision 60517a1edbc8ecf509223e9af94a7adec7d736b8)
1import pathlib
2import sys
3import zlib
4
5
6def main(args):
7    in_path = pathlib.Path(args.pop(0))
8    out_path = pathlib.Path(args.pop(0))
9
10    data = in_path.read_bytes()
11    offset = 0
12    for _ in range(4):
13        offset = data.index(b"\n", offset) + 1
14
15    compressed_bytes = zlib.compress(data[offset:])
16    with out_path.open(mode="bw") as fp:
17        fp.write(data[:offset])
18        fp.write(compressed_bytes)
19
20    return 0
21
22
23if __name__ == "__main__":
24    sys.exit(main(sys.argv[1:]))
25