# 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)