diff --git a/src/iam/dependencies.py b/src/iam/dependencies.py index d5632c7..24ae1ab 100644 --- a/src/iam/dependencies.py +++ b/src/iam/dependencies.py @@ -15,7 +15,7 @@ from fastapi import Depends, Query from src.database import db_dependency -from src.iam.models import Group +from src.iam.models import Group, Permission from src.iam.exceptions import GroupNotFoundException, PermNotFoundException from src.iam.schemas import GroupIDMixin, PermIDMixin @@ -43,14 +43,14 @@ def get_group_model_body(db: db_dependency, request_model: Optional[GroupIDMixin group_model_body_dependency = Annotated[type[Group], Depends(get_group_model_body)] -def get_perm_model_body(db: db_dependency, request_model: Optional[PermIDMixin] = None) -> type[Group]: +def get_perm_model_body(db: db_dependency, request_model: Optional[PermIDMixin] = None) -> type[Permission]: perm_id = getattr(request_model, "permission_id", None) if perm_id is None: raise PermNotFoundException - group_model = db.get(Group, perm_id) - if group_model is None: + perm_model = db.get(Permission, perm_id) + if perm_model is None: raise PermNotFoundException(perm_id) - return group_model + return perm_model -perm_model_body_dependency = Annotated[type[Group], Depends(get_perm_model_body)] +perm_model_body_dependency = Annotated[type[Permission], Depends(get_perm_model_body)]