766 lines
20 KiB
Python
766 lines
20 KiB
Python
# Django imports
|
|
from django.utils import timezone
|
|
from django.core.validators import URLValidator
|
|
from django.core.exceptions import ValidationError
|
|
|
|
# Third Party imports
|
|
from rest_framework import serializers
|
|
|
|
# Module imports
|
|
from .base import BaseSerializer, DynamicBaseSerializer
|
|
from .user import UserLiteSerializer
|
|
from .state import StateLiteSerializer
|
|
from .project import ProjectLiteSerializer
|
|
from .workspace import WorkspaceLiteSerializer
|
|
from plane.db.models import (
|
|
User,
|
|
Issue,
|
|
IssueActivity,
|
|
IssueComment,
|
|
IssueUserProperty,
|
|
IssueAssignee,
|
|
IssueSubscriber,
|
|
IssueLabel,
|
|
Label,
|
|
CycleIssue,
|
|
Cycle,
|
|
Module,
|
|
ModuleIssue,
|
|
IssueLink,
|
|
IssueAttachment,
|
|
IssueReaction,
|
|
CommentReaction,
|
|
IssueVote,
|
|
IssueRelation,
|
|
State,
|
|
)
|
|
|
|
|
|
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):
|
|
# ids
|
|
state_id = serializers.PrimaryKeyRelatedField(
|
|
source="state",
|
|
queryset=State.objects.all(),
|
|
required=False,
|
|
allow_null=True,
|
|
)
|
|
parent_id = serializers.PrimaryKeyRelatedField(
|
|
source="parent",
|
|
queryset=Issue.objects.all(),
|
|
required=False,
|
|
allow_null=True,
|
|
)
|
|
label_ids = serializers.ListField(
|
|
child=serializers.PrimaryKeyRelatedField(queryset=Label.objects.all()),
|
|
write_only=True,
|
|
required=False,
|
|
)
|
|
assignee_ids = serializers.ListField(
|
|
child=serializers.PrimaryKeyRelatedField(queryset=User.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)
|
|
assignee_ids = self.initial_data.get("assignee_ids")
|
|
data["assignee_ids"] = assignee_ids if assignee_ids else []
|
|
label_ids = self.initial_data.get("label_ids")
|
|
data["label_ids"] = label_ids if label_ids else []
|
|
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("assignee_ids", None)
|
|
labels = validated_data.pop("label_ids", None)
|
|
|
|
project_id = self.context["project_id"]
|
|
workspace_id = self.context["workspace_id"]
|
|
default_assignee_id = self.context["default_assignee_id"]
|
|
|
|
# Create Issue
|
|
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("assignee_ids", None)
|
|
labels = validated_data.pop("label_ids", 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")
|
|
workspace_detail = WorkspaceLiteSerializer(
|
|
read_only=True, source="workspace"
|
|
)
|
|
|
|
class Meta:
|
|
model = IssueActivity
|
|
fields = "__all__"
|
|
|
|
|
|
class IssueUserPropertySerializer(BaseSerializer):
|
|
class Meta:
|
|
model = IssueUserProperty
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"user",
|
|
"workspace",
|
|
"project",
|
|
]
|
|
|
|
|
|
class LabelSerializer(BaseSerializer):
|
|
class Meta:
|
|
model = Label
|
|
fields = [
|
|
"parent",
|
|
"name",
|
|
"color",
|
|
"id",
|
|
"project_id",
|
|
"workspace_id",
|
|
"sort_order",
|
|
]
|
|
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):
|
|
id = serializers.UUIDField(source="related_issue.id", read_only=True)
|
|
project_id = serializers.PrimaryKeyRelatedField(
|
|
source="related_issue.project_id", read_only=True
|
|
)
|
|
sequence_id = serializers.IntegerField(
|
|
source="related_issue.sequence_id", read_only=True
|
|
)
|
|
name = serializers.CharField(source="related_issue.name", read_only=True)
|
|
relation_type = serializers.CharField(read_only=True)
|
|
|
|
class Meta:
|
|
model = IssueRelation
|
|
fields = [
|
|
"id",
|
|
"project_id",
|
|
"sequence_id",
|
|
"relation_type",
|
|
"name",
|
|
]
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
]
|
|
|
|
|
|
class RelatedIssueSerializer(BaseSerializer):
|
|
id = serializers.UUIDField(source="issue.id", read_only=True)
|
|
project_id = serializers.PrimaryKeyRelatedField(
|
|
source="issue.project_id", read_only=True
|
|
)
|
|
sequence_id = serializers.IntegerField(
|
|
source="issue.sequence_id", read_only=True
|
|
)
|
|
name = serializers.CharField(source="issue.name", read_only=True)
|
|
relation_type = serializers.CharField(read_only=True)
|
|
|
|
class Meta:
|
|
model = IssueRelation
|
|
fields = [
|
|
"id",
|
|
"project_id",
|
|
"sequence_id",
|
|
"relation_type",
|
|
"name",
|
|
]
|
|
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",
|
|
]
|
|
|
|
def validate_url(self, value):
|
|
# Check URL format
|
|
validate_url = URLValidator()
|
|
try:
|
|
validate_url(value)
|
|
except ValidationError:
|
|
raise serializers.ValidationError("Invalid URL format.")
|
|
|
|
# Check URL scheme
|
|
if not value.startswith(("http://", "https://")):
|
|
raise serializers.ValidationError("Invalid URL scheme.")
|
|
|
|
return value
|
|
|
|
# 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)
|
|
|
|
def update(self, instance, validated_data):
|
|
if (
|
|
IssueLink.objects.filter(
|
|
url=validated_data.get("url"),
|
|
issue_id=instance.issue_id,
|
|
)
|
|
.exclude(pk=instance.id)
|
|
.exists()
|
|
):
|
|
raise serializers.ValidationError(
|
|
{"error": "URL already exists for this Issue"}
|
|
)
|
|
|
|
return super().update(instance, validated_data)
|
|
|
|
|
|
class IssueLinkLiteSerializer(BaseSerializer):
|
|
class Meta:
|
|
model = IssueLink
|
|
fields = [
|
|
"id",
|
|
"issue_id",
|
|
"title",
|
|
"url",
|
|
"metadata",
|
|
"created_by_id",
|
|
"created_at",
|
|
]
|
|
read_only_fields = fields
|
|
|
|
|
|
class IssueAttachmentSerializer(BaseSerializer):
|
|
class Meta:
|
|
model = IssueAttachment
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"created_by",
|
|
"updated_by",
|
|
"created_at",
|
|
"updated_at",
|
|
"workspace",
|
|
"project",
|
|
"issue",
|
|
]
|
|
|
|
|
|
class IssueAttachmentLiteSerializer(DynamicBaseSerializer):
|
|
class Meta:
|
|
model = IssueAttachment
|
|
fields = [
|
|
"id",
|
|
"asset",
|
|
"attributes",
|
|
"issue_id",
|
|
"updated_at",
|
|
"updated_by",
|
|
]
|
|
read_only_fields = fields
|
|
|
|
|
|
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 IssueReactionLiteSerializer(DynamicBaseSerializer):
|
|
class Meta:
|
|
model = IssueReaction
|
|
fields = [
|
|
"id",
|
|
"actor",
|
|
"issue",
|
|
"reaction",
|
|
]
|
|
|
|
|
|
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 = CommentReactionSerializer(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(DynamicBaseSerializer):
|
|
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)
|
|
attachment_count = serializers.IntegerField(read_only=True)
|
|
link_count = serializers.IntegerField(read_only=True)
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = "__all__"
|
|
|
|
|
|
class IssueInboxSerializer(DynamicBaseSerializer):
|
|
label_ids = serializers.ListField(
|
|
child=serializers.UUIDField(),
|
|
required=False,
|
|
)
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = [
|
|
"id",
|
|
"name",
|
|
"priority",
|
|
"sequence_id",
|
|
"project_id",
|
|
"created_at",
|
|
"label_ids",
|
|
"created_by",
|
|
]
|
|
read_only_fields = fields
|
|
|
|
|
|
class IssueSerializer(DynamicBaseSerializer):
|
|
# ids
|
|
cycle_id = serializers.PrimaryKeyRelatedField(read_only=True)
|
|
module_ids = serializers.ListField(
|
|
child=serializers.UUIDField(),
|
|
required=False,
|
|
)
|
|
|
|
# Many to many
|
|
label_ids = serializers.ListField(
|
|
child=serializers.UUIDField(),
|
|
required=False,
|
|
)
|
|
assignee_ids = serializers.ListField(
|
|
child=serializers.UUIDField(),
|
|
required=False,
|
|
)
|
|
|
|
# Count items
|
|
sub_issues_count = serializers.IntegerField(read_only=True)
|
|
attachment_count = serializers.IntegerField(read_only=True)
|
|
link_count = serializers.IntegerField(read_only=True)
|
|
|
|
class Meta:
|
|
model = Issue
|
|
fields = [
|
|
"id",
|
|
"name",
|
|
"state_id",
|
|
"sort_order",
|
|
"completed_at",
|
|
"estimate_point",
|
|
"priority",
|
|
"start_date",
|
|
"target_date",
|
|
"sequence_id",
|
|
"project_id",
|
|
"parent_id",
|
|
"cycle_id",
|
|
"module_ids",
|
|
"label_ids",
|
|
"assignee_ids",
|
|
"sub_issues_count",
|
|
"created_at",
|
|
"updated_at",
|
|
"created_by",
|
|
"updated_by",
|
|
"attachment_count",
|
|
"link_count",
|
|
"is_draft",
|
|
"archived_at",
|
|
]
|
|
read_only_fields = fields
|
|
|
|
|
|
class IssueLiteSerializer(DynamicBaseSerializer):
|
|
class Meta:
|
|
model = Issue
|
|
fields = [
|
|
"id",
|
|
"sequence_id",
|
|
"project_id",
|
|
]
|
|
read_only_fields = fields
|
|
|
|
|
|
class IssueDetailSerializer(IssueSerializer):
|
|
description_html = serializers.CharField()
|
|
is_subscribed = serializers.BooleanField(read_only=True)
|
|
|
|
class Meta(IssueSerializer.Meta):
|
|
fields = IssueSerializer.Meta.fields + [
|
|
"description_html",
|
|
"is_subscribed",
|
|
]
|
|
read_only_fields = fields
|
|
|
|
|
|
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",
|
|
]
|