21 lines
639 B
Python
21 lines
639 B
Python
from collections.abc import Iterable
|
|
|
|
from domain.user import User
|
|
from usecase.interface import UserRepository
|
|
|
|
|
|
class InMemoryUserRepository(UserRepository):
|
|
def __init__(self, users: Iterable[User] | None = None) -> None:
|
|
self._users = {user.id: user for user in users or ()}
|
|
|
|
def get(self, user_id: str) -> User | None:
|
|
return self._users.get(user_id)
|
|
|
|
def get_by_email(self, email: str) -> User | None:
|
|
for user in self._users.values():
|
|
if user.email == email:
|
|
return user
|
|
return None
|
|
|
|
def save(self, user: User) -> None:
|
|
self._users[user.id] = user
|