40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import json
|
||
|
|
from pathlib import Path
|
||
|
|
from os import getenv
|
||
|
|
from sys import argv
|
||
|
|
|
||
|
|
if len(argv) != 2:
|
||
|
|
print("JSON info files script requires ouput file as argument")
|
||
|
|
exit(1)
|
||
|
|
|
||
|
|
output_path = Path(argv[1])
|
||
|
|
|
||
|
|
assert getenv("WORK_DIR"), "$WORK_DIR required"
|
||
|
|
|
||
|
|
work_dir = Path(getenv("WORK_DIR"))
|
||
|
|
|
||
|
|
assert work_dir.is_dir(), "$WORK_DIR not a directory"
|
||
|
|
|
||
|
|
output = {}
|
||
|
|
|
||
|
|
for json_file in work_dir.glob("*.json"):
|
||
|
|
image_info = json.loads(json_file.read_text())
|
||
|
|
if not output:
|
||
|
|
output.update(image_info)
|
||
|
|
else:
|
||
|
|
# get first (and only) profile in json file
|
||
|
|
device_id = next(iter(image_info["profiles"].keys()))
|
||
|
|
if device_id not in output["profiles"]:
|
||
|
|
output["profiles"].update(image_info["profiles"])
|
||
|
|
else:
|
||
|
|
output["profiles"][device_id]["images"].append(
|
||
|
|
image_info["profiles"][device_id]["images"][0]
|
||
|
|
)
|
||
|
|
|
||
|
|
if output:
|
||
|
|
output_path.write_text(json.dumps(output, sort_keys=True, separators=(",", ":")))
|
||
|
|
else:
|
||
|
|
print("JSON info file script could not find any JSON files for target")
|