25f7d813b9
* Crud for wuick links * Validate quick link existence * Add custom method for destroy and retrieve * Add List method * Remove print statements * List all the workspace quick links * Filter by user
67 lines
No EOL
2.9 KiB
Python
67 lines
No EOL
2.9 KiB
Python
# Third party imports
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
|
|
# Module imports
|
|
from plane.db.models import WorkspaceUserLink, Workspace
|
|
from plane.app.serializers import WorkspaceUserLinkSerializer
|
|
from ..base import BaseViewSet
|
|
from plane.app.permissions import allow_permission, ROLE
|
|
|
|
class QuickLinkViewSet(BaseViewSet):
|
|
model = WorkspaceUserLink
|
|
|
|
def get_serializer_class(self):
|
|
return WorkspaceUserLinkSerializer
|
|
|
|
@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)
|
|
|
|
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()
|
|
|
|
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)
|
|
|
|
@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
|
|
)
|
|
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
|
|
)
|
|
|
|
@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()
|
|
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)
|
|
|
|
serializer = WorkspaceUserLinkSerializer(quick_links, many=True)
|
|
return Response(serializer.data, status=status.HTTP_200_OK) |