80 lines
2.7 KiB
Python
80 lines
2.7 KiB
Python
from decimal import Decimal
|
|
|
|
from rest_framework import serializers
|
|
from core.serializers.base import BaseModelSerializer
|
|
from apps.projects.models import Project
|
|
from apps.workspaces.models import PriceUnit
|
|
|
|
|
|
class ProjectSerializer(BaseModelSerializer):
|
|
class Meta:
|
|
model = Project
|
|
fields = BaseModelSerializer.Meta.fields + (
|
|
"workspace",
|
|
"name",
|
|
"client",
|
|
"description",
|
|
"is_archived",
|
|
"color",
|
|
)
|
|
read_only_fields = fields
|
|
|
|
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="")
|
|
|
|
|
|
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)
|
|
|
|
|
|
class ProjectAccessQuerySerializer(serializers.Serializer):
|
|
workspace = serializers.UUIDField()
|
|
user = serializers.UUIDField()
|
|
|
|
|
|
class ProjectAccessMutationSerializer(serializers.Serializer):
|
|
workspace = serializers.UUIDField()
|
|
user = serializers.UUIDField()
|
|
project_ids = serializers.ListField(
|
|
child=serializers.UUIDField(),
|
|
allow_empty=False,
|
|
)
|
|
|
|
|
|
class ProjectAccessRateMutationSerializer(serializers.Serializer):
|
|
workspace = serializers.UUIDField()
|
|
user = serializers.UUIDField()
|
|
project = serializers.UUIDField()
|
|
hourly_rate = serializers.DecimalField(
|
|
max_digits=10,
|
|
decimal_places=2,
|
|
min_value=Decimal("0.01"),
|
|
required=False,
|
|
allow_null=True,
|
|
)
|
|
currency = serializers.CharField(max_length=3, required=False, default="USD")
|
|
|
|
def validate_currency(self, value):
|
|
code = value.upper()
|
|
if not PriceUnit.objects.filter(code=code, is_deleted=False).exists():
|
|
raise serializers.ValidationError("Selected price unit is invalid.")
|
|
return code
|