handle shell warning when the lease has been transferred

Signed-off-by: Benny Zlotnik <bzlotnik@redhat.com>
Assisted-by: claude-sonnet-4.6

Author: bennyz

python/packages/jumpstarter-cli/jumpstarter_cli/shell.py

@@ -275,6 +275,10 @@ async def _shell_with_signal_handling(  # noqa: C901
                         # Lease expired naturally (e.g. during beforeLease hook)
                         # — exit gracefully instead of showing a scary error
                         pass
+                    elif lease_used.lease_transferred:
+                        raise ExporterOfflineError(
+                            "Lease has been transferred to another client. Session is no longer valid."
+                        ) from None
                     else:
                         raise ExporterOfflineError("Connection to exporter lost") from None
                 else:

python/packages/jumpstarter-cli/jumpstarter_cli/update.py

@@ -43,15 +43,15 @@ def update_lease(
     Update a lease
 
     Update the duration, begin time, and/or owner of an existing lease.
-    At least one of --duration, --begin-time, or --to_client must be specified.
+    At least one of --duration, --begin-time, or --to-client must be specified.
 
-    To transfer a lease to another client in the same namespace, use the --to_client option.
+    To transfer a lease to another client in the same namespace, use the --to-client option.
 
     Updating the begin time of an already active lease is not allowed.
     """
 
     if duration is None and begin_time is None and to_client is None:
-        raise click.UsageError("At least one of --duration, --begin-time, or --client must be specified")
+        raise click.UsageError("At least one of --duration, --begin-time, or --to-client must be specified")
 
     client_path = f"namespaces/{config.metadata.namespace}/clients/{to_client}" if to_client else None
 

python/packages/jumpstarter/jumpstarter/client/lease.py

@@ -63,6 +63,7 @@ class Lease(ContextManagerMixin, AsyncContextManagerMixin):
         default=None, init=False
     )  # Called when lease is ending
     lease_ended: bool = field(default=False, init=False)  # Set when lease expires naturally
+    lease_transferred: bool = field(default=False, init=False)  # Set when lease is transferred to another client
 
     def __post_init__(self):
         if hasattr(super(), "__post_init__"):
@@ -293,7 +294,15 @@ async def handle_async(self, stream):
                     attempt += 1
                     continue
                 # Exporter went offline or lease ended - log and exit gracefully
-                logger.warning("Connection to exporter lost: %s", e.details())
+                if "permission denied" in str(e.details()).lower():
+                    self.lease_transferred = True
+                    logger.warning(
+                        "Lease %s has been transferred to another client. "
+                        "Your session is no longer valid.",
+                        self.name,
+                    )
+                else:
+                    logger.warning("Connection to exporter lost: %s", e.details())
                 return
         async with connect_router_stream(
             response.router_endpoint, response.router_token, stream, self.tls_config, self.grpc_options