meilisearch-python-sdk 2.0.0__py3-none-any.whl

meilisearch_python_sdk/_http_requests.py

@@ -0,0 +1,138 @@
+from __future__ import annotations
+
+from functools import lru_cache
+from typing import Any, Callable
+
+from httpx import (
+    AsyncClient,
+    Client,
+    ConnectError,
+    ConnectTimeout,
+    HTTPError,
+    RemoteProtocolError,
+    Response,
+)
+
+from meilisearch_python_sdk._version import VERSION
+from meilisearch_python_sdk.errors import (
+    MeilisearchApiError,
+    MeilisearchCommunicationError,
+    MeilisearchError,
+)
+
+
+class AsyncHttpRequests:
+    def __init__(self, http_client: AsyncClient) -> None:
+        self.http_client = http_client
+
+    async def _send_request(
+        self,
+        http_method: Callable,
+        path: str,
+        body: Any | None = None,
+        content_type: str = "applicaton/json",
+    ) -> Response:
+        headers = build_headers(content_type)
+        try:
+            if not body:
+                response = await http_method(path)
+            elif content_type == "application/json":
+                response = await http_method(path, json=body, headers=headers)
+            else:
+                response = await http_method(path, content=body, headers=headers)
+
+            response.raise_for_status()
+            return response
+
+        except (ConnectError, ConnectTimeout, RemoteProtocolError) as err:
+            raise MeilisearchCommunicationError(str(err)) from err
+        except HTTPError as err:
+            if "response" in locals():
+                raise MeilisearchApiError(str(err), response) from err
+            else:
+                # Fail safe just in case error happens before response is created
+                raise MeilisearchError(str(err))  # pragma: no cover
+
+    async def get(self, path: str) -> Response:
+        return await self._send_request(self.http_client.get, path)
+
+    async def patch(
+        self, path: str, body: Any | None = None, content_type: str = "application/json"
+    ) -> Response:
+        return await self._send_request(self.http_client.patch, path, body, content_type)
+
+    async def post(
+        self, path: str, body: Any | None = None, content_type: str = "application/json"
+    ) -> Response:
+        return await self._send_request(self.http_client.post, path, body, content_type)
+
+    async def put(
+        self, path: str, body: Any | None = None, content_type: str = "application/json"
+    ) -> Response:
+        return await self._send_request(self.http_client.put, path, body, content_type)
+
+    async def delete(self, path: str, body: dict | None = None) -> Response:
+        return await self._send_request(self.http_client.delete, path, body)
+
+
+class HttpRequests:
+    def __init__(self, http_client: Client) -> None:
+        self.http_client = http_client
+
+    def _send_request(
+        self,
+        http_method: Callable,
+        path: str,
+        body: Any | None = None,
+        content_type: str = "applicaton/json",
+    ) -> Response:
+        headers = build_headers(content_type)
+        try:
+            if not body:
+                response = http_method(path)
+            elif content_type == "application/json":
+                response = http_method(path, json=body, headers=headers)
+            else:
+                response = http_method(path, content=body, headers=headers)
+
+            response.raise_for_status()
+            return response
+
+        except (ConnectError, ConnectTimeout, RemoteProtocolError) as err:
+            raise MeilisearchCommunicationError(str(err)) from err
+        except HTTPError as err:
+            if "response" in locals():
+                raise MeilisearchApiError(str(err), response) from err
+            else:
+                # Fail safe just in case error happens before response is created
+                raise MeilisearchError(str(err))  # pragma: no cover
+
+    def get(self, path: str) -> Response:
+        return self._send_request(self.http_client.get, path)
+
+    def patch(
+        self, path: str, body: Any | None = None, content_type: str = "application/json"
+    ) -> Response:
+        return self._send_request(self.http_client.patch, path, body, content_type)
+
+    def post(
+        self, path: str, body: Any | None = None, content_type: str = "application/json"
+    ) -> Response:
+        return self._send_request(self.http_client.post, path, body, content_type)
+
+    def put(
+        self, path: str, body: Any | None = None, content_type: str = "application/json"
+    ) -> Response:
+        return self._send_request(self.http_client.put, path, body, content_type)
+
+    def delete(self, path: str, body: dict | None = None) -> Response:
+        return self._send_request(self.http_client.delete, path, body)
+
+
+def build_headers(content_type: str) -> dict[str, str]:
+    return {"user-agent": user_agent(), "Content-Type": content_type}
+
+
+@lru_cache(maxsize=1)
+def user_agent() -> str:
+    return f"Meilisearch Python Async (v{VERSION})"