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

Only create KhojUser on login if doesn't already exist

Browse source
This commit is contained in:
Debanjum Singh Solanky 2023-11-14 18:02:31 -08:00
parent 3d8d6145f2
commit f8e5e118e1
1 changed files with 7 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
src/database/adapters/__init__.py
View file

@ -75,13 +75,16 @@ async def delete_khoj_token(user: KhojUser, token: str):
async def get_or_create_user(token: dict) -> KhojUser: async def get_or_create_user(token: dict) -> KhojUser:
user = await get_user_by_token(token) user = await get_user_by_token(token)
if not user: if not user:
user = await create_google_user(token) user = await create_user_by_token(token)
return user return user
async def create_google_user(token: dict) -> KhojUser: async def create_user_by_token(token: dict) -> KhojUser:
user = await KhojUser.objects.filter(email=token.get("email")).aget_or_none()
if not user:
user = await KhojUser.objects.acreate(username=token.get("email"), email=token.get("email")) user = await KhojUser.objects.acreate(username=token.get("email"), email=token.get("email"))
await user.asave() await user.asave()
await GoogleUser.objects.acreate( await GoogleUser.objects.acreate(
sub=token.get("sub"), sub=token.get("sub"),
azp=token.get("azp"), azp=token.get("azp"),