volcano-sdk-python 0.9.2__py3-none-any.whl → 0.9.3__py3-none-any.whl

volcano_sdk/_transport.py

@@ -99,10 +99,12 @@ from ._generated.api.database_queries.query_database_update import (
 from ._generated.api.database_queries.query_database_update import (
     _get_kwargs as database_update_kwargs,
 )
-from ._generated.api.functions import resolve_function_for_invocation
 from ._generated.api.functions.invoke_function import (
     _get_kwargs as invoke_function_kwargs,
 )
+from ._generated.api.functions.resolve_function_for_invocation import (
+    _get_kwargs as resolve_function_kwargs,
+)
 from ._generated.api.locks import (
     force_release_project_lock,
     get_project_lock,
@@ -1643,11 +1645,10 @@ class GeneratedTransport:
         name: str,
     ) -> TransportResponse:
         with self._client(authorization) as client:
-            response = resolve_function_for_invocation.sync_detailed(
-                client=client,
-                name=name,
+            response = client.get_httpx_client().request(
+                **resolve_function_kwargs(name=name)
             )
-        return self._response(response)
+        return self._raw_response(response)
 
     def invoke_function(
         self,

volcano_sdk/functions.py

@@ -4,14 +4,19 @@ from __future__ import annotations
 
 import json
 import re
-from collections.abc import Mapping
-from typing import TYPE_CHECKING, Protocol, cast
+from collections.abc import Callable, Mapping
+from typing import TYPE_CHECKING, Protocol, TypeVar, cast
 
 from . import _function_resolution
 from ._function_resolution import FunctionResolution
 from ._transport import TransportResponse, invoke, response_payload
-from .errors import NotFoundError
-from .models import FunctionResponse, JSONValue
+from .errors import (
+    AuthenticationError,
+    NotFoundError,
+    SessionChangedError,
+    VolcanoError,
+)
+from .models import FunctionResponse, JSONValue, _freeze_json
 
 if TYPE_CHECKING:
     from .client import VolcanoClient
@@ -27,10 +32,57 @@ _UNKNOWN_FUNCTION = "Function was not found"
 _HTTP_SUCCESS_MIN = 200
 _HTTP_SUCCESS_MAX = 300
 _HTTP_NOT_FOUND = 404
+_HTTP_UNAUTHORIZED = 401
 # Present only once the platform has dispatched to the function.
 _FUNCTION_INVOKED_HEADER = "X-Volcano-Function-Invoked"
 
 
+_Result = TypeVar("_Result")
+
+
+class _FunctionAuth:
+    def __init__(self, client: VolcanoClient) -> None:
+        self._client = client
+        self._binding = client._capture_session_binding()
+        self._fallback_token = client._function_token()
+
+    def run(self, operation: Callable[[str], _Result]) -> _Result:
+        if self._binding[2] is not None:
+            self._binding = self._client.auth._owned_refresh_session(self._binding)
+        try:
+            return self._run(operation)
+        finally:
+            if self._binding[2] is not None:
+                self._client.auth._validate_read_failure(self._binding)
+            elif self._client._capture_session_binding()[1] != self._binding[1]:
+                raise SessionChangedError
+
+    def _run(self, operation: Callable[[str], _Result]) -> _Result:
+        try:
+            return operation(self._token())
+        except AuthenticationError as original:
+            if self._binding[2] is None or original.status != _HTTP_UNAUTHORIZED:
+                raise
+            try:
+                # Resolve has released its cache lock before refresh callbacks run.
+                self._client.auth._refresh_session_for_binding(self._binding)
+            except SessionChangedError:
+                raise
+            except VolcanoError:
+                raise original from None
+            return operation(self._token())
+
+    def _token(self) -> str:
+        if self._binding[2] is None:
+            if self._client._capture_session_binding()[1] != self._binding[1]:
+                raise SessionChangedError
+            return self._fallback_token
+        session = self._client.auth._owned_refresh_session(self._binding)[2]
+        if session is None:
+            raise SessionChangedError
+        return session.access_token
+
+
 class FunctionsTransport(Protocol):
     """Transport operations required by the functions facade."""
 
@@ -77,13 +129,17 @@ class Functions:
         payload: Mapping[str, JSONValue] | None = None,
     ) -> FunctionResponse:
         """Resolve and invoke a function with an optional JSON object payload."""
+        auth = _FunctionAuth(self._client)
         name = _function_name(name)
         request_payload = _function_payload(payload)
-        authorization = self._client._function_token()
         transport = cast("FunctionsTransport", self._client._transport)
-        resolution = self._resolve(transport, authorization, name)
-        response = self._invoke_resolved(
-            transport, authorization, resolution, request_payload
+        authorization, resolution = auth.run(
+            lambda token: (token, self._resolve(transport, token, name))
+        )
+        response = auth.run(
+            lambda token: self._invoke_resolved(
+                transport, token, resolution, request_payload
+            )
         )
         if _stale_mapping(response):
             # The function was deleted and recreated, so the cached identity no
@@ -91,9 +147,13 @@ class Functions:
             _function_resolution.forget(
                 self._client._api_base_url(), authorization, name
             )
-            resolution = self._resolve(transport, authorization, name)
-            response = self._invoke_resolved(
-                transport, authorization, resolution, request_payload
+            _, resolution = auth.run(
+                lambda token: (token, self._resolve(transport, token, name))
+            )
+            response = auth.run(
+                lambda token: self._invoke_resolved(
+                    transport, token, resolution, request_payload
+                )
             )
         return self._response(response)
 
@@ -118,7 +178,13 @@ class Functions:
                 invoke_url=resolution.invoke_url,
                 payload=payload,
             )
-        return cast("TransportResponse", response)
+        response = cast("TransportResponse", response)
+        if (
+            response.status_code == _HTTP_UNAUTHORIZED
+            and _header(response.headers, _FUNCTION_INVOKED_HEADER) is None
+        ):
+            response_payload(response, _HTTP_SUCCESS_MIN)
+        return response
 
     def _resolve(
         self,
@@ -269,9 +335,9 @@ def _function_name(value: object) -> str:
     return value
 
 
-def _function_payload(value: object) -> dict[str, JSONValue]:
+def _function_payload(value: object) -> Mapping[str, JSONValue]:
     if value is None:
         return {}
     if not isinstance(value, Mapping):
         raise TypeError(_INVALID_FUNCTION_PAYLOAD)
-    return dict(cast("Mapping[str, JSONValue]", value))
+    return cast("Mapping[str, JSONValue]", _freeze_json(cast("JSONValue", value)))

{volcano_sdk_python-0.9.2.dist-info → volcano_sdk_python-0.9.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: volcano-sdk-python
-Version: 0.9.2
+Version: 0.9.3
 Summary: Official Python SDK for Volcano
 Project-URL: Homepage, https://volcano.dev
 Project-URL: Repository, https://github.com/Kong/volcano-sdk-python
@@ -224,8 +224,16 @@ user session; the function receives no user identity. The immutable result
 includes the response body, status, headers, and `X-Volcano-Version`. The body
 can be a JSON object, array, scalar, or text; an empty body returns `None`.
 JSON arrays become immutable tuples. Invalid JSON is returned as text. A
-function's own non-2xx response is returned when the version header proves it
-ran; platform failures raise typed SDK errors.
+function's own non-2xx response is returned when Volcano confirms it ran;
+non-success platform HTTP responses raise typed SDK errors.
+
+
+Function resolution and invocation recover from a platform HTTP 401 before dispatch:
+the SDK refreshes the captured session and retries the rejected request once.
+Concurrent calls share successful recovery. Replacing or signing out that session
+prevents replay under another identity. The call preserves its original payload values.
+A function's own response, HTTP 403, or a network failure never triggers this retry.
+Anonymous and service keys do not refresh.
 
 `logs.search()` returns an immutable page of retained runtime or deployment log
 events. Pass `next_cursor` back as `cursor` to continue a search. `logs.activity()`

{volcano_sdk_python-0.9.2.dist-info → volcano_sdk_python-0.9.3.dist-info}/RECORD

@@ -6,13 +6,13 @@ volcano_sdk/_lock_worker.py,sha256=QhSmwR0WcwgT-7ueJMlbCogLqOakQ5eRAtyWV74fgV0,3
 volcano_sdk/_realtime_fetch_worker.py,sha256=zQTgFF4cIUBeAL4J2OpWAlz_sIynv2BLfu588EogAdo,9269
 volcano_sdk/_session.py,sha256=LYCv2NCDlL5Slh2YLEG8VQ8z-kDaKDYWW4vGdMvUf1U,2474
 volcano_sdk/_session_operations.py,sha256=1vI9gtwjxG2eUGT-nIYnMEgRldJqgIj2lRLukBbG64E,5518
-volcano_sdk/_transport.py,sha256=AFd47C3X1WAHwq3f9V2ux0WNzwGIn4NBxrwgaIN-xl0,58296
+volcano_sdk/_transport.py,sha256=ox_wKKEW42gm2307uK9vIiQFB5l8pzb0ifWXxkzPiHQ,58331
 volcano_sdk/auth.py,sha256=iKq8o0K8geVKXrnLzhvOjAQoX_Nd0yQDJqDH4bemWaM,47366
 volcano_sdk/client.py,sha256=CJJnf_iTcXnyY2Y57LXJVAn39mTUsepblUg9op-Thzo,12283
 volcano_sdk/connection_string.py,sha256=LlTZigsrCF4IfAqQ8SDTWJv35hnnzNu1MfeVv5JW-Kk,2331
 volcano_sdk/database.py,sha256=p_5OaqlBjKAtIUfKQCflXM_CGqokrVmpOMsp6KHiJ8A,9493
 volcano_sdk/errors.py,sha256=ffs66FIN_jhyh5SMj7KdXb-zVV3DSgVK6xJEPNA5h9Y,1613
-volcano_sdk/functions.py,sha256=zDeAKyCieNnI6wzb9iZMlg9_cbjaYWXYbXHb7jzCle4,10126
+volcano_sdk/functions.py,sha256=bk523Ss80nREwMD1wk6_pO2XKqOIXqDKylq0fQvMmb0,12540
 volcano_sdk/locks.py,sha256=eSkidfHQ_wml5-1LT2hBtHxeALP6E1cGYK0FzHjbrNk,9241
 volcano_sdk/logs.py,sha256=lxORRQBvsnJ0BdMau19MXrIeig5_mEucQSyJ4F1CZxU,4666
 volcano_sdk/models.py,sha256=n5EdzkWzmOstXsuFlvfGOILRemQAqG2P4VCLMelW8SA,8690
@@ -775,7 +775,7 @@ volcano_sdk/_generated/models/usage_data_point.py,sha256=d4U08JbVTiIMM_l2BmRILL1
 volcano_sdk/_generated/models/variable.py,sha256=tvGDOMv77gW6j5skufnfrlNAhzCzA1F7nuPCQWx79yY,5926
 volcano_sdk/_generated/models/variable_deploy_source.py,sha256=G_aCiFHzQx8WhJFs0GvYZEFclfTrfkDvd3kT1e3vty8,463
 volcano_sdk/_generated/models/variable_status.py,sha256=THSsKBSiAYG4kAtBzGA4TZqxjgx8hdL0Q4qCdZfOj00,391
-volcano_sdk_python-0.9.2.dist-info/METADATA,sha256=rJFF7ZXI76Kjkc_CyNAl6Dg5ryRKy2VjiOV4HBtIGLQ,35536
-volcano_sdk_python-0.9.2.dist-info/WHEEL,sha256=THafob7ofN-NsuMN7Mg4qZyHaQI7KkD-QlcQatYhXPo,87
-volcano_sdk_python-0.9.2.dist-info/licenses/LICENSE,sha256=AIHJy-5MFqM8fHfuGFTTb9a8rvJGypcVgN4aZ-7XTWw,11387
-volcano_sdk_python-0.9.2.dist-info/RECORD,,
+volcano_sdk_python-0.9.3.dist-info/METADATA,sha256=AmqFIPDJDOmZgEWKZUIn0jW8EK4ITcvAG92gKMDl1Mk,36008
+volcano_sdk_python-0.9.3.dist-info/WHEEL,sha256=THafob7ofN-NsuMN7Mg4qZyHaQI7KkD-QlcQatYhXPo,87
+volcano_sdk_python-0.9.3.dist-info/licenses/LICENSE,sha256=AIHJy-5MFqM8fHfuGFTTb9a8rvJGypcVgN4aZ-7XTWw,11387
+volcano_sdk_python-0.9.3.dist-info/RECORD,,