581 lines
22 KiB
Python
581 lines
22 KiB
Python
# Python imports
|
|
import json
|
|
|
|
# Django import
|
|
from django.utils import timezone
|
|
from django.db.models import Q, Count, OuterRef, Func, F, Prefetch, Subquery
|
|
from django.core.serializers.json import DjangoJSONEncoder
|
|
from django.contrib.postgres.aggregates import ArrayAgg
|
|
from django.contrib.postgres.fields import ArrayField
|
|
from django.db.models import Value, UUIDField
|
|
from django.db.models.functions import Coalesce
|
|
|
|
# Third party imports
|
|
from rest_framework import status
|
|
from rest_framework.response import Response
|
|
|
|
# Module imports
|
|
from ..base import BaseViewSet
|
|
from plane.app.permissions import allow_permission, ROLE
|
|
from plane.db.models import (
|
|
Intake,
|
|
IntakeIssue,
|
|
Issue,
|
|
State,
|
|
IssueLink,
|
|
FileAsset,
|
|
Project,
|
|
ProjectMember,
|
|
CycleIssue,
|
|
)
|
|
from plane.app.serializers import (
|
|
IssueCreateSerializer,
|
|
IssueSerializer,
|
|
IntakeSerializer,
|
|
IntakeIssueSerializer,
|
|
IntakeIssueDetailSerializer,
|
|
)
|
|
from plane.utils.issue_filters import issue_filters
|
|
from plane.bgtasks.issue_activities_task import issue_activity
|
|
|
|
|
|
class IntakeViewSet(BaseViewSet):
|
|
serializer_class = IntakeSerializer
|
|
model = Intake
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
super()
|
|
.get_queryset()
|
|
.filter(
|
|
workspace__slug=self.kwargs.get("slug"),
|
|
project_id=self.kwargs.get("project_id"),
|
|
)
|
|
.annotate(
|
|
pending_issue_count=Count(
|
|
"issue_intake", filter=Q(issue_intake__status=-2)
|
|
)
|
|
)
|
|
.select_related("workspace", "project")
|
|
)
|
|
|
|
@allow_permission([ROLE.ADMIN, ROLE.MEMBER])
|
|
def list(self, request, slug, project_id):
|
|
intake = self.get_queryset().first()
|
|
return Response(IntakeSerializer(intake).data, status=status.HTTP_200_OK)
|
|
|
|
@allow_permission([ROLE.ADMIN, ROLE.MEMBER])
|
|
def perform_create(self, serializer):
|
|
serializer.save(project_id=self.kwargs.get("project_id"))
|
|
|
|
@allow_permission([ROLE.ADMIN, ROLE.MEMBER])
|
|
def destroy(self, request, slug, project_id, pk):
|
|
intake = Intake.objects.filter(
|
|
workspace__slug=slug, project_id=project_id, pk=pk
|
|
).first()
|
|
# Handle default intake delete
|
|
if intake.is_default:
|
|
return Response(
|
|
{"error": "You cannot delete the default intake"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
intake.delete()
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class IntakeIssueViewSet(BaseViewSet):
|
|
serializer_class = IntakeIssueSerializer
|
|
model = IntakeIssue
|
|
|
|
filterset_fields = ["statulls"]
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
Issue.objects.filter(
|
|
project_id=self.kwargs.get("project_id"),
|
|
workspace__slug=self.kwargs.get("slug"),
|
|
)
|
|
.select_related("workspace", "project", "state", "parent")
|
|
.prefetch_related("assignees", "labels", "issue_module__module")
|
|
.prefetch_related(
|
|
Prefetch(
|
|
"issue_intake",
|
|
queryset=IntakeIssue.objects.only(
|
|
"status", "duplicate_to", "snoozed_till", "source"
|
|
),
|
|
)
|
|
)
|
|
.annotate(
|
|
cycle_id=Subquery(
|
|
CycleIssue.objects.filter(
|
|
issue=OuterRef("id"), deleted_at__isnull=True
|
|
).values("cycle_id")[:1]
|
|
)
|
|
)
|
|
.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")
|
|
)
|
|
.annotate(
|
|
sub_issues_count=Issue.issue_objects.filter(parent=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.annotate(
|
|
label_ids=Coalesce(
|
|
ArrayAgg(
|
|
"labels__id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(labels__id__isnull=True)
|
|
& Q(label_issue__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
assignee_ids=Coalesce(
|
|
ArrayAgg(
|
|
"assignees__id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(assignees__id__isnull=True)
|
|
& Q(assignees__member_project__is_active=True)
|
|
& Q(issue_assignee__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
module_ids=Coalesce(
|
|
ArrayAgg(
|
|
"issue_module__module_id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(issue_module__module_id__isnull=True)
|
|
& Q(issue_module__module__archived_at__isnull=True)
|
|
& Q(issue_module__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
)
|
|
).distinct()
|
|
|
|
@allow_permission([ROLE.ADMIN, ROLE.MEMBER, ROLE.GUEST])
|
|
def list(self, request, slug, project_id):
|
|
intake_id = Intake.objects.filter(
|
|
workspace__slug=slug, project_id=project_id
|
|
).first()
|
|
project = Project.objects.get(pk=project_id)
|
|
filters = issue_filters(request.GET, "GET", "issue__")
|
|
intake_issue = (
|
|
IntakeIssue.objects.filter(
|
|
intake_id=intake_id.id, project_id=project_id, **filters
|
|
)
|
|
.select_related("issue")
|
|
.prefetch_related("issue__labels")
|
|
.annotate(
|
|
label_ids=Coalesce(
|
|
ArrayAgg(
|
|
"issue__labels__id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(issue__labels__id__isnull=True)
|
|
& Q(issue__label_issue__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
)
|
|
)
|
|
).order_by(request.GET.get("order_by", "-issue__created_at"))
|
|
# Intake status filter
|
|
intake_status = [
|
|
item
|
|
for item in request.GET.get("status", "-2").split(",")
|
|
if item != "null"
|
|
]
|
|
if intake_status:
|
|
intake_issue = intake_issue.filter(status__in=intake_status)
|
|
|
|
if (
|
|
ProjectMember.objects.filter(
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
member=request.user,
|
|
role=5,
|
|
is_active=True,
|
|
).exists()
|
|
and not project.guest_view_all_features
|
|
):
|
|
intake_issue = intake_issue.filter(created_by=request.user)
|
|
return self.paginate(
|
|
request=request,
|
|
queryset=(intake_issue),
|
|
on_results=lambda intake_issues: IntakeIssueSerializer(
|
|
intake_issues, many=True
|
|
).data,
|
|
)
|
|
|
|
@allow_permission([ROLE.ADMIN, ROLE.MEMBER, ROLE.GUEST])
|
|
def create(self, request, slug, project_id):
|
|
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
|
|
)
|
|
|
|
# create an issue
|
|
project = Project.objects.get(pk=project_id)
|
|
serializer = IssueCreateSerializer(
|
|
data=request.data.get("issue"),
|
|
context={
|
|
"project_id": project_id,
|
|
"workspace_id": project.workspace_id,
|
|
"default_assignee_id": project.default_assignee_id,
|
|
},
|
|
)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
intake_id = Intake.objects.filter(
|
|
workspace__slug=slug, project_id=project_id
|
|
).first()
|
|
# create an intake issue
|
|
intake_issue = IntakeIssue.objects.create(
|
|
intake_id=intake_id.id,
|
|
project_id=project_id,
|
|
issue_id=serializer.data["id"],
|
|
source=request.data.get("source", "IN-APP"),
|
|
)
|
|
# 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(serializer.data["id"]),
|
|
project_id=str(project_id),
|
|
current_instance=None,
|
|
epoch=int(timezone.now().timestamp()),
|
|
notification=True,
|
|
origin=request.META.get("HTTP_ORIGIN"),
|
|
intake=str(intake_issue.id),
|
|
)
|
|
intake_issue = (
|
|
IntakeIssue.objects.select_related("issue")
|
|
.prefetch_related("issue__labels", "issue__assignees")
|
|
.annotate(
|
|
label_ids=Coalesce(
|
|
ArrayAgg(
|
|
"issue__labels__id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(issue__labels__id__isnull=True)
|
|
& Q(issue__label_issue__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
assignee_ids=Coalesce(
|
|
ArrayAgg(
|
|
"issue__assignees__id",
|
|
distinct=True,
|
|
filter=~Q(issue__assignees__id__isnull=True)
|
|
& Q(issue__assignees__member_project__is_active=True),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
)
|
|
.get(
|
|
intake_id=intake_id.id,
|
|
issue_id=serializer.data["id"],
|
|
project_id=project_id,
|
|
)
|
|
)
|
|
serializer = IntakeIssueDetailSerializer(intake_issue)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
else:
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
@allow_permission(allowed_roles=[ROLE.ADMIN], creator=True, model=Issue)
|
|
def partial_update(self, request, slug, project_id, pk):
|
|
intake_id = Intake.objects.filter(
|
|
workspace__slug=slug, project_id=project_id
|
|
).first()
|
|
intake_issue = IntakeIssue.objects.get(
|
|
issue_id=pk,
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
intake_id=intake_id,
|
|
)
|
|
# Get the project member
|
|
project_member = ProjectMember.objects.get(
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
member=request.user,
|
|
is_active=True,
|
|
)
|
|
# Only project members admins and created_by users can access this endpoint
|
|
if project_member.role <= 5 and 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)
|
|
if bool(issue_data):
|
|
issue = Issue.objects.annotate(
|
|
label_ids=Coalesce(
|
|
ArrayAgg(
|
|
"labels__id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(labels__id__isnull=True)
|
|
& Q(label_issue__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
assignee_ids=Coalesce(
|
|
ArrayAgg(
|
|
"assignees__id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(assignees__id__isnull=True)
|
|
& Q(issue_assignee__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
).get(pk=intake_issue.issue_id, workspace__slug=slug, project_id=project_id)
|
|
# Only allow guests to edit name and description
|
|
if project_member.role <= 5:
|
|
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
|
|
)
|
|
|
|
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_id),
|
|
current_instance=json.dumps(
|
|
IssueSerializer(current_instance).data,
|
|
cls=DjangoJSONEncoder,
|
|
),
|
|
epoch=int(timezone.now().timestamp()),
|
|
notification=True,
|
|
origin=request.META.get("HTTP_ORIGIN"),
|
|
intake=str(intake_issue.id),
|
|
)
|
|
issue_serializer.save()
|
|
else:
|
|
return Response(
|
|
issue_serializer.errors, status=status.HTTP_400_BAD_REQUEST
|
|
)
|
|
|
|
# Only project admins and members can edit intake issue attributes
|
|
if project_member.role > 15:
|
|
serializer = IntakeIssueSerializer(
|
|
intake_issue, data=request.data, partial=True
|
|
)
|
|
current_instance = json.dumps(
|
|
IntakeIssueSerializer(intake_issue).data, cls=DjangoJSONEncoder
|
|
)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
# Update the issue state if the issue is rejected or marked as duplicate
|
|
if serializer.data["status"] in [-1, 2]:
|
|
issue = Issue.objects.get(
|
|
pk=intake_issue.issue_id,
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
)
|
|
state = State.objects.filter(
|
|
group="cancelled", workspace__slug=slug, project_id=project_id
|
|
).first()
|
|
if state is not None:
|
|
issue.state = state
|
|
issue.save()
|
|
|
|
# Update the issue state if it is accepted
|
|
if serializer.data["status"] in [1]:
|
|
issue = Issue.objects.get(
|
|
pk=intake_issue.issue_id,
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
)
|
|
|
|
# Update the issue state only if it is in triage state
|
|
if issue.state.is_triage:
|
|
# Move to default state
|
|
state = State.objects.filter(
|
|
workspace__slug=slug, project_id=project_id, default=True
|
|
).first()
|
|
if state is not None:
|
|
issue.state = state
|
|
issue.save()
|
|
# create a activity for status change
|
|
issue_activity.delay(
|
|
type="intake.activity.created",
|
|
requested_data=json.dumps(request.data, cls=DjangoJSONEncoder),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(pk),
|
|
project_id=str(project_id),
|
|
current_instance=current_instance,
|
|
epoch=int(timezone.now().timestamp()),
|
|
notification=False,
|
|
origin=request.META.get("HTTP_ORIGIN"),
|
|
intake=(intake_issue.id),
|
|
)
|
|
|
|
intake_issue = (
|
|
IntakeIssue.objects.select_related("issue")
|
|
.prefetch_related("issue__labels", "issue__assignees")
|
|
.annotate(
|
|
label_ids=Coalesce(
|
|
ArrayAgg(
|
|
"issue__labels__id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(issue__labels__id__isnull=True)
|
|
& Q(issue__label_issue__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
assignee_ids=Coalesce(
|
|
ArrayAgg(
|
|
"issue__assignees__id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(issue__assignees__id__isnull=True)
|
|
& Q(issue__issue_assignee__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
)
|
|
.get(intake_id=intake_id.id, issue_id=pk, project_id=project_id)
|
|
)
|
|
serializer = IntakeIssueDetailSerializer(intake_issue).data
|
|
return Response(serializer, status=status.HTTP_200_OK)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
else:
|
|
serializer = IntakeIssueDetailSerializer(intake_issue).data
|
|
return Response(serializer, status=status.HTTP_200_OK)
|
|
|
|
@allow_permission(
|
|
allowed_roles=[ROLE.ADMIN, ROLE.MEMBER, ROLE.GUEST], creator=True, model=Issue
|
|
)
|
|
def retrieve(self, request, slug, project_id, pk):
|
|
intake_id = Intake.objects.filter(
|
|
workspace__slug=slug, project_id=project_id
|
|
).first()
|
|
project = Project.objects.get(pk=project_id)
|
|
intake_issue = (
|
|
IntakeIssue.objects.select_related("issue")
|
|
.prefetch_related("issue__labels", "issue__assignees")
|
|
.annotate(
|
|
label_ids=Coalesce(
|
|
ArrayAgg(
|
|
"issue__labels__id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(issue__labels__id__isnull=True)
|
|
& Q(issue__label_issue__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
assignee_ids=Coalesce(
|
|
ArrayAgg(
|
|
"issue__assignees__id",
|
|
distinct=True,
|
|
filter=Q(
|
|
~Q(issue__assignees__id__isnull=True)
|
|
& Q(issue__issue_assignee__deleted_at__isnull=True)
|
|
),
|
|
),
|
|
Value([], output_field=ArrayField(UUIDField())),
|
|
),
|
|
)
|
|
.get(intake_id=intake_id.id, issue_id=pk, project_id=project_id)
|
|
)
|
|
if (
|
|
ProjectMember.objects.filter(
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
member=request.user,
|
|
role=5,
|
|
is_active=True,
|
|
).exists()
|
|
and not project.guest_view_all_features
|
|
and not intake_issue.created_by == request.user
|
|
):
|
|
return Response(
|
|
{"error": "You are not allowed to view this issue"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
issue = IntakeIssueDetailSerializer(intake_issue).data
|
|
return Response(issue, status=status.HTTP_200_OK)
|
|
|
|
@allow_permission(allowed_roles=[ROLE.ADMIN], creator=True, model=Issue)
|
|
def destroy(self, request, slug, project_id, pk):
|
|
intake_id = Intake.objects.filter(
|
|
workspace__slug=slug, project_id=project_id
|
|
).first()
|
|
intake_issue = IntakeIssue.objects.get(
|
|
issue_id=pk,
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
intake_id=intake_id,
|
|
)
|
|
|
|
# Check the issue status
|
|
if intake_issue.status in [-2, -1, 0, 2]:
|
|
# Delete the issue also
|
|
issue = Issue.objects.filter(
|
|
workspace__slug=slug, project_id=project_id, pk=pk
|
|
).first()
|
|
issue.delete()
|
|
|
|
intake_issue.delete()
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|