146 lines
5.1 KiB
Python
146 lines
5.1 KiB
Python
from rest_framework import serializers
|
|
from core.serializers.base import BaseModelSerializer
|
|
from apps.projects.models import (
|
|
Project,
|
|
ProjectMembership,
|
|
ProjectRate,
|
|
ProjectUserRate,
|
|
)
|
|
from core.serializers.mini import UserMiniSerializer
|
|
|
|
|
|
class ProjectMemberInputSerializer(serializers.Serializer):
|
|
user_id = serializers.UUIDField()
|
|
role = serializers.ChoiceField(choices=ProjectMembership.Role.choices, default=ProjectMembership.Role.MEMBER)
|
|
|
|
|
|
class ProjectSerializer(BaseModelSerializer):
|
|
my_role = serializers.SerializerMethodField()
|
|
members = serializers.SerializerMethodField()
|
|
|
|
class Meta:
|
|
model = Project
|
|
fields = BaseModelSerializer.Meta.fields + (
|
|
"workspace",
|
|
"name",
|
|
"client",
|
|
"description",
|
|
"is_archived",
|
|
"color",
|
|
"my_role",
|
|
"members",
|
|
)
|
|
read_only_fields = fields
|
|
|
|
def get_my_role(self, obj):
|
|
request = self.context.get("request")
|
|
if not request or not request.user.is_authenticated:
|
|
return None
|
|
membership = obj.memberships.filter(user=request.user, is_active=True, is_deleted=False).first()
|
|
return getattr(membership, "role", None)
|
|
|
|
def get_members(self, obj):
|
|
"""
|
|
Returns active project members in the response
|
|
"""
|
|
active_memberships = obj.memberships.filter(is_active=True, is_deleted=False)
|
|
return ProjectMembershipSerializer(active_memberships, many=True).data
|
|
|
|
def to_representation(self, instance):
|
|
representation = super().to_representation(instance)
|
|
if instance.client:
|
|
representation['client'] = {
|
|
'id': instance.client.id,
|
|
'name': instance.client.name
|
|
}
|
|
return representation
|
|
|
|
|
|
class ProjectCreateSerializer(serializers.Serializer):
|
|
workspace = serializers.UUIDField()
|
|
name = serializers.CharField(max_length=255)
|
|
client = serializers.UUIDField(required=False, allow_null=True)
|
|
description = serializers.CharField(required=False, allow_blank=True, default="")
|
|
color = serializers.CharField(max_length=7, required=False, allow_blank=True, default="")
|
|
members = ProjectMemberInputSerializer(many=True, required=False)
|
|
|
|
|
|
class ProjectUpdateSerializer(serializers.Serializer):
|
|
name = serializers.CharField(max_length=255, required=False)
|
|
client = serializers.UUIDField(required=False, allow_null=True)
|
|
description = serializers.CharField(required=False, allow_blank=True)
|
|
color = serializers.CharField(max_length=7, required=False, allow_blank=True)
|
|
is_archived = serializers.BooleanField(required=False)
|
|
members = ProjectMemberInputSerializer(many=True, required=False)
|
|
|
|
class ProjectMembershipSerializer(BaseModelSerializer):
|
|
user_details = UserMiniSerializer(read_only=True)
|
|
|
|
class Meta:
|
|
model = ProjectMembership
|
|
fields = BaseModelSerializer.Meta.fields + (
|
|
"project",
|
|
"user",
|
|
"user_details",
|
|
"role",
|
|
"is_active",
|
|
)
|
|
read_only_fields = fields
|
|
|
|
|
|
class ProjectMembershipCreateSerializer(serializers.Serializer):
|
|
project_id = serializers.UUIDField()
|
|
user_id = serializers.UUIDField()
|
|
role = serializers.ChoiceField(choices=ProjectMembership.Role.choices)
|
|
|
|
|
|
class ProjectMembershipUpdateSerializer(serializers.Serializer):
|
|
role = serializers.ChoiceField(choices=ProjectMembership.Role.choices, required=False)
|
|
is_active = serializers.BooleanField(required=False)
|
|
|
|
|
|
class ProjectRateSerializer(BaseModelSerializer):
|
|
class Meta:
|
|
model = ProjectRate
|
|
fields = BaseModelSerializer.Meta.fields + (
|
|
"project",
|
|
"hourly_rate",
|
|
"currency",
|
|
)
|
|
read_only_fields = fields
|
|
|
|
|
|
class ProjectRateCreateSerializer(serializers.Serializer):
|
|
project_id = serializers.UUIDField()
|
|
hourly_rate = serializers.DecimalField(max_digits=10, decimal_places=2)
|
|
currency = serializers.CharField(max_length=3, default="USD")
|
|
|
|
|
|
class ProjectRateUpdateSerializer(serializers.Serializer):
|
|
hourly_rate = serializers.DecimalField(max_digits=10, decimal_places=2, required=False)
|
|
currency = serializers.CharField(max_length=3, required=False)
|
|
|
|
|
|
class ProjectUserRateSerializer(BaseModelSerializer):
|
|
class Meta:
|
|
model = ProjectUserRate
|
|
fields = BaseModelSerializer.Meta.fields + (
|
|
"project",
|
|
"user",
|
|
"hourly_rate",
|
|
"currency",
|
|
)
|
|
read_only_fields = fields
|
|
|
|
|
|
class ProjectUserRateCreateSerializer(serializers.Serializer):
|
|
project_id = serializers.UUIDField()
|
|
user_id = serializers.UUIDField()
|
|
hourly_rate = serializers.DecimalField(max_digits=10, decimal_places=2)
|
|
currency = serializers.CharField(max_length=3, default="USD")
|
|
|
|
|
|
class ProjectUserRateUpdateSerializer(serializers.Serializer):
|
|
hourly_rate = serializers.DecimalField(max_digits=10, decimal_places=2, required=False)
|
|
currency = serializers.CharField(max_length=3, required=False)
|