volcano-sdk-python 0.11.0__py3-none-any.whl → 0.11.2__py3-none-any.whl

volcano_sdk/_function_resolution.py

@@ -196,7 +196,7 @@ def _store(
 def forget(api_url: str, authorization: str, name: str) -> None:
     """Drop one cached resolution that turned out to be stale."""
     with _lock:
-        _entries.pop((api_url, authorization, name), None)
+        _ = _entries.pop((api_url, authorization, name), None)
 
 
 def clear() -> None:

volcano_sdk/_lock_guard.py

@@ -9,22 +9,31 @@ from typing import TYPE_CHECKING
 from ._lock_renewer import renewal_delay as _calculate_renewal_delay
 
 if TYPE_CHECKING:
+    from _thread import LockType
+
     from .models import LockLease
 
 MAX_LOCK_LIFETIME_SECONDS = 7_776_000
 LOSS_POLL_INTERVAL_SECONDS = 1.0
-SUSPEND_AWARE_CLOCK_ID = getattr(time, "CLOCK_BOOTTIME", None)
 _LEASE_EXPIRED = "lock lease expired before renewal completed"
 _NO_SAFE_RENEWAL_WINDOW = "lock renewal returned no safe lease window"
 
 
+def _suspend_aware_clock_id(value: object) -> int | None:
+    return value if isinstance(value, int) else None
+
+
+_clock_id_value: object = getattr(time, "CLOCK_BOOTTIME", None)
+SUSPEND_AWARE_CLOCK_ID = _suspend_aware_clock_id(_clock_id_value)
+
+
 class _FallbackClock:
     """Combine monotonic progress with suspend-aware wall time."""
 
     def __init__(self) -> None:
-        self._lock = threading.Lock()
-        self._monotonic = time.monotonic()
-        self._value = time.time()
+        self._lock: LockType = threading.Lock()
+        self._monotonic: float = time.monotonic()
+        self._value: float = time.time()
 
     def __call__(self) -> float:
         with self._lock:
@@ -38,11 +47,11 @@ class _FallbackClock:
 _FALLBACK_CLOCK = _FallbackClock()
 
 
-def _lease_now() -> float:
-    clock_gettime = getattr(time, "clock_gettime", None)
+def lease_now() -> float:
+    clock_gettime: object = getattr(time, "clock_gettime", None)
     if clock_gettime is None or SUSPEND_AWARE_CLOCK_ID is None:
         return _FALLBACK_CLOCK()
-    return float(clock_gettime(SUSPEND_AWARE_CLOCK_ID))
+    return float(time.clock_gettime(SUSPEND_AWARE_CLOCK_ID))
 
 
 class LockGuard:
@@ -57,15 +66,17 @@ class LockGuard:
         lease_started_at: float | None = None,
     ) -> None:
         """Track one acquired lease against its local monotonic deadline."""
-        self._state_lock = threading.Lock()
-        self._lease = lease
-        self._ttl = ttl
-        self._absolute_deadline = started_at + MAX_LOCK_LIFETIME_SECONDS
+        self._state_lock: LockType = threading.Lock()
+        self._lease: LockLease = lease
+        self._ttl: int = ttl
+        self._absolute_deadline: float = started_at + MAX_LOCK_LIFETIME_SECONDS
         # A successful retry renews the TTL, not the maximum ownership lifetime.
         lease_started_at = started_at if lease_started_at is None else lease_started_at
-        self._lease_deadline = min(lease_started_at + ttl, self._absolute_deadline)
+        self._lease_deadline: float = min(
+            lease_started_at + ttl, self._absolute_deadline
+        )
         self._failure: Exception | None = None
-        self._lost = threading.Event()
+        self._lost: threading.Event = threading.Event()
 
     @property
     def lease(self) -> LockLease:
@@ -77,7 +88,7 @@ class LockGuard:
     def lost(self) -> bool:
         """Whether renewal failed, the lease expired, or the guard closed."""
         with self._state_lock:
-            self._expire_if_needed_locked(_lease_now())
+            self._expire_if_needed_locked(lease_now())
             return self._lost.is_set()
 
     def wait_lost(self, timeout: float | None = None) -> bool:
