104 lines
3.2 KiB
Python
104 lines
3.2 KiB
Python
from ..AGame import AGame
|
|
|
|
from channels.generic.websocket import WebsocketConsumer
|
|
|
|
from .TicTacToePlayer import TicTacToePlayer
|
|
|
|
from .TicTacToeSpectator import TicTacToeSpectator
|
|
|
|
import time
|
|
|
|
import sys
|
|
|
|
class TicTacToeGame(AGame):
|
|
|
|
|
|
def __init__(self, game_id: int, game_manager):
|
|
super().__init__("tictactoe", game_id, game_manager)
|
|
|
|
players = self.model.get_players()
|
|
|
|
self.players: list[TicTacToePlayer] = [TicTacToePlayer(player, None, self, ["x", "o"][i]) for i, player in enumerate(players)]
|
|
|
|
self.time = -1
|
|
|
|
self.turn = 'x'
|
|
|
|
self._map = [[-1 for _ in range(9)] for _ in range(9)]
|
|
|
|
self.winner = None
|
|
|
|
def _everbody_is_here(self):
|
|
return len(self.players) == len(self.get_players_connected())
|
|
|
|
def get_players_by_sign(self, sign):
|
|
for player in self.players:
|
|
if (player.sign == sign):
|
|
return player
|
|
return None
|
|
|
|
def _player_join(self, user_id: int, socket: WebsocketConsumer):
|
|
|
|
player = self.get_player_by_user_id(user_id)
|
|
if (player is None):
|
|
return None
|
|
|
|
# check if player is already connected
|
|
if (player.is_connected()):
|
|
player.disconnect(1001)
|
|
|
|
player.socket = socket
|
|
return player
|
|
|
|
def add(self, newmove, player):
|
|
if (self.checkMove(newmove, player) and self.checkWin() == False):
|
|
self._map[newmove.get("targetMorpion")][newmove.get("targetCase")] = newmove.get("sign")
|
|
player.currentMorpion = int(newmove.get("targetCase"))
|
|
self.changeTurn()
|
|
self.time = time.time()
|
|
return True
|
|
|
|
return False
|
|
|
|
def changeTurn(self):
|
|
self.turn = 'x' if (self.turn == 'o') else 'o'
|
|
|
|
def checkMove(self, newmove, player):
|
|
if (int(newmove.get("targetMorpion")) != player.currentMorpion or newmove.get("sign") != self.turn):
|
|
return False
|
|
|
|
if (self._map[newmove.get("targetMorpion")][newmove.get("targetCase")] != -1):
|
|
return False
|
|
|
|
return True
|
|
|
|
def checkWin(self):
|
|
for tab in self._map:
|
|
for i in range(3):
|
|
if tab[i] != -1 and tab[i] == tab[i + 3] and tab[i + 3] == tab[i + 6]:
|
|
return
|
|
for i in range(0, 9, 3):
|
|
if tab[i] != -1 and tab[i] == tab[i + 1] and tab[i + 1] == tab[i + 2]:
|
|
return tab[i]
|
|
if tab[0] != -1 and tab[0] == tab[4] and tab[4] == tab[8]:
|
|
return tab[0]
|
|
if tab[6] != -1 and tab[6] == tab[4] and tab[4] == tab[2]:
|
|
return tab[6]
|
|
return False
|
|
|
|
|
|
def _spectator_join(self, user_id: int, socket: WebsocketConsumer):
|
|
|
|
spectator:TicTacToeSpectator = TicTacToeSpectator(user_id, socket, self)
|
|
|
|
self.spectators.append(spectator)
|
|
|
|
return spectator
|
|
|
|
def join(self, user_id: int, socket: WebsocketConsumer) -> TicTacToeSpectator | TicTacToePlayer:
|
|
member: TicTacToePlayer = self._player_join(user_id, socket)
|
|
|
|
if (member is None):
|
|
member: TicTacToeSpectator = self._spectator_join(user_id, socket)
|
|
|
|
return member |