47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from ..AGame import AGame
|
|
|
|
from channels.generic.websocket import WebsocketConsumer
|
|
|
|
from .TicTacToePlayer import TicTacToePlayer
|
|
|
|
from .TicTacToeSpectator import TicTacToeSpectator
|
|
|
|
|
|
class TicTacToeGame(AGame):
|
|
|
|
def __init__(self, game_id: int, game_manager):
|
|
super().__init__("tictactoe", game_id, game_manager)
|
|
|
|
def _everbody_is_here(self):
|
|
return len(self.players) == len(self.get_players_connected())
|
|
|
|
def _player_join(self, user_id: int, socket: WebsocketConsumer):
|
|
|
|
if (self.model.started):
|
|
return None
|
|
|
|
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 _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 |