tournament: add: player can join tournament now

tournament/models.py

@@ -1,7 +1,15 @@
 from django.db import models
 
+from channels.generic.websocket import WebsocketConsumer
+
 from games.models import GameModel
 
+import json
+
+from transcendence.abstract.AbstractRoomMember import AbstractRoomMember
+from transcendence.abstract.AbstractRoom import AbstractRoom
+from transcendence.abstract.AbstractRoomManager import AbstractRoomManager
+
 # Create your models here.tu
 class TournamentModel(models.Model):
 
@@ -33,4 +41,80 @@ class TournamentGamesModel(models.Model):
 
 	tournament_id = models.IntegerField()
 	tournament_level = models.IntegerField()
-	game_id = models.IntegerField()
+	game_id = models.IntegerField()
+
+class TournamentMember(AbstractRoomMember):
+
+	def __init__(self, user_id: int, socket: WebsocketConsumer, tournament):
+		super().__init__(user_id, socket)
+		self.participate = False
+		self.tournament = tournament
+
+	def receive(self, text_data: str = None, byte_dates: bytes = None):
+		
+		if (text_data is None):
+			return
+
+		data: dict = json.loads(text_data)
+
+		self.update_participate(data.get("participate", self.participate))
+
+	def update_participate(self, new_participate: bool):
+
+		if (self.participate == new_participate):
+			return
+		self.participate = new_participate
+		self.tournament.update_participants()
+
+class TournamentRoom(AbstractRoom):
+
+	def __init__(self, room_manager, tournament_id: int):
+		super().__init__(room_manager)
+		self.tournament_id = tournament_id
+		self.definitive_participant_list = []
+		self.started = False
+		self.model = TournamentModel.objects.get(pk=tournament_id)
+
+	def start(self):
+		self.broadcast("tournament_start")
+
+	def update_participants(self):
+		nb_participants = self.get_nb_participants()
+		self.broadcast("update_participants", {"nb_participants": nb_participants})
+		if (nb_participants == self.model.nb_players):
+			self.start()
+
+	def get_nb_participants(self):
+		nb_participants = 0
+		for member in self._member_list:
+			member: TournamentMember
+			if (member.participate):
+				nb_participants += 1
+		return nb_participants 
+
+	def get_participants(self):
+		return [member.user_id for member in self._member_list if member.participate]
+
+	def start(self):
+		self.started = True
+	
+	def append(self, member: TournamentMember):
+		super().append(member)
+		member.send("nb_participants", {"nb_participants": self.get_nb_participants()})
+
+class TournamentRoomManager(AbstractRoomManager):
+
+	def get(self, tournament_id: int):
+
+		for room in self._room_list:
+			if (room.tournament_id == tournament_id):
+				return room
+		
+		if (TournamentModel.objects.filter(pk = tournament_id).exists()):
+			room = TournamentRoom(self, tournament_id)
+			self.append(room)
+			return room
+		
+		return None
+
+tournament_manager: TournamentRoomManager = TournamentRoomManager()