tournament: add: player can join tournament now

tournament/consumers.py

@@ -6,9 +6,9 @@ from games.models import GameModel
 
 import json
 
-from .models import Waiter, WaitingRoom, WaitingRoomManager, normal
+from .models import tournament_manager, TournamentMember, TournamentRoom, TournamentRoomManager
 
-class TournamentWaitingRoom(WebsocketConsumer):
+class TournamentWebConsumer(WebsocketConsumer):
 
 	def __init__(self, *args, **kwargs):
 		super().__init__(*args, **kwargs)
@@ -17,25 +17,27 @@ class TournamentWaitingRoom(WebsocketConsumer):
 
 	def connect(self):
 
-		user: User = self.scope["user"]
-		if (user.is_anonymous or not user.is_authenticated):
+		self.user: User = self.scope["user"]
+		if (self.user.is_anonymous or not self.user.is_authenticated):
 			return
 
 		self.channel_layer.group_add(self.group_name, self.channel_name)
 
 		self.tournament_id = int(self.scope['url_route']['kwargs']['tournament_id'])
 
-		waiting_room: WaitingRoom = normal.get(self.mode)
-		waiting_room.append(Waiter(user.pk, self))
+		self.room = tournament_manager.get(self.tournament_id)
+		self.participant = TournamentMember(self.user.pk, self, self.room)
 
-		print(len(waiting_room), "/", self.mode, [len(waiting_room),self.mode])
-		if (len(waiting_room) == self.mode):
-			game_id: int = GameModel().create(waiting_room.get_users_id())
-			waiting_room.broadcast({"detail": "Game found !", "game_id": game_id})
-			waiting_room.clear()
+		if (self.room is None):
+			self.participant.send("Tournament not found")
+			self.disconnect(1017)
+	
+		self.room.append(self.participant)
+
+	def receive(self, text_data: str = None, bytes_data: bytes = None):
+		self.participant.receive(text_data, bytes_data)
 
 	def disconnect(self, close_code):
-		waiting_room: WaitingRoom = normal.get(self.mode)
-		waiter: Waiter = waiting_room.get_waiter_by_socket(self)
-		if (waiter is not None):
-			waiting_room.remove(self.scope["user"].pk)
+		member = self.room.get_member_by_socket(self)
+		if (member is not None):
+			self.room.remove(self.participant, close_code)

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()

tournament/routing.py

@@ -0,0 +1,6 @@
+from django.urls import re_path
+from . import consumers
+
+websocket_urlpatterns = [
+	re_path(r'ws/tournaments/(?P<tournament_id>\d+)$', consumers.TournamentWebConsumer.as_asgi())
+]