Only returns board game when game is finished

This commit is contained in:
2020-11-08 00:26:08 -03:00
parent 956008f736
commit 6fdd9ac743

View File

@@ -1,12 +1,19 @@
from rest_framework import serializers from rest_framework import serializers
from game.models import Game, GameEvent from game.models import Game, GameEvent, GameStatuses
class GameSerializer(serializers.ModelSerializer): class GameSerializer(serializers.ModelSerializer):
def to_representation(self, instance):
response = super().to_representation(instance)
if not response["status"] == GameStatuses.FINISHED:
del response["board"]
return response
class Meta: class Meta:
model = Game model = Game
exclude = ["board"] fields = "__all__"
class GameEventSerializer(serializers.ModelSerializer): class GameEventSerializer(serializers.ModelSerializer):