update

Author: reliveyy

images/utils/launcher/node/image.py

@@ -17,7 +17,8 @@
 from docker import DockerClient
 from docker.errors import ImageNotFound
 
-from launcher.errors import FatalError
+from launcher.errors import FatalError, NoWaiting
+from launcher.utils import yes_or_no
 
 if TYPE_CHECKING:
     from .base import Node
@@ -352,11 +353,17 @@ def check_for_updates(self) -> Dict[Image, str]:
         executor = self.config.executor
 
         futs = {executor.submit(img.get_update_action): img for img in images}
-        done, not_done = wait(futs, 30)
-        for f in not_done:
-            logger.debug("Failed future %s", f)
-        if len(not_done) > 0:
-            raise RuntimeError("Failed to check for image updates")
+
+        while True:
+            done, not_done = wait(futs, 30)
+            if len(not_done) > 0:
+                names = ", ".join([futs[f].name for f in not_done])
+                print("Still waiting for update checking results of image(s): %s" % names)
+                reply = yes_or_no("Would you like to keep waiting?")
+                if reply == "no":
+                    raise NoWaiting
+            else:
+                break
 
         result = {}
         for f in done:

images/utils/launcher/node/xud.py

@@ -202,7 +202,7 @@ def _ensure_dependencies_ready(self, stop: threading.Event):
             names_str = ", ".join(names)
             reply = yes_or_no("Keep waiting for {} to be ready?".format(names_str))
             if reply == "no":
-                raise Exception("Give up waiting for {}".format(names_str))
+                raise NoWaiting
 
     def _save_seed(self, output):
         s = output