序列化 DRF 中的嵌套对象

Question

我的网站上有一个测试对象，我正试图正确地进入数据库。 它是一个带有嵌套产品数据的订单对象。 下面是前端发布的 JSON 示例：

{
    "phone": "123456789",
    "first_name": "name",
    "delivery_date": "2020-01-06",
    "delivery_time": 2,
    "address": "address",
    "comment": "comment",
    "payment": 0,
    "order_items": [
      {
        "quantity": 2,
        "pizza": 1
      },
      {
        "quantity": 3,
        "pizza": 2
      }
    ]
}

这是我的serializers.py

class OrderItemSerializer(serializers.ModelSerializer):
    quantity = serializers.IntegerField()
    pizza = serializers.SerializerMethodField()
    print(pizza, quantity)

    def get_pizza(self, obj):
        print(obj.pizza.id)
        return obj.pizza.id

    class Meta:
        model = OrderItem
        fields = ('quantity',
                  'pizza',)


class OrderSerializer(serializers.ModelSerializer):
    order_items = serializers.SerializerMethodField()

    """
    Calculate order_items field 
    """
    def get_order_items(self, obj):
        items = obj.orderitem_set.all()
        print(obj)
        print(items)
        return OrderItemSerializer(items, many=True).data

    class Meta:
        model = Order
        fields = ('phone',
                  'first_name',
                  'delivery_date',
                  'delivery_time',
                  'address',
                  'comment',
                  'payment',
                  'order_items',)

因此，order_items 始终为空： "order_items": []

我正在使用 Swagger 来测试和记录 API。我可以直接从那里发布订单而没有错误，但即使我手动添加订单项目，它们仍然不会出现在服务器响应正文中。这一定意味着我没有正确处理嵌套对象。

发布

{
    "phone": "123456",
    "first_name": "string",
    "delivery_date": "2000-10-21",
    "delivery_time": 2,
    "address": "string",
    "comment": "string",
    "payment": 0,
    "order_items": [
        {"pizza": 2,"quantity": 3},
        {"pizza": 1,"quantity": 4}]
}

201，响应正文：

{
  "phone": "123456",
  "first_name": "string",
  "delivery_date": "2000-10-21",
  "delivery_time": 2,
  "address": "string",
  "comment": "string",
  "payment": 0,
  "order_items": []
}

我还在上面的代码中设置了一些print 行，看起来OrderItemSerializer 根本没有被使用。 get_order_items 打印订单 id 和一个空数组，而 get_pizza 在网站下订单时不打印任何内容。

Answer 1

最终我想出了这个。 我不得不将order_items 变量添加到Order 对象，它没有以任何方式被使用，只会浪费数据库中的空间。虽然不是很大的浪费，但我觉得无论如何都不是写代码的好方法。

如果有人可以在 cmets 或其他答案中提出更好的版本，我也会尝试一下。

from rest_framework import serializers
from .models import Order, OrderItem
import copy


class OrderItemSerializer(serializers.ModelSerializer):
    class Meta:
        model = OrderItem
        fields = ('quantity',
                  'pizza',)


class OrderSerializer(serializers.ModelSerializer):
    order_items = OrderItemSerializer(many=True)

    class Meta:
        model = Order
        fields = ('phone',
                  'first_name',
                  'delivery_date',
                  'delivery_time',
                  'address',
                  'comment',
                  'payment',
                  'order_items')

    def create(self, validated_data, **kwargs):
        for_items = copy.deepcopy(validated_data)

        # create order object
        del validated_data['order_items']
        print('VALID', validated_data)
        order = Order.objects.create(**validated_data)

        # create order items
        for item in for_items.pop('order_items'):
            order_item = dict(item.items())
            print('ITEM', order_item)
            OrderItem.objects.create(order=order, **order_item)

        return order

Answer 2

您需要重写序列化程序的create() 方法以生成writable nested serializers。

试试这样的：

class OrderSerializer(serializers.ModelSerializer):
    order_items = serializers.SerializerMethodField()

    """
    Calculate order_items field 
    """
    def get_order_items(self, obj):
        items = obj.orderitem_set.all()
        print(obj)
        print(items)
        return OrderItemSerializer(items, many=True).data

    class Meta:
        model = Order
        fields = ('phone',
                  'first_name',
                  'delivery_date',
                  'delivery_time',
                  'address',
                  'comment',
                  'payment',
                  'order_items',)

    def create(self, validated_data):
        items_data = validated_data.pop('order_items')
        order = Order.objects.create(**validated_data)
        for item_data in items_data:
            OrderItem.objects.create(order=order, **item_data)
        return order

您也可以只使用OrderItemSerializer 作为字段而不使用SerializerMethodField。您可以使用source 参数指定字段source：

order_items = serializers.OrderItemSerializer(many=True, source="orderitem_set")