[docker] Catch errors. Fail graciously

ManuelSchneid3r · ManuelSchneid3r · commit 0911582b526f · 2023-08-09T07:17:35.000+02:00
diff --git a/docker/__init__.py b/docker/__init__.py
@@ -8,7 +8,7 @@
 from albert import *
 
 md_iid = "2.0"
-md_version = "1.5"
+md_version = "1.6"
 md_name = "Docker"
 md_description = "Manage docker images and containers"
 md_license = "BSD-3"
@@ -27,60 +27,61 @@ def __init__(self):
                                     defaultTrigger='d ',
                                     synopsis='<image tag|container name>')
         PluginInstance.__init__(self, extensions=[self])
-
         self.icon_urls_running = [f"file:{Path(__file__).parent}/running.png"]
         self.icon_urls_stopped = [f"file:{Path(__file__).parent}/stopped.png"]
-        self.client = docker.from_env()
-        if not self.client:
-            self.client = docker.DockerClient(base_url='unix://var/run/docker.sock')
-            if not self.client:
-                raise "Failed to initialize client."
+        self.client = None
 
     def handleGlobalQuery(self, query):
         rank_items = []
+        try:
+            if not self.client:
+                self.client = docker.from_env()
 
-        for container in self.client.containers.list(all=True):
-            if query.string in container.name:
-                # Create dynamic actions
-                if container.status == 'running':
-                    actions = [Action("stop", "Stop container", lambda c=container: c.stop()),
-                               Action("restart", "Restart container", lambda c=container: c.restart())]
-                else:
-                    actions = [Action("start", "Start container", lambda c=container: c.start())]
-                actions.extend([
-                    Action("logs", "Logs",
-                           lambda c=container.id: runTerminal("docker logs -f %s" % c, close_on_exit=False)),
-                    Action("remove", "Remove (forced, with volumes)",
-                           lambda c=container: c.remove(v=True, force=True)),
-                    Action("copy-id", "Copy id to clipboard",
-                           lambda id=container.id: setClipboardText(id))
-                ])
-
-                rank_items.append(RankItem(
-                    item=StandardItem(
-                        id=container.id,
-                        text="%s (%s)" % (container.name, ", ".join(container.image.tags)),
-                        subtext="Container: %s" % container.id,
-                        iconUrls=self.icon_urls_running if container.status == 'running' else self.icon_urls_stopped,
-                        actions=actions
-                    ),
-                    score=len(query.string)/len(container.name)
-                ))
+            for container in self.client.containers.list(all=True):
+                if query.string in container.name:
+                    # Create dynamic actions
+                    if container.status == 'running':
+                        actions = [Action("stop", "Stop container", lambda c=container: c.stop()),
+                                   Action("restart", "Restart container", lambda c=container: c.restart())]
+                    else:
+                        actions = [Action("start", "Start container", lambda c=container: c.start())]
+                    actions.extend([
+                        Action("logs", "Logs",
+                               lambda c=container.id: runTerminal("docker logs -f %s" % c, close_on_exit=False)),
+                        Action("remove", "Remove (forced, with volumes)",
+                               lambda c=container: c.remove(v=True, force=True)),
+                        Action("copy-id", "Copy id to clipboard",
+                               lambda id=container.id: setClipboardText(id))
+                    ])
 
-        for image in reversed(self.client.images.list()):
-            for tag in sorted(image.tags, key=len):  # order by resulting score
-                if query.string in tag:
                     rank_items.append(RankItem(
                         item=StandardItem(
-                            id=image.short_id,
-                            text=", ".join(image.tags),
-                            subtext="Image: %s" % image.id,
-                            iconUrls=self.icon_urls_stopped,
-                            actions=[Action("run", "Run with command: %s" % query.string,
-                                            lambda i=image, s=query.string: client.containers.run(i, s)),
-                                     Action("rmi", "Remove image", lambda i=image: i.remove())]
+                            id=container.id,
+                            text="%s (%s)" % (container.name, ", ".join(container.image.tags)),
+                            subtext="Container: %s" % container.id,
+                            iconUrls=self.icon_urls_running if container.status == 'running' else self.icon_urls_stopped,
+                            actions=actions
                         ),
-                        score=len(query.string)/len(tag)
+                        score=len(query.string)/len(container.name)
                     ))
 
+            for image in reversed(self.client.images.list()):
+                for tag in sorted(image.tags, key=len):  # order by resulting score
+                    if query.string in tag:
+                        rank_items.append(RankItem(
+                            item=StandardItem(
+                                id=image.short_id,
+                                text=", ".join(image.tags),
+                                subtext="Image: %s" % image.id,
+                                iconUrls=self.icon_urls_stopped,
+                                actions=[Action("run", "Run with command: %s" % query.string,
+                                                lambda i=image, s=query.string: client.containers.run(i, s)),
+                                         Action("rmi", "Remove image", lambda i=image: i.remove())]
+                            ),
+                            score=len(query.string)/len(tag)
+                        ))
+        except Exception as e:
+            warning(e)
+            self.client = None
+
         return rank_items
