lambdadb 0.1.2__py3-none-any.whl

lambdadb/errors/__init__.py

@@ -0,0 +1,74 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from typing import TYPE_CHECKING
+from importlib import import_module
+
+if TYPE_CHECKING:
+    from .apierror import APIError
+    from .badrequest_error import BadRequestError, BadRequestErrorData
+    from .internalservererror import InternalServerError, InternalServerErrorData
+    from .resourcealreadyexists_error import (
+        ResourceAlreadyExistsError,
+        ResourceAlreadyExistsErrorData,
+    )
+    from .resourcenotfound_error import ResourceNotFoundError, ResourceNotFoundErrorData
+    from .toomanyrequests_error import TooManyRequestsError, TooManyRequestsErrorData
+    from .unauthenticated_error import UnauthenticatedError, UnauthenticatedErrorData
+
+__all__ = [
+    "APIError",
+    "BadRequestError",
+    "BadRequestErrorData",
+    "InternalServerError",
+    "InternalServerErrorData",
+    "ResourceAlreadyExistsError",
+    "ResourceAlreadyExistsErrorData",
+    "ResourceNotFoundError",
+    "ResourceNotFoundErrorData",
+    "TooManyRequestsError",
+    "TooManyRequestsErrorData",
+    "UnauthenticatedError",
+    "UnauthenticatedErrorData",
+]
+
+_dynamic_imports: dict[str, str] = {
+    "APIError": ".apierror",
+    "BadRequestError": ".badrequest_error",
+    "BadRequestErrorData": ".badrequest_error",
+    "InternalServerError": ".internalservererror",
+    "InternalServerErrorData": ".internalservererror",
+    "ResourceAlreadyExistsError": ".resourcealreadyexists_error",
+    "ResourceAlreadyExistsErrorData": ".resourcealreadyexists_error",
+    "ResourceNotFoundError": ".resourcenotfound_error",
+    "ResourceNotFoundErrorData": ".resourcenotfound_error",
+    "TooManyRequestsError": ".toomanyrequests_error",
+    "TooManyRequestsErrorData": ".toomanyrequests_error",
+    "UnauthenticatedError": ".unauthenticated_error",
+    "UnauthenticatedErrorData": ".unauthenticated_error",
+}
+
+
+def __getattr__(attr_name: str) -> object:
+    module_name = _dynamic_imports.get(attr_name)
+    if module_name is None:
+        raise AttributeError(
+            f"No {attr_name} found in _dynamic_imports for module name -> {__name__} "
+        )
+
+    try:
+        module = import_module(module_name, __package__)
+        result = getattr(module, attr_name)
+        return result
+    except ImportError as e:
+        raise ImportError(
+            f"Failed to import {attr_name} from {module_name}: {e}"
+        ) from e
+    except AttributeError as e:
+        raise AttributeError(
+            f"Failed to get {attr_name} from {module_name}: {e}"
+        ) from e
+
+
+def __dir__():
+    lazy_attrs = list(_dynamic_imports.keys())
+    return sorted(lazy_attrs)

lambdadb/errors/apierror.py

@@ -0,0 +1,22 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from dataclasses import dataclass
+from typing import Optional
+import httpx
+
+
+@dataclass
+class APIError(Exception):
+    """Represents an error returned by the API."""
+
+    message: str
+    status_code: int = -1
+    body: str = ""
+    raw_response: Optional[httpx.Response] = None
+
+    def __str__(self):
+        body = ""
+        if len(self.body) > 0:
+            body = f"\n{self.body}"
+
+        return f"{self.message}: Status {self.status_code}{body}"

lambdadb/errors/badrequest_error.py

@@ -0,0 +1,20 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from lambdadb import utils
+from lambdadb.types import BaseModel
+from typing import Optional
+
+
+class BadRequestErrorData(BaseModel):
+    message: Optional[str] = None
+
+
+class BadRequestError(Exception):
+    data: BadRequestErrorData
+
+    def __init__(self, data: BadRequestErrorData):
+        self.data = data
+
+    def __str__(self) -> str:
+        return utils.marshal_json(self.data, BadRequestErrorData)

lambdadb/errors/internalservererror.py

@@ -0,0 +1,20 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from lambdadb import utils
+from lambdadb.types import BaseModel
+from typing import Optional
+
+
+class InternalServerErrorData(BaseModel):
+    message: Optional[str] = None
+
+
+class InternalServerError(Exception):
+    data: InternalServerErrorData
+
+    def __init__(self, data: InternalServerErrorData):
+        self.data = data
+
+    def __str__(self) -> str:
+        return utils.marshal_json(self.data, InternalServerErrorData)

lambdadb/errors/resourcealreadyexists_error.py

@@ -0,0 +1,20 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from lambdadb import utils
+from lambdadb.types import BaseModel
+from typing import Optional
+
+
+class ResourceAlreadyExistsErrorData(BaseModel):
+    message: Optional[str] = None
+
+
+class ResourceAlreadyExistsError(Exception):
+    data: ResourceAlreadyExistsErrorData
+
+    def __init__(self, data: ResourceAlreadyExistsErrorData):
+        self.data = data
+
+    def __str__(self) -> str:
+        return utils.marshal_json(self.data, ResourceAlreadyExistsErrorData)

lambdadb/errors/resourcenotfound_error.py

@@ -0,0 +1,20 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from lambdadb import utils
+from lambdadb.types import BaseModel
+from typing import Optional
+
+
+class ResourceNotFoundErrorData(BaseModel):
+    message: Optional[str] = None
+
+
+class ResourceNotFoundError(Exception):
+    data: ResourceNotFoundErrorData
+
+    def __init__(self, data: ResourceNotFoundErrorData):
+        self.data = data
+
+    def __str__(self) -> str:
+        return utils.marshal_json(self.data, ResourceNotFoundErrorData)

