scripts/json_add_image_info.py

   1 #!/usr/bin/env python3
   2
   3 from os import getenv
   4 from pathlib import Path
   5 from sys import argv
   6 import hashlib
   7 import json
   8
   9 if len(argv) != 2:
  10     print("ERROR: JSON info script requires output arg")
  11     exit(1)
  12
  13 json_path = Path(argv[1])
  14 bin_dir = Path(getenv("BIN_DIR"))
  15 image_file = bin_dir / getenv("IMAGE_NAME")
  16
  17 if not image_file.is_file():
  18     print("Skip JSON creation for non existing image ", image_file)
  19     exit(0)
  20
  21
  22 def get_titles():
  23     titles = []
  24     for prefix in ["", "ALT0_", "ALT1_", "ALT2_"]:
  25         title = {}
  26         for var in ["vendor", "model", "variant"]:
  27             if getenv("DEVICE_{}{}".format(prefix, var.upper())):
  28                 title[var] = getenv("DEVICE_{}{}".format(prefix, var.upper()))
  29
  30         if title:
  31             titles.append(title)
  32
  33     if not titles:
  34         titles.append({"title": getenv("DEVICE_TITLE")})
  35
  36     return titles
  37
  38
  39 device_id = getenv("DEVICE_ID")
  40 image_hash = hashlib.sha256(image_file.read_bytes()).hexdigest()
  41
  42 image_info = {
  43     "metadata_version": 1,
  44     "target": "{}/{}".format(getenv("TARGET"), getenv("SUBTARGET")),
  45     "version_code": getenv("VERSION_CODE"),
  46     "version_number": getenv("VERSION_NUMBER"),
  47     "profiles": {
  48         device_id: {
  49             "image_prefix": getenv("IMAGE_PREFIX"),
  50             "images": [
  51                 {
  52                     "type": getenv("IMAGE_TYPE"),
  53                     "name": getenv("IMAGE_NAME"),
  54                     "sha256": image_hash,
  55                 }
  56             ],
  57             "supported_devices": getenv("SUPPORTED_DEVICES").split(),
  58             "titles": get_titles(),
  59         }
  60     },
  61 }
  62
  63 json_path.write_text(json.dumps(image_info, separators=(",", ":")))