7249f84e18
* dev: remove len for empty comparison * dev: using in instead of multiple ors * dev: assign expression to empty variables * dev: use f-string * dev: remove list comprehension and use generators * dev: remove assert from paginator * dev: use is for identity comparison with singleton * dev: remove unnecessary else statements * dev: fix does not exists error for both project and workspace * dev: remove reimports * dev: iterate a dictionary * dev: remove unused commented code * dev: remove redefinition * dev: remove unused imports * dev: remove unused imports * dev: remove unnecessary f strings * dev: remove unused variables * dev: use literal structure to create the data structure * dev: add empty lines at the end of the file * dev: remove user middleware * dev: remove unnecessary default None
615 lines
18 KiB
Python
615 lines
18 KiB
Python
# Django imports
|
|
from django.utils import timezone
|
|
|
|
# Third Party imports
|
|
from rest_framework import serializers
|
|
|
|
# Module imports
|
|
from .base import BaseSerializer
|
|
from .user import UserLiteSerializer
|
|
from .state import StateSerializer, StateLiteSerializer
|
|
from .project import ProjectLiteSerializer
|
|
from .workspace import WorkspaceLiteSerializer
|
|
from plane.db.models import (
|
|
User,
|
|
Issue,
|
|
IssueActivity,
|
|
IssueComment,
|
|
IssueProperty,
|
|
IssueAssignee,
|
|
IssueSubscriber,
|
|
IssueLabel,
|
|
Label,
|
|
CycleIssue,
|
|
Cycle,
|
|
Module,
|
|
ModuleIssue,
|
|
IssueLink,
|
|
IssueAttachment,
|
|
IssueReaction,
|
|
CommentReaction,
|
|
IssueVote,
|
|
IssueRelation,
|
|
)
|
|
|
|
|
|
class IssueFlatSerializer(BaseSerializer):
|
|
## Contain only flat fields
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = [
|
|
"id",
|
|
"name",
|
|
"description",
|
|
"description_html",
|
|
"priority",
|
|
"start_date",
|
|
"target_date",
|
|
"sequence_id",
|
|
"sort_order",
|
|
"is_draft",
|
|
]
|
|
|
|
|
|
class IssueProjectLiteSerializer(BaseSerializer):
|
|
project_detail = ProjectLiteSerializer(source="project", read_only=True)
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = [
|
|
"id",
|
|
"project_detail",
|
|
"name",
|
|
"sequence_id",
|
|
]
|
|
read_only_fields = fields
|
|
|
|
|
|
##TODO: Find a better way to write this serializer
|
|
## Find a better approach to save manytomany?
|
|
class IssueCreateSerializer(BaseSerializer):
|
|
state_detail = StateSerializer(read_only=True, source="state")
|
|
created_by_detail = UserLiteSerializer(read_only=True, source="created_by")
|
|
project_detail = ProjectLiteSerializer(read_only=True, source="project")
|
|
workspace_detail = WorkspaceLiteSerializer(read_only=True, source="workspace")
|
|
|
|
assignees = serializers.ListField(
|
|
child=serializers.PrimaryKeyRelatedField(queryset=User.objects.all()),
|
|
write_only=True,
|
|
required=False,
|
|
)
|
|
|
|
labels = serializers.ListField(
|
|
child=serializers.PrimaryKeyRelatedField(queryset=Label.objects.all()),
|
|
write_only=True,
|
|
required=False,
|
|
)
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
]
|
|
|
|
def to_representation(self, instance):
|
|
data = super().to_representation(instance)
|
|
data['assignees'] = [str(assignee.id) for assignee in instance.assignees.all()]
|
|
data['labels'] = [str(label.id) for label in instance.labels.all()]
|
|
return data
|
|
|
|
def validate(self, data):
|
|
if (
|
|
data.get("start_date", None) is not None
|
|
and data.get("target_date", None) is not None
|
|
and data.get("start_date", None) > data.get("target_date", None)
|
|
):
|
|
raise serializers.ValidationError("Start date cannot exceed target date")
|
|
return data
|
|
|
|
def create(self, validated_data):
|
|
assignees = validated_data.pop("assignees", None)
|
|
labels = validated_data.pop("labels", None)
|
|
|
|
project_id = self.context["project_id"]
|
|
workspace_id = self.context["workspace_id"]
|
|
default_assignee_id = self.context["default_assignee_id"]
|
|
|
|
issue = Issue.objects.create(**validated_data, project_id=project_id)
|
|
|
|
# Issue Audit Users
|
|
created_by_id = issue.created_by_id
|
|
updated_by_id = issue.updated_by_id
|
|
|
|
if assignees is not None and len(assignees):
|
|
IssueAssignee.objects.bulk_create(
|
|
[
|
|
IssueAssignee(
|
|
assignee=user,
|
|
issue=issue,
|
|
project_id=project_id,
|
|
workspace_id=workspace_id,
|
|
created_by_id=created_by_id,
|
|
updated_by_id=updated_by_id,
|
|
)
|
|
for user in assignees
|
|
],
|
|
batch_size=10,
|
|
)
|
|
else:
|
|
# Then assign it to default assignee
|
|
if default_assignee_id is not None:
|
|
IssueAssignee.objects.create(
|
|
assignee_id=default_assignee_id,
|
|
issue=issue,
|
|
project_id=project_id,
|
|
workspace_id=workspace_id,
|
|
created_by_id=created_by_id,
|
|
updated_by_id=updated_by_id,
|
|
)
|
|
|
|
if labels is not None and len(labels):
|
|
IssueLabel.objects.bulk_create(
|
|
[
|
|
IssueLabel(
|
|
label=label,
|
|
issue=issue,
|
|
project_id=project_id,
|
|
workspace_id=workspace_id,
|
|
created_by_id=created_by_id,
|
|
updated_by_id=updated_by_id,
|
|
)
|
|
for label in labels
|
|
],
|
|
batch_size=10,
|
|
)
|
|
|
|
return issue
|
|
|
|
def update(self, instance, validated_data):
|
|
assignees = validated_data.pop("assignees", None)
|
|
labels = validated_data.pop("labels", None)
|
|
|
|
# Related models
|
|
project_id = instance.project_id
|
|
workspace_id = instance.workspace_id
|
|
created_by_id = instance.created_by_id
|
|
updated_by_id = instance.updated_by_id
|
|
|
|
if assignees is not None:
|
|
IssueAssignee.objects.filter(issue=instance).delete()
|
|
IssueAssignee.objects.bulk_create(
|
|
[
|
|
IssueAssignee(
|
|
assignee=user,
|
|
issue=instance,
|
|
project_id=project_id,
|
|
workspace_id=workspace_id,
|
|
created_by_id=created_by_id,
|
|
updated_by_id=updated_by_id,
|
|
)
|
|
for user in assignees
|
|
],
|
|
batch_size=10,
|
|
)
|
|
|
|
if labels is not None:
|
|
IssueLabel.objects.filter(issue=instance).delete()
|
|
IssueLabel.objects.bulk_create(
|
|
[
|
|
IssueLabel(
|
|
label=label,
|
|
issue=instance,
|
|
project_id=project_id,
|
|
workspace_id=workspace_id,
|
|
created_by_id=created_by_id,
|
|
updated_by_id=updated_by_id,
|
|
)
|
|
for label in labels
|
|
],
|
|
batch_size=10,
|
|
)
|
|
|
|
# Time updation occues even when other related models are updated
|
|
instance.updated_at = timezone.now()
|
|
return super().update(instance, validated_data)
|
|
|
|
|
|
class IssueActivitySerializer(BaseSerializer):
|
|
actor_detail = UserLiteSerializer(read_only=True, source="actor")
|
|
issue_detail = IssueFlatSerializer(read_only=True, source="issue")
|
|
project_detail = ProjectLiteSerializer(read_only=True, source="project")
|
|
|
|
class Meta:
|
|
model = IssueActivity
|
|
fields = "__all__"
|
|
|
|
|
|
|
|
class IssuePropertySerializer(BaseSerializer):
|
|
class Meta:
|
|
model = IssueProperty
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"user",
|
|
"workspace",
|
|
"project",
|
|
]
|
|
|
|
|
|
class LabelSerializer(BaseSerializer):
|
|
workspace_detail = WorkspaceLiteSerializer(source="workspace", read_only=True)
|
|
project_detail = ProjectLiteSerializer(source="project", read_only=True)
|
|
|
|
class Meta:
|
|
model = Label
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
]
|
|
|
|
|
|
class LabelLiteSerializer(BaseSerializer):
|
|
class Meta:
|
|
model = Label
|
|
fields = [
|
|
"id",
|
|
"name",
|
|
"color",
|
|
]
|
|
|
|
|
|
class IssueLabelSerializer(BaseSerializer):
|
|
|
|
class Meta:
|
|
model = IssueLabel
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
]
|
|
|
|
|
|
class IssueRelationSerializer(BaseSerializer):
|
|
issue_detail = IssueProjectLiteSerializer(read_only=True, source="related_issue")
|
|
|
|
class Meta:
|
|
model = IssueRelation
|
|
fields = [
|
|
"issue_detail",
|
|
"relation_type",
|
|
"related_issue",
|
|
"issue",
|
|
"id"
|
|
]
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
]
|
|
|
|
class RelatedIssueSerializer(BaseSerializer):
|
|
issue_detail = IssueProjectLiteSerializer(read_only=True, source="issue")
|
|
|
|
class Meta:
|
|
model = IssueRelation
|
|
fields = [
|
|
"issue_detail",
|
|
"relation_type",
|
|
"related_issue",
|
|
"issue",
|
|
"id"
|
|
]
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
]
|
|
|
|
|
|
class IssueAssigneeSerializer(BaseSerializer):
|
|
assignee_details = UserLiteSerializer(read_only=True, source="assignee")
|
|
|
|
class Meta:
|
|
model = IssueAssignee
|
|
fields = "__all__"
|
|
|
|
|
|
class CycleBaseSerializer(BaseSerializer):
|
|
class Meta:
|
|
model = Cycle
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
]
|
|
|
|
|
|
class IssueCycleDetailSerializer(BaseSerializer):
|
|
cycle_detail = CycleBaseSerializer(read_only=True, source="cycle")
|
|
|
|
class Meta:
|
|
model = CycleIssue
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
]
|
|
|
|
|
|
class ModuleBaseSerializer(BaseSerializer):
|
|
class Meta:
|
|
model = Module
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
]
|
|
|
|
|
|
class IssueModuleDetailSerializer(BaseSerializer):
|
|
module_detail = ModuleBaseSerializer(read_only=True, source="module")
|
|
|
|
class Meta:
|
|
model = ModuleIssue
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
]
|
|
|
|
|
|
class IssueLinkSerializer(BaseSerializer):
|
|
created_by_detail = UserLiteSerializer(read_only=True, source="created_by")
|
|
|
|
class Meta:
|
|
model = IssueLink
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
"issue",
|
|
]
|
|
|
|
# Validation if url already exists
|
|
def create(self, validated_data):
|
|
if IssueLink.objects.filter(
|
|
url=validated_data.get("url"), issue_id=validated_data.get("issue_id")
|
|
).exists():
|
|
raise serializers.ValidationError(
|
|
{"error": "URL already exists for this Issue"}
|
|
)
|
|
return IssueLink.objects.create(**validated_data)
|
|
|
|
|
|
class IssueAttachmentSerializer(BaseSerializer):
|
|
class Meta:
|
|
model = IssueAttachment
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
"workspace",
|
|
"project",
|
|
"issue",
|
|
]
|
|
|
|
|
|
class IssueReactionSerializer(BaseSerializer):
|
|
|
|
actor_detail = UserLiteSerializer(read_only=True, source="actor")
|
|
|
|
class Meta:
|
|
model = IssueReaction
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"issue",
|
|
"actor",
|
|
]
|
|
|
|
|
|
class CommentReactionLiteSerializer(BaseSerializer):
|
|
actor_detail = UserLiteSerializer(read_only=True, source="actor")
|
|
|
|
class Meta:
|
|
model = CommentReaction
|
|
fields = [
|
|
"id",
|
|
"reaction",
|
|
"comment",
|
|
"actor_detail",
|
|
]
|
|
|
|
|
|
class CommentReactionSerializer(BaseSerializer):
|
|
class Meta:
|
|
model = CommentReaction
|
|
fields = "__all__"
|
|
read_only_fields = ["workspace", "project", "comment", "actor"]
|
|
|
|
|
|
class IssueVoteSerializer(BaseSerializer):
|
|
|
|
actor_detail = UserLiteSerializer(read_only=True, source="actor")
|
|
|
|
class Meta:
|
|
model = IssueVote
|
|
fields = ["issue", "vote", "workspace", "project", "actor", "actor_detail"]
|
|
read_only_fields = fields
|
|
|
|
|
|
class IssueCommentSerializer(BaseSerializer):
|
|
actor_detail = UserLiteSerializer(read_only=True, source="actor")
|
|
issue_detail = IssueFlatSerializer(read_only=True, source="issue")
|
|
project_detail = ProjectLiteSerializer(read_only=True, source="project")
|
|
workspace_detail = WorkspaceLiteSerializer(read_only=True, source="workspace")
|
|
comment_reactions = CommentReactionLiteSerializer(read_only=True, many=True)
|
|
is_member = serializers.BooleanField(read_only=True)
|
|
|
|
class Meta:
|
|
model = IssueComment
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"issue",
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
]
|
|
|
|
|
|
class IssueStateFlatSerializer(BaseSerializer):
|
|
state_detail = StateLiteSerializer(read_only=True, source="state")
|
|
project_detail = ProjectLiteSerializer(read_only=True, source="project")
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = [
|
|
"id",
|
|
"sequence_id",
|
|
"name",
|
|
"state_detail",
|
|
"project_detail",
|
|
]
|
|
|
|
|
|
# Issue Serializer with state details
|
|
class IssueStateSerializer(BaseSerializer):
|
|
label_details = LabelLiteSerializer(read_only=True, source="labels", many=True)
|
|
state_detail = StateLiteSerializer(read_only=True, source="state")
|
|
project_detail = ProjectLiteSerializer(read_only=True, source="project")
|
|
assignee_details = UserLiteSerializer(read_only=True, source="assignees", many=True)
|
|
sub_issues_count = serializers.IntegerField(read_only=True)
|
|
bridge_id = serializers.UUIDField(read_only=True)
|
|
attachment_count = serializers.IntegerField(read_only=True)
|
|
link_count = serializers.IntegerField(read_only=True)
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = "__all__"
|
|
|
|
|
|
class IssueSerializer(BaseSerializer):
|
|
project_detail = ProjectLiteSerializer(read_only=True, source="project")
|
|
state_detail = StateSerializer(read_only=True, source="state")
|
|
parent_detail = IssueStateFlatSerializer(read_only=True, source="parent")
|
|
label_details = LabelSerializer(read_only=True, source="labels", many=True)
|
|
assignee_details = UserLiteSerializer(read_only=True, source="assignees", many=True)
|
|
related_issues = IssueRelationSerializer(read_only=True, source="issue_relation", many=True)
|
|
issue_relations = RelatedIssueSerializer(read_only=True, source="issue_related", many=True)
|
|
issue_cycle = IssueCycleDetailSerializer(read_only=True)
|
|
issue_module = IssueModuleDetailSerializer(read_only=True)
|
|
issue_link = IssueLinkSerializer(read_only=True, many=True)
|
|
issue_attachment = IssueAttachmentSerializer(read_only=True, many=True)
|
|
sub_issues_count = serializers.IntegerField(read_only=True)
|
|
issue_reactions = IssueReactionSerializer(read_only=True, many=True)
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
]
|
|
|
|
|
|
class IssueLiteSerializer(BaseSerializer):
|
|
workspace_detail = WorkspaceLiteSerializer(read_only=True, source="workspace")
|
|
project_detail = ProjectLiteSerializer(read_only=True, source="project")
|
|
state_detail = StateLiteSerializer(read_only=True, source="state")
|
|
label_details = LabelLiteSerializer(read_only=True, source="labels", many=True)
|
|
assignee_details = UserLiteSerializer(read_only=True, source="assignees", many=True)
|
|
sub_issues_count = serializers.IntegerField(read_only=True)
|
|
cycle_id = serializers.UUIDField(read_only=True)
|
|
module_id = serializers.UUIDField(read_only=True)
|
|
attachment_count = serializers.IntegerField(read_only=True)
|
|
link_count = serializers.IntegerField(read_only=True)
|
|
issue_reactions = IssueReactionSerializer(read_only=True, many=True)
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"start_date",
|
|
"target_date",
|
|
"completed_at",
|
|
"workspace",
|
|
"project",
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
]
|
|
|
|
|
|
class IssuePublicSerializer(BaseSerializer):
|
|
project_detail = ProjectLiteSerializer(read_only=True, source="project")
|
|
state_detail = StateLiteSerializer(read_only=True, source="state")
|
|
reactions = IssueReactionSerializer(read_only=True, many=True, source="issue_reactions")
|
|
votes = IssueVoteSerializer(read_only=True, many=True)
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = [
|
|
"id",
|
|
"name",
|
|
"description_html",
|
|
"sequence_id",
|
|
"state",
|
|
"state_detail",
|
|
"project",
|
|
"project_detail",
|
|
"workspace",
|
|
"priority",
|
|
"target_date",
|
|
"reactions",
|
|
"votes",
|
|
]
|
|
read_only_fields = fields
|
|
|
|
|
|
|
|
class IssueSubscriberSerializer(BaseSerializer):
|
|
class Meta:
|
|
model = IssueSubscriber
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"issue",
|
|
]
|