@@ -89,10 +100,10 @@ class LockGuard:
             True when the guard reports lost ownership, False on timeout.
 
         """
-        timeout_deadline = None if timeout is None else _lease_now() + timeout
+        timeout_deadline = None if timeout is None else lease_now() + timeout
         while True:
             with self._state_lock:
-                now = _lease_now()
+                now = lease_now()
                 self._expire_if_needed_locked(now)
                 if self._lost.is_set():
                     return True
@@ -101,15 +112,15 @@ class LockGuard:
                     self._remaining_seconds_locked(now),
                 )
             if timeout_deadline is not None:
-                timeout_remaining = timeout_deadline - _lease_now()
+                timeout_remaining = timeout_deadline - lease_now()
                 if timeout_remaining <= 0:
                     return False
                 wait = min(wait, timeout_remaining)
-            self._lost.wait(wait)
+            _ = self._lost.wait(wait)
 
     def replace_lease(self, lease: LockLease, *, started_at: float) -> bool:
         with self._state_lock:
-            now = _lease_now()
+            now = lease_now()
             self._expire_if_needed_locked(now)
             if self._lost.is_set():
                 return False
@@ -126,18 +137,18 @@ class LockGuard:
 
     def mark_lost(self, failure: Exception) -> None:
         with self._state_lock:
-            self._expire_if_needed_locked(_lease_now())
+            self._expire_if_needed_locked(lease_now())
             self._mark_lost_locked(failure)
 
     def _remaining_seconds(self) -> float:
         with self._state_lock:
-            now = _lease_now()
+            now = lease_now()
             self._expire_if_needed_locked(now)
             return self._remaining_seconds_locked(now)
 
     def renewal_delay(self) -> float:
         with self._state_lock:
-            now = _lease_now()
+            now = lease_now()
             self._expire_if_needed_locked(now)
             if self._lost.is_set():
                 return 0.0
@@ -148,12 +159,12 @@ class LockGuard:
 
     def _renewal_failure(self) -> Exception | None:
         with self._state_lock:
-            self._expire_if_needed_locked(_lease_now())
+            self._expire_if_needed_locked(lease_now())
             return self._failure
 
     def _close(self) -> None:
         with self._state_lock:
-            self._expire_if_needed_locked(_lease_now())
+            self._expire_if_needed_locked(lease_now())
             self._lost.set()
 
     def _mark_lost_locked(self, failure: Exception) -> None:

volcano_sdk/_lock_worker.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import threading
 from typing import TYPE_CHECKING, Protocol
 
-from ._lock_guard import LockGuard, _lease_now
+from ._lock_guard import LockGuard, lease_now
 
 if TYPE_CHECKING:
     from types import TracebackType
@@ -27,7 +27,7 @@ class LockRenewalClient(Protocol):
 
 class _RenewalFailureRecorder:
     def __init__(self, guard: LockGuard) -> None:
-        self._guard = guard
+        self._guard: LockGuard = guard
 
     def __enter__(self) -> None:
         return None
@@ -56,12 +56,12 @@ class LockRenewer:
         *,
         ttl: int,
     ) -> None:
-        self._locks = locks
-        self._key = key
-        self._guard = guard
-        self._ttl = ttl
-        self._stop = threading.Event()
-        self._thread = threading.Thread(target=self._run, daemon=True)
+        self._locks: LockRenewalClient = locks
+        self._key: str = key
+        self._guard: LockGuard = guard
+        self._ttl: int = ttl
+        self._stop: threading.Event = threading.Event()
+        self._thread: threading.Thread = threading.Thread(target=self._run, daemon=True)
 
     def start(self) -> None:
         """Start renewing the guarded lease."""
@@ -81,11 +81,11 @@ class LockRenewer:
                     return
 
     def _wait_until_renewal(self) -> bool:
-        renew_at = _lease_now() + self._guard.renewal_delay()
+        renew_at = lease_now() + self._guard.renewal_delay()
         while not self._stop.is_set():
             if self._guard.lost:
                 return False
-            remaining = renew_at - _lease_now()
+            remaining = renew_at - lease_now()
             if remaining <= 0:
                 return True
             if self._stop.wait(min(remaining, MAX_RENEWAL_WAIT_SLICE_SECONDS)):
@@ -93,7 +93,7 @@ class LockRenewer:
         return False
 
     def _renew_once(self) -> bool:
-        started_at = _lease_now()
+        started_at = lease_now()
         with _RenewalFailureRecorder(self._guard):
             lease = self._locks.renew(
                 self._key,

volcano_sdk/_log_response.py

@@ -14,12 +14,12 @@ INVALID_LOG_RESPONSE = "Expected a complete log response"
 def response_values(payload: object) -> Mapping[str, object]:
     """Require a JSON object for the response envelope.
 
