112 lines
4.1 KiB
Python
112 lines
4.1 KiB
Python
from decimal import Decimal
|
|
|
|
from rest_framework import serializers
|
|
from apps.utils.serializers import CustomModelSerializer
|
|
from apps.sam.models import Customer, Contract, Order, OrderItem, ContractRecord
|
|
from apps.utils.constants import EXCLUDE_FIELDS, EXCLUDE_FIELDS_BASE
|
|
from rest_framework.exceptions import ValidationError
|
|
from apps.mtm.serializers import MaterialSerializer
|
|
|
|
|
|
class CustomerSerializer(CustomModelSerializer):
|
|
class Meta:
|
|
model = Customer
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS + ['belong_dept']
|
|
|
|
|
|
class ContractSerializer(CustomModelSerializer):
|
|
customer_name = serializers.CharField(
|
|
source='customer.name', read_only=True)
|
|
create_by_name = serializers.CharField(
|
|
source='create_by.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = Contract
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS + ['belong_dept', 'invoice']
|
|
|
|
|
|
class OrderSerializer(CustomModelSerializer):
|
|
customer_name = serializers.CharField(
|
|
source='customer.name', read_only=True)
|
|
contract_number = serializers.CharField(
|
|
source='contract.number', read_only=True)
|
|
create_by_name = serializers.CharField(
|
|
source='create_by.name', read_only=True)
|
|
update_by_name = serializers.CharField(
|
|
source='update_by.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = Order
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS + \
|
|
['belong_dept', 'state', 'submit_time']
|
|
|
|
def validate(self, attrs):
|
|
contract = attrs.get('contract', None)
|
|
if contract:
|
|
attrs['customer'] = contract.customer
|
|
if attrs.get('customer', None) is None:
|
|
raise ValidationError('未选择客户')
|
|
return attrs
|
|
|
|
def update(self, instance, validated_data):
|
|
if instance.state != Order.ORDER_CREATE:
|
|
raise ValidationError('订单信息不可编辑')
|
|
return super().update(instance, validated_data)
|
|
|
|
|
|
class OrderItemSerializer(CustomModelSerializer):
|
|
material_ = MaterialSerializer(source='material', read_only=True)
|
|
order_ = OrderSerializer(source='order', read_only=True)
|
|
|
|
class Meta:
|
|
model = OrderItem
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS_BASE + ['delivered_count']
|
|
|
|
def validate(self, attrs):
|
|
order = attrs['order']
|
|
if order.state != Order.ORDER_CREATE:
|
|
raise ValidationError('该订单状态下不可创建')
|
|
return attrs
|
|
|
|
def create(self, validated_data):
|
|
order = validated_data['order']
|
|
material = validated_data['material']
|
|
if OrderItem.objects.filter(order=order, material=material).exists():
|
|
raise ValidationError('该产品已选择!')
|
|
return super().create(validated_data)
|
|
|
|
def update(self, instance, validated_data):
|
|
validated_data.pop('product', None)
|
|
validated_data.pop('order', None)
|
|
return super().update(instance, validated_data)
|
|
|
|
|
|
class ContractRecordSerializer(CustomModelSerializer):
|
|
contract_number = serializers.CharField(source='contract.number', read_only=True)
|
|
customer_name = serializers.CharField(source='contract.customer.name', read_only=True)
|
|
|
|
class Meta:
|
|
model = ContractRecord
|
|
fields = '__all__'
|
|
read_only_fields = EXCLUDE_FIELDS + ['belong_dept']
|
|
|
|
def validate(self, attrs):
|
|
contract = attrs.get('contract', getattr(self.instance, 'contract', None))
|
|
amount = attrs.get('amount', getattr(self.instance, 'amount', None))
|
|
if contract is None or amount is None:
|
|
return attrs
|
|
if contract.status == Contract.STATUS_TERMINATED:
|
|
raise ValidationError('合同已终止，不可操作到款流水')
|
|
qs = ContractRecord.objects.filter(contract=contract)
|
|
if self.instance is not None:
|
|
qs = qs.exclude(id=self.instance.id)
|
|
total = sum((item.amount for item in qs), Decimal('0.00')) + amount
|
|
contract_amount = Decimal(str(contract.amount or 0))
|
|
if total > contract_amount:
|
|
raise ValidationError('累计到款金额不可超过合同金额')
|
|
return attrs
|