V2 list image by region #594

src/cloudimagedirectory/transform/transform.py

@@ -577,10 +577,39 @@ def run(self, data: list[connection.DataEntry]) -> list:
             region = filename[8]
 
             # Build the API path that corresponds to this entry.
+            # NOTE: Add /list suffix to prevent collision with "region" folder.
             api_path = f"v2/os/{os}/provider/{provider}/version/{version}/region/list"
 
             # Increment the count for this region at this API path.
             regions[api_path][region] += 1
 
         # Convert the API path and region counts into DataEntry objects.
         return [connection.DataEntry(x, dict(y)) for x, y in regions.items()]
+
+
+class TransformerV2ListImageByRegion(Transformer):
+    """Generate a list of all images for one region."""
+
+    def run(self, data: list[connection.DataEntry]) -> list:
+        # NOTE: check that its the v2 data entries.
+        entries = [x for x in data if x.is_API("v2")]
+
+        images: defaultdict = defaultdict(list)
+
+        for e in entries:
+            entry = copy.deepcopy(e)
+            filename = entry.filename.split("/")
+            os = filename[2]
+            provider = filename[4]
+            version = filename[6]
+            region = filename[8]
+            image = filename[10]
+
+            # Build the API path that corresponds to this entry.
+            # NOTE: Add /list suffix to prevent collision with "image" folder.
+            api_path = f"v2/os/{os}/provider/{provider}/version/{version}/region/{region}/image/list"
+
+            images[api_path].append(image)
+
+        # Convert the API path and appendend image into DataEntry objects.
+        return [connection.DataEntry(x, y) for x, y in images.items()]

src/cloudimagedirectory/transformer.py

@@ -101,6 +101,7 @@ def run(origin_path: str, destination_path: str, arg_files: str, filter_until: s
             transform.TransformerV2ListProviderByOS,
             transform.TransformerV2ListVersionByProvider,
             transform.TransformerV2ListRegionByVersion,
+            transform.TransformerV2ListImageByRegion,
         ],
     )
     print("run pipeline v2")

tests/transformer/testdata/expected/v2/os/rhel/provider/google/version/7/region/global/image/list

@@ -0,0 +1 @@
+["a2f9b1c21e096445099c419aa0c0c9bc32657059"]