27 lines
813 B
Python
27 lines
813 B
Python
from collections.abc import Iterable
|
|
|
|
from domain.user import User
|
|
from usecase.interface import Tracer, UserRepository
|
|
|
|
|
|
class InMemoryUserRepository(UserRepository):
|
|
def __init__(
|
|
self,
|
|
tracer: Tracer,
|
|
users: Iterable[User] | None = None,
|
|
) -> None:
|
|
self._users = {user.id: user for user in users or ()}
|
|
self._tracer = tracer
|
|
|
|
def get(self, user_id: str) -> User | None:
|
|
with self._tracer.start_span('repository.user', attrs={'user.id': user_id}):
|
|
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
|