matchmaking/consumers.py

@@ -28,14 +28,14 @@ class MatchMaking(WebsocketConsumer):
 
 		waiting_room: WaitingRoom = normal.get(self.mode)
 		waiting_room.append(Waiter(user.pk, self))
-		waiting_room.broadcast(f"{len(waiting_room)} / {waiting_room.mode}")
+		waiting_room.broadcast(f"{len(waiting_room)} / {waiting_room._mode}")
-		if (len(waiting_room) == waiting_room.mode):
+		if (len(waiting_room) == self.mode):
 			game_id: int = GameModel().create(waiting_room.get_users_id())
 			waiting_room.broadcast("game_found", {"game_id": game_id})
 			waiting_room.clear()
 
 	def disconnect(self, close_code):
 		waiting_room: WaitingRoom = normal.get(self.mode)
-		waiter: Waiter = waiting_room.get_member_by_socket(self)
+		waiter: Waiter = waiting_room.get_waiter_by_socket(self)
 		if (waiter is not None):
 			waiting_room.remove(waiter)

matchmaking/models.py

@@ -3,37 +3,94 @@ from django.db import models
 from channels.generic.websocket import WebsocketConsumer
 import json
 
-from transcendence.abstract.AbstractRoom import AbstractRoom
-from transcendence.abstract.AbstractRoomManager import AbstractRoomManager
-from transcendence.abstract.AbstractRoomMember import AbstractRoomMember
-
 # Create your models here.
-class Waiter(AbstractRoomMember):
+class Waiter:
-	pass
 
-class WaitingRoom(AbstractRoom):
+	def __init__(self, user_id: int, socket: WebsocketConsumer):
+		self.user_id: int = user_id
+		self.socket: WebsocketConsumer = socket
 	
-	def __init__(self, room_manager,mode):
+	def send(self, detail: str, data: dict = {}):
-		super().__init__(room_manager)
+		raw_data: dict = {"detail": detail}
-		self.mode = mode
+		raw_data.update(data)
+		self.socket.send(text_data=json.dumps(raw_data))
+
+	def accept(self):
+		self.socket.accept()
+
+	def disconnect(self):
+		self.socket.disconnect(200)
+
+class WaitingRoom:
+	
+	def __init__(self, waiting_room_manager, mode: int):
+		self._waiter_list: [Waiter] = [] 
+		self._mode: int = mode
+		self._waiting_room_manager = waiting_room_manager
+
+	def broadcast(self, detail: str, data: dict = {}):
+		for waiter in self._waiter_list:
+			waiter: Waiter
+			waiter.send(detail, data)
+	
+	def clear(self):
+		self._waiter_list.clear()
+
+	def get_waiter_by_socket(self, socket: WebsocketConsumer):
+		for waiter in self._waiter_list:
+			waiter: Waiter
+			if (waiter.socket is socket):
+				return waiter
+		return None
+
+	def get_waiter_by_user_id(self, user_id: int):
+		for waiter in self._waiter_list:
+			waiter: Waiter
+			if (waiter.user_id == user_id):
+				return waiter
+		return None
+
 
 	def append(self, waiter: Waiter):
-		tmp: Waiter = self.get_member_by_user_id(waiter.user_id)
+		tmp: Waiter = self.get_waiter_by_user_id(waiter.user_id)
 		if (tmp is not None):
 			tmp.send("Connection close: Another connection open with the same user id.")
 			self.remove(tmp)
 		waiter.accept()
-		self._member_list.append(waiter)
+		self._waiter_list.append(waiter)
 
-class WaitingRoomManager(AbstractRoomManager):
+	def remove(self, waiter: Waiter):
+		self._waiter_list.remove(waiter)
+		waiter.disconnect()
+
+	def empty(self):
+		for _ in self._waiter_list:
+			return False
+		return True
+
+	def get_users_id(self):
+		return [waiter.user_id for waiter in self._waiter_list]
+
+	def __len__(self):
+		return len(self._waiter_list)
+	
+
+
+class WaitingRoomManager:
+
+	def __init__(self):
+		self._waiting_rooms: [WaitingRoom] = []
 
 	def get(self, mode: int):
-		for waiting_room in self._room_list:
+		for waiting_room in self._waiting_rooms:
-			waiting_room: WaitingRoom
+			waiting_room: WaitingRoom = waiting_room
-			if (waiting_room.mode == mode):
+			if (waiting_room._mode == mode):
 				return waiting_room
 		tmp: WaitingRoom = WaitingRoom(self, mode)
-		super().append(tmp)
+		self._waiting_rooms.append(tmp)
 		return tmp
+	
+	def remove(self, waiting_room: WaitingRoom):
+		self._waiting_rooms.remove(waiting_room)
 
 normal: WaitingRoomManager = WaitingRoomManager()

transcendence/abstract/AbstractRoom.py

@@ -1,5 +1,3 @@
-from channels.generic.websocket import WebsocketConsumer
-
 from .AbstractRoomMember import AbstractRoomMember
 
 class AbstractRoom:
@@ -32,12 +30,16 @@ class AbstractRoom:
 
 
 	def append(self, member: AbstractRoomMember):
-		self._member_list.append(waiter)
+		tmp: AbstractRoomMember = self.get_member_by_user_id(member.user_id)
+		if (tmp is not None):
+			tmp.send("Connection close: Another connection open with the same user id.")
+			self.remove(tmp)
 		member.accept()
+		self._member_list.append(waiter)
 
 	def remove(self, member: AbstractRoomMember):
 		self._member_list.remove(member)
-		member.disconnect()
+		waiter.disconnect()
 
 	def empty(self):
 		for _ in self._waiter_list:
@@ -45,7 +47,7 @@ class AbstractRoom:
 		return True
 
 	def get_users_id(self):
-		return [member.user_id for member in self._member_list]
+		return [waiter.user_id for waiter in self._waiter_list]
 
 	def __len__(self):
-		return len(self._member_list)
+		return len(self._waiter_list)

transcendence/abstract/AbstractRoomManager.py

@@ -15,4 +15,4 @@ class AbstractRoomManager:
 		return tmp
 	
 	def remove(self, waiting_room: WaitingRoom):
-		self._waiting_rooms.remove(waiting_room)
+		self._waiting_rooms.remove(waiting_room)

transcendence/abstract/AbstractRoomMember.py

@@ -1,7 +1,3 @@
-from channels.generic.websocket import WebsocketConsumer
-
-import json
-
 class AbstractRoomMember:
 
 	def __init__(self, user_id: int, socket: WebsocketConsumer):