Django rest框架将数据添加到序列化程序

时间:2018-06-07 20:36:14

标签: python django django-rest-framework

我正在尝试将数据添加到序列化程序中。我拥有的是Game模型,而Game可以附加多个Match模型。问题是,Game模型并不了解这种关系。该关系在Match模型上作为Game的外键。

但是,有时我会收到Game我希望将Match模型包含在内。这是我的Game模型:

import uuid
from django.db import models
from django.utils import timezone

class Game(models.Model):
    created_at = models.DateTimeField(editable=False)
    id = models.UUIDField(primary_key=True, default=uuid.uuid4)
    updated_at = models.DateTimeField(editable=False)

    def __str__(self):
        return self.title

    def save(self, *args, **kwargs):
        ''' On save, update timestamps '''
        if not self.created_at:
            self.created_at = timezone.now()
        self.updated_at = timezone.now()
        return super(Analysis, self).save(*args, **kwargs)

    class Meta:
        db_table = "games"

我的Match型号:

import uuid
from django.db import models
from django.utils import timezone
from games.models import Game

class Match(models.Model):
    game = models.ForeignKey(Game, on_delete=models.CASCADE)
    created_at = models.DateTimeField(editable=False)
    id = models.UUIDField(primary_key=True, default=uuid.uuid4)
    updated_at = models.DateTimeField(editable=False)

    def save(self, *args, **kwargs):
        ''' On save, update timestamps '''
        if not self.created_at:
            self.created_at = timezone.now()
        self.updated_at = timezone.now()
        return super(Match, self).save(*args, **kwargs)

    class Meta:
        db_table = "matches"

我的GameSerializer

from rest_framework import serializers
from games.models import Game
from games.serializers import GameSerializer

class GameSerializer(serializers.ModelSerializer):
    class Meta:
        model = Game
        fields = ('id', 'created_at', 'updated_at')

我假设我需要调整GameSerializer以使其正常工作并发送相关数据。我该怎么做?

1 个答案:

答案 0 :(得分:2)

您需要将匹配字段添加到GameSerializer,并自行处理保存。

class GameSerializer(serializers.ModelSerializer):
    matches = MatchSerializer(many=True, source='match_set')

    def save(self):
        matches = self.validated_data.get('matches', [])
        game = super(GameSerializer, self).save()
        for match in matches:
            match['game'] = game
            Match.objects.create(**match)

为您的字段添加匹配项。在序列化程序的validated_data中,matches将是来自MatchSerializer的经过验证的数据列表。因此,您可以直接在GameSerializer保存中创建Match对象。

参考:http://www.django-rest-framework.org/api-guide/relations/#writable-nested-serializers

或者,如果您的用例涉及发送已存在的Match对象的匹配项,您只需要在数据中发送ID,然后使用:

class GameSerializer(serializers.ModelSerializer):
    matches = serializers.ListField(
        child=serializers.PrimaryKeyRelatedField(Match))

    def save(self):
        matches = self.validated_data.get('matches')
        game = super(GameSerializer, self).save()
        for match in matches:
            match.game = game
            match.save()
相关问题