klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-04 05:39:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
Download Patch File Download Diff File Expand all files Collapse all files

11
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:
user = await get_user_by_token(token)
if not user:
user = await create_google_user(token)
user = await create_user_by_token(token)
return user
async def create_google_user(token: dict) -> KhojUser:
user = await KhojUser.objects.acreate(username=token.get("email"), email=token.get("email"))
await user.asave()
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"))
await user.asave()
await GoogleUser.objects.acreate(
sub=token.get("sub"),
azp=token.get("azp"),