Skip to content

Commit

Permalink
fix: usertype system user get or create
Browse files Browse the repository at this point in the history
  • Loading branch information
stevegerrits committed Nov 25, 2024
1 parent f4de895 commit 27e2d25
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions vespadb/users/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,12 @@
from vespadb.users.models import UserType, VespaUser


def get_system_user(user_type: Literal[UserType.SYNC, UserType.IMPORT]) -> VespaUser:
"""Get System Users based on usertype."""
def get_system_user(user_type: Literal[UserType.SYNC]) -> VespaUser:
"""Get the system user specifically for SYNC."""
if user_type != UserType.SYNC:
raise ValueError("This function only supports UserType.SYNC.")

system_user, _ = VespaUser.objects.get_or_create(
username=user_type.value, first_name="vespadb", defaults={"user_type": user_type.value}
username="sync", defaults={"user_type": UserType.SYNC.value}
)
return cast(VespaUser, system_user) # make mypy happy

0 comments on commit 27e2d25

Please sign in to comment.