diff --git a/apiserver/plane/app/views/workspace/quick_link.py b/apiserver/plane/app/views/workspace/quick_link.py index 30a21ed459..beb3d8c761 100644 --- a/apiserver/plane/app/views/workspace/quick_link.py +++ b/apiserver/plane/app/views/workspace/quick_link.py @@ -8,7 +8,8 @@ from plane.app.serializers import WorkspaceUserLinkSerializer from ..base import BaseViewSet from plane.app.permissions import allow_permission, ROLE -class QuickLinkViewSet(BaseViewSet): + +class QuickLinkViewSet(BaseViewSet): model = WorkspaceUserLink def get_serializer_class(self): @@ -16,52 +17,58 @@ class QuickLinkViewSet(BaseViewSet): @allow_permission([ROLE.ADMIN, ROLE.MEMBER, ROLE.GUEST], level="WORKSPACE") def create(self, request, slug): - workspace = Workspace.objects.get(slug=slug, owner=request.user) - serializer = WorkspaceUserLinkSerializer(data=request.data) + workspace = Workspace.objects.get(slug=slug) + serializer = WorkspaceUserLinkSerializer(data=request.data) if serializer.is_valid(): serializer.save(workspace_id=workspace.id, owner=request.user) return Response(serializer.data, status=status.HTTP_201_CREATED) return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) - - + @allow_permission([ROLE.ADMIN, ROLE.MEMBER, ROLE.GUEST], level="WORKSPACE") def partial_update(self, request, slug, pk): - quick_link = WorkspaceUserLink.objects.filter(pk=pk, workspace__slug=slug, owner=request.user).first() + quick_link = WorkspaceUserLink.objects.filter( + pk=pk, workspace__slug=slug, owner=request.user + ).first() - if quick_link: - serializer = WorkspaceUserLinkSerializer(quick_link, data=request.data, partial=True) + if quick_link: + serializer = WorkspaceUserLinkSerializer( + quick_link, data=request.data, partial=True + ) if serializer.is_valid(): serializer.save() return Response(serializer.data, status=status.HTTP_200_OK) - return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) - return Response({"detail": "Quick link not found."}, status=status.HTTP_404_NOT_FOUND) + return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST) + return Response( + {"detail": "Quick link not found."}, status=status.HTTP_404_NOT_FOUND + ) @allow_permission([ROLE.ADMIN, ROLE.MEMBER, ROLE.GUEST], level="WORKSPACE") def retrieve(self, request, slug, pk): try: quick_link = WorkspaceUserLink.objects.get( - pk=pk, - workspace__slug=slug, - owner=request.user + pk=pk, workspace__slug=slug, owner=request.user ) serializer = WorkspaceUserLinkSerializer(quick_link) return Response(serializer.data, status=status.HTTP_200_OK) except WorkspaceUserLink.DoesNotExist: return Response( - {"error": "Quick link not found."}, - status=status.HTTP_404_NOT_FOUND + {"error": "Quick link not found."}, status=status.HTTP_404_NOT_FOUND ) @allow_permission([ROLE.ADMIN, ROLE.MEMBER, ROLE.GUEST], level="WORKSPACE") def destroy(self, request, slug, pk): - quick_link = WorkspaceUserLink.objects.get(pk=pk, workspace__slug=slug, owner=request.user) - quick_link.delete() + quick_link = WorkspaceUserLink.objects.get( + pk=pk, workspace__slug=slug, owner=request.user + ) + quick_link.delete() return Response(status=status.HTTP_204_NO_CONTENT) @allow_permission([ROLE.ADMIN, ROLE.MEMBER, ROLE.GUEST], level="WORKSPACE") def list(self, request, slug): - quick_links = WorkspaceUserLink.objects.filter(workspace__slug=slug, owner=request.user) - + quick_links = WorkspaceUserLink.objects.filter( + workspace__slug=slug, owner=request.user + ) + serializer = WorkspaceUserLinkSerializer(quick_links, many=True) - return Response(serializer.data, status=status.HTTP_200_OK) \ No newline at end of file + return Response(serializer.data, status=status.HTTP_200_OK)