from rest_framework import serializers

from game.models import Game, GameEvent, GameStatuses


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:
        model = Game
        fields = "__all__"


class GameEventSerializer(serializers.ModelSerializer):
    def to_representation(self, instance):
        response = super().to_representation(instance)
        response["active_game"] = GameSerializer(instance.game).data
        return response

    class Meta:
        model = GameEvent
        fields = "__all__"