scj/apps/product/serializers.py

#coding=utf-8

from rest_framework import serializers
from django.db.models import Q

from apps.exceptions import CustomError
from apps.serializer_errors import dump_serializer_errors
from apps.warehouse.models import WarehouseStock
from models import ProductBase

class ProductBaseSerializer(serializers.ModelSerializer):
    class Meta:
        model = ProductBase
        fields = '__all__'

    @staticmethod
    def factory(user, data, id=None):
        if id:
            instance = ProductBase.getById(id)
        else:
            instance = None

        serializer = ProductBaseSerializer(instance, data=data)
        serializer.user = user
        return serializer

    def create(self, validated_data):
        item = ProductBase.objects.filter(type=validated_data['type'], model=validated_data['model']).first()
        if item:
            raise CustomError(u'代码为[%s]的%s已存在' % (validated_data['model'],ProductBase.getTypeText(validated_data['type'])))

        instance = super(ProductBaseSerializer, self).create(validated_data)
        WarehouseStock.createByProduct(instance)
        return instance

    def update(self, instance, validated_data):
        if 'type' in validated_data:
            if validated_data['type'] != instance.type:
                raise CustomError(u'禁止修改产品的类型')

        item = ProductBase.objects.filter(Q(type=instance.type), Q(model=validated_data['model']),~Q(id=instance.id)).first()
        if item:
            raise CustomError(u'代码为[%s]的%s已存在' % (validated_data['model'], ProductBase.getTypeText(instance.type)))

        instance = super(ProductBaseSerializer, self).update(instance, validated_data)
        return instance

    def validSave(self):
        if self.is_valid():
            return self.save()
        else:
            raise CustomError(dump_serializer_errors(self))