From 6fdd9ac743eaeec912c2e3cd7624cdd169addb57 Mon Sep 17 00:00:00 2001
From: Michel Wilhelm <michelwilhelm@gmail.com>
Date: Sun, 8 Nov 2020 00:26:08 -0300
Subject: [PATCH] Only returns board game when game is finished

---
 api/serializers/game.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/api/serializers/game.py b/api/serializers/game.py
index 2539085..f537bdd 100644
--- a/api/serializers/game.py
+++ b/api/serializers/game.py
@@ -1,12 +1,19 @@
 from rest_framework import serializers
 
-from game.models import Game, GameEvent
+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
-        exclude = ["board"]
+        fields = "__all__"
 
 
 class GameEventSerializer(serializers.ModelSerializer):