from rest_framework import serializers
from rest_framework import exceptions
from apps.inm.models import IProduct
from apps.inm.serializers import IProductListSerializer
from apps.mtm.serializers import MaterialSimpleSerializer
from apps.sam.models import Sale, SaleProduct
from apps.sam.serializers import CustomerSimpleSerializer, OrderSimpleSerializer
class SaleCreateSerializer(serializers.ModelSerializer):
    iproducts = serializers.PrimaryKeyRelatedField(queryset=
    IProduct.objects.all(), many=True)
    class Meta:
        model = Sale
        fields = ['customer', 'order', 'product', 'iproducts']
    
    def validate(self, attrs):
        order = attrs.get('order', None)
        if order:
            if order.customer:
                attrs['customer'] = order.customer
            attrs['product'] = order.product
        for i in attrs['iproducts']:
            if i.material !=  attrs['product']:
                raise exceptions.APIException('产品选取错误')
        return super().validate(attrs)


class SaleListSerializer(serializers.ModelSerializer):
    customer_ = CustomerSimpleSerializer(source='customer', read_only=True)
    order_ = OrderSimpleSerializer(source='order', read_only=True)
    product_ = MaterialSimpleSerializer(source='product', read_only=True)
    class Meta:
        model = Sale
        fields = '__all__'

class SaleProductListSerializer(serializers.ModelSerializer):
    iproduct_ = IProductListSerializer(source='iproduct', read_only=True)
    class Meta:
        model = SaleProduct
        fields = '__all__'

class SaleProductCreateSerializer(serializers.ModelSerializer):
    class Meta:
        model = SaleProduct
        fields = ['sale', 'iproduct']

    def create(self, validated_data):
        validated_data['number'] = validated_data['iproduct'].number
        instance = SaleProduct.objects.create(**validated_data)
        instance.sale.count = SaleProduct.objects.filter(sale=instance.sale).count()
        instance.sale.save()
        return instance