antfly-sdk 0.0.2.dev2__py3-none-any.whl

antfly/client_generated/api/data_operations/begin_transaction.py

@@ -0,0 +1,172 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.error import Error
+from ...models.transaction_begin_request import TransactionBeginRequest
+from ...models.transaction_begin_response import TransactionBeginResponse
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    *,
+    body: TransactionBeginRequest | Unset = UNSET,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/db/v1/transactions/begin",
+    }
+
+    if not isinstance(body, Unset):
+        _kwargs["json"] = body.to_dict()
+
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Error | TransactionBeginResponse | None:
+    if response.status_code == 201:
+        response_201 = TransactionBeginResponse.from_dict(response.json())
+
+        return response_201
+
+    if response.status_code == 400:
+        response_400 = Error.from_dict(response.json())
+
+        return response_400
+
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Error | TransactionBeginResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient,
+    body: TransactionBeginRequest | Unset = UNSET,
+) -> Response[Error | TransactionBeginResponse]:
+    """Begin a transaction session
+
+    Args:
+        body (TransactionBeginRequest | Unset):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Error | TransactionBeginResponse]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient,
+    body: TransactionBeginRequest | Unset = UNSET,
+) -> Error | TransactionBeginResponse | None:
+    """Begin a transaction session
+
+    Args:
+        body (TransactionBeginRequest | Unset):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Error | TransactionBeginResponse
+    """
+
+    return sync_detailed(
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+    body: TransactionBeginRequest | Unset = UNSET,
+) -> Response[Error | TransactionBeginResponse]:
+    """Begin a transaction session
+
+    Args:
+        body (TransactionBeginRequest | Unset):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Error | TransactionBeginResponse]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient,
+    body: TransactionBeginRequest | Unset = UNSET,
+) -> Error | TransactionBeginResponse | None:
+    """Begin a transaction session
+
+    Args:
+        body (TransactionBeginRequest | Unset):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Error | TransactionBeginResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            body=body,
+        )
+    ).parsed

antfly/client_generated/api/data_operations/cleanup_transaction_sessions.py

@@ -0,0 +1,171 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.error import Error
+from ...models.transaction_session_cleanup_response import TransactionSessionCleanupResponse
+from ...types import UNSET, Response, Unset
+
+
+def _get_kwargs(
+    *,
+    cutoff_ns: str | Unset = UNSET,
+) -> dict[str, Any]:
+
+    params: dict[str, Any] = {}
+
+    params["cutoff_ns"] = cutoff_ns
+
+    params = {k: v for k, v in params.items() if v is not UNSET and v is not None}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/db/v1/transactions/cleanup",
+        "params": params,
+    }
+
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Error | TransactionSessionCleanupResponse | None:
+    if response.status_code == 200:
+        response_200 = TransactionSessionCleanupResponse.from_dict(response.json())
+
+        return response_200
+
+    if response.status_code == 400:
+        response_400 = Error.from_dict(response.json())
+
+        return response_400
+
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Error | TransactionSessionCleanupResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient,
+    cutoff_ns: str | Unset = UNSET,
+) -> Response[Error | TransactionSessionCleanupResponse]:
+    """Clean up expired transaction sessions
+
+    Args:
+        cutoff_ns (str | Unset):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Error | TransactionSessionCleanupResponse]
+    """
+
+    kwargs = _get_kwargs(
+        cutoff_ns=cutoff_ns,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient,
+    cutoff_ns: str | Unset = UNSET,
+) -> Error | TransactionSessionCleanupResponse | None:
+    """Clean up expired transaction sessions
+
+    Args:
+        cutoff_ns (str | Unset):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Error | TransactionSessionCleanupResponse
+    """
+
+    return sync_detailed(
+        client=client,
+        cutoff_ns=cutoff_ns,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+    cutoff_ns: str | Unset = UNSET,
+) -> Response[Error | TransactionSessionCleanupResponse]:
+    """Clean up expired transaction sessions
+
+    Args:
+        cutoff_ns (str | Unset):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Error | TransactionSessionCleanupResponse]
+    """
+
+    kwargs = _get_kwargs(
+        cutoff_ns=cutoff_ns,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient,
+    cutoff_ns: str | Unset = UNSET,
+) -> Error | TransactionSessionCleanupResponse | None:
+    """Clean up expired transaction sessions
+
+    Args:
+        cutoff_ns (str | Unset):
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Error | TransactionSessionCleanupResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            cutoff_ns=cutoff_ns,
+        )
+    ).parsed

