如何在Django Rest Framework中显示嵌套对象的属性

问题描述 投票:0回答:1

我正在使用Django和Django-Rest-Framework构建战斗系统的API。在我的代码中,我有2个模型:父模型Battle和子模型RoundRound具有一些@property字段(start_time, end_time, score),这些字段是根据不同的值计算的。当我直接访问Round路由时,会得到所需的输出:

http://127.0.0.1:8001/battle/rounds/1/

{
    "id": 1,
    "battle": "http://127.0.0.1:8001/battle/battles/1/",
    "index": 0,
    "contender_entry": null,
    "opponent_entry": null,
    "start_time": "2019-12-11T17:38:00Z",
    "end_time": "2019-12-11T17:39:40Z",
    "score": [
        0,
        0
    ]
}

但是,当我访问Battle路由时,将返回嵌套的Round,但是only数据库字段,而不是属性:

http://127.0.0.1:8001/battle/battles/1/

{
    "url": "http://127.0.0.1:8001/battle/battles/1/",
    "id": 1,
    "status": "live",
    "start_time": "2019-12-11T17:38:00Z",
    "round_length": "00:01:40",
    ...
    "rounds": [
        {
            "url": "http://127.0.0.1:8001/battle/rounds/1/",
            "beat": null,
            "index": 0,
            "battle": "http://127.0.0.1:8001/battle/battles/1/",
            "contender_entry": null,
            "opponent_entry": null
        },
        {
            "url": "http://127.0.0.1:8001/battle/rounds/2/",
            "beat": null,
            "index": 1,
            "battle": "http://127.0.0.1:8001/battle/battles/1/",
            "contender_entry": null,
            "opponent_entry": null
        },
        {
            "url": "http://127.0.0.1:8001/battle/rounds/3/",
            "beat": null,
            "index": 2,
            "battle": "http://127.0.0.1:8001/battle/battles/1/",
            "contender_entry": null,
            "opponent_entry": null
        }
    ],
    "current_round": null
}

我希望属性显示在Round中的嵌套Battle对象中。但是我无法使它正常工作。

这些是我的模特:

class Round(models.Model):
    battle = models.ForeignKey(Battle, on_delete=models.CASCADE, related_name="rounds")
    index = models.IntegerField()
    contender_entry = models.OneToOneField(Entry, on_delete=models.DO_NOTHING, related_name="round_contender",
                                           null=True)
    opponent_entry = models.OneToOneField(Entry, on_delete=models.DO_NOTHING, related_name="round_opponent", null=True)

    @property
    def start_time(self):
        return self.battle.start_time + (self.index * self.battle.round_length)

    @property
    def end_time(self):
        return self.start_time + self.battle.round_length

    @property
    def score(self):
        opponent_votes = self.votes.filter(favors="opponent").count()
        contender_votes = self.votes.filter(favors="contender").count()
        draw_votes = self.votes.filter(favors="draw").count()
        return (opponent_votes + draw_votes, contender_votes + draw_votes)

class Battle(models.Model):
    status = models.CharField(max_length=32, choices=BATTLE_STATUS_CHOICES, default="awaiting_approval")
    contender = models.ForeignKey(User, on_delete=models.CASCADE, related_name="contender_battles")
    opponent = models.ForeignKey(User, on_delete=models.CASCADE, related_name="opponent_battles")
    start_time = models.DateTimeField(default=timezone.now)
    round_length = models.DurationField(default=timedelta(days=3))

和序列化器:

class RoundSerializer(serializers.HyperlinkedModelSerializer):
    id = serializers.IntegerField(source="pk", read_only=True)

    class Meta:
        model = Round
        fields = ["id", "battle", "index", "contender_entry", "opponent_entry", "start_time", "end_time", "score"]
        read_only_fields = ["id", "battle", "index", "start_time", "end_time", "score"]

class BattleSerializer(serializers.HyperlinkedModelSerializer):
    id = serializers.IntegerField(source='pk', read_only=True)
    current_round = RoundSerializer(read_only=True)

    class Meta:
        model = Battle
        fields = ["url", "id", "status", "start_time", "round_length",
                  "opponent", "contender", "rounds", "current_round"]
        read_only_fields = ["contender", "rounds", "status"]


class BattleReadSerializer(BattleSerializer):
    contender = UserSerializer(read_only=True)
    opponent = UserSerializer(read_only=True)

    class Meta:
        model = Battle
        fields = ["url", "id", "status", "start_time", "round_length",
                  "opponent", "contender", "rounds", "current_round"]
        read_only_fields = ["contender", "rounds"]
        depth = 1

[请注意,我有2个Battle序列化程序:BattleSerializer用于POST和PUT,以及用户超链接而不是嵌套字段。 BattleReadSerializer用于GET并嵌套输出。 BattleReadSerializer是上面示例中使用的那个。

我已经尝试将字段显式添加到RoundSerializer,如下所示:

class RoundSerializer(serializers.HyperlinkedModelSerializer):
    id = serializers.IntegerField(source="pk", read_only=True)
    start_time = serializers.DateTimeField(read_only=True)
    score = serializers.ListField(read_only=True)

但是那没什么改变。有什么方法可以使属性字段显示(除了使它们成为数据库字段并经常重新计算它们)?

python django django-models django-rest-framework django-serializer
1个回答
1
投票

尝试如下将RoundSerializer序列化器显式添加到BattleReadSerializer序列化器,

class BattleReadSerializer(BattleSerializer):
    contender = UserSerializer(read_only=True)
    opponent = UserSerializer(read_only=True)
    rounds = RoundSerializer(read_only=True, many=True)

    class Meta:
        model = Battle
        fields = ["url", "id", "status", "start_time", "round_length",
                  "opponent", "contender", "rounds", "current_round"]
        read_only_fields = ["contender", "rounds"]
        depth = 1  # remove this
© www.soinside.com 2019 - 2024. All rights reserved.