refactor: decoupled strike run status literal from cli

dreadnode · Nov 19, 2024 · abc521a · abc521a
1 parent 73f489b
commit abc521a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/dreadnode_cli/agent/cli.py b/dreadnode_cli/agent/cli.py
@@ -203,7 +203,7 @@ def deploy(
         return
 
     with Live(formatted, refresh_per_second=2) as live:
-        while run.status not in ["completed", "failed", "timeout"]:
+        while run.is_running():
             time.sleep(1)
             run = client.get_strike_run(run.id)
             live.update(format_run(run))

diff --git a/dreadnode_cli/api.py b/dreadnode_cli/api.py
@@ -353,6 +353,9 @@ class _StrikeRun(BaseModel):
         start: datetime | None
         end: datetime | None
 
+        def is_running(self) -> bool:
+            return self.status not in ["completed", "failed", "timeout", "terminated"]
+
     class StrikeRunSummaryResponse(_StrikeRun):
         zones: list["Client.StrikeRunZoneSummary"]