from django.http import JsonResponse from django.shortcuts import render # Create your views here. from django.utils import timezone from battleship.models import Game, Player, generate_token from battleship.utils import check_field def new_game(request): game = Game.objects.create() player1 = Player.objects.create( game=game, number=0, token=generate_token() ) player2 = Player.objects.create( game=game, number=1 ) return JsonResponse({ 'game_id': game.id, 'player_token': player2.attend_token, 'my_token': player1.token }) def attend_game(request): game_id = request.POST['game_id'] attend_token = request.POST['attend_token'] player = Player.objects.get(game_id=game_id, attend_token=attend_token) if player.token is not None: return JsonResponse({}, status=403) player.token = generate_token() player.save() return JsonResponse({ 'token': player.token }) def place_ships(request): game_id = request.POST['game_id'] token = request.POST['token'] player = Player.objects.get(game_id=game_id, token=token) if player.field != ' ' * 100: return JsonResponse({}, status=403) player.field = request.POST['field'] if check_field(player.field): player.save() return JsonResponse({}) else: return JsonResponse({}, status=403) def check_opponent(request): game_id = request.POST['game_id'] token = request.POST['token'] player = Player.objects.get(game_id=game_id, token=token) player2 = Player.objects.filter(game_id=game_id, number=(1 - player.number)).first() if player2 is None: return JsonResponse({"attend": False, "ready": False}) return JsonResponse({"attend": True, "ready": player2.field != ' ' * 100}) def shoot(request): game_id = request.POST['game_id'] token = request.POST['token'] player = Player.objects.get(game_id=game_id, token=token) if player.game.turn != player.number: return JsonResponse({}, status=403) player2 = Player.objects.get(game_id=game_id, number=(1 - player.number)) h = request.POST['h'] v = request.POST['v'] pos = h * 10 + v if player2.field[pos] == 'x' or player2.field[pos] == '.': return JsonResponse({}, status=403) if player2.field[pos] == 'o': new_symb = 'x' else: new_symb = '.' player.game.turn = (1 - player.game.turn) player.game.last_move_ts = timezone.now() player.game.save() if pos == 0: player2.field = new_symb + player2.field[1:] elif pos == 99: player2.field[:99] + new_symb else: player2.field = player2.field[:pos] + new_symb + player2.field[pos + 1:] player2.save() return JsonResponse({ 'shot': new_symb == 'x', 'game_finish': 'o' not in player2.field }) def check_status(request): game_id = request.POST['game_id'] token = request.POST['token'] player = Player.objects.get(game_id=game_id, token=token) player2 = Player.objects.get(game_id=game_id, number=1 - player.number) return JsonResponse({ 'my_turn': player.game.turn == player.number, 'game_finished': 'o' not in player.field or 'o' not in player2.field }) def check_field_correct(request): return JsonResponse({ 'correct': check_field(request.POST['field']) })