1 files changed, 15 insertions, 7 deletions
diff --git a/config.py b/config.py
index bb18474..5a31013 100644
--- a/config.py
+++ b/config.py
@@ -1,18 +1,26 @@
 import os
-
+import docker
+from docker.errors import NotFound
+import traceback
 
 def autodiscover():
     container_names = ["rmq", "pdrmq", "snekbox_pdrmq_1"]
-    try:
-        import docker
-        client = docker.from_env()
-        for name in container_names:
+
+    client = docker.from_env()
+    for name in container_names:
+        try:
             container = client.containers.get(name)
             if container.status == "running":
                 host = list(container.attrs.get('NetworkSettings').get('Networks').values())[0]['IPAddress']
                 return host
-    except Exception:
-        return '127.0.0.1'
+        except NotFound:
+            #print(traceback.format_exc())
+            print("retrying...")
+            continue
+        except Exception:
+            print(traceback.format_exc())
+            return '127.0.0.1'
+
 
 
 USERNAME = os.environ.get('RMQ_USERNAME', 'guest')