plane/apiserver/plane/api/views/api_token.py

# Python import
from uuid import uuid4

# Third party
from rest_framework.response import Response
from rest_framework import status
from sentry_sdk import capture_exception

# Module import
from .base import BaseAPIView
from plane.db.models import APIToken
from plane.api.serializers import APITokenSerializer


class ApiTokenEndpoint(BaseAPIView):
    def post(self, request):
        try:
            label = request.data.get("label", str(uuid4().hex))
            workspace = request.data.get("workspace", False)

            if not workspace:
                return Response(
                    {"error": "Workspace is required"}, status=status.HTTP_200_OK
                )

            api_token = APIToken.objects.create(
                label=label, user=request.user, workspace_id=workspace
            )

            serializer = APITokenSerializer(api_token)
            return Response(serializer.data, status=status.HTTP_201_CREATED)

        except Exception as e:
            capture_exception(e)
            return Response(
                {"error": "Something went wrong please try again later"},
                status=status.HTTP_400_BAD_REQUEST,
            )

    def get(self, request):
        try:
            api_tokens = APIToken.objects.filter(user=request.user)
            serializer = APITokenSerializer(api_tokens, many=True)
            return Response(serializer.data, status=status.HTTP_200_OK)
        except Exception as e:
            capture_exception(e)
            return Response(
                {"error": "Something went wrong please try again later"},
                status=status.HTTP_400_BAD_REQUEST,
            )

    def delete(self, request, pk):
        try:
            api_token = APIToken.objects.get(pk=pk)
            api_token.delete()
            return Response(status=status.HTTP_204_NO_CONTENT)
        except APIToken.DoesNotExist:
            return Response(
                {"error": "Token does not exists"}, status=status.HTTP_400_BAD_REQUEST
            )
        except Exception as e:
            capture_exception(e)
            return Response(
                {"error": "Something went wrong please try again later"},
                status=status.HTTP_400_BAD_REQUEST,
            )