Working on RBAC.

This commit is contained in:
charles-gauthereau
2024-10-27 18:31:54 +01:00
parent 7a8a8fb6d3
commit ff7f864ec2
+3 -2
View File
@@ -1,17 +1,18 @@
from fastapi import Depends, APIRouter, Request from fastapi import Depends, APIRouter, Request
from sqlalchemy.orm import Session from sqlalchemy.orm import Session
from apps.users.roles import Permission from apps.users.roles import Permission
from apps.users.services import permission_required, get_current_user, get_current_active_user from apps.users.services import permission_required, get_current_user, get_current_active_user
from pydantic import BaseModel from pydantic import BaseModel
from middleware.db_connection import get_db from middleware.db_connection import get_db
router = APIRouter(prefix="/agents", tags=["agents"]) router = APIRouter(prefix="/agents", tags=["agents"])
class Agent(BaseModel): class Agent(BaseModel):
name: str name: str
@router.post("/", dependencies=[Depends(get_current_active_user)]) @router.post("/", dependencies=[Depends(get_current_active_user)])
@permission_required(Permission.CREATE, Permission.READ) @permission_required(Permission.CREATE, Permission.READ)
async def create_agent(agent: Agent, request: Request, db: Session = Depends(get_db)): async def create_agent(agent: Agent, request: Request, db: Session = Depends(get_db)):