mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-15 01:21:48 +03:00
101 lines
2.9 KiB
Python
101 lines
2.9 KiB
Python
import os
|
|
from typing import Optional
|
|
from uuid import UUID
|
|
|
|
import resend
|
|
from logger import get_logger
|
|
from models.settings import CommonsDep, common_dependencies
|
|
from pydantic import BaseModel
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
class BrainSubscription(BaseModel):
|
|
brain_id: Optional[UUID] = None
|
|
inviter_email: Optional[str]
|
|
email: Optional[str]
|
|
rights: Optional[str]
|
|
|
|
class Config:
|
|
arbitrary_types_allowed = True
|
|
|
|
@property
|
|
def commons(self) -> CommonsDep:
|
|
return common_dependencies()
|
|
|
|
def create_subscription_invitation(self):
|
|
logger.info("Creating subscription invitation")
|
|
response = (
|
|
self.commons["supabase"]
|
|
.table("brain_subscription_invitations")
|
|
.insert(
|
|
{
|
|
"brain_id": str(self.brain_id),
|
|
"email": self.email,
|
|
"rights": self.rights,
|
|
}
|
|
)
|
|
.execute()
|
|
)
|
|
return response.data
|
|
|
|
def update_subscription_invitation(self):
|
|
logger.info("Updating subscription invitation")
|
|
response = (
|
|
self.commons["supabase"]
|
|
.table("brain_subscription_invitations")
|
|
.update({"rights": self.rights})
|
|
.eq("brain_id", str(self.brain_id))
|
|
.eq("email", self.email)
|
|
.execute()
|
|
)
|
|
return response.data
|
|
|
|
def create_or_update_subscription_invitation(self):
|
|
response = (
|
|
self.commons["supabase"]
|
|
.table("brain_subscription_invitations")
|
|
.select("*")
|
|
.eq("brain_id", str(self.brain_id))
|
|
.eq("email", self.email)
|
|
.execute()
|
|
)
|
|
|
|
if response.data:
|
|
response = self.update_subscription_invitation()
|
|
else:
|
|
response = self.create_subscription_invitation()
|
|
|
|
return response
|
|
|
|
def get_brain_url(self) -> str:
|
|
"""Generates the brain URL based on the brain_id."""
|
|
base_url = "https://www.quivr.app/chat"
|
|
return f"{base_url}?brain_subscription_invitation={self.brain_id}"
|
|
|
|
def resend_invitation_email(self):
|
|
resend.api_key = os.getenv("RESEND_API_KEY")
|
|
|
|
brain_url = self.get_brain_url()
|
|
|
|
html_body = f"""
|
|
<p>This brain has been shared with you by {self.inviter_email}.</p>
|
|
<p><a href='{brain_url}'>Click here</a> to access your brain.</p>
|
|
"""
|
|
|
|
try:
|
|
r = resend.Emails.send(
|
|
{
|
|
"from": "onboarding@resend.dev",
|
|
"to": self.email,
|
|
"subject": "Quivr - Brain Shared With You",
|
|
"html": html_body,
|
|
}
|
|
)
|
|
print("Resend response", r)
|
|
except Exception as e:
|
|
logger.error(f"Error sending email: {e}")
|
|
return
|
|
|
|
return r
|