sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

If a user subscribes to Khoj with an email address that's not present in the DB, create an account

Browse source
This commit is contained in:
sabaimran 2023-12-03 17:28:40 -05:00
parent c5d297a9ed
commit 21f8d63e89
1 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
src/khoj/database/adapters/__init__.py
View file

@ -41,6 +41,7 @@ from khoj.search_filter.word_filter import WordFilter
from khoj.utils import state
from khoj.utils.config import GPT4AllProcessorModel
from khoj.utils.helpers import generate_random_name
from khoj.database.adapters import get_or_create_user_by_email
class SubscriptionState(Enum):
@ -85,6 +86,15 @@ async def get_or_create_user(token: dict) -> KhojUser:
return user
async def get_or_create_user_by_email(email: str) -> KhojUser:
user, _ = await KhojUser.objects.filter(email=email).aupdate_or_create(defaults={"username": email, "email": email})
await user.asave()
await Subscription.objects.acreate(user=user, type="trial")
return user
async def create_user_by_google_token(token: dict) -> KhojUser:
user, _ = await KhojUser.objects.filter(email=token.get("email")).aupdate_or_create(
defaults={"username": token.get("email"), "email": token.get("email")}
@ -115,7 +125,10 @@ def get_user_subscription(email: str) -> Optional[Subscription]:
async def set_user_subscription(
email: str, is_recurring=None, renewal_date=None, type="standard"
) -> Optional[Subscription]:
user_subscription = await Subscription.objects.filter(user__email=email).afirst()
# Get or create the user object
user = await get_or_create_user_by_email(email)
user_subscription = await Subscription.objects.filter(user=user).afirst()
if not user_subscription:
user = await get_user_by_email(email)
if not user: