107 lines
3.2 KiB
Python
107 lines
3.2 KiB
Python
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
|
|
})
|