-    Raises:
-        TypeError: The response envelope is not an object.
-
     Returns:
         The response fields without normalizing their values.
 
+    Raises:
+        TypeError: The response envelope is not an object.
+
     """
     if not isinstance(payload, Mapping):
         raise TypeError(INVALID_LOG_RESPONSE)
@@ -29,12 +29,12 @@ def response_values(payload: object) -> Mapping[str, object]:
 def response_data(values: Mapping[str, object]) -> tuple[Mapping[str, JSONValue], ...]:
     """Require a list of objects without coercing empty mappings into lists.
 
-    Raises:
-        TypeError: The rows are not a list of objects.
-
     Returns:
         The validated rows in their original order.
 
+    Raises:
+        TypeError: The rows are not a list of objects.
+
     """
     raw_data = values.get("data")
     if not isinstance(raw_data, list):
@@ -48,12 +48,12 @@ def response_data(values: Mapping[str, object]) -> tuple[Mapping[str, JSONValue]
 def search_metadata(values: Mapping[str, object]) -> tuple[int, bool, str | None]:
     """Require search pagination fields without coercing their values.
 
-    Raises:
-        TypeError: Required metadata is missing or has an invalid type.
-
     Returns:
         The page limit, continuation flag, and optional cursor.
 
+    Raises:
+        TypeError: Required metadata is missing or has an invalid type.
+
     """
     limit = values.get("limit")
     has_more = values.get("has_more")
@@ -71,12 +71,12 @@ def search_metadata(values: Mapping[str, object]) -> tuple[int, bool, str | None
 def activity_total(values: Mapping[str, object]) -> int:
     """Require the activity total without treating booleans as integers.
 
-    Raises:
-        TypeError: The total is missing or is not an integer.
-
     Returns:
         The total reported by the server.
 
+    Raises:
+        TypeError: The total is missing or is not an integer.
+
     """
     total = values.get("total")
     if not isinstance(total, int) or isinstance(total, bool):

volcano_sdk/_session.py

@@ -33,8 +33,8 @@ def session_id_from_access_token(access_token: str) -> str | None:
         return None
     padding = "=" * (-len(parts[1]) % 4)
     try:
-        payload: object = json.loads(
-            base64.urlsafe_b64decode(parts[1] + padding).decode()
+        payload = cast(
+            "object", json.loads(base64.urlsafe_b64decode(parts[1] + padding).decode())
         )
     except (ValueError, UnicodeDecodeError, RecursionError):
         return None

volcano_sdk/_session_operations.py

@@ -9,6 +9,7 @@ from typing import TYPE_CHECKING, TypeVar
 from .errors import SessionChangedError, VolcanoError
 
 if TYPE_CHECKING:
+    from _thread import LockType
     from collections.abc import Callable
 
     from .models import Session
@@ -40,11 +41,11 @@ class SessionOperations:
     """Retain only this session's latest refresh and shared sign-out outcome."""
 
     def __init__(self, verified: Session | None = None) -> None:
-        self._lock = Lock()
+        self._lock: LockType = Lock()
         self.refreshing: Future[Session] | None = None
         self.signing_out: Future[BaseException | None] | None = None
