gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: change endpoint to use url data

Browse source
This commit is contained in:
Igor Ilic 2025-06-02 13:34:38 +02:00
parent a500aeba80
commit 939aaf1b5e
1 changed files with 7 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
cognee/api/v1/permissions/routers/get_permissions_router.py
View file

@ -6,13 +6,6 @@ from fastapi.responses import JSONResponse
from cognee.modules.users.models import User
from cognee.modules.users.methods import get_authenticated_user
from cognee.api.DTO import InDTO
class DatasetPermissionsPayloadDTO(InDTO):
principal_id: UUID
permission_name: str
dataset_ids: List[UUID]
def get_permissions_router() -> APIRouter:
@ -20,14 +13,17 @@ def get_permissions_router() -> APIRouter:
@permissions_router.post("/datasets/{principal_id}/")
async def give_datasets_permission_to_principal(
payload: DatasetPermissionsPayloadDTO, user: User = Depends(get_authenticated_user)
permission_name: str,
dataset_ids: List[UUID],
principal_id: UUID,
user: User = Depends(get_authenticated_user),
):
from cognee.modules.users.permissions.methods import authorized_give_permission_on_datasets
await authorized_give_permission_on_datasets(
payload.principal_id,
[dataset_id for dataset_id in payload.dataset_ids],
payload.permission_name,
principal_id,
[dataset_id for dataset_id in dataset_ids],
permission_name,
user.id,
)