From 76c853a5fb6078a01c2b94190e7825597098cf3d Mon Sep 17 00:00:00 2001
From: Dalena <mimidalena@gmail.com>
Date: Mon, 28 Oct 2024 12:26:54 -0500
Subject: [PATCH] feat: add cluster update (#158)
---
silverback/_cli.py | 52 ++++++++++++++++++++++++++++++++++++
silverback/cluster/client.py | 19 +++++++++++++
2 files changed, 71 insertions(+)
diff --git a/silverback/_cli.py b/silverback/_cli.py
index b7568e87..28dbd0cc 100644
--- a/silverback/_cli.py
+++ b/silverback/_cli.py
@@ -378,6 +378,58 @@ def new_cluster(
)
+@cluster.command(name="update", section="Platform Commands (https://silverback.apeworx.io)")
+@click.option(
+ "-n",
+ "--name",
+ "name",
+ default=None,
+ help="Update name for cluster",
+)
+@click.option(
+ "-s",
+ "--slug",
+ "slug",
+ default=None,
+ help="Update slug for cluster",
+)
+@click.argument("cluster_path")
+@platform_client
+def update_cluster(
+ platform: PlatformClient,
+ cluster_path: str,
+ name: str | None,
+ slug: str | None,
+):
+ """Update name and slug for a CLUSTER"""
+
+ if "/" not in cluster_path or len(cluster_path.split("/")) > 2:
+ raise click.BadArgumentUsage(f"Invalid cluster path: '{cluster_path}'")
+
+ workspace_name, cluster_name = cluster_path.split("/")
+ if not (workspace_client := platform.workspaces.get(workspace_name)):
+ raise click.BadArgumentUsage(f"Unknown workspace: '{workspace_name}'")
+
+ elif not (cluster := workspace_client.clusters.get(cluster_name)):
+ raise click.BadArgumentUsage(
+ f"Unknown cluster in workspace '{workspace_name}': '{cluster_name}'"
+ )
+
+ elif name is None and slug is None:
+ raise click.UsageError(
+ "No update name or slug found. Please enter a name or slug to update."
+ )
+ elif name == "" or slug == "":
+ raise click.UsageError("Empty string value found for name or slug.")
+
+ updated_cluster = workspace_client.update_cluster(
+ cluster_id=str(cluster.id),
+ name=name,
+ slug=slug,
+ )
+ click.echo(f"{click.style('SUCCESS', fg='green')}: Updated '{updated_cluster.name}'")
+
+
@cluster.group(cls=SectionedHelpGroup, section="Platform Commands (https://silverback.apeworx.io)")
def pay():
"""Pay for CLUSTER with Crypto using ApePay streaming payments"""
diff --git a/silverback/cluster/client.py b/silverback/cluster/client.py
index 6000fe6d..ff4cfc5b 100644
--- a/silverback/cluster/client.py
+++ b/silverback/cluster/client.py
@@ -385,6 +385,25 @@ def create_cluster(
self.clusters.update({new_cluster.slug: new_cluster}) # NOTE: Update cache
return new_cluster
+ def update_cluster(
+ self,
+ cluster_id: str,
+ name: str | None = None,
+ slug: str | None = None,
+ ) -> ClusterInfo:
+ data = dict()
+ if name:
+ data["name"] = name
+ if slug:
+ data["slug"] = slug
+ response = self.client.patch(
+ f"/clusters/{cluster_id}",
+ params=dict(workspace=str(self.id), cluster_id=cluster_id),
+ data=data,
+ )
+ handle_error_with_response(response)
+ return ClusterInfo.model_validate(response.json())
+
def get_payment_stream(self, cluster: ClusterInfo, chain_id: int) -> Stream | None:
response = self.client.get(
f"/clusters/{cluster.id}/stream",