feat: adding converter from dict to User entity for user repository
This commit is contained in:
parent
d63bf3e713
commit
7f5479188e
|
@ -11,18 +11,18 @@ class UserRepository:
|
|||
def get_all_users(self) -> list[UserEntity]:
|
||||
"""Mengambil semua user dari database."""
|
||||
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]:
|
||||
"""Mendapatkan user berdasarkan email."""
|
||||
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]:
|
||||
"""Mendapatkan user berdasarkan ID."""
|
||||
object_id = ObjectId(user_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]:
|
||||
user_data = self.collection.find_one({"google_id": google_id})
|
||||
|
|
Loading…
Reference in New Issue