allow to terminate instances

Author: sanderegg

scripts/maintenance/computational-clusters/autoscaled_monitor/cli.py

@@ -209,5 +209,19 @@ def check_database_connection() -> None:
     asyncio.run(api.check_database_connection(state))
 
 
+@app.command()
+def terminate_instances(
+    user_id: Annotated[int, typer.Option(help="the user ID")],
+    wallet_id: Annotated[
+        Optional[int | None],  # noqa: UP007 # typer does not understand | syntax
+        typer.Option(help="the wallet ID"),
+    ] = None,
+    *,
+    force: Annotated[bool, typer.Option(help="will not ask for confirmation")] = False,
+) -> None:
+    """this will terminate the instance(s) used for the given user/wallet"""
+    asyncio.run(api.terminate_instances(state, user_id, wallet_id, force=force))
+
+
 if __name__ == "__main__":
     app()

scripts/maintenance/computational-clusters/autoscaled_monitor/core.py

@@ -728,7 +728,7 @@ async def cancel_jobs(  # noqa: C901, PLR0912
 
 
 async def trigger_cluster_termination(
-    state: AppState, user_id: int, wallet_id: int, *, force: bool
+    state: AppState, user_id: int, wallet_id: int | None, *, force: bool
 ) -> None:
     assert state.ec2_resource_clusters_keeper
     computational_instances = await ec2.list_computational_instances_from_ec2(
@@ -746,6 +746,7 @@ async def trigger_cluster_termination(
         computational_clusters,
         state.environment,
         state.ec2_resource_clusters_keeper.meta.client.meta.region_name,
+        output=None,
     )
     if (force is True) or typer.confirm(
         "Are you sure you want to trigger termination of that cluster?"
@@ -775,3 +776,31 @@ async def trigger_cluster_termination(
 
 async def check_database_connection(state: AppState) -> None:
     await db.check_db_connection(state)
+
+
+async def terminate_instances(
+    state: AppState, user_id: int | None, wallet_id: int | None, *, force: bool
+) -> None:
+    assert state.ec2_resource_autoscaling
+    dynamic_instances = await ec2.list_dynamic_instances_from_ec2(
+        state, user_id, wallet_id
+    )
+    dynamic_autoscaled_instances = await _parse_dynamic_instances(
+        state, dynamic_instances, state.ssh_key_path, user_id, wallet_id
+    )
+
+    if not dynamic_autoscaled_instances:
+        rich.print("no instances found")
+        raise typer.Exit(0)
+
+    for instance in dynamic_autoscaled_instances:
+        rich.print(
+            f"terminating instance {instance.ec2_instance.instance_id} with name {utils.get_instance_name(instance)}"
+        )
+        if force is True or typer.confirm(
+            f"Are you sure you want to terminate instance {instance.ec2_instance.instance_id}?"
+        ):
+            instance.ec2_instance.terminate()
+            rich.print(f"terminated instance {instance.ec2_instance.instance_id}")
+        else:
+            rich.print("not terminating anything")