Kraken: allow installing extension if pull fails but image is available offline

Author: Williangalvani

core/services/kraken/extension/extension.py

@@ -177,6 +177,16 @@ def _prepare_docker_auth(self) -> Optional[str]:
         docker_auth = f"{self.source.auth.username}:{self.source.auth.password}"
         return base64.b64encode(docker_auth.encode("utf-8")).decode("utf-8")
 
+    async def _image_is_available_locally(self) -> bool:
+        """Check if the Docker image is available locally."""
+        try:
+            image_ref = f"{self.source.docker}:{self.tag}" + (f"@{self.digest}" if self.digest else "")
+            async with DockerCtx() as client:
+                await client.images.inspect(image_ref)
+                return True
+        except Exception:
+            return False
+
     async def _pull_docker_image(self, docker_auth: Optional[str]) -> AsyncGenerator[bytes, None]:
         """Pull Docker image and yield progress updates."""
         tag = f"{self.source.docker}:{self.tag}" + (f"@{self.digest}" if self.digest else "")
@@ -214,12 +224,14 @@ async def install(self, clear_remaining_tags: bool = True, atomic: bool = False)
             # In case of some external installs kraken shouldn't try to install it again so we remove from settings
             if atomic:
                 should_raise = False
-                if not running_ext or self.unique_entry != running_ext.unique_entry:
-                    should_raise = True
-                    await self.uninstall()
-
-                if running_ext:
-                    await running_ext.enable()
+                if await self._image_is_available_locally():
+                    logger.info(f"Pull failed but image {self.identifier}:{self.tag} is already available locally")
+                else:
+                    if not running_ext or self.unique_entry != running_ext.unique_entry:
+                        should_raise = True
+                        await self.uninstall()
+                    if running_ext:
+                        await running_ext.enable()
 
                 if should_raise:
                     raise ExtensionPullFailed(f"Failed to pull extension {self.identifier}:{self.tag}") from error