Coverage for app/domain/repository/user_repository.py: 100%

1from sqlmodel import Session, select 

2 

3from migrations.models import TUser 

4 

5class UserRepository: 

6 def __init__(self, session: Session) -> None: 

7 self.__session: Session = session 

8 

9 def _fetch_by_id(self, _id: str) -> TUser | None: 

10 statement = select(TUser).where(TUser.user_id == _id) 

11 return self.__session.exec(statement).first() 

12 

13 def find_by_id(self, _id: str): 

14 model: TUser | None = self._fetch_by_id(_id) 

15 return model 

16 # if model is None: 

17 # raise UserNotFoundError(user_id=_id.root) 

18 # return User.model_validate(model)