add image info export

Author: antheas

.github/workflows/online_test_deck.yml

@@ -58,6 +58,8 @@ jobs:
             org.universal-blue.pkg.hhd=<version:hhd>
             org.universal-blue.pkg.adjustor=<version:adjustor>
             org.universal-blue.pkg.hhd-ui=<version:hhd-ui>
+
+            org.universal-blue.info=<imginfo>
           description: >
             Bazzite Deck <pretty>
             [Kernel: <relver:kernel>,

src/rechunk/model.py

@@ -1,5 +1,5 @@
-from typing import NamedTuple
 from datetime import datetime
+from typing import Literal, NamedTuple, Sequence, TypedDict
 
 
 class File(NamedTuple):
@@ -16,11 +16,33 @@ class Package(NamedTuple):
     version: str = ""
     release: str = ""
 
+
 class MetaPackage(NamedTuple):
     index: int
     name: str
     nevra: tuple[str, ...]
     size: int
     updates: tuple[datetime, ...] = tuple()
     dedicated: bool = False
-    meta: bool = False
+    meta: bool = False
+
+
+class ExportInfoV1(TypedDict):
+    version: Literal[1]
+
+    uniq: str
+    packages: dict[str, str]
+
+
+def export_v1(
+    uniq: str | None,
+    base_pkg: Sequence[Package] | None,
+) -> str:
+    import json
+
+    packages = {}
+    if base_pkg:
+        for p in base_pkg:
+            packages[p.name] = f"{p.version}-{p.release}"
+
+    return json.dumps(ExportInfoV1(version=1, uniq=uniq or "", packages=packages))

src/rechunk/utils.py

@@ -8,7 +8,7 @@
 import numpy as np
 from tqdm.auto import tqdm as tqdm_orig
 
-from .model import MetaPackage, Package
+from .model import MetaPackage, Package, export_v1
 
 logger = logging.getLogger(__name__)
 
@@ -180,6 +180,9 @@ def get_labels(
             with open(version_fn, "w") as f:
                 f.write(new_version)
 
+    imginfo = None
+    blacklist = dict()
+
     if labels:
         for line in labels:
             if not "=" in line:
@@ -198,6 +201,11 @@ def get_labels(
                 value = value.replace("<pretty>", pretty)
             if "<previous>" in value and prev_version:
                 value = value.replace("<previous>", prev_version)
+            if "<imginfo>" in value:
+                if not imginfo:
+                    imginfo = export_v1(uniq=new_version, base_pkg=base_pkg)
+                value = value.replace("<imginfo>", imginfo)
+                blacklist[key] = "> IMGINFO V1 INSERTED"
 
             if base_pkg:
                 for pkg in base_pkg:
@@ -222,6 +230,8 @@ def get_labels(
     if new_labels:
         log = "Writing labels:\n"
         for key, value in new_labels.items():
+            if key in blacklist:
+                value = blacklist[key]
             log += f" - {key} =\n'{value}'\n"
         logger.info(log)
     else: