carwin_activity/apps/customer/order/serializers.py

# coding=utf-8

from rest_framework import serializers
from apps.activity.models import Order


class ActivityOrderSerializer(serializers.ModelSerializer):

    class Meta:
        model = Order
        fields = ('id', 'number', 'create_time', )


class OrderSerializer(serializers.ModelSerializer):

    class Meta:
        model = Order
        fields = '__all__'

    def validate(self, attrs):
        attrs['customer'] = self.context['request'].customer
        if 'activity' in attrs:
            attrs['activity'].checkStatus()
        return attrs

    def create(self, validated_data):
        validated_data['branch'] = validated_data['activity'].branch
        instance = super(OrderSerializer, self).create(validated_data)
        return instance