forked from github/plane
1753 lines
62 KiB
Python
1753 lines
62 KiB
Python
# Python imports
|
|
import json
|
|
import random
|
|
from itertools import chain
|
|
|
|
# Django imports
|
|
from django.utils import timezone
|
|
from django.db.models import (
|
|
Prefetch,
|
|
OuterRef,
|
|
Func,
|
|
F,
|
|
Q,
|
|
Count,
|
|
Case,
|
|
Value,
|
|
CharField,
|
|
When,
|
|
Exists,
|
|
Max,
|
|
IntegerField,
|
|
)
|
|
from django.core.serializers.json import DjangoJSONEncoder
|
|
from django.utils.decorators import method_decorator
|
|
from django.views.decorators.gzip import gzip_page
|
|
from django.db import IntegrityError
|
|
|
|
# Third Party imports
|
|
from rest_framework.response import Response
|
|
from rest_framework import status
|
|
from rest_framework.parsers import MultiPartParser, FormParser
|
|
|
|
# Module imports
|
|
from . import BaseViewSet, BaseAPIView, WebhookMixin
|
|
from plane.app.serializers import (
|
|
IssueCreateSerializer,
|
|
IssueActivitySerializer,
|
|
IssueCommentSerializer,
|
|
IssuePropertySerializer,
|
|
IssueSerializer,
|
|
LabelSerializer,
|
|
IssueFlatSerializer,
|
|
IssueLinkSerializer,
|
|
IssueLiteSerializer,
|
|
IssueAttachmentSerializer,
|
|
IssueSubscriberSerializer,
|
|
ProjectMemberLiteSerializer,
|
|
IssueReactionSerializer,
|
|
CommentReactionSerializer,
|
|
IssueVoteSerializer,
|
|
IssueRelationSerializer,
|
|
RelatedIssueSerializer,
|
|
IssuePublicSerializer,
|
|
)
|
|
from plane.app.permissions import (
|
|
ProjectEntityPermission,
|
|
WorkSpaceAdminPermission,
|
|
ProjectMemberPermission,
|
|
ProjectLitePermission,
|
|
)
|
|
from plane.db.models import (
|
|
Project,
|
|
Issue,
|
|
IssueActivity,
|
|
IssueComment,
|
|
IssueProperty,
|
|
Label,
|
|
IssueLink,
|
|
IssueAttachment,
|
|
State,
|
|
IssueSubscriber,
|
|
ProjectMember,
|
|
IssueReaction,
|
|
CommentReaction,
|
|
ProjectDeployBoard,
|
|
IssueVote,
|
|
IssueRelation,
|
|
ProjectPublicMember,
|
|
)
|
|
from plane.bgtasks.issue_activites_task import issue_activity
|
|
from plane.utils.grouper import group_results
|
|
from plane.utils.issue_filters import issue_filters
|
|
|
|
|
|
class IssueViewSet(WebhookMixin, BaseViewSet):
|
|
def get_serializer_class(self):
|
|
return (
|
|
IssueCreateSerializer
|
|
if self.action in ["create", "update", "partial_update"]
|
|
else IssueSerializer
|
|
)
|
|
|
|
model = Issue
|
|
webhook_event = "issue"
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
search_fields = [
|
|
"name",
|
|
]
|
|
|
|
filterset_fields = [
|
|
"state__name",
|
|
"assignees__id",
|
|
"workspace__id",
|
|
]
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
Issue.issue_objects.annotate(
|
|
sub_issues_count=Issue.issue_objects.filter(parent=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.filter(project_id=self.kwargs.get("project_id"))
|
|
.filter(workspace__slug=self.kwargs.get("slug"))
|
|
.select_related("project")
|
|
.select_related("workspace")
|
|
.select_related("state")
|
|
.select_related("parent")
|
|
.prefetch_related("assignees")
|
|
.prefetch_related("labels")
|
|
.prefetch_related(
|
|
Prefetch(
|
|
"issue_reactions",
|
|
queryset=IssueReaction.objects.select_related("actor"),
|
|
)
|
|
)
|
|
).distinct()
|
|
|
|
@method_decorator(gzip_page)
|
|
def list(self, request, slug, project_id):
|
|
filters = issue_filters(request.query_params, "GET")
|
|
|
|
# Custom ordering for priority and state
|
|
priority_order = ["urgent", "high", "medium", "low", "none"]
|
|
state_order = ["backlog", "unstarted", "started", "completed", "cancelled"]
|
|
|
|
order_by_param = request.GET.get("order_by", "-created_at")
|
|
|
|
issue_queryset = (
|
|
self.get_queryset()
|
|
.filter(**filters)
|
|
.annotate(cycle_id=F("issue_cycle__cycle_id"))
|
|
.annotate(module_id=F("issue_module__module_id"))
|
|
.annotate(
|
|
link_count=IssueLink.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.annotate(
|
|
attachment_count=IssueAttachment.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
)
|
|
|
|
# Priority Ordering
|
|
if order_by_param == "priority" or order_by_param == "-priority":
|
|
priority_order = (
|
|
priority_order if order_by_param == "priority" else priority_order[::-1]
|
|
)
|
|
issue_queryset = issue_queryset.annotate(
|
|
priority_order=Case(
|
|
*[
|
|
When(priority=p, then=Value(i))
|
|
for i, p in enumerate(priority_order)
|
|
],
|
|
output_field=CharField(),
|
|
)
|
|
).order_by("priority_order")
|
|
|
|
# State Ordering
|
|
elif order_by_param in [
|
|
"state__name",
|
|
"state__group",
|
|
"-state__name",
|
|
"-state__group",
|
|
]:
|
|
state_order = (
|
|
state_order
|
|
if order_by_param in ["state__name", "state__group"]
|
|
else state_order[::-1]
|
|
)
|
|
issue_queryset = issue_queryset.annotate(
|
|
state_order=Case(
|
|
*[
|
|
When(state__group=state_group, then=Value(i))
|
|
for i, state_group in enumerate(state_order)
|
|
],
|
|
default=Value(len(state_order)),
|
|
output_field=CharField(),
|
|
)
|
|
).order_by("state_order")
|
|
# assignee and label ordering
|
|
elif order_by_param in [
|
|
"labels__name",
|
|
"-labels__name",
|
|
"assignees__first_name",
|
|
"-assignees__first_name",
|
|
]:
|
|
issue_queryset = issue_queryset.annotate(
|
|
max_values=Max(
|
|
order_by_param[1::]
|
|
if order_by_param.startswith("-")
|
|
else order_by_param
|
|
)
|
|
).order_by(
|
|
"-max_values" if order_by_param.startswith("-") else "max_values"
|
|
)
|
|
else:
|
|
issue_queryset = issue_queryset.order_by(order_by_param)
|
|
|
|
issues = IssueLiteSerializer(issue_queryset, many=True).data
|
|
|
|
## Grouping the results
|
|
group_by = request.GET.get("group_by", False)
|
|
sub_group_by = request.GET.get("sub_group_by", False)
|
|
if sub_group_by and sub_group_by == group_by:
|
|
return Response(
|
|
{"error": "Group by and sub group by cannot be same"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
if group_by:
|
|
grouped_results = group_results(issues, group_by, sub_group_by)
|
|
return Response(
|
|
grouped_results,
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
|
|
return Response(issues, status=status.HTTP_200_OK)
|
|
|
|
def create(self, request, slug, project_id):
|
|
project = Project.objects.get(pk=project_id)
|
|
|
|
serializer = IssueCreateSerializer(
|
|
data=request.data,
|
|
context={
|
|
"project_id": project_id,
|
|
"workspace_id": project.workspace_id,
|
|
"default_assignee_id": project.default_assignee_id,
|
|
},
|
|
)
|
|
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
|
|
# Track the issue
|
|
issue_activity.delay(
|
|
type="issue.activity.created",
|
|
requested_data=json.dumps(self.request.data, cls=DjangoJSONEncoder),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(serializer.data.get("id", None)),
|
|
project_id=str(project_id),
|
|
current_instance=None,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def retrieve(self, request, slug, project_id, pk=None):
|
|
issue = Issue.issue_objects.annotate(
|
|
sub_issues_count=Issue.issue_objects.filter(parent=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
).get(workspace__slug=slug, project_id=project_id, pk=pk)
|
|
return Response(IssueSerializer(issue).data, status=status.HTTP_200_OK)
|
|
|
|
def partial_update(self, request, slug, project_id, pk=None):
|
|
issue = Issue.objects.get(workspace__slug=slug, project_id=project_id, pk=pk)
|
|
current_instance = json.dumps(
|
|
IssueSerializer(issue).data, cls=DjangoJSONEncoder
|
|
)
|
|
requested_data = json.dumps(self.request.data, cls=DjangoJSONEncoder)
|
|
serializer = IssueCreateSerializer(issue, data=request.data, partial=True)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
issue_activity.delay(
|
|
type="issue.activity.updated",
|
|
requested_data=requested_data,
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(pk),
|
|
project_id=str(project_id),
|
|
current_instance=current_instance,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def destroy(self, request, slug, project_id, pk=None):
|
|
issue = Issue.objects.get(workspace__slug=slug, project_id=project_id, pk=pk)
|
|
current_instance = json.dumps(
|
|
IssueSerializer(issue).data, cls=DjangoJSONEncoder
|
|
)
|
|
issue.delete()
|
|
issue_activity.delay(
|
|
type="issue.activity.deleted",
|
|
requested_data=json.dumps({"issue_id": str(pk)}),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(pk),
|
|
project_id=str(project_id),
|
|
current_instance=current_instance,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class IssueListEndpoint(BaseAPIView):
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
def get(self, request, slug, project_id):
|
|
fields = [field for field in request.GET.get("fields", "").split(",") if field]
|
|
filters = issue_filters(request.query_params, "GET")
|
|
|
|
issue_queryset = (
|
|
Issue.objects.filter(workspace__slug=slug, project_id=project_id)
|
|
.select_related("project")
|
|
.select_related("workspace")
|
|
.select_related("state")
|
|
.select_related("parent")
|
|
.prefetch_related("assignees")
|
|
.prefetch_related("labels")
|
|
.prefetch_related(
|
|
Prefetch(
|
|
"issue_reactions",
|
|
queryset=IssueReaction.objects.select_related("actor"),
|
|
)
|
|
)
|
|
.filter(**filters)
|
|
.annotate(cycle_id=F("issue_cycle__cycle_id"))
|
|
.annotate(module_id=F("issue_module__module_id"))
|
|
.annotate(
|
|
link_count=IssueLink.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.annotate(
|
|
attachment_count=IssueAttachment.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.distinct()
|
|
)
|
|
|
|
serializer = IssueLiteSerializer(
|
|
issue_queryset, many=True, fields=fields if fields else None
|
|
)
|
|
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
|
|
class IssueListGroupedEndpoint(BaseAPIView):
|
|
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
def get(self, request, slug, project_id):
|
|
filters = issue_filters(request.query_params, "GET")
|
|
fields = [field for field in request.GET.get("fields", "").split(",") if field]
|
|
|
|
issue_queryset = (
|
|
Issue.objects.filter(workspace__slug=slug, project_id=project_id)
|
|
.select_related("project")
|
|
.select_related("workspace")
|
|
.select_related("state")
|
|
.select_related("parent")
|
|
.prefetch_related("assignees")
|
|
.prefetch_related("labels")
|
|
.prefetch_related(
|
|
Prefetch(
|
|
"issue_reactions",
|
|
queryset=IssueReaction.objects.select_related("actor"),
|
|
)
|
|
)
|
|
.filter(**filters)
|
|
.annotate(cycle_id=F("issue_cycle__cycle_id"))
|
|
.annotate(module_id=F("issue_module__module_id"))
|
|
.annotate(
|
|
link_count=IssueLink.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.annotate(
|
|
attachment_count=IssueAttachment.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.distinct()
|
|
)
|
|
|
|
issues = IssueLiteSerializer(issue_queryset, many=True, fields=fields if fields else None).data
|
|
issue_dict = {str(issue["id"]): issue for issue in issues}
|
|
return Response(
|
|
issue_dict,
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
|
|
|
|
class UserWorkSpaceIssues(BaseAPIView):
|
|
@method_decorator(gzip_page)
|
|
def get(self, request, slug):
|
|
filters = issue_filters(request.query_params, "GET")
|
|
# Custom ordering for priority and state
|
|
priority_order = ["urgent", "high", "medium", "low", "none"]
|
|
state_order = ["backlog", "unstarted", "started", "completed", "cancelled"]
|
|
|
|
order_by_param = request.GET.get("order_by", "-created_at")
|
|
|
|
issue_queryset = (
|
|
Issue.issue_objects.filter(
|
|
(
|
|
Q(assignees__in=[request.user])
|
|
| Q(created_by=request.user)
|
|
| Q(issue_subscribers__subscriber=request.user)
|
|
),
|
|
workspace__slug=slug,
|
|
)
|
|
.annotate(
|
|
sub_issues_count=Issue.issue_objects.filter(parent=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.select_related("project")
|
|
.select_related("workspace")
|
|
.select_related("state")
|
|
.select_related("parent")
|
|
.prefetch_related("assignees")
|
|
.prefetch_related("labels")
|
|
.order_by(order_by_param)
|
|
.annotate(
|
|
link_count=IssueLink.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.annotate(
|
|
attachment_count=IssueAttachment.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.prefetch_related(
|
|
Prefetch(
|
|
"issue_reactions",
|
|
queryset=IssueReaction.objects.select_related("actor"),
|
|
)
|
|
)
|
|
.filter(**filters)
|
|
).distinct()
|
|
|
|
# Priority Ordering
|
|
if order_by_param == "priority" or order_by_param == "-priority":
|
|
priority_order = (
|
|
priority_order if order_by_param == "priority" else priority_order[::-1]
|
|
)
|
|
issue_queryset = issue_queryset.annotate(
|
|
priority_order=Case(
|
|
*[
|
|
When(priority=p, then=Value(i))
|
|
for i, p in enumerate(priority_order)
|
|
],
|
|
output_field=CharField(),
|
|
)
|
|
).order_by("priority_order")
|
|
|
|
# State Ordering
|
|
elif order_by_param in [
|
|
"state__name",
|
|
"state__group",
|
|
"-state__name",
|
|
"-state__group",
|
|
]:
|
|
state_order = (
|
|
state_order
|
|
if order_by_param in ["state__name", "state__group"]
|
|
else state_order[::-1]
|
|
)
|
|
issue_queryset = issue_queryset.annotate(
|
|
state_order=Case(
|
|
*[
|
|
When(state__group=state_group, then=Value(i))
|
|
for i, state_group in enumerate(state_order)
|
|
],
|
|
default=Value(len(state_order)),
|
|
output_field=CharField(),
|
|
)
|
|
).order_by("state_order")
|
|
# assignee and label ordering
|
|
elif order_by_param in [
|
|
"labels__name",
|
|
"-labels__name",
|
|
"assignees__first_name",
|
|
"-assignees__first_name",
|
|
]:
|
|
issue_queryset = issue_queryset.annotate(
|
|
max_values=Max(
|
|
order_by_param[1::]
|
|
if order_by_param.startswith("-")
|
|
else order_by_param
|
|
)
|
|
).order_by(
|
|
"-max_values" if order_by_param.startswith("-") else "max_values"
|
|
)
|
|
else:
|
|
issue_queryset = issue_queryset.order_by(order_by_param)
|
|
|
|
issues = IssueLiteSerializer(issue_queryset, many=True).data
|
|
|
|
## Grouping the results
|
|
group_by = request.GET.get("group_by", False)
|
|
sub_group_by = request.GET.get("sub_group_by", False)
|
|
if sub_group_by and sub_group_by == group_by:
|
|
return Response(
|
|
{"error": "Group by and sub group by cannot be same"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
if group_by:
|
|
grouped_results = group_results(issues, group_by, sub_group_by)
|
|
return Response(
|
|
grouped_results,
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
|
|
return Response(issues, status=status.HTTP_200_OK)
|
|
|
|
|
|
class WorkSpaceIssuesEndpoint(BaseAPIView):
|
|
permission_classes = [
|
|
WorkSpaceAdminPermission,
|
|
]
|
|
|
|
@method_decorator(gzip_page)
|
|
def get(self, request, slug):
|
|
issues = (
|
|
Issue.issue_objects.filter(workspace__slug=slug)
|
|
.filter(project__project_projectmember__member=self.request.user)
|
|
.order_by("-created_at")
|
|
)
|
|
serializer = IssueSerializer(issues, many=True)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
|
|
class IssueActivityEndpoint(BaseAPIView):
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
@method_decorator(gzip_page)
|
|
def get(self, request, slug, project_id, issue_id):
|
|
issue_activities = (
|
|
IssueActivity.objects.filter(issue_id=issue_id)
|
|
.filter(
|
|
~Q(field__in=["comment", "vote", "reaction", "draft"]),
|
|
project__project_projectmember__member=self.request.user,
|
|
)
|
|
.select_related("actor", "workspace", "issue", "project")
|
|
).order_by("created_at")
|
|
issue_comments = (
|
|
IssueComment.objects.filter(issue_id=issue_id)
|
|
.filter(project__project_projectmember__member=self.request.user)
|
|
.order_by("created_at")
|
|
.select_related("actor", "issue", "project", "workspace")
|
|
.prefetch_related(
|
|
Prefetch(
|
|
"comment_reactions",
|
|
queryset=CommentReaction.objects.select_related("actor"),
|
|
)
|
|
)
|
|
)
|
|
issue_activities = IssueActivitySerializer(issue_activities, many=True).data
|
|
issue_comments = IssueCommentSerializer(issue_comments, many=True).data
|
|
|
|
result_list = sorted(
|
|
chain(issue_activities, issue_comments),
|
|
key=lambda instance: instance["created_at"],
|
|
)
|
|
|
|
return Response(result_list, status=status.HTTP_200_OK)
|
|
|
|
|
|
class IssueCommentViewSet(WebhookMixin, BaseViewSet):
|
|
serializer_class = IssueCommentSerializer
|
|
model = IssueComment
|
|
webhook_event = "issue_comment"
|
|
permission_classes = [
|
|
ProjectLitePermission,
|
|
]
|
|
|
|
filterset_fields = [
|
|
"issue__id",
|
|
"workspace__id",
|
|
]
|
|
|
|
def get_queryset(self):
|
|
return self.filter_queryset(
|
|
super()
|
|
.get_queryset()
|
|
.filter(workspace__slug=self.kwargs.get("slug"))
|
|
.filter(project_id=self.kwargs.get("project_id"))
|
|
.filter(issue_id=self.kwargs.get("issue_id"))
|
|
.filter(project__project_projectmember__member=self.request.user)
|
|
.select_related("project")
|
|
.select_related("workspace")
|
|
.select_related("issue")
|
|
.annotate(
|
|
is_member=Exists(
|
|
ProjectMember.objects.filter(
|
|
workspace__slug=self.kwargs.get("slug"),
|
|
project_id=self.kwargs.get("project_id"),
|
|
member_id=self.request.user.id,
|
|
is_active=True,
|
|
)
|
|
)
|
|
)
|
|
.distinct()
|
|
)
|
|
|
|
def create(self, request, slug, project_id, issue_id):
|
|
serializer = IssueCommentSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save(
|
|
project_id=project_id,
|
|
issue_id=issue_id,
|
|
actor=request.user,
|
|
)
|
|
issue_activity.delay(
|
|
type="comment.activity.created",
|
|
requested_data=json.dumps(serializer.data, cls=DjangoJSONEncoder),
|
|
actor_id=str(self.request.user.id),
|
|
issue_id=str(self.kwargs.get("issue_id")),
|
|
project_id=str(self.kwargs.get("project_id")),
|
|
current_instance=None,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def partial_update(self, request, slug, project_id, issue_id, pk):
|
|
issue_comment = IssueComment.objects.get(
|
|
workspace__slug=slug, project_id=project_id, issue_id=issue_id, pk=pk
|
|
)
|
|
requested_data = json.dumps(self.request.data, cls=DjangoJSONEncoder)
|
|
current_instance = json.dumps(
|
|
IssueCommentSerializer(issue_comment).data,
|
|
cls=DjangoJSONEncoder,
|
|
)
|
|
serializer = IssueCommentSerializer(
|
|
issue_comment, data=request.data, partial=True
|
|
)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
issue_activity.delay(
|
|
type="comment.activity.updated",
|
|
requested_data=requested_data,
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(issue_id),
|
|
project_id=str(project_id),
|
|
current_instance=current_instance,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def destroy(self, request, slug, project_id, issue_id, pk):
|
|
issue_comment = IssueComment.objects.get(
|
|
workspace__slug=slug, project_id=project_id, issue_id=issue_id, pk=pk
|
|
)
|
|
current_instance = json.dumps(
|
|
IssueCommentSerializer(issue_comment).data,
|
|
cls=DjangoJSONEncoder,
|
|
)
|
|
issue_comment.delete()
|
|
issue_activity.delay(
|
|
type="comment.activity.deleted",
|
|
requested_data=json.dumps({"comment_id": str(pk)}),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(issue_id),
|
|
project_id=str(project_id),
|
|
current_instance=current_instance,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class IssueUserDisplayPropertyEndpoint(BaseAPIView):
|
|
permission_classes = [
|
|
ProjectLitePermission,
|
|
]
|
|
|
|
def post(self, request, slug, project_id):
|
|
issue_property, created = IssueProperty.objects.get_or_create(
|
|
user=request.user,
|
|
project_id=project_id,
|
|
)
|
|
|
|
if not created:
|
|
issue_property.properties = request.data.get("properties", {})
|
|
issue_property.save()
|
|
issue_property.properties = request.data.get("properties", {})
|
|
issue_property.save()
|
|
serializer = IssuePropertySerializer(issue_property)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
|
|
def get(self, request, slug, project_id):
|
|
issue_property, _ = IssueProperty.objects.get_or_create(
|
|
user=request.user, project_id=project_id
|
|
)
|
|
serializer = IssuePropertySerializer(issue_property)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
|
|
class LabelViewSet(BaseViewSet):
|
|
serializer_class = LabelSerializer
|
|
model = Label
|
|
permission_classes = [
|
|
ProjectMemberPermission,
|
|
]
|
|
|
|
def create(self, request, slug, project_id):
|
|
try:
|
|
serializer = LabelSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save(project_id=project_id)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
except IntegrityError:
|
|
return Response(
|
|
{"error": "Label with the same name already exists in the project"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
def get_queryset(self):
|
|
return self.filter_queryset(
|
|
super()
|
|
.get_queryset()
|
|
.filter(workspace__slug=self.kwargs.get("slug"))
|
|
.filter(project_id=self.kwargs.get("project_id"))
|
|
.filter(project__project_projectmember__member=self.request.user)
|
|
.select_related("project")
|
|
.select_related("workspace")
|
|
.select_related("parent")
|
|
.distinct()
|
|
.order_by("sort_order")
|
|
)
|
|
|
|
|
|
class BulkDeleteIssuesEndpoint(BaseAPIView):
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
def delete(self, request, slug, project_id):
|
|
issue_ids = request.data.get("issue_ids", [])
|
|
|
|
if not len(issue_ids):
|
|
return Response(
|
|
{"error": "Issue IDs are required"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
issues = Issue.issue_objects.filter(
|
|
workspace__slug=slug, project_id=project_id, pk__in=issue_ids
|
|
)
|
|
|
|
total_issues = len(issues)
|
|
|
|
issues.delete()
|
|
|
|
return Response(
|
|
{"message": f"{total_issues} issues were deleted"},
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
|
|
|
|
class SubIssuesEndpoint(BaseAPIView):
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
@method_decorator(gzip_page)
|
|
def get(self, request, slug, project_id, issue_id):
|
|
sub_issues = (
|
|
Issue.issue_objects.filter(parent_id=issue_id, workspace__slug=slug)
|
|
.select_related("project")
|
|
.select_related("workspace")
|
|
.select_related("state")
|
|
.select_related("parent")
|
|
.prefetch_related("assignees")
|
|
.prefetch_related("labels")
|
|
.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=IssueAttachment.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.prefetch_related(
|
|
Prefetch(
|
|
"issue_reactions",
|
|
queryset=IssueReaction.objects.select_related("actor"),
|
|
)
|
|
)
|
|
)
|
|
|
|
state_distribution = (
|
|
State.objects.filter(workspace__slug=slug, state_issue__parent_id=issue_id)
|
|
.annotate(state_group=F("group"))
|
|
.values("state_group")
|
|
.annotate(state_count=Count("state_group"))
|
|
.order_by("state_group")
|
|
)
|
|
|
|
result = {
|
|
item["state_group"]: item["state_count"] for item in state_distribution
|
|
}
|
|
|
|
serializer = IssueLiteSerializer(
|
|
sub_issues,
|
|
many=True,
|
|
)
|
|
return Response(
|
|
{
|
|
"sub_issues": serializer.data,
|
|
"state_distribution": result,
|
|
},
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
|
|
# Assign multiple sub issues
|
|
def post(self, request, slug, project_id, issue_id):
|
|
parent_issue = Issue.issue_objects.get(pk=issue_id)
|
|
sub_issue_ids = request.data.get("sub_issue_ids", [])
|
|
|
|
if not len(sub_issue_ids):
|
|
return Response(
|
|
{"error": "Sub Issue IDs are required"},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
sub_issues = Issue.issue_objects.filter(id__in=sub_issue_ids)
|
|
|
|
for sub_issue in sub_issues:
|
|
sub_issue.parent = parent_issue
|
|
|
|
_ = Issue.objects.bulk_update(sub_issues, ["parent"], batch_size=10)
|
|
|
|
updated_sub_issues = Issue.issue_objects.filter(id__in=sub_issue_ids)
|
|
|
|
# Track the issue
|
|
_ = [
|
|
issue_activity.delay(
|
|
type="issue.activity.updated",
|
|
requested_data=json.dumps({"parent": str(issue_id)}),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(sub_issue_id),
|
|
project_id=str(project_id),
|
|
current_instance=json.dumps({"parent": str(sub_issue_id)}),
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
for sub_issue_id in sub_issue_ids
|
|
]
|
|
|
|
return Response(
|
|
IssueFlatSerializer(updated_sub_issues, many=True).data,
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
|
|
|
|
class IssueLinkViewSet(BaseViewSet):
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
model = IssueLink
|
|
serializer_class = IssueLinkSerializer
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
super()
|
|
.get_queryset()
|
|
.filter(workspace__slug=self.kwargs.get("slug"))
|
|
.filter(project_id=self.kwargs.get("project_id"))
|
|
.filter(issue_id=self.kwargs.get("issue_id"))
|
|
.filter(project__project_projectmember__member=self.request.user)
|
|
.order_by("-created_at")
|
|
.distinct()
|
|
)
|
|
|
|
def create(self, request, slug, project_id, issue_id):
|
|
serializer = IssueLinkSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save(
|
|
project_id=project_id,
|
|
issue_id=issue_id,
|
|
)
|
|
issue_activity.delay(
|
|
type="link.activity.created",
|
|
requested_data=json.dumps(serializer.data, cls=DjangoJSONEncoder),
|
|
actor_id=str(self.request.user.id),
|
|
issue_id=str(self.kwargs.get("issue_id")),
|
|
project_id=str(self.kwargs.get("project_id")),
|
|
current_instance=None,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def partial_update(self, request, slug, project_id, issue_id, pk):
|
|
issue_link = IssueLink.objects.get(
|
|
workspace__slug=slug, project_id=project_id, issue_id=issue_id, pk=pk
|
|
)
|
|
requested_data = json.dumps(request.data, cls=DjangoJSONEncoder)
|
|
current_instance = json.dumps(
|
|
IssueLinkSerializer(issue_link).data,
|
|
cls=DjangoJSONEncoder,
|
|
)
|
|
serializer = IssueLinkSerializer(issue_link, data=request.data, partial=True)
|
|
if serializer.is_valid():
|
|
serializer.save()
|
|
issue_activity.delay(
|
|
type="link.activity.updated",
|
|
requested_data=requested_data,
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(issue_id),
|
|
project_id=str(project_id),
|
|
current_instance=current_instance,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def destroy(self, request, slug, project_id, issue_id, pk):
|
|
issue_link = IssueLink.objects.get(
|
|
workspace__slug=slug, project_id=project_id, issue_id=issue_id, pk=pk
|
|
)
|
|
current_instance = json.dumps(
|
|
IssueLinkSerializer(issue_link).data,
|
|
cls=DjangoJSONEncoder,
|
|
)
|
|
issue_activity.delay(
|
|
type="link.activity.deleted",
|
|
requested_data=json.dumps({"link_id": str(pk)}),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(issue_id),
|
|
project_id=str(project_id),
|
|
current_instance=current_instance,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
issue_link.delete()
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class BulkCreateIssueLabelsEndpoint(BaseAPIView):
|
|
def post(self, request, slug, project_id):
|
|
label_data = request.data.get("label_data", [])
|
|
project = Project.objects.get(pk=project_id)
|
|
|
|
labels = Label.objects.bulk_create(
|
|
[
|
|
Label(
|
|
name=label.get("name", "Migrated"),
|
|
description=label.get("description", "Migrated Issue"),
|
|
color="#" + "%06x" % random.randint(0, 0xFFFFFF),
|
|
project_id=project_id,
|
|
workspace_id=project.workspace_id,
|
|
created_by=request.user,
|
|
updated_by=request.user,
|
|
)
|
|
for label in label_data
|
|
],
|
|
batch_size=50,
|
|
ignore_conflicts=True,
|
|
)
|
|
|
|
return Response(
|
|
{"labels": LabelSerializer(labels, many=True).data},
|
|
status=status.HTTP_201_CREATED,
|
|
)
|
|
|
|
|
|
class IssueAttachmentEndpoint(BaseAPIView):
|
|
serializer_class = IssueAttachmentSerializer
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
model = IssueAttachment
|
|
parser_classes = (MultiPartParser, FormParser)
|
|
|
|
def post(self, request, slug, project_id, issue_id):
|
|
serializer = IssueAttachmentSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save(project_id=project_id, issue_id=issue_id)
|
|
issue_activity.delay(
|
|
type="attachment.activity.created",
|
|
requested_data=None,
|
|
actor_id=str(self.request.user.id),
|
|
issue_id=str(self.kwargs.get("issue_id", None)),
|
|
project_id=str(self.kwargs.get("project_id", None)),
|
|
current_instance=json.dumps(
|
|
serializer.data,
|
|
cls=DjangoJSONEncoder,
|
|
),
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def delete(self, request, slug, project_id, issue_id, pk):
|
|
issue_attachment = IssueAttachment.objects.get(pk=pk)
|
|
issue_attachment.asset.delete(save=False)
|
|
issue_attachment.delete()
|
|
issue_activity.delay(
|
|
type="attachment.activity.deleted",
|
|
requested_data=None,
|
|
actor_id=str(self.request.user.id),
|
|
issue_id=str(self.kwargs.get("issue_id", None)),
|
|
project_id=str(self.kwargs.get("project_id", None)),
|
|
current_instance=None,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
def get(self, request, slug, project_id, issue_id):
|
|
issue_attachments = IssueAttachment.objects.filter(
|
|
issue_id=issue_id, workspace__slug=slug, project_id=project_id
|
|
)
|
|
serializer = IssueAttachmentSerializer(issue_attachments, many=True)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
|
|
class IssueArchiveViewSet(BaseViewSet):
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
serializer_class = IssueFlatSerializer
|
|
model = Issue
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
Issue.objects.annotate(
|
|
sub_issues_count=Issue.objects.filter(parent=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.filter(archived_at__isnull=False)
|
|
.filter(project_id=self.kwargs.get("project_id"))
|
|
.filter(workspace__slug=self.kwargs.get("slug"))
|
|
.select_related("project")
|
|
.select_related("workspace")
|
|
.select_related("state")
|
|
.select_related("parent")
|
|
.prefetch_related("assignees")
|
|
.prefetch_related("labels")
|
|
)
|
|
|
|
@method_decorator(gzip_page)
|
|
def list(self, request, slug, project_id):
|
|
filters = issue_filters(request.query_params, "GET")
|
|
show_sub_issues = request.GET.get("show_sub_issues", "true")
|
|
|
|
# Custom ordering for priority and state
|
|
priority_order = ["urgent", "high", "medium", "low", "none"]
|
|
state_order = ["backlog", "unstarted", "started", "completed", "cancelled"]
|
|
|
|
order_by_param = request.GET.get("order_by", "-created_at")
|
|
|
|
issue_queryset = (
|
|
self.get_queryset()
|
|
.filter(**filters)
|
|
.annotate(cycle_id=F("issue_cycle__cycle_id"))
|
|
.annotate(module_id=F("issue_module__module_id"))
|
|
.annotate(
|
|
link_count=IssueLink.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.annotate(
|
|
attachment_count=IssueAttachment.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
)
|
|
|
|
# Priority Ordering
|
|
if order_by_param == "priority" or order_by_param == "-priority":
|
|
priority_order = (
|
|
priority_order if order_by_param == "priority" else priority_order[::-1]
|
|
)
|
|
issue_queryset = issue_queryset.annotate(
|
|
priority_order=Case(
|
|
*[
|
|
When(priority=p, then=Value(i))
|
|
for i, p in enumerate(priority_order)
|
|
],
|
|
output_field=CharField(),
|
|
)
|
|
).order_by("priority_order")
|
|
|
|
# State Ordering
|
|
elif order_by_param in [
|
|
"state__name",
|
|
"state__group",
|
|
"-state__name",
|
|
"-state__group",
|
|
]:
|
|
state_order = (
|
|
state_order
|
|
if order_by_param in ["state__name", "state__group"]
|
|
else state_order[::-1]
|
|
)
|
|
issue_queryset = issue_queryset.annotate(
|
|
state_order=Case(
|
|
*[
|
|
When(state__group=state_group, then=Value(i))
|
|
for i, state_group in enumerate(state_order)
|
|
],
|
|
default=Value(len(state_order)),
|
|
output_field=CharField(),
|
|
)
|
|
).order_by("state_order")
|
|
# assignee and label ordering
|
|
elif order_by_param in [
|
|
"labels__name",
|
|
"-labels__name",
|
|
"assignees__first_name",
|
|
"-assignees__first_name",
|
|
]:
|
|
issue_queryset = issue_queryset.annotate(
|
|
max_values=Max(
|
|
order_by_param[1::]
|
|
if order_by_param.startswith("-")
|
|
else order_by_param
|
|
)
|
|
).order_by(
|
|
"-max_values" if order_by_param.startswith("-") else "max_values"
|
|
)
|
|
else:
|
|
issue_queryset = issue_queryset.order_by(order_by_param)
|
|
|
|
issue_queryset = (
|
|
issue_queryset
|
|
if show_sub_issues == "true"
|
|
else issue_queryset.filter(parent__isnull=True)
|
|
)
|
|
|
|
issues = IssueLiteSerializer(issue_queryset, many=True).data
|
|
|
|
## Grouping the results
|
|
group_by = request.GET.get("group_by", False)
|
|
if group_by:
|
|
return Response(group_results(issues, group_by), status=status.HTTP_200_OK)
|
|
|
|
return Response(issues, status=status.HTTP_200_OK)
|
|
|
|
def retrieve(self, request, slug, project_id, pk=None):
|
|
issue = Issue.objects.get(
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
archived_at__isnull=False,
|
|
pk=pk,
|
|
)
|
|
return Response(IssueSerializer(issue).data, status=status.HTTP_200_OK)
|
|
|
|
def unarchive(self, request, slug, project_id, pk=None):
|
|
issue = Issue.objects.get(
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
archived_at__isnull=False,
|
|
pk=pk,
|
|
)
|
|
issue_activity.delay(
|
|
type="issue.activity.updated",
|
|
requested_data=json.dumps({"archived_at": None}),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(issue.id),
|
|
project_id=str(project_id),
|
|
current_instance=json.dumps(
|
|
IssueSerializer(issue).data, cls=DjangoJSONEncoder
|
|
),
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
issue.archived_at = None
|
|
issue.save()
|
|
|
|
return Response(IssueSerializer(issue).data, status=status.HTTP_200_OK)
|
|
|
|
|
|
class IssueSubscriberViewSet(BaseViewSet):
|
|
serializer_class = IssueSubscriberSerializer
|
|
model = IssueSubscriber
|
|
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
def get_permissions(self):
|
|
if self.action in ["subscribe", "unsubscribe", "subscription_status"]:
|
|
self.permission_classes = [
|
|
ProjectLitePermission,
|
|
]
|
|
else:
|
|
self.permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
return super(IssueSubscriberViewSet, self).get_permissions()
|
|
|
|
def perform_create(self, serializer):
|
|
serializer.save(
|
|
project_id=self.kwargs.get("project_id"),
|
|
issue_id=self.kwargs.get("issue_id"),
|
|
)
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
super()
|
|
.get_queryset()
|
|
.filter(workspace__slug=self.kwargs.get("slug"))
|
|
.filter(project_id=self.kwargs.get("project_id"))
|
|
.filter(issue_id=self.kwargs.get("issue_id"))
|
|
.filter(project__project_projectmember__member=self.request.user)
|
|
.order_by("-created_at")
|
|
.distinct()
|
|
)
|
|
|
|
def list(self, request, slug, project_id, issue_id):
|
|
members = (
|
|
ProjectMember.objects.filter(
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
is_active=True,
|
|
)
|
|
.annotate(
|
|
is_subscribed=Exists(
|
|
IssueSubscriber.objects.filter(
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
issue_id=issue_id,
|
|
subscriber=OuterRef("member"),
|
|
)
|
|
)
|
|
)
|
|
.select_related("member")
|
|
)
|
|
serializer = ProjectMemberLiteSerializer(members, many=True)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
|
|
def destroy(self, request, slug, project_id, issue_id, subscriber_id):
|
|
issue_subscriber = IssueSubscriber.objects.get(
|
|
project=project_id,
|
|
subscriber=subscriber_id,
|
|
workspace__slug=slug,
|
|
issue=issue_id,
|
|
)
|
|
issue_subscriber.delete()
|
|
return Response(
|
|
status=status.HTTP_204_NO_CONTENT,
|
|
)
|
|
|
|
def subscribe(self, request, slug, project_id, issue_id):
|
|
if IssueSubscriber.objects.filter(
|
|
issue_id=issue_id,
|
|
subscriber=request.user,
|
|
workspace__slug=slug,
|
|
project=project_id,
|
|
).exists():
|
|
return Response(
|
|
{"message": "User already subscribed to the issue."},
|
|
status=status.HTTP_400_BAD_REQUEST,
|
|
)
|
|
|
|
subscriber = IssueSubscriber.objects.create(
|
|
issue_id=issue_id,
|
|
subscriber_id=request.user.id,
|
|
project_id=project_id,
|
|
)
|
|
serializer = IssueSubscriberSerializer(subscriber)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
|
|
def unsubscribe(self, request, slug, project_id, issue_id):
|
|
issue_subscriber = IssueSubscriber.objects.get(
|
|
project=project_id,
|
|
subscriber=request.user,
|
|
workspace__slug=slug,
|
|
issue=issue_id,
|
|
)
|
|
issue_subscriber.delete()
|
|
return Response(
|
|
status=status.HTTP_204_NO_CONTENT,
|
|
)
|
|
|
|
def subscription_status(self, request, slug, project_id, issue_id):
|
|
issue_subscriber = IssueSubscriber.objects.filter(
|
|
issue=issue_id,
|
|
subscriber=request.user,
|
|
workspace__slug=slug,
|
|
project=project_id,
|
|
).exists()
|
|
return Response({"subscribed": issue_subscriber}, status=status.HTTP_200_OK)
|
|
|
|
|
|
class IssueReactionViewSet(BaseViewSet):
|
|
serializer_class = IssueReactionSerializer
|
|
model = IssueReaction
|
|
permission_classes = [
|
|
ProjectLitePermission,
|
|
]
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
super()
|
|
.get_queryset()
|
|
.filter(workspace__slug=self.kwargs.get("slug"))
|
|
.filter(project_id=self.kwargs.get("project_id"))
|
|
.filter(issue_id=self.kwargs.get("issue_id"))
|
|
.filter(project__project_projectmember__member=self.request.user)
|
|
.order_by("-created_at")
|
|
.distinct()
|
|
)
|
|
|
|
def create(self, request, slug, project_id, issue_id):
|
|
serializer = IssueReactionSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save(
|
|
issue_id=issue_id,
|
|
project_id=project_id,
|
|
actor=request.user,
|
|
)
|
|
issue_activity.delay(
|
|
type="issue_reaction.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_id),
|
|
current_instance=None,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def destroy(self, request, slug, project_id, issue_id, reaction_code):
|
|
issue_reaction = IssueReaction.objects.get(
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
issue_id=issue_id,
|
|
reaction=reaction_code,
|
|
actor=request.user,
|
|
)
|
|
issue_activity.delay(
|
|
type="issue_reaction.activity.deleted",
|
|
requested_data=None,
|
|
actor_id=str(self.request.user.id),
|
|
issue_id=str(self.kwargs.get("issue_id", None)),
|
|
project_id=str(self.kwargs.get("project_id", None)),
|
|
current_instance=json.dumps(
|
|
{
|
|
"reaction": str(reaction_code),
|
|
"identifier": str(issue_reaction.id),
|
|
}
|
|
),
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
issue_reaction.delete()
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class CommentReactionViewSet(BaseViewSet):
|
|
serializer_class = CommentReactionSerializer
|
|
model = CommentReaction
|
|
permission_classes = [
|
|
ProjectLitePermission,
|
|
]
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
super()
|
|
.get_queryset()
|
|
.filter(workspace__slug=self.kwargs.get("slug"))
|
|
.filter(project_id=self.kwargs.get("project_id"))
|
|
.filter(comment_id=self.kwargs.get("comment_id"))
|
|
.filter(project__project_projectmember__member=self.request.user)
|
|
.order_by("-created_at")
|
|
.distinct()
|
|
)
|
|
|
|
def create(self, request, slug, project_id, comment_id):
|
|
serializer = CommentReactionSerializer(data=request.data)
|
|
if serializer.is_valid():
|
|
serializer.save(
|
|
project_id=project_id,
|
|
actor_id=request.user.id,
|
|
comment_id=comment_id,
|
|
)
|
|
issue_activity.delay(
|
|
type="comment_reaction.activity.created",
|
|
requested_data=json.dumps(request.data, cls=DjangoJSONEncoder),
|
|
actor_id=str(request.user.id),
|
|
issue_id=None,
|
|
project_id=str(project_id),
|
|
current_instance=None,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def destroy(self, request, slug, project_id, comment_id, reaction_code):
|
|
comment_reaction = CommentReaction.objects.get(
|
|
workspace__slug=slug,
|
|
project_id=project_id,
|
|
comment_id=comment_id,
|
|
reaction=reaction_code,
|
|
actor=request.user,
|
|
)
|
|
issue_activity.delay(
|
|
type="comment_reaction.activity.deleted",
|
|
requested_data=None,
|
|
actor_id=str(self.request.user.id),
|
|
issue_id=None,
|
|
project_id=str(self.kwargs.get("project_id", None)),
|
|
current_instance=json.dumps(
|
|
{
|
|
"reaction": str(reaction_code),
|
|
"identifier": str(comment_reaction.id),
|
|
"comment_id": str(comment_id),
|
|
}
|
|
),
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
comment_reaction.delete()
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class IssueRelationViewSet(BaseViewSet):
|
|
serializer_class = IssueRelationSerializer
|
|
model = IssueRelation
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
|
|
def get_queryset(self):
|
|
return self.filter_queryset(
|
|
super()
|
|
.get_queryset()
|
|
.filter(workspace__slug=self.kwargs.get("slug"))
|
|
.filter(project_id=self.kwargs.get("project_id"))
|
|
.filter(issue_id=self.kwargs.get("issue_id"))
|
|
.filter(project__project_projectmember__member=self.request.user)
|
|
.select_related("project")
|
|
.select_related("workspace")
|
|
.select_related("issue")
|
|
.distinct()
|
|
)
|
|
|
|
def create(self, request, slug, project_id, issue_id):
|
|
related_list = request.data.get("related_list", [])
|
|
relation = request.data.get("relation", None)
|
|
project = Project.objects.get(pk=project_id)
|
|
|
|
issue_relation = IssueRelation.objects.bulk_create(
|
|
[
|
|
IssueRelation(
|
|
issue_id=related_issue["issue"],
|
|
related_issue_id=related_issue["related_issue"],
|
|
relation_type=related_issue["relation_type"],
|
|
project_id=project_id,
|
|
workspace_id=project.workspace_id,
|
|
created_by=request.user,
|
|
updated_by=request.user,
|
|
)
|
|
for related_issue in related_list
|
|
],
|
|
batch_size=10,
|
|
ignore_conflicts=True,
|
|
)
|
|
|
|
issue_activity.delay(
|
|
type="issue_relation.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_id),
|
|
current_instance=None,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
|
|
if relation == "blocking":
|
|
return Response(
|
|
RelatedIssueSerializer(issue_relation, many=True).data,
|
|
status=status.HTTP_201_CREATED,
|
|
)
|
|
else:
|
|
return Response(
|
|
IssueRelationSerializer(issue_relation, many=True).data,
|
|
status=status.HTTP_201_CREATED,
|
|
)
|
|
|
|
def destroy(self, request, slug, project_id, issue_id, pk):
|
|
issue_relation = IssueRelation.objects.get(
|
|
workspace__slug=slug, project_id=project_id, issue_id=issue_id, pk=pk
|
|
)
|
|
current_instance = json.dumps(
|
|
IssueRelationSerializer(issue_relation).data,
|
|
cls=DjangoJSONEncoder,
|
|
)
|
|
issue_relation.delete()
|
|
issue_activity.delay(
|
|
type="issue_relation.activity.deleted",
|
|
requested_data=json.dumps({"related_list": None}),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(issue_id),
|
|
project_id=str(project_id),
|
|
current_instance=current_instance,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(status=status.HTTP_204_NO_CONTENT)
|
|
|
|
|
|
class IssueDraftViewSet(BaseViewSet):
|
|
permission_classes = [
|
|
ProjectEntityPermission,
|
|
]
|
|
serializer_class = IssueFlatSerializer
|
|
model = Issue
|
|
|
|
def get_queryset(self):
|
|
return (
|
|
Issue.objects.annotate(
|
|
sub_issues_count=Issue.issue_objects.filter(parent=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.filter(project_id=self.kwargs.get("project_id"))
|
|
.filter(workspace__slug=self.kwargs.get("slug"))
|
|
.filter(is_draft=True)
|
|
.select_related("project")
|
|
.select_related("workspace")
|
|
.select_related("state")
|
|
.select_related("parent")
|
|
.prefetch_related("assignees")
|
|
.prefetch_related("labels")
|
|
.prefetch_related(
|
|
Prefetch(
|
|
"issue_reactions",
|
|
queryset=IssueReaction.objects.select_related("actor"),
|
|
)
|
|
)
|
|
)
|
|
|
|
@method_decorator(gzip_page)
|
|
def list(self, request, slug, project_id):
|
|
filters = issue_filters(request.query_params, "GET")
|
|
|
|
# Custom ordering for priority and state
|
|
priority_order = ["urgent", "high", "medium", "low", "none"]
|
|
state_order = ["backlog", "unstarted", "started", "completed", "cancelled"]
|
|
|
|
order_by_param = request.GET.get("order_by", "-created_at")
|
|
|
|
issue_queryset = (
|
|
self.get_queryset()
|
|
.filter(**filters)
|
|
.annotate(cycle_id=F("issue_cycle__cycle_id"))
|
|
.annotate(module_id=F("issue_module__module_id"))
|
|
.annotate(
|
|
link_count=IssueLink.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
.annotate(
|
|
attachment_count=IssueAttachment.objects.filter(issue=OuterRef("id"))
|
|
.order_by()
|
|
.annotate(count=Func(F("id"), function="Count"))
|
|
.values("count")
|
|
)
|
|
)
|
|
|
|
# Priority Ordering
|
|
if order_by_param == "priority" or order_by_param == "-priority":
|
|
priority_order = (
|
|
priority_order if order_by_param == "priority" else priority_order[::-1]
|
|
)
|
|
issue_queryset = issue_queryset.annotate(
|
|
priority_order=Case(
|
|
*[
|
|
When(priority=p, then=Value(i))
|
|
for i, p in enumerate(priority_order)
|
|
],
|
|
output_field=CharField(),
|
|
)
|
|
).order_by("priority_order")
|
|
|
|
# State Ordering
|
|
elif order_by_param in [
|
|
"state__name",
|
|
"state__group",
|
|
"-state__name",
|
|
"-state__group",
|
|
]:
|
|
state_order = (
|
|
state_order
|
|
if order_by_param in ["state__name", "state__group"]
|
|
else state_order[::-1]
|
|
)
|
|
issue_queryset = issue_queryset.annotate(
|
|
state_order=Case(
|
|
*[
|
|
When(state__group=state_group, then=Value(i))
|
|
for i, state_group in enumerate(state_order)
|
|
],
|
|
default=Value(len(state_order)),
|
|
output_field=CharField(),
|
|
)
|
|
).order_by("state_order")
|
|
# assignee and label ordering
|
|
elif order_by_param in [
|
|
"labels__name",
|
|
"-labels__name",
|
|
"assignees__first_name",
|
|
"-assignees__first_name",
|
|
]:
|
|
issue_queryset = issue_queryset.annotate(
|
|
max_values=Max(
|
|
order_by_param[1::]
|
|
if order_by_param.startswith("-")
|
|
else order_by_param
|
|
)
|
|
).order_by(
|
|
"-max_values" if order_by_param.startswith("-") else "max_values"
|
|
)
|
|
else:
|
|
issue_queryset = issue_queryset.order_by(order_by_param)
|
|
|
|
issues = IssueLiteSerializer(issue_queryset, many=True).data
|
|
|
|
## Grouping the results
|
|
group_by = request.GET.get("group_by", False)
|
|
if group_by:
|
|
grouped_results = group_results(issues, group_by)
|
|
return Response(
|
|
grouped_results,
|
|
status=status.HTTP_200_OK,
|
|
)
|
|
|
|
return Response(issues, status=status.HTTP_200_OK)
|
|
|
|
def create(self, request, slug, project_id):
|
|
project = Project.objects.get(pk=project_id)
|
|
|
|
serializer = IssueCreateSerializer(
|
|
data=request.data,
|
|
context={
|
|
"project_id": project_id,
|
|
"workspace_id": project.workspace_id,
|
|
"default_assignee_id": project.default_assignee_id,
|
|
},
|
|
)
|
|
|
|
if serializer.is_valid():
|
|
serializer.save(is_draft=True)
|
|
|
|
# Track the issue
|
|
issue_activity.delay(
|
|
type="issue_draft.activity.created",
|
|
requested_data=json.dumps(self.request.data, cls=DjangoJSONEncoder),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(serializer.data.get("id", None)),
|
|
project_id=str(project_id),
|
|
current_instance=None,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_201_CREATED)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def partial_update(self, request, slug, project_id, pk):
|
|
issue = Issue.objects.get(workspace__slug=slug, project_id=project_id, pk=pk)
|
|
serializer = IssueSerializer(issue, data=request.data, partial=True)
|
|
|
|
if serializer.is_valid():
|
|
if request.data.get("is_draft") is not None and not request.data.get(
|
|
"is_draft"
|
|
):
|
|
serializer.save(created_at=timezone.now(), updated_at=timezone.now())
|
|
else:
|
|
serializer.save()
|
|
issue_activity.delay(
|
|
type="issue_draft.activity.updated",
|
|
requested_data=json.dumps(request.data, cls=DjangoJSONEncoder),
|
|
actor_id=str(self.request.user.id),
|
|
issue_id=str(self.kwargs.get("pk", None)),
|
|
project_id=str(self.kwargs.get("project_id", None)),
|
|
current_instance=json.dumps(
|
|
IssueSerializer(issue).data,
|
|
cls=DjangoJSONEncoder,
|
|
),
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(serializer.data, status=status.HTTP_200_OK)
|
|
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
|
|
|
|
def retrieve(self, request, slug, project_id, pk=None):
|
|
issue = Issue.objects.get(
|
|
workspace__slug=slug, project_id=project_id, pk=pk, is_draft=True
|
|
)
|
|
return Response(IssueSerializer(issue).data, status=status.HTTP_200_OK)
|
|
|
|
def destroy(self, request, slug, project_id, pk=None):
|
|
issue = Issue.objects.get(workspace__slug=slug, project_id=project_id, pk=pk)
|
|
current_instance = json.dumps(
|
|
IssueSerializer(issue).data, cls=DjangoJSONEncoder
|
|
)
|
|
issue.delete()
|
|
issue_activity.delay(
|
|
type="issue_draft.activity.deleted",
|
|
requested_data=json.dumps({"issue_id": str(pk)}),
|
|
actor_id=str(request.user.id),
|
|
issue_id=str(pk),
|
|
project_id=str(project_id),
|
|
current_instance=current_instance,
|
|
epoch=int(timezone.now().timestamp()),
|
|
)
|
|
return Response(status=status.HTTP_204_NO_CONTENT) |