-        self._locally_cleared = False
-        self._verified_pair = (
+        self._locally_cleared: bool = False
+        self._verified_pair: tuple[str, str | None] | None = (
             (verified.access_token, verified.refresh_token)
             if verified is not None
             else None

volcano_sdk/_transport.py

@@ -7,7 +7,15 @@ from collections.abc import Mapping
 from dataclasses import dataclass
 from io import BytesIO
 from pathlib import PurePosixPath
-from typing import TYPE_CHECKING, Any, ParamSpec, Protocol, TypeVar, cast
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    ParamSpec,
+    Protocol,
+    TypeVar,
+    cast,
+    runtime_checkable,
+)
 from uuid import UUID, uuid4
 
 import httpx
@@ -375,6 +383,7 @@ class _GeneratedTransportResponse:
     headers: Mapping[str, str]
 
 
+@runtime_checkable
 class AuthRefreshTransport(Protocol):
     def auth_refresh(
         self,
@@ -384,6 +393,7 @@ class AuthRefreshTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthLogoutTransport(Protocol):
     def auth_logout(
         self,
@@ -393,6 +403,7 @@ class AuthLogoutTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthSignUpTransport(Protocol):
     def auth_signup(
         self,
@@ -404,6 +415,7 @@ class AuthSignUpTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthSignUpAnonymousTransport(Protocol):
     def auth_signup_anonymous(
         self,
@@ -413,6 +425,7 @@ class AuthSignUpAnonymousTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthConvertAnonymousTransport(Protocol):
     def auth_convert_anonymous(
         self,
@@ -424,6 +437,7 @@ class AuthConvertAnonymousTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthForgotPasswordTransport(Protocol):
     def auth_forgot_password(
         self,
@@ -433,6 +447,7 @@ class AuthForgotPasswordTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthConfirmEmailTransport(Protocol):
     def auth_confirm_email(
         self,
@@ -442,6 +457,7 @@ class AuthConfirmEmailTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthResetPasswordTransport(Protocol):
     def auth_reset_password(
         self,
@@ -452,6 +468,7 @@ class AuthResetPasswordTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthResendConfirmationTransport(Protocol):
     def auth_resend_confirmation(
         self,
@@ -461,6 +478,7 @@ class AuthResendConfirmationTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthRequestEmailChangeTransport(Protocol):
     def auth_request_email_change(
         self,
@@ -470,6 +488,7 @@ class AuthRequestEmailChangeTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthCancelEmailChangeTransport(Protocol):
     def auth_cancel_email_change(
         self,
@@ -478,6 +497,7 @@ class AuthCancelEmailChangeTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthConfirmEmailChangeTransport(Protocol):
     def auth_confirm_email_change(
         self,
@@ -487,6 +507,7 @@ class AuthConfirmEmailChangeTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthDeleteAllMySessionsTransport(Protocol):
     def auth_delete_all_my_sessions(
         self,
@@ -495,6 +516,7 @@ class AuthDeleteAllMySessionsTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthDeleteMySessionTransport(Protocol):
     def auth_delete_my_session(
         self,
@@ -504,6 +526,7 @@ class AuthDeleteMySessionTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthGetMySessionsTransport(Protocol):
     def auth_get_my_sessions(
         self,
@@ -514,6 +537,7 @@ class AuthGetMySessionsTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthListOAuthProvidersTransport(Protocol):
     def auth_list_oauth_providers(
         self,
@@ -522,6 +546,7 @@ class AuthListOAuthProvidersTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthOAuthAuthorizationURLTransport(Protocol):
     def auth_oauth_authorization_url(
         self,
@@ -533,6 +558,7 @@ class AuthOAuthAuthorizationURLTransport(Protocol):
     ) -> str: ...
 
 
+@runtime_checkable
 class AuthOAuthExchangeTransport(Protocol):
     def auth_oauth_exchange(
         self,
@@ -543,6 +569,7 @@ class AuthOAuthExchangeTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthLinkOAuthProviderTransport(Protocol):
     def auth_link_oauth_provider(
         self,
@@ -552,6 +579,7 @@ class AuthLinkOAuthProviderTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthUnlinkOAuthProviderTransport(Protocol):
     def auth_unlink_oauth_provider(
         self,
@@ -561,6 +589,7 @@ class AuthUnlinkOAuthProviderTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthGetOAuthProviderTokenTransport(Protocol):
     def auth_get_oauth_provider_token(
         self,
@@ -570,6 +599,7 @@ class AuthGetOAuthProviderTokenTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthRefreshOAuthProviderTokenTransport(Protocol):
     def auth_refresh_oauth_provider_token(
         self,
@@ -579,6 +609,7 @@ class AuthRefreshOAuthProviderTokenTransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthCallOAuthAPITransport(Protocol):
     def auth_call_oauth_api(
         self,
@@ -591,10 +622,12 @@ class AuthCallOAuthAPITransport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthGetUserTransport(Protocol):
     def auth_get_user(self, *, authorization: str) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AuthUpdateUserTransport(Protocol):
     def auth_update_user(
         self,
@@ -685,6 +718,7 @@ class Transport(Protocol):
     ) -> TransportResponse: ...
 
 
+@runtime_checkable
 class AsyncDatabaseSelectTransport(Protocol):
     """Async database query capability used by cancellable realtime fetches."""