feat: add csv() modifier

supabase · Sep 30, 2023 · 2394306 · 2394306
1 parent 17e9f09
commit 2394306
Show file tree

Hide file tree

Showing 3 changed files with 31 additions and 1 deletion.
diff --git a/postgrest/_async/request_builder.py b/postgrest/_async/request_builder.py
@@ -252,6 +252,18 @@ def text_search(
             session=self.session,  # type: ignore
         )
 
+    def csv(self) -> AsyncSingleRequestBuilder[str]:
+        """Specify that the query must retrieve data as a single CSV string."""
+        self.headers["Accept"] = "text/csv"
+        return AsyncSingleRequestBuilder[str](
+            session=self.session,  # type: ignore
+            path=self.path,
+            http_method=self.http_method,
+            headers=self.headers,
+            params=self.params,
+            json=self.json,
+        )
+
 
 class AsyncRequestBuilder(Generic[_ReturnT]):
     def __init__(self, session: AsyncClient, path: str) -> None:

diff --git a/postgrest/_sync/request_builder.py b/postgrest/_sync/request_builder.py
@@ -252,6 +252,18 @@ def text_search(
             session=self.session,  # type: ignore
         )
 
+    def csv(self) -> SyncSingleRequestBuilder[str]:
+        """Specify that the query must retrieve data as a single CSV string."""
+        self.headers["Accept"] = "text/csv"
+        return SyncSingleRequestBuilder[str](
+            session=self.session,  # type: ignore
+            path=self.path,
+            http_method=self.http_method,
+            headers=self.headers,
+            params=self.params,
+            json=self.json,
+        )
+
 
 class SyncRequestBuilder(Generic[_ReturnT]):
     def __init__(self, session: SyncClient, path: str) -> None:

diff --git a/postgrest/base_request_builder.py b/postgrest/base_request_builder.py
@@ -200,8 +200,14 @@ class SingleAPIResponse(APIResponse[_ReturnT], Generic[_ReturnT]):
     def from_http_request_response(
         cls: Type[Self], request_response: RequestResponse
     ) -> Self:
-        data = request_response.json()
         count = cls._get_count_from_http_request_response(request_response)
+        try:
+            data = request_response.json()
+        except JSONDecodeError:
+            if len(request_response.text) > 0:
+                data = request_response.text
+            else:
+                data = []
         return cls[_ReturnT](data=data, count=count)  # type: ignore
 
     @classmethod