lambdadb/errors/toomanyrequests_error.py

@@ -0,0 +1,20 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from lambdadb import utils
+from lambdadb.types import BaseModel
+from typing import Optional
+
+
+class TooManyRequestsErrorData(BaseModel):
+    message: Optional[str] = None
+
+
+class TooManyRequestsError(Exception):
+    data: TooManyRequestsErrorData
+
+    def __init__(self, data: TooManyRequestsErrorData):
+        self.data = data
+
+    def __str__(self) -> str:
+        return utils.marshal_json(self.data, TooManyRequestsErrorData)

lambdadb/errors/unauthenticated_error.py

@@ -0,0 +1,20 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from lambdadb import utils
+from lambdadb.types import BaseModel
+from typing import Optional
+
+
+class UnauthenticatedErrorData(BaseModel):
+    message: Optional[str] = None
+
+
+class UnauthenticatedError(Exception):
+    data: UnauthenticatedErrorData
+
+    def __init__(self, data: UnauthenticatedErrorData):
+        self.data = data
+
+    def __str__(self) -> str:
+        return utils.marshal_json(self.data, UnauthenticatedErrorData)

lambdadb/httpclient.py

@@ -0,0 +1,126 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+# pyright: reportReturnType = false
+import asyncio
+from typing_extensions import Protocol, runtime_checkable
+import httpx
+from typing import Any, Optional, Union
+
+
+@runtime_checkable
+class HttpClient(Protocol):
+    def send(
+        self,
+        request: httpx.Request,
+        *,
+        stream: bool = False,
+        auth: Union[
+            httpx._types.AuthTypes, httpx._client.UseClientDefault, None
+        ] = httpx.USE_CLIENT_DEFAULT,
+        follow_redirects: Union[
+            bool, httpx._client.UseClientDefault
+        ] = httpx.USE_CLIENT_DEFAULT,
+    ) -> httpx.Response:
+        pass
+
+    def build_request(
+        self,
+        method: str,
+        url: httpx._types.URLTypes,
+        *,
+        content: Optional[httpx._types.RequestContent] = None,
+        data: Optional[httpx._types.RequestData] = None,
+        files: Optional[httpx._types.RequestFiles] = None,
+        json: Optional[Any] = None,
+        params: Optional[httpx._types.QueryParamTypes] = None,
+        headers: Optional[httpx._types.HeaderTypes] = None,
+        cookies: Optional[httpx._types.CookieTypes] = None,
+        timeout: Union[
+            httpx._types.TimeoutTypes, httpx._client.UseClientDefault
+        ] = httpx.USE_CLIENT_DEFAULT,
+        extensions: Optional[httpx._types.RequestExtensions] = None,
+    ) -> httpx.Request:
+        pass
+
+    def close(self) -> None:
+        pass
+
+
+@runtime_checkable
+class AsyncHttpClient(Protocol):
+    async def send(
+        self,
+        request: httpx.Request,
+        *,
+        stream: bool = False,
+        auth: Union[
+            httpx._types.AuthTypes, httpx._client.UseClientDefault, None
+        ] = httpx.USE_CLIENT_DEFAULT,
+        follow_redirects: Union[
+            bool, httpx._client.UseClientDefault
+        ] = httpx.USE_CLIENT_DEFAULT,
+    ) -> httpx.Response:
+        pass
+
+    def build_request(
+        self,
+        method: str,
+        url: httpx._types.URLTypes,
+        *,
+        content: Optional[httpx._types.RequestContent] = None,
+        data: Optional[httpx._types.RequestData] = None,
+        files: Optional[httpx._types.RequestFiles] = None,
+        json: Optional[Any] = None,
+        params: Optional[httpx._types.QueryParamTypes] = None,
+        headers: Optional[httpx._types.HeaderTypes] = None,
+        cookies: Optional[httpx._types.CookieTypes] = None,
+        timeout: Union[
+            httpx._types.TimeoutTypes, httpx._client.UseClientDefault
+        ] = httpx.USE_CLIENT_DEFAULT,
+        extensions: Optional[httpx._types.RequestExtensions] = None,
+    ) -> httpx.Request:
+        pass
+
+    async def aclose(self) -> None:
+        pass
+
+
+class ClientOwner(Protocol):
+    client: Union[HttpClient, None]
+    async_client: Union[AsyncHttpClient, None]
+
+
+def close_clients(
+    owner: ClientOwner,
+    sync_client: Union[HttpClient, None],
+    sync_client_supplied: bool,
+    async_client: Union[AsyncHttpClient, None],
+    async_client_supplied: bool,
+) -> None:
+    """
+    A finalizer function that is meant to be used with weakref.finalize to close
+    httpx clients used by an SDK so that underlying resources can be garbage
+    collected.
+    """
+
+    # Unset the client/async_client properties so there are no more references
+    # to them from the owning SDK instance and they can be reaped.
+    owner.client = None
+    owner.async_client = None
+
+    if sync_client is not None and not sync_client_supplied:
+        try:
+            sync_client.close()
+        except Exception:
+            pass
+
+    if async_client is not None and not async_client_supplied:
+        try:
+            loop = asyncio.get_running_loop()
+            asyncio.run_coroutine_threadsafe(async_client.aclose(), loop)
+        except RuntimeError:
+            try:
+                asyncio.run(async_client.aclose())
+            except RuntimeError:
+                # best effort
+                pass