mirror of
https://github.com/StanGirard/quivr.git
synced 2024-12-15 13:42:08 +03:00
8af6d61e76
* reorganize import level * add __init__, reorganize import from __init__ * reorganize import level * reorganize import level * fix circular import error by keep the import deep as "from models.settings" * fix the relative import * restor unwanted staged files * add backend/venv and backend/.env to gitignore * clean importing
85 lines
3.0 KiB
Python
85 lines
3.0 KiB
Python
from logger import get_logger
|
|
from models import BrainSubscription, get_supabase_client
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class SubscriptionInvitationService:
|
|
def __init__(self):
|
|
self.supabase_client = get_supabase_client()
|
|
|
|
def create_subscription_invitation(self, brain_subscription: BrainSubscription):
|
|
logger.info("Creating subscription invitation")
|
|
response = (
|
|
self.supabase_client.table("brain_subscription_invitations")
|
|
.insert(
|
|
{
|
|
"brain_id": str(brain_subscription.brain_id),
|
|
"email": brain_subscription.email,
|
|
"rights": brain_subscription.rights,
|
|
}
|
|
)
|
|
.execute()
|
|
)
|
|
return response.data
|
|
|
|
def update_subscription_invitation(self, brain_subscription: BrainSubscription):
|
|
logger.info("Updating subscription invitation")
|
|
response = (
|
|
self.supabase_client.table("brain_subscription_invitations")
|
|
.update({"rights": brain_subscription.rights})
|
|
.eq("brain_id", str(brain_subscription.brain_id))
|
|
.eq("email", brain_subscription.email)
|
|
.execute()
|
|
)
|
|
return response.data
|
|
|
|
def create_or_update_subscription_invitation(
|
|
self, brain_subscription: BrainSubscription
|
|
):
|
|
response = (
|
|
self.supabase_client.table("brain_subscription_invitations")
|
|
.select("*")
|
|
.eq("brain_id", str(brain_subscription.brain_id))
|
|
.eq("email", brain_subscription.email)
|
|
.execute()
|
|
)
|
|
|
|
if response.data:
|
|
response = self.update_subscription_invitation(brain_subscription)
|
|
else:
|
|
response = self.create_subscription_invitation(brain_subscription)
|
|
|
|
return response
|
|
|
|
def fetch_invitation(self, subscription: BrainSubscription):
|
|
logger.info("Fetching subscription invitation")
|
|
response = (
|
|
self.supabase_client.table("brain_subscription_invitations")
|
|
.select("*")
|
|
.eq("brain_id", str(subscription.brain_id))
|
|
.eq("email", subscription.email)
|
|
.execute()
|
|
)
|
|
if response.data:
|
|
return response.data[0] # return the first matching invitation
|
|
else:
|
|
return None
|
|
|
|
def remove_invitation(self, subscription: BrainSubscription):
|
|
logger.info(
|
|
f"Removing subscription invitation for email {subscription.email} and brain {subscription.brain_id}"
|
|
)
|
|
response = (
|
|
self.supabase_client.table("brain_subscription_invitations")
|
|
.delete()
|
|
.eq("brain_id", str(subscription.brain_id))
|
|
.eq("email", subscription.email)
|
|
.execute()
|
|
)
|
|
logger.info(
|
|
f"Removed subscription invitation for email {subscription.email} and brain {subscription.brain_id}"
|
|
)
|
|
logger.info(response)
|
|
return response.data
|