cli: support lease transfer

depends on jumpstarter-dev/jumpstarter-controller#199

Signed-off-by: Benny Zlotnik <bzlotnik@redhat.com>

Author: bennyz

packages/jumpstarter-cli/jumpstarter_cli/update.py

@@ -22,20 +22,32 @@ def update():
 @click.argument("name")
 @opt_duration_partial(required=False)
 @opt_begin_time
+@click.option(
+    "--client",
+    type=str,
+    default=None,
+    help="Transfer lease to a different client (format: namespaces/{namespace}/clients/{client})",
+)
 @opt_output_all
 @handle_exceptions_with_reauthentication(relogin_client)
-def update_lease(config, name: str, duration: timedelta | None, begin_time: datetime | None, output: OutputType):
+def update_lease(
+    config, name: str, duration: timedelta | None, begin_time: datetime | None, client: str | None, output: OutputType
+):
     """
     Update a lease
 
-    Update the duration and/or begin time of an existing lease.
-    At least one of --duration or --begin-time must be specified.
+    Update the duration, begin time, and/or owner of an existing lease.
+    At least one of --duration, --begin-time, or --client must be specified.
+
+    To transfer a lease to another client, use the --client option with the
+    full client identifier (e.g., namespaces/default/clients/other-client).
+
     Updating the begin time of an already active lease is not allowed.
     """
 
-    if duration is None and begin_time is None:
-        raise click.UsageError("At least one of --duration or --begin-time must be specified")
+    if duration is None and begin_time is None and client is None:
+        raise click.UsageError("At least one of --duration, --begin-time, or --client must be specified")
 
-    lease = config.update_lease(name, duration=duration, begin_time=begin_time)
+    lease = config.update_lease(name, duration=duration, begin_time=begin_time, client=client)
 
     model_print(lease, output)

packages/jumpstarter/jumpstarter/client/grpc.py

@@ -275,11 +275,7 @@ def model_dump_json(self, **kwargs):
         if not self.include_online:
             exclude_fields.add("online")
 
-        data = {
-            "exporters": [
-                exporter.model_dump(mode="json", exclude=exclude_fields) for exporter in self.exporters
-            ]
-        }
+        data = {"exporters": [exporter.model_dump(mode="json", exclude=exclude_fields) for exporter in self.exporters]}
         return json.dumps(data, **json_kwargs)
 
     def model_dump(self, **kwargs):
@@ -289,11 +285,8 @@ def model_dump(self, **kwargs):
         if not self.include_online:
             exclude_fields.add("online")
 
-        return {
-            "exporters": [
-                exporter.model_dump(mode="json", exclude=exclude_fields) for exporter in self.exporters
-            ]
-        }
+        return {"exporters": [exporter.model_dump(mode="json", exclude=exclude_fields) for exporter in self.exporters]}
+
 
 class LeaseList(BaseModel):
     leases: list[Lease]
@@ -419,6 +412,7 @@ async def UpdateLease(
         name: str,
         duration: timedelta | None = None,
         begin_time: datetime | None = None,
+        client: str | None = None,
     ):
         lease_pb = client_pb2.Lease(
             name="namespaces/{}/leases/{}".format(self.namespace, name),
@@ -438,8 +432,12 @@ async def UpdateLease(
             lease_pb.begin_time.CopyFrom(timestamp_pb)
             update_fields.append("begin_time")
 
+        if client is not None:
+            lease_pb.client = client
+            update_fields.append("client")
+
         if not update_fields:
-            raise ValueError("At least one of duration or begin_time must be provided")
+            raise ValueError("At least one of duration, begin_time, or client must be provided")
 
         update_mask = field_mask_pb2.FieldMask()
         update_mask.FromJsonString(",".join(update_fields))

packages/jumpstarter/jumpstarter/config/client.py

@@ -244,9 +244,10 @@ async def update_lease(
         name: str,
         duration: timedelta | None = None,
         begin_time: datetime | None = None,
+        client: str | None = None,
     ):
         svc = ClientService(channel=await self.channel(), namespace=self.metadata.namespace)
-        return await svc.UpdateLease(name=name, duration=duration, begin_time=begin_time)
+        return await svc.UpdateLease(name=name, duration=duration, begin_time=begin_time, client=client)
 
     @asynccontextmanager
     async def lease_async(