status: check connections (#557)

Author: pinheadmz

src/warnet/network.py

@@ -63,7 +63,7 @@ def copy_scenario_defaults(directory: Path):
     )
 
 
-def _connected():
+def _connected(end="\n"):
     tanks = get_mission("tank")
     for tank in tanks:
         # Get actual
@@ -73,11 +73,11 @@ def _connected():
             if peer["connection_type"] == "manual":
                 actual += 1
         expected = int(tank.metadata.annotations["init_peers"])
-        print(f"Tank {tank.metadata.name} peers expected: {expected}, actual: {actual}")
+        print(f"Tank {tank.metadata.name} peers expected: {expected}, actual: {actual}", end=end)
         # Even if more edges are specified, bitcoind only allows
         # 8 manual outbound connections
         if min(8, expected) > actual:
-            print("Network not connected")
+            print("\nNetwork not connected")
             return False
-    print("Network connected")
+    print("Network connected                                                           ")
     return True

src/warnet/status.py

@@ -5,6 +5,7 @@
 from rich.text import Text
 
 from .k8s import get_mission
+from .network import _connected
 
 
 @click.command()
@@ -53,6 +54,7 @@ def status():
     summary.append(f"\nTotal Tanks: {len(tanks)}", style="bold cyan")
     summary.append(f" | Active Scenarios: {len(scenarios)}", style="bold green")
     console.print(summary)
+    _connected(end="\r")
 
 
 def _get_tank_status():