iteration of spawn

Author: victormlg

src/cfengine_cli/up.py

@@ -9,9 +9,10 @@
 from cf_remote import commands
 from cf_remote import utils
 from cf_remote import paths
+from cf_remote import aramid
 
 import itertools
-
+import time
 
 # Forces pydantic to throw validation error if config contains unknown keys
 class NoExtra(BaseModel, extra="forbid"):
@@ -233,6 +234,10 @@ def validate_config(content):
         raise UserError("\n".join(msgs))
     return state
 
+def three_dots():
+    for _ in range(3):
+        print(".", end="", flush=True)
+        time.sleep(0.3) 
 
 def spawn_from_config(state):
 
@@ -246,12 +251,14 @@ def spawn_from_config(state):
     for old, new in itertools.zip_longest(data.keys(), state.keys()):
 
         if old is not None:
-            if old not in state:
+            if old[1:] not in state:
                 destroy.append(old)
+                print("<<< {}".format(old[1:]))
 
         if new is not None:
             if "@{}".format(new) not in data:
                 spawn.append(new)
+                print(">>> {}".format(new))
 
     for k in spawn:
         config = state[k].config
@@ -297,3 +304,33 @@ def spawn_from_config(state):
 
     for k in destroy:
         commands.destroy(k)
+
+    data = utils.read_json(paths.CLOUD_STATE_FPATH)
+    if data is None:
+        data = {}
+
+    if spawn:
+        print("Waiting for VMs to be accessible", end="")
+        three_dots()
+        print()
+
+    for k,v in data.items():
+
+        if k[1:] not in spawn:
+            continue
+        
+        ips = [h_info["public_ips"][0] for h, h_info in v.items() if h != "meta"]
+        hosts = [aramid.Host(host_name=ip, user="admin") for ip in ips]
+
+        while True:
+            try:
+                aramid.execute(hosts, "echo", ignore_failed=False, echo=False, echo_cmd=False)
+                break
+            except aramid.ExecutionError:
+                three_dots()                
+                print("\r\033[2K", end="", flush=True)
+
+        commands.info(ips)
+    
+    print("done")
+