mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-15 09:32:22 +03:00
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
from models.databases.repository import Repository
|
||
|
|
||
|
from logger import get_logger
|
||
|
|
||
|
logger = get_logger(__name__)
|
||
|
|
||
|
|
||
|
class BrainSubscription(Repository):
|
||
|
def __init__(self, supabase_client):
|
||
|
self.db = supabase_client
|
||
|
|
||
|
def create_subscription_invitation(self, brain_id, user_email, rights):
|
||
|
logger.info("Creating subscription invitation")
|
||
|
response = (
|
||
|
self.db.table("brain_subscription_invitations")
|
||
|
.insert({"brain_id": str(brain_id), "email": user_email, "rights": rights})
|
||
|
.execute()
|
||
|
)
|
||
|
return response.data
|
||
|
|
||
|
def update_subscription_invitation(self, brain_id, user_email, rights):
|
||
|
logger.info("Updating subscription invitation")
|
||
|
response = (
|
||
|
self.db.table("brain_subscription_invitations")
|
||
|
.update({"rights": rights})
|
||
|
.eq("brain_id", str(brain_id))
|
||
|
.eq("email", user_email)
|
||
|
.execute()
|
||
|
)
|
||
|
return response.data
|
||
|
|
||
|
def get_subscription_invitations_by_brain_id_and_email(self, brain_id, user_email):
|
||
|
response = (
|
||
|
self.db.table("brain_subscription_invitations")
|
||
|
.select("*")
|
||
|
.eq("brain_id", str(brain_id))
|
||
|
.eq("email", user_email)
|
||
|
.execute()
|
||
|
)
|
||
|
|
||
|
return response
|