276 lines
11 KiB
Python
276 lines
11 KiB
Python
# Python imports
|
|
import json
|
|
|
|
# Django import
|
|
from django.utils import timezone
|
|
from django.db.models import Q, OuterRef, Func, F, Prefetch
|
|
from django.core.serializers.json import DjangoJSONEncoder
|
|
|
|
# Third party imports
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
|
|
# Module imports
|
|
from .base import BaseViewSet
|
|
from plane.db.models import IntakeIssue, Issue, IssueLink, FileAsset, DeployBoard, State, StateGroup
|
|
from plane.app.serializers import (
|
|
IssueSerializer,
|
|
IntakeIssueSerializer,
|
|
IssueCreateSerializer,
|
|
IssueStateIntakeSerializer,
|
|
)
|
|
from plane.utils.issue_filters import issue_filters
|
|
from plane.bgtasks.issue_activities_task import issue_activity
|
|
from plane.db.models.intake import SourceType
|
|
|
|
|
|
class IntakeIssuePublicViewSet(BaseViewSet):
|
|
serializer_class = IntakeIssueSerializer
|
|
model = IntakeIssue
|
|
|
|
filterset_fields = ["status"]
|
|
|
|
def get_queryset(self):
|
|
project_deploy_board = DeployBoard.objects.get(
|
|
workspace__slug=self.kwargs.get("slug"),
|
|
project_id=self.kwargs.get("project_id"),
|
|
)
|
|
if project_deploy_board is not None:
|
|
return self.filter_queryset(
|
|
super()
|
|
.get_queryset()
|
|
.filter(
|
|
Q(snoozed_till__gte=timezone.now()) | Q(snoozed_till__isnull=True),
|
|
project_id=self.kwargs.get("project_id"),
|
|
workspace__slug=self.kwargs.get("slug"),
|
|
intake_id=self.kwargs.get("intake_id"),
|
|
)
|
|
.select_related("issue", "workspace", "project")
|
|
)
|
|
return IntakeIssue.objects.none()
|
|
|
|
def list(self, request, anchor, intake_id):
|
|
project_deploy_board = DeployBoard.objects.get(anchor=anchor, entity_name="project")
|
|
if project_deploy_board.intake is None:
|
|
return Response(
|
|
{"error": "Intake is not enabled for this Project Board"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
filters = issue_filters(request.query_params, "GET")
|
|
issues = (
|
|
Issue.objects.filter(
|
|
issue_intake__intake_id=intake_id,
|
|
workspace_id=project_deploy_board.workspace_id,
|
|
project_id=project_deploy_board.project_id,
|
|
)
|
|
.filter(**filters)
|
|
.annotate(bridge_id=F("issue_intake__id"))
|
|
.select_related("workspace", "project", "state", "parent")
|
|
.prefetch_related("assignees", "labels")
|
|
.order_by("issue_intake__snoozed_till", "issue_intake__status")
|
|
.annotate(
|
|
sub_issues_count=Issue.issue_objects.filter(parent=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.annotate(
|
|
link_count=IssueLink.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.annotate(
|
|
attachment_count=FileAsset.objects.filter(
|
|
issue_id=OuterRef("id"),
|
|
entity_type=FileAsset.EntityTypeContext.ISSUE_ATTACHMENT,
|
|
)
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.prefetch_related(
|
|
Prefetch(
|
|
"issue_intake",
|
|
queryset=IntakeIssue.objects.only("status", "duplicate_to", "snoozed_till", "source"),
|
|
)
|
|
)
|
|
)
|
|
issues_data = IssueStateIntakeSerializer(issues, many=True).data
|
|
return Response(issues_data, status=status.HTTP_200_OK)
|
|
|
|
def create(self, request, anchor, intake_id):
|
|
project_deploy_board = DeployBoard.objects.get(anchor=anchor, entity_name="project")
|
|
if project_deploy_board.intake is None:
|
|
return Response(
|
|
{"error": "Intake is not enabled for this Project Board"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
if not request.data.get("issue", {}).get("name", False):
|
|
return Response({"error": "Name is required"}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
# Check for valid priority
|
|
if request.data.get("issue", {}).get("priority", "none") not in [
|
|
"low",
|
|
"medium",
|
|
"high",
|
|
"urgent",
|
|
"none",
|
|
]:
|
|
return Response({"error": "Invalid priority"}, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
# get the triage state
|
|
triage_state = State.triage_objects.filter(
|
|
project_id=project_deploy_board.project_id, workspace_id=project_deploy_board.workspace_id
|
|
).first()
|
|
|
|
if not triage_state:
|
|
triage_state = State.objects.create(
|
|
name="Triage",
|
|
group=StateGroup.TRIAGE.value,
|
|
project_id=project_deploy_board.project_id,
|
|
workspace_id=project_deploy_board.workspace_id,
|
|
color="#4E5355",
|
|
sequence=65000,
|
|
default=False,
|
|
)
|
|
|
|
# create an issue
|
|
issue = Issue.objects.create(
|
|
name=request.data.get("issue", {}).get("name"),
|
|
description=request.data.get("issue", {}).get("description", {}),
|
|
description_html=request.data.get("issue", {}).get("description_html", "<p></p>"),
|
|
priority=request.data.get("issue", {}).get("priority", "low"),
|
|
project_id=project_deploy_board.project_id,
|
|
state_id=triage_state.id,
|
|
)
|
|
|
|
# Create an Issue Activity
|
|
issue_activity.delay(
|
|
type="issue.activity.created",
|
|
requested_data=json.dumps(request.data, cls=DjangoJSONEncoder),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(issue.id),
|
|
project_id=str(project_deploy_board.project_id),
|
|
current_instance=None,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
# create an intake issue
|
|
IntakeIssue.objects.create(
|
|
intake_id=intake_id,
|
|
project_id=project_deploy_board.project_id,
|
|
issue=issue,
|
|
source=SourceType.IN_APP,
|
|
)
|
|
|
|
serializer = IssueStateIntakeSerializer(issue)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
def partial_update(self, request, anchor, intake_id, pk):
|
|
project_deploy_board = DeployBoard.objects.get(anchor=anchor, entity_name="project")
|
|
if project_deploy_board.intake is None:
|
|
return Response(
|
|
{"error": "Intake is not enabled for this Project Board"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
intake_issue = IntakeIssue.objects.get(
|
|
pk=pk,
|
|
workspace_id=project_deploy_board.workspace_id,
|
|
project_id=project_deploy_board.project_id,
|
|
intake_id=intake_id,
|
|
)
|
|
# Get the project member
|
|
if str(intake_issue.created_by_id) != str(request.user.id):
|
|
return Response(
|
|
{"error": "You cannot edit intake issues"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
# Get issue data
|
|
issue_data = request.data.pop("issue", False)
|
|
|
|
issue = Issue.objects.get(
|
|
pk=intake_issue.issue_id,
|
|
workspace_id=project_deploy_board.workspace_id,
|
|
project_id=project_deploy_board.project_id,
|
|
)
|
|
# viewers and guests since only viewers and guests
|
|
issue_data = {
|
|
"name": issue_data.get("name", issue.name),
|
|
"description_html": issue_data.get("description_html", issue.description_html),
|
|
"description": issue_data.get("description", issue.description),
|
|
}
|
|
|
|
issue_serializer = IssueCreateSerializer(
|
|
issue,
|
|
data=issue_data,
|
|
partial=True,
|
|
context={"project_id": project_deploy_board.project_id, "allow_triage_state": True},
|
|
)
|
|
|
|
if issue_serializer.is_valid():
|
|
current_instance = issue
|
|
# Log all the updates
|
|
requested_data = json.dumps(issue_data, cls=DjangoJSONEncoder)
|
|
if issue is not None:
|
|
issue_activity.delay(
|
|
type="issue.activity.updated",
|
|
requested_data=requested_data,
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(issue.id),
|
|
project_id=str(project_deploy_board.project_id),
|
|
current_instance=json.dumps(IssueSerializer(current_instance).data, cls=DjangoJSONEncoder),
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
issue_serializer.save()
|
|
return Response(issue_serializer.data, status=status.HTTP_200_OK)
|
|
return Response(issue_serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def retrieve(self, request, anchor, intake_id, pk):
|
|
project_deploy_board = DeployBoard.objects.get(anchor=anchor, entity_name="project")
|
|
if project_deploy_board.intake is None:
|
|
return Response(
|
|
{"error": "Intake is not enabled for this Project Board"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
intake_issue = IntakeIssue.objects.get(
|
|
pk=pk,
|
|
workspace_id=project_deploy_board.workspace_id,
|
|
project_id=project_deploy_board.project_id,
|
|
intake_id=intake_id,
|
|
)
|
|
issue = Issue.objects.get(
|
|
pk=intake_issue.issue_id,
|
|
workspace_id=project_deploy_board.workspace_id,
|
|
project_id=project_deploy_board.project_id,
|
|
)
|
|
serializer = IssueStateIntakeSerializer(issue)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
def destroy(self, request, anchor, intake_id, pk):
|
|
project_deploy_board = DeployBoard.objects.get(anchor=anchor, entity_name="project")
|
|
if project_deploy_board.intake is None:
|
|
return Response(
|
|
{"error": "Intake is not enabled for this Project Board"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
intake_issue = IntakeIssue.objects.get(
|
|
pk=pk,
|
|
workspace_id=project_deploy_board.workspace_id,
|
|
project_id=project_deploy_board.project_id,
|
|
intake_id=intake_id,
|
|
)
|
|
|
|
if str(intake_issue.created_by_id) != str(request.user.id):
|
|
return Response(
|
|
{"error": "You cannot delete intake issue"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
intake_issue.delete()
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|