akhdanre
/
TIF_E41211115_Genso_quiz_backend
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: adding converter from dict to User entity for user repository

This commit is contained in:
akhdanre 2025-03-19 13:53:11 +07:00
parent d63bf3e713
commit 7f5479188e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/repositories/user_repository.py
View File

@ -11,18 +11,18 @@ class UserRepository:
def get_all_users(self) -> list[UserEntity]: def get_all_users(self) -> list[UserEntity]:
"""Mengambil semua user dari database.""" """Mengambil semua user dari database."""
users = list(self.collection.find({}, {"_id": 0})) users = list(self.collection.find({}, {"_id": 0}))
return users if users else [] return [UserEntity(**user) for user in users]
def get_user_by_email(self, email: str) -> Optional[UserEntity]: def get_user_by_email(self, email: str) -> Optional[UserEntity]:
"""Mendapatkan user berdasarkan email.""" """Mendapatkan user berdasarkan email."""
user = self.collection.find_one({"email": email}, {"_id": 0}) user = self.collection.find_one({"email": email}, {"_id": 0})
return user return UserEntity(**user) if user else None
def get_user_by_id(self, user_id: str) -> Optional[UserEntity]: def get_user_by_id(self, user_id: str) -> Optional[UserEntity]:
"""Mendapatkan user berdasarkan ID.""" """Mendapatkan user berdasarkan ID."""
object_id = ObjectId(user_id) object_id = ObjectId(user_id)
user = self.collection.find_one({"_id": object_id}) user = self.collection.find_one({"_id": object_id})
return user return UserEntity(**user) if user else None
def get_by_google_id(self, google_id: str) -> Optional[UserEntity]: def get_by_google_id(self, google_id: str) -> Optional[UserEntity]:
user_data = self.collection.find_one({"google_id": google_id}) user_data = self.collection.find_one({"google_id": google_id})