trying to get postgrest working

Author: fedden

Diff for: pyproject.toml

@@ -9,7 +9,6 @@ license = "MIT"
 python = "^3.7.1"
 postgrest-py = "0.4.0"
 realtime-py="^0.1.2"
-realtime-py="^0.1.0"
 gotrue="0.2.0"
 pytest="6.2.2"
 

Diff for: supabase_py/client.py

@@ -60,7 +60,8 @@ def __init__(
         #  )
         self.realtime = None
         self.postgrest: PostgrestClient = self._init_postgrest_client(
-            rest_url=self.rest_url
+            rest_url=self.rest_url,
+            supabase_key=supabase_key,
         )
 
     def table(self, table_name: str) -> SupabaseQueryBuilder:
@@ -161,9 +162,11 @@ def _init_supabase_auth_client(
         )
 
     @staticmethod
-    def _init_postgrest_client(rest_url: str) -> PostgrestClient:
+    def _init_postgrest_client(rest_url: str, supabase_key: str) -> PostgrestClient:
         """Private helper for creating an instance of the Postgrest client."""
-        return PostgrestClient(rest_url)
+        client = PostgrestClient(rest_url)
+        client.auth(token=supabase_key)
+        return client
 
     def _get_auth_headers(self) -> Dict[str, str]:
         """Helper method to get auth headers."""

Diff for: supabase_py/lib/query_builder.py

@@ -1,4 +1,6 @@
+from httpx import AsyncClient
 from postgrest_py.client import PostgrestClient
+
 from .realtime_client import SupabaseRealtimeClient
 
 
@@ -23,7 +25,15 @@ def __init__(self, url, headers, schema, realtime, table):
         -------
         None
         """
-        super().__init__(url)
+        super().__init__(base_url=url)
+        headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+            "Accept-Profile": schema,
+            "Content-Profile": schema,
+            **headers,
+        }
+        self.session = AsyncClient(base_url=url, headers=headers)
         # self._subscription = SupabaseRealtimeClient(realtime, schema, table)
         # self._realtime = realtime
 

Diff for: tests/test_client.py

@@ -56,7 +56,8 @@ def test_client_auth():
     _assert_authenticated_user(user)
 
 
-def test_client_select():
+@pytest.mark.asyncio
+async def test_client_select():
     """Ensure we can select data from a table."""
     from supabase_py import create_client, Client
 
@@ -65,7 +66,8 @@ def test_client_select():
     supabase: Client = create_client(url, key)
     # TODO(fedden): Add this set back in (and expand on it) when postgrest and
     #               realtime libs are working.
+    query = supabase.table("countries").select("*")
     import ipdb
 
     ipdb.set_trace()
-    data = supabase.table("countries").select("*")
+    data = await query.execute()