make lease aquisition timeout configurable

add new client config for timeout on lease acquisition.
leases:
  acquisition_timeout: 300

Author: michalskrivanek

packages/jumpstarter/jumpstarter/client/lease.py

@@ -47,6 +47,7 @@ class Lease(ContextManagerMixin, AsyncContextManagerMixin):
     controller: jumpstarter_pb2_grpc.ControllerServiceStub = field(init=False)
     tls_config: TLSConfigV1Alpha1 = field(default_factory=TLSConfigV1Alpha1)
     grpc_options: dict[str, Any] = field(default_factory=dict)
+    acquisition_timeout: int = field(default=300)  # Timeout in seconds for lease acquisition, polled in 5s intervals
     exporter_name: str = field(default="remote", init=False)  # Populated during acquisition
 
     def __post_init__(self):
@@ -115,7 +116,7 @@ async def _acquire(self):
 
         Makes sure the lease is ready, and returns the lease object.
         """
-        with fail_after(60):  # TODO: configurable timeout
+        with fail_after(self.acquisition_timeout):
             while True:
                 logger.debug("Polling Lease %s", self.name)
                 result = await self.get()

packages/jumpstarter/jumpstarter/config/client.py

@@ -86,6 +86,17 @@ def decode_unsafe(self) -> Self:
         return self
 
 
+class ClientConfigV1Alpha1Lease(BaseSettings):
+    """Configuration for lease operations."""
+
+    acquisition_timeout: int = Field(
+        default=300,
+        description="Timeout in seconds for lease acquisition",
+        ge=5,  # Must be at least 5 seconds (polling interval)
+        le=43200,  # Maximum 12 hours
+    )
+
+
 class ClientConfigV1Alpha1(BaseSettings):
     CLIENT_CONFIGS_PATH: ClassVar[Path] = CONFIG_PATH / "clients"
 
@@ -108,6 +119,8 @@ class ClientConfigV1Alpha1(BaseSettings):
 
     shell: ShellConfigV1Alpha1 = Field(default_factory=ShellConfigV1Alpha1)
 
+    leases: ClientConfigV1Alpha1Lease = Field(default_factory=ClientConfigV1Alpha1Lease)
+
     async def channel(self):
         if self.endpoint is None or self.token is None:
             raise ConfigurationError("endpoint or token not set in client config")
@@ -258,6 +271,7 @@ async def lease_async(
                 release=release_lease,
                 tls_config=self.tls,
                 grpc_options=self.grpcOptions,
+                acquisition_timeout=self.leases.acquisition_timeout,
             ) as lease:
                 yield lease