From 5c0eccd12f479e9c8097b5307fee87d5394e8d4c Mon Sep 17 00:00:00 2001
From: Pascal Vizeli <pascal.vizeli@syshack.ch>
Date: Tue, 16 May 2017 00:07:43 +0200
Subject: [PATCH] Bugfix attach container/image (#52)

---
 hassio/dock/addon.py | 16 ++++++++++++++--
 1 file changed, 14 insertions(+), 2 deletions(-)

diff --git a/hassio/dock/addon.py b/hassio/dock/addon.py
index d98fdbab9..679b594bf 100644
--- a/hassio/dock/addon.py
+++ b/hassio/dock/addon.py
@@ -97,15 +97,27 @@ class DockerAddon(DockerBase):
 
         Need run inside executor.
         """
+        # read container
         try:
             self.container = self.dock.containers.get(self.docker_name)
             self.process_metadata()
 
-            _LOGGER.info(
-                "Attach to image %s with version %s", self.image, self.version)
+            _LOGGER.info("Attach to container %s with version %s",
+                         self.image, self.version)
+            return
         except (docker.errors.DockerException, KeyError):
             pass
 
+        # read image
+        try:
+            image = self.dock.images.get(self.image)
+            self.process_metadata(metadata=image.attrs)
+
+            _LOGGER.info("Attach to image %s with version %s",
+                         self.image, self.version)
+        except (docker.errors.DockerException, KeyError):
+            _LOGGER.error("No container/image found for %s", self.image)
+
     def _install(self, tag):
         """Pull docker image or build it.