Delete project command

CHANGELOG.D/581.feature

@@ -0,0 +1 @@
+Added command `neuro-flow delete-project <project-id>` that allows complete removal of project.

neuro_flow/cli/click_types.py

@@ -373,3 +373,39 @@ def extract_attempt_no(args: Sequence[str]) -> int:
     args_to_bake_id=operator.itemgetter(-1),
     args_to_attempt=extract_attempt_no,
 )
+
+
+class ProjectType(AsyncType[str]):
+    name = "project"
+
+    async def async_convert(
+        self,
+        root: Root,
+        value: str,
+        param: Optional[click.Parameter],
+        ctx: Optional[click.Context],
+    ) -> str:
+        return value
+
+    async def async_complete(
+        self,
+        root: Root,
+        ctx: click.Context,
+        args: Sequence[str],
+        incomplete: str,
+    ) -> List[Tuple[str, Optional[str]]]:
+        variants = []
+        async with AsyncExitStack() as stack:
+            client = await stack.enter_async_context(neuro_sdk.get())
+            storage: Storage = await stack.enter_async_context(ApiStorage(client))
+            try:
+                async for project in storage.list_projects():
+                    variants.append(project.yaml_id)
+            except ValueError:
+                pass
+        return [
+            (yaml_id, None) for yaml_id in variants if yaml_id.startswith(incomplete)
+        ]
+
+
+PROJECT = ProjectType()

neuro_flow/cli/live.py

@@ -1,9 +1,14 @@
 import click
 import neuro_sdk
 import sys
-from typing import List, Optional, Tuple
+from typing import List, Optional, Sequence, Tuple
 
-from neuro_flow.cli.click_types import LIVE_JOB, LIVE_JOB_OR_ALL, SUFFIX_AFTER_LIVE_JOB
+from neuro_flow.cli.click_types import (
+    LIVE_JOB,
+    LIVE_JOB_OR_ALL,
+    PROJECT,
+    SUFFIX_AFTER_LIVE_JOB,
+)
 from neuro_flow.cli.utils import argument, option, wrap_async
 from neuro_flow.live_runner import LiveRunner
 
@@ -150,3 +155,20 @@ async def kill(
                     "Suffix is not supported when killing ALL jobs"
                 )
             await runner.kill_all()
+
+
+@click.command()
+@argument("project_ids", type=PROJECT, nargs=-1, required=True)
+@wrap_async()
+async def delete_project(
+    root: Root,
+    project_ids: Sequence[str],
+) -> None:
+    """Completely remove project with all related entities"""
+    async with AsyncExitStack() as stack:
+        client = await stack.enter_async_context(neuro_sdk.get())
+        storage: Storage = await stack.enter_async_context(ApiStorage(client))
+        for project_id in project_ids:
+            await storage.project(yaml_id=project_id).delete()
+            if root.verbosity >= 0:
+                root.console.print(f"Project '{project_id}' was successfully removed.")

neuro_flow/cli/main.py

@@ -163,6 +163,7 @@ def cli(
 cli.add_command(live.logs)
 cli.add_command(live.status)
 cli.add_command(live.kill)
+cli.add_command(live.delete_project)
 
 # Batch commands
 cli.add_command(batch.bake)

neuro_flow/storage/api.py

@@ -363,9 +363,14 @@ async def create_project(
         )
 
     def list_projects(
-        self, name: str, cluster: Optional[str] = None
+        self, name: Optional[str] = None, cluster: Optional[str] = None
     ) -> AsyncIterator[Project]:
-        return self._raw_client.list("flow/projects", _parse_project_payload)
+        params = []
+        if name is not None:
+            params += [("name", name)]
+        if cluster is not None:
+            params += [("cluster", cluster)]
+        return self._raw_client.list("flow/projects", _parse_project_payload, params)
 
     def bake(self, *, id: str) -> "BakeStorage":
         return ApiBakeStorage(self._raw_client, id=id)

neuro_flow/storage/base.py

@@ -198,7 +198,7 @@ async def create_project(
 
     @abc.abstractmethod
     def list_projects(
-        self, name: str, cluster: Optional[str] = None
+        self, name: Optional[str] = None, cluster: Optional[str] = None
     ) -> AsyncIterator[Project]:
         pass
 

neuro_flow/storage/in_memory.py

@@ -107,10 +107,10 @@ async def create_project(
         return project
 
     async def list_projects(
-        self, name: str, cluster: Optional[str] = None
+        self, name: Optional[str] = None, cluster: Optional[str] = None
     ) -> AsyncIterator[Project]:
         for project in self._db.projects.values():
-            if project.yaml_id != name:
+            if name and project.yaml_id != name:
                 continue
             if cluster and project.cluster != cluster:
                 continue

reference/cli.md

@@ -35,6 +35,7 @@ neuro-flow [OPTIONS] COMMAND [ARGS]...
 | [_neuro-flow cancel_](cli.md#neuro-flow-cancel) | Cancel a bake. |
 | [_neuro-flow clean_](cli.md#neuro-flow-clean) | Clean volume. |
 | [_neuro-flow clear-cache_](cli.md#neuro-flow-clear-cache) | Clear cache. |
+| [_neuro-flow delete-project_](cli.md#neuro-flow-delete-project) | Completely remove project with all related entities |
 | [_neuro-flow download_](cli.md#neuro-flow-download) | Download volume. |
 | [_neuro-flow inspect_](cli.md#neuro-flow-inspect) | Inspect a bake. |
 | [_neuro-flow kill_](cli.md#neuro-flow-kill) | Kill a job. |
@@ -221,6 +222,22 @@ neuro-flow clear-cache [OPTIONS] BATCH [TASK_ID]
 | :--- | :--- |
 | _--help_ | Show this message and exit. |
 
+### neuro-flow delete-project
+
+Completely remove project with all related entities
+
+**Usage:**
+
+```bash
+neuro-flow delete-project [OPTIONS] PROJECT_IDS...
+```
+
+**Options:**
+
+| Name | Description |
+| :--- | :--- |
+| _--help_ | Show this message and exit. |
+
 ### neuro-flow download
 
 Download volume.