antfly/client_generated/api/data_operations/commit_transaction.py

@@ -0,0 +1,289 @@
+from http import HTTPStatus
+from typing import Any
+
+import httpx
+
+from ... import errors
+from ...client import AuthenticatedClient, Client
+from ...models.error import Error
+from ...models.transaction_commit_request import TransactionCommitRequest
+from ...models.transaction_commit_response import TransactionCommitResponse
+from ...types import Response
+
+
+def _get_kwargs(
+    *,
+    body: TransactionCommitRequest,
+) -> dict[str, Any]:
+    headers: dict[str, Any] = {}
+
+    _kwargs: dict[str, Any] = {
+        "method": "post",
+        "url": "/db/v1/transactions/commit",
+    }
+
+    _kwargs["json"] = body.to_dict()
+
+    headers["Content-Type"] = "application/json"
+
+    _kwargs["headers"] = headers
+    return _kwargs
+
+
+def _parse_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Error | TransactionCommitResponse | None:
+    if response.status_code == 200:
+        response_200 = TransactionCommitResponse.from_dict(response.json())
+
+        return response_200
+
+    if response.status_code == 400:
+        response_400 = Error.from_dict(response.json())
+
+        return response_400
+
+    if response.status_code == 404:
+        response_404 = Error.from_dict(response.json())
+
+        return response_404
+
+    if response.status_code == 409:
+        response_409 = TransactionCommitResponse.from_dict(response.json())
+
+        return response_409
+
+    if response.status_code == 500:
+        response_500 = Error.from_dict(response.json())
+
+        return response_500
+
+    if client.raise_on_unexpected_status:
+        raise errors.UnexpectedStatus(response.status_code, response.content)
+    else:
+        return None
+
+
+def _build_response(
+    *, client: AuthenticatedClient | Client, response: httpx.Response
+) -> Response[Error | TransactionCommitResponse]:
+    return Response(
+        status_code=HTTPStatus(response.status_code),
+        content=response.content,
+        headers=response.headers,
+        parsed=_parse_response(client=client, response=response),
+    )
+
+
+def sync_detailed(
+    *,
+    client: AuthenticatedClient,
+    body: TransactionCommitRequest,
+) -> Response[Error | TransactionCommitResponse]:
+    """Commit an OCC transaction
+
+     Commit a stateless OCC (Optimistic Concurrency Control) transaction.
+
+    **Workflow**:
+    1. Read documents using regular lookup endpoints, capturing the
+       `X-Antfly-Version` response header for each read
+    2. Compute writes locally based on the read values
+    3. Submit this commit request with the read set (keys + versions)
+       and the write set (batch operations per table)
+
+    The server validates that all read versions still match current state.
+    If any version has changed, the transaction is aborted with a 409 Conflict
+    response containing details about which key conflicted.
+
+    If all versions match, writes are executed atomically via 2PC.
+
+    Stateless clients manage their own read set. For server-managed
+    read-modify-write workflows, use the transaction session endpoints.
+
+    Args:
+        body (TransactionCommitRequest): Stateless OCC (Optimistic Concurrency Control)
+            transaction commit request.
+
+            The client reads documents (capturing version tokens from the X-Antfly-Version
+            response header on lookups), computes writes locally, then submits everything
+            in this single commit request. The server validates that all read versions
+            still match before executing writes atomically via 2PC.
+
+            **No server-side state**: There is no "begin" endpoint. The client manages
+            its own read set and submits the full transaction in one request.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Error | TransactionCommitResponse]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = client.get_httpx_client().request(
+        **kwargs,
+    )
+
+    return _build_response(client=client, response=response)
+
+
+def sync(
+    *,
+    client: AuthenticatedClient,
+    body: TransactionCommitRequest,
+) -> Error | TransactionCommitResponse | None:
+    """Commit an OCC transaction
+
+     Commit a stateless OCC (Optimistic Concurrency Control) transaction.
+
+    **Workflow**:
+    1. Read documents using regular lookup endpoints, capturing the
+       `X-Antfly-Version` response header for each read
+    2. Compute writes locally based on the read values
+    3. Submit this commit request with the read set (keys + versions)
+       and the write set (batch operations per table)
+
+    The server validates that all read versions still match current state.
+    If any version has changed, the transaction is aborted with a 409 Conflict
+    response containing details about which key conflicted.
+
+    If all versions match, writes are executed atomically via 2PC.
+
+    Stateless clients manage their own read set. For server-managed
+    read-modify-write workflows, use the transaction session endpoints.
+
+    Args:
+        body (TransactionCommitRequest): Stateless OCC (Optimistic Concurrency Control)
+            transaction commit request.
+
+            The client reads documents (capturing version tokens from the X-Antfly-Version
+            response header on lookups), computes writes locally, then submits everything
+            in this single commit request. The server validates that all read versions
+            still match before executing writes atomically via 2PC.
+
+            **No server-side state**: There is no "begin" endpoint. The client manages
+            its own read set and submits the full transaction in one request.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Error | TransactionCommitResponse
+    """
+
+    return sync_detailed(
+        client=client,
+        body=body,
+    ).parsed
+
+
+async def asyncio_detailed(
+    *,
+    client: AuthenticatedClient,
+    body: TransactionCommitRequest,
+) -> Response[Error | TransactionCommitResponse]:
+    """Commit an OCC transaction
+
+     Commit a stateless OCC (Optimistic Concurrency Control) transaction.
+
+    **Workflow**:
+    1. Read documents using regular lookup endpoints, capturing the
+       `X-Antfly-Version` response header for each read
+    2. Compute writes locally based on the read values
+    3. Submit this commit request with the read set (keys + versions)
+       and the write set (batch operations per table)
+
+    The server validates that all read versions still match current state.
+    If any version has changed, the transaction is aborted with a 409 Conflict
+    response containing details about which key conflicted.
+
+    If all versions match, writes are executed atomically via 2PC.
+
+    Stateless clients manage their own read set. For server-managed
+    read-modify-write workflows, use the transaction session endpoints.
+
+    Args:
+        body (TransactionCommitRequest): Stateless OCC (Optimistic Concurrency Control)
+            transaction commit request.
+
+            The client reads documents (capturing version tokens from the X-Antfly-Version
+            response header on lookups), computes writes locally, then submits everything
+            in this single commit request. The server validates that all read versions
+            still match before executing writes atomically via 2PC.
+
+            **No server-side state**: There is no "begin" endpoint. The client manages
+            its own read set and submits the full transaction in one request.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Response[Error | TransactionCommitResponse]
+    """
+
+    kwargs = _get_kwargs(
+        body=body,
+    )
+
+    response = await client.get_async_httpx_client().request(**kwargs)
+
+    return _build_response(client=client, response=response)
+
+
+async def asyncio(
+    *,
+    client: AuthenticatedClient,
+    body: TransactionCommitRequest,
+) -> Error | TransactionCommitResponse | None:
+    """Commit an OCC transaction
+
+     Commit a stateless OCC (Optimistic Concurrency Control) transaction.
+
+    **Workflow**:
+    1. Read documents using regular lookup endpoints, capturing the
+       `X-Antfly-Version` response header for each read
+    2. Compute writes locally based on the read values
+    3. Submit this commit request with the read set (keys + versions)
+       and the write set (batch operations per table)
+
+    The server validates that all read versions still match current state.
+    If any version has changed, the transaction is aborted with a 409 Conflict
+    response containing details about which key conflicted.
+
+    If all versions match, writes are executed atomically via 2PC.
+
+    Stateless clients manage their own read set. For server-managed
+    read-modify-write workflows, use the transaction session endpoints.
+
+    Args:
+        body (TransactionCommitRequest): Stateless OCC (Optimistic Concurrency Control)
+            transaction commit request.
+
+            The client reads documents (capturing version tokens from the X-Antfly-Version
+            response header on lookups), computes writes locally, then submits everything
+            in this single commit request. The server validates that all read versions
+            still match before executing writes atomically via 2PC.
+
+            **No server-side state**: There is no "begin" endpoint. The client manages
+            its own read set and submits the full transaction in one request.
+
+    Raises:
+        errors.UnexpectedStatus: If the server returns an undocumented status code and Client.raise_on_unexpected_status is True.
+        httpx.TimeoutException: If the request takes longer than Client.timeout.
+
+    Returns:
+        Error | TransactionCommitResponse
+    """
+
+    return (
+        await asyncio_detailed(
+            client=client,
+            body=body,
+        )
+    ).parsed