feat: add schema method to the client (#809)

Author: silentworks

supabase/_async/client.py

@@ -74,7 +74,6 @@ def __init__(
         self.auth_url = f"{supabase_url}/auth/v1"
         self.storage_url = f"{supabase_url}/storage/v1"
         self.functions_url = f"{supabase_url}/functions/v1"
-        self.schema = options.schema
 
         # Instantiate clients.
         self.auth = self._init_supabase_auth_client(
@@ -110,6 +109,19 @@ def table(self, table_name: str) -> AsyncRequestBuilder:
         """
         return self.from_(table_name)
 
+    def schema(self, schema: str) -> AsyncPostgrestClient:
+        """Select a schema to query or perform an function (rpc) call.
+
+        The schema needs to be on the list of exposed schemas inside Supabase.
+        """
+        self._postgrest = self._init_postgrest_client(
+            rest_url=self.rest_url,
+            headers=self.options.headers,
+            schema=schema,
+            timeout=self.options.postgrest_client_timeout,
+        )
+        return self._postgrest
+
     def from_(self, table_name: str) -> AsyncRequestBuilder:
         """Perform a table operation.
 

supabase/_sync/client.py

@@ -74,7 +74,6 @@ def __init__(
         self.auth_url = f"{supabase_url}/auth/v1"
         self.storage_url = f"{supabase_url}/storage/v1"
         self.functions_url = f"{supabase_url}/functions/v1"
-        self.schema = options.schema
 
         # Instantiate clients.
         self.auth = self._init_supabase_auth_client(
@@ -110,6 +109,19 @@ def table(self, table_name: str) -> SyncRequestBuilder:
         """
         return self.from_(table_name)
 
+    def schema(self, schema: str) -> SyncPostgrestClient:
+        """Select a schema to query or perform an function (rpc) call.
+
+        The schema needs to be on the list of exposed schemas inside Supabase.
+        """
+        self._postgrest = self._init_postgrest_client(
+            rest_url=self.rest_url,
+            headers=self.options.headers,
+            schema=schema,
+            timeout=self.options.postgrest_client_timeout,
+        )
+        return self._postgrest
+
     def from_(self, table_name: str) -> SyncRequestBuilder:
         """Perform a table operation.