Tomas Valenta
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
from typing import Dict
|
|
|
|
from pydantic import BaseModel, PrivateAttr
|
|
|
|
from postgrest._async.client import AsyncPostgrestClient
|
|
from postgrest._async.request_builder import (
|
|
AsyncRequestBuilder,
|
|
)
|
|
|
|
class Client(BaseModel):
|
|
_supabase_url: str = PrivateAttr()
|
|
_supabase_key: str = PrivateAttr()
|
|
_rest_url: str = PrivateAttr()
|
|
|
|
_client: AsyncPostgrestClient = PrivateAttr()
|
|
|
|
def __init__(self, supabase_url: str, supabase_key: str) -> None:
|
|
self._supabase_url = supabase_url
|
|
self._supabase_key = supabase_key
|
|
|
|
self._rest_url = f"{supabase_url}/rest/v1"
|
|
|
|
# Initialize AsyncPostgrestClient
|
|
self._client = AsyncPostgrestClient(
|
|
base_url=self._rest_url,
|
|
headers=self._get_auth_headers(),
|
|
)
|
|
self._client.auth(token=self._supabase_key)
|
|
|
|
def table(self, name: str) -> AsyncRequestBuilder:
|
|
return self._client.table(name)
|
|
|
|
def _get_auth_headers(self) -> Dict[str, str]:
|
|
return {
|
|
"apiKey": self._supabase_key,
|
|
"Authorization": f"Bearer {self._supabase_key}",
|
|
}
|