Coverage for src/cstlcore/users/dependencies.py: 47%

1import uuid 

2 

3from fastapi import Depends, HTTPException 

4from sqlmodel import Session, select 

5 

6from cstlcore.database.dependencies import get_session 

7from cstlcore.users.models import User 

8 

9 

10def get_user_by_email(email: str, session: Session = Depends(get_session)) -> User: 

11 user = session.exec(select(User).where(User.email == email)).first() 

12 if not user: 

13 raise HTTPException(status_code=404, detail="User not found") 

14 return user 

15 

16 

17def get_user_by_id(user_id: uuid.UUID, session: Session = Depends(get_session)) -> User: 

18 user = session.get(User, user_id) 

19 if not user: 

20 raise HTTPException(status_code=404, detail="User not found") 

21 return user