forked from sr2/cloud-api
feat: user dependencies
In addition to the by-query and by-body db fetch dependencies. Users also have a by-claim dependency. Issue #6
This commit is contained in:
parent
657f91d73d
commit
748544fb82
3 changed files with 59 additions and 27 deletions
|
|
@ -8,4 +8,47 @@ Classes:
|
|||
Functions:
|
||||
- List: Description
|
||||
- Functions: Description
|
||||
"""
|
||||
"""
|
||||
from typing import Annotated
|
||||
from fastapi import Depends, Query
|
||||
|
||||
from src.user.exceptions import UserNotFoundException
|
||||
from src.user.models import User
|
||||
|
||||
from src.auth.service import claims_dependency
|
||||
from src.database import db_dependency
|
||||
from src.user.schemas import UserIDMixin
|
||||
|
||||
|
||||
async def get_user_model_claims(claims: claims_dependency, db: db_dependency):
|
||||
user_id = claims.get("db_id", None)
|
||||
if user_id is None:
|
||||
raise UserNotFoundException()
|
||||
|
||||
user_model = db.get(User, user_id)
|
||||
if user_model is None:
|
||||
raise UserNotFoundException(user_id=user_id)
|
||||
|
||||
return user_model
|
||||
|
||||
user_model_claims_dependency = Annotated[type[User], Depends(get_user_model_claims)]
|
||||
|
||||
|
||||
async def get_user_model_query(db: db_dependency, user_id: Annotated[int, Query(gt=0)]):
|
||||
user_model = db.get(User, user_id)
|
||||
if user_model is None:
|
||||
raise UserNotFoundException(user_id=user_id)
|
||||
|
||||
return user_model
|
||||
|
||||
user_model_query_dependency = Annotated[type[User], Depends(get_user_model_query)]
|
||||
|
||||
|
||||
async def get_user_model_body(db: db_dependency, request_model: UserIDMixin):
|
||||
user_model = db.get(User, request_model.user_id)
|
||||
if user_model is None:
|
||||
raise UserNotFoundException(user_id=request_model.user_id)
|
||||
|
||||
return user_model
|
||||
|
||||
user_model_body_dependency = Annotated[type[User], Depends(get_user_model_body)]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue