Allow setting headers in PostgrestClient's constructor (#10)

Author: lqmanh

CHANGELOG.md

@@ -1,5 +1,11 @@
 ## CHANGELOG
 
+### _Unreleased_
+
+#### Added
+
+- Allow setting headers in `PostgrestClient`'s constructor
+
 ### v0.4.0
 
 #### Added

postgrest_py/__init__.py

@@ -1,2 +1,2 @@
 from postgrest_py.__version__ import __version__
-from postgrest_py.client import Client, PostgrestClient
+from postgrest_py.client import DEFAULT_POSTGREST_CLIENT_HEADERS, Client, PostgrestClient

postgrest_py/client.py

@@ -1,19 +1,29 @@
-from typing import Union
+from typing import Dict, Union
 
 from deprecation import deprecated
 from httpx import AsyncClient, BasicAuth, Response
 
 from postgrest_py.__version__ import __version__
 from postgrest_py.request_builder import RequestBuilder
 
+DEFAULT_POSTGREST_CLIENT_HEADERS: Dict[str, str] = {
+    "Accept": "application/json",
+    "Content-Type": "application/json",
+}
+
 
 class PostgrestClient:
     """PostgREST client."""
 
-    def __init__(self, base_url: str, *, schema="public") -> None:
+    def __init__(
+        self,
+        base_url: str,
+        *,
+        schema: str = "public",
+        headers: Dict[str, str] = DEFAULT_POSTGREST_CLIENT_HEADERS,
+    ) -> None:
         headers = {
-            "Accept": "application/json",
-            "Content-Type": "application/json",
+            **headers,
             "Accept-Profile": schema,
             "Content-Profile": schema,
         }
@@ -32,7 +42,7 @@ def auth(
         self,
         token: str,
         *,
-        username: Union[str, bytes] = None,
+        username: Union[str, bytes, None] = None,
         password: Union[str, bytes] = "",
     ):
         """Authenticate the client with either bearer token or basic authentication."""
@@ -44,9 +54,7 @@ def auth(
 
     def schema(self, schema: str):
         """Switch to another schema."""
-        self.session.headers.update(
-            {"Accept-Profile": schema, "Content-Profile": schema}
-        )
+        self.session.headers.update({"Accept-Profile": schema, "Content-Profile": schema})
         return self
 
     def from_(self, table: str) -> RequestBuilder: