diff --git a/battleship/views.py b/battleship/views.py index 9cbc943..63fbee0 100644 --- a/battleship/views.py +++ b/battleship/views.py @@ -144,12 +144,15 @@ def get_fields(request): player = Player.objects.get(game_id=game_id, token=token) my_field = list(player.field) player2 = Player.objects.get(game_id=game_id, number=(1 - player.number)) - another_field = list(player2.field.replace('o', ' ')) + another_field = player2.field + game_finished = 'o' not in player.field or 'o' not in player2.field + if not game_finished: + another_field = list(another_field.replace('o', ' ')) return JsonResponse({ 'my_field': [my_field[x * 10: (x + 1) * 10] for x in range(0, 10)], 'opponent_field': [another_field[x * 10: (x + 1) * 10] for x in range(0, 10)], 'my_turn': player.game.turn == player.number, - 'game_finished': 'o' not in player.field or 'o' not in player2.field + 'game_finished': game_finished })