flowmesh-cli 0.1.9__tar.gz → 0.1.10rc2__tar.gz

{flowmesh_cli-0.1.9/src/flowmesh_cli.egg-info → flowmesh_cli-0.1.10rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flowmesh-cli
-Version: 0.1.9
+Version: 0.1.10rc2
 Summary: FlowMesh command line interface
 License-Expression: Apache-2.0
 Requires-Python: >=3.12
@@ -10,9 +10,9 @@ Requires-Dist: typer>=0.12.5
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: rich>=14.2.0
 Requires-Dist: websockets>=15.0
-Requires-Dist: flowmesh-sdk==0.1.9
+Requires-Dist: flowmesh-sdk==0.1.10rc2
 Provides-Extra: stack
-Requires-Dist: flowmesh-cli-stack==0.1.9; extra == "stack"
+Requires-Dist: flowmesh-cli-stack==0.1.10rc2; extra == "stack"
 Dynamic: license-file
 
 # FlowMesh CLI

{flowmesh_cli-0.1.9 → flowmesh_cli-0.1.10rc2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "flowmesh-cli"
-version = "0.1.9"
+version = "0.1.10rc2"
 description = "FlowMesh command line interface"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -15,11 +15,11 @@ dependencies = [
     "pyyaml>=6.0.2",
     "rich>=14.2.0",
     "websockets>=15.0",
-    "flowmesh-sdk==0.1.9",
+    "flowmesh-sdk==0.1.10rc2",
 ]
 
 [project.optional-dependencies]
-stack = ["flowmesh-cli-stack==0.1.9"]
+stack = ["flowmesh-cli-stack==0.1.10rc2"]
 
 [project.scripts]
 flowmesh = "flowmesh_cli.cli:main"

{flowmesh_cli-0.1.9 → flowmesh_cli-0.1.10rc2}/src/flowmesh_cli/_version.py

@@ -3,7 +3,7 @@
 from importlib.metadata import PackageNotFoundError, version
 
 _PACKAGE_NAME = "flowmesh-cli"
-_STATIC_VERSION = "0.1.9"
+_STATIC_VERSION = "0.1.10rc2"
 
 
 def _resolve_version() -> str:

{flowmesh_cli-0.1.9 → flowmesh_cli-0.1.10rc2}/src/flowmesh_cli/commands/node.py

@@ -67,7 +67,7 @@ app.add_typer(worker_app, name="worker")
 def list_workers(
     node_id: str | None = typer.Argument(None, help="Node identifier"),
     worker_id: str | None = typer.Option(None, "--id", help="Filter by worker id"),
-    name: str | None = typer.Option(None, "--name", help="Filter by worker name"),
+    alias: str | None = typer.Option(None, "--alias", help="Filter by worker alias"),
     namespace: str | None = typer.Option(
         None, "--namespace", help="Filter by worker namespace"
     ),
@@ -94,7 +94,7 @@ def list_workers(
     client = FlowMesh()
     query_params = parse_query_filters(query)
     append_param(query_params, "id", worker_id)
-    append_param(query_params, "name", name)
+    append_param(query_params, "alias", alias)
     append_param(query_params, "namespace", namespace)
     append_param(query_params, "cluster", cluster)
     append_param(query_params, "provider", provider)
@@ -115,28 +115,28 @@ def list_workers(
 @worker_app.command("start")
 def start_worker(
     node_id: str = typer.Argument(..., help="Node identifier"),
-    worker_name: str = typer.Argument(..., help="Worker name"),
+    alias: str = typer.Argument(..., help="Worker alias"),
 ) -> None:
     """Start a worker on a specific node."""
     client = FlowMesh()
     try:
-        client.nodes.start_worker(node_id, worker_name)
+        client.nodes.start_worker(node_id, alias)
     except FlowMeshError as exc:
         logging.error(str(exc))
         raise typer.Exit(code=1)
-    logging.success(f"Worker '{worker_name}' started on node '{node_id}'")
+    logging.success(f"Worker '{alias}' started on node '{node_id}'")
 
 
 @worker_app.command("stop")
 def stop_worker(
     node_id: str = typer.Argument(..., help="Node identifier"),
-    worker_name: str = typer.Argument(..., help="Worker name"),
+    alias: str = typer.Argument(..., help="Worker alias"),
 ) -> None:
     """Stop a worker on a specific node."""
     client = FlowMesh()
     try:
-        client.nodes.stop_worker(node_id, worker_name)
+        client.nodes.stop_worker(node_id, alias)
     except FlowMeshError as exc:
         logging.error(str(exc))
         raise typer.Exit(code=1)
-    logging.success(f"Worker '{worker_name}' stopped on node '{node_id}'")
+    logging.success(f"Worker '{alias}' stopped on node '{node_id}'")

{flowmesh_cli-0.1.9 → flowmesh_cli-0.1.10rc2}/src/flowmesh_cli/commands/worker.py

@@ -73,3 +73,70 @@ def list_workers(
         logging.error(str(exc))
         raise typer.Exit(code=1)
     logging.log(json.dumps([w.model_dump(mode="json") for w in workers], indent=2))
+
+
+def _set_cordon(
+    cordoned: bool,
+    worker_id: str | None,
+    node_alias: str | None,
+    alias: str | None,
+) -> None:
+    client = FlowMesh()
+    try:
+        if worker_id is not None and node_alias is None and alias is None:
+            set_by_id = client.workers.cordon if cordoned else client.workers.uncordon
+            result = set_by_id(worker_id)
+        elif worker_id is None and node_alias and alias:
+            set_by_alias = (
+                client.workers.cordon_alias
+                if cordoned
+                else client.workers.uncordon_alias
+            )
+            result = set_by_alias(node_alias, alias)
+        else:
+            logging.error("Pass a WORKER_ID, or both --node-alias and --alias.")
+            raise typer.Exit(code=2)
+    except FlowMeshError as exc:
+        logging.error(str(exc))
+        raise typer.Exit(code=1)
+    logging.log(result.model_dump_json(indent=2))
+
+
+@app.command()
+def cordon(
+    worker_id: str | None = typer.Argument(None, help="Worker identifier"),
+    node_alias: str | None = typer.Option(
+        None, "--node-alias", help="Node alias, with --alias instead of WORKER_ID"
+    ),
+    alias: str | None = typer.Option(
+        None, "--alias", help="Worker alias, with --node-alias instead of WORKER_ID"
+    ),
+) -> None:
+    """Stop offering new tasks to a worker without stopping it."""
+    _set_cordon(True, worker_id, node_alias, alias)
+
+
+@app.command()
+def uncordon(
+    worker_id: str | None = typer.Argument(None, help="Worker identifier"),
+    node_alias: str | None = typer.Option(
+        None, "--node-alias", help="Node alias, with --alias instead of WORKER_ID"
+    ),
+    alias: str | None = typer.Option(
+        None, "--alias", help="Worker alias, with --node-alias instead of WORKER_ID"
+    ),
+) -> None:
+    """Allow a cordoned worker to receive tasks again."""
+    _set_cordon(False, worker_id, node_alias, alias)
+
+
+@app.command("cordons")
+def list_cordons() -> None:
+    """List the cordons visible to the caller."""
+    client = FlowMesh()
+    try:
+        cordons = client.workers.list_cordons()
+    except FlowMeshError as exc:
+        logging.error(str(exc))
+        raise typer.Exit(code=1)
+    logging.log(json.dumps([c.model_dump(mode="json") for c in cordons], indent=2))

{flowmesh_cli-0.1.9 → flowmesh_cli-0.1.10rc2/src/flowmesh_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flowmesh-cli
-Version: 0.1.9
+Version: 0.1.10rc2
 Summary: FlowMesh command line interface
 License-Expression: Apache-2.0
 Requires-Python: >=3.12
@@ -10,9 +10,9 @@ Requires-Dist: typer>=0.12.5
 Requires-Dist: pyyaml>=6.0.2
 Requires-Dist: rich>=14.2.0
 Requires-Dist: websockets>=15.0
-Requires-Dist: flowmesh-sdk==0.1.9
+Requires-Dist: flowmesh-sdk==0.1.10rc2
 Provides-Extra: stack
-Requires-Dist: flowmesh-cli-stack==0.1.9; extra == "stack"
+Requires-Dist: flowmesh-cli-stack==0.1.10rc2; extra == "stack"
 Dynamic: license-file
 
 # FlowMesh CLI

{flowmesh_cli-0.1.9 → flowmesh_cli-0.1.10rc2}/src/flowmesh_cli.egg-info/requires.txt

@@ -2,7 +2,7 @@ typer>=0.12.5
 pyyaml>=6.0.2
 rich>=14.2.0
 websockets>=15.0
-flowmesh-sdk==0.1.9
+flowmesh-sdk==0.1.10rc2
 
 [stack]
-flowmesh-cli-stack==0.1.9
+flowmesh-cli-stack==0.1.10rc2