diff --git a/battleship/views.py b/battleship/views.py index db531aa..1babb96 100644 --- a/battleship/views.py +++ b/battleship/views.py @@ -5,7 +5,7 @@ from django.http import JsonResponse from django.utils import timezone from django.views.decorators.csrf import csrf_exempt -from battleship.models import Game, Player, generate_token +from battleship.models import Game, Player from battleship.utils import check_field @@ -38,7 +38,7 @@ def attend_game(request): game_id = data['game_id'] attend_token = data['attend_token'] player = Player.objects.get(game_id=game_id, token=attend_token) - if player.token is not None: + if player.attended: return JsonResponse({}, status=403) player.attended = True player.save() @@ -75,7 +75,6 @@ def check_opponent(request): player = Player.objects.get(game_id=game_id, token=token) player2 = Player.objects.filter(game_id=game_id, number=(1 - player.number)).first() value = {"attend": player2.attended, "ready": 'o' in player2.field} - print(value) return JsonResponse(value)