ec2: fetch image serial directly from aws

Lucas Moura · Lucas Moura · commit e0bf1979739f · 2022-06-20T12:06:06.000-03:00
diff --git a/pycloudlib/ec2/cloud.py b/pycloudlib/ec2/cloud.py
@@ -1,5 +1,7 @@
 # This file is part of pycloudlib. See LICENSE file for license information.
 """AWS EC2 Cloud type."""
+import re
+
 import botocore
 
 from pycloudlib.cloud import BaseCloud, ImageType
@@ -111,13 +113,11 @@ def _get_name_for_image_type(
             )
             if release in LTS_RELEASES:
                 return "{}/ubuntu-{}{}-*-server-*".format(
-                    base_location, release,
-                    "-daily" if daily else ""
+                    base_location, release, "-daily" if daily else ""
                 )
 
             return "{}/ubuntu-{}{}-*".format(
-                base_location, release,
-                "-daily" if daily else ""
+                base_location, release, "-daily" if daily else ""
             )
 
         if image_type == ImageType.PRO:
@@ -196,6 +196,36 @@ def daily_image(
         )
         return image["ImageId"]
 
+    def _find_image_serial(
+        self, image_id, image_type: ImageType = ImageType.GENERIC
+    ):
+        owner = self._get_owner(image_type=image_type)
+        filters = [
+            {
+                "Name": "image-id",
+                "Values": (image_id,),
+            }
+        ]
+
+        images = self.client.describe_images(
+            Owners=[owner],
+            Filters=filters,
+        )
+
+        if not images.get("Images"):
+            raise Exception("Could not find image: {}".format(image_id))
+
+        image_name = images["Images"][0].get("Name", "")
+        serial_regex = r"ubuntu/.*/.*/.*-(?P<serial>\d+)$"
+        serial_match = re.match(serial_regex, image_name)
+
+        if not serial_match:
+            raise Exception(
+                "Could not find image serial for image: {}".format(image_id)
+            )
+
+        return serial_match.groupdict().get("serial")
+
     def image_serial(self, image_id):
         """Find the image serial of a given EC2 image ID.
 
