繁体   English   中英

Django Rest Framework将数据添加到序列化器

[英]Django rest framework add data to serializer

我正在尝试将数据添加到我的序列化器。 我所拥有的是一个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才能使其正常工作并发送相关数据。 我该怎么做呢?

您需要将match字段添加到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()

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM