993cf3faba
* chore: add assignee avatar in cycle endpoint * dev: update the structure to return avatar and firstname * dev: return distinct users * dev: update the structure to return id * dev: update the prefetch queryset to distinct by id * dev: remove id from distinct * dev: add unique condition
79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
# Third party imports
|
|
from rest_framework import serializers
|
|
|
|
# Module imports
|
|
from .base import BaseSerializer
|
|
from .user import UserLiteSerializer
|
|
from .issue import IssueStateSerializer
|
|
from .workspace import WorkspaceLiteSerializer
|
|
from .project import ProjectLiteSerializer
|
|
from plane.db.models import Cycle, CycleIssue, CycleFavorite
|
|
|
|
|
|
class CycleSerializer(BaseSerializer):
|
|
owned_by = UserLiteSerializer(read_only=True)
|
|
is_favorite = serializers.BooleanField(read_only=True)
|
|
total_issues = serializers.IntegerField(read_only=True)
|
|
cancelled_issues = serializers.IntegerField(read_only=True)
|
|
completed_issues = serializers.IntegerField(read_only=True)
|
|
started_issues = serializers.IntegerField(read_only=True)
|
|
unstarted_issues = serializers.IntegerField(read_only=True)
|
|
backlog_issues = serializers.IntegerField(read_only=True)
|
|
assignees = serializers.SerializerMethodField()
|
|
|
|
workspace_detail = WorkspaceLiteSerializer(read_only=True, source="workspace")
|
|
project_detail = ProjectLiteSerializer(read_only=True, source="project")
|
|
|
|
def get_assignees(self, obj):
|
|
members = [
|
|
{
|
|
"avatar": assignee.avatar,
|
|
"first_name": assignee.first_name,
|
|
"id": assignee.id,
|
|
}
|
|
for issue_cycle in obj.issue_cycle.all()
|
|
for assignee in issue_cycle.issue.assignees.all()
|
|
]
|
|
# Use a set comprehension to return only the unique objects
|
|
unique_objects = {frozenset(item.items()) for item in members}
|
|
|
|
# Convert the set back to a list of dictionaries
|
|
unique_list = [dict(item) for item in unique_objects]
|
|
|
|
return unique_list
|
|
|
|
class Meta:
|
|
model = Cycle
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"owned_by",
|
|
]
|
|
|
|
|
|
class CycleIssueSerializer(BaseSerializer):
|
|
issue_detail = IssueStateSerializer(read_only=True, source="issue")
|
|
sub_issues_count = serializers.IntegerField(read_only=True)
|
|
|
|
class Meta:
|
|
model = CycleIssue
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"cycle",
|
|
]
|
|
|
|
|
|
class CycleFavoriteSerializer(BaseSerializer):
|
|
cycle_detail = CycleSerializer(source="cycle", read_only=True)
|
|
|
|
class Meta:
|
|
model = CycleFavorite
|
|
fields = "__all__"
|
|
read_only_fields = [
|
|
"workspace",
|
|
"project",
|
|
"user",
|
|
]
|