From f3a76b6e93e758bd702aeb99c1d76ca579147f95 Mon Sep 17 00:00:00 2001
From: Andrea <rctandrew100@gmail.com>
Date: Sun, 19 May 2024 11:03:48 +0200
Subject: [PATCH] wip - handled excess rooms case for autofill

---
 admin.py | 21 ++++++++++++++++++---
 ext.py   | 12 ++++++++++++
 2 files changed, 30 insertions(+), 3 deletions(-)

diff --git a/admin.py b/admin.py
index cdd9c42..140d6c7 100644
--- a/admin.py
+++ b/admin.py
@@ -128,19 +128,34 @@ async def room_wizard(request, order:Order):
 			'room_type': 5,
 			'room_name': 'generated 1',
 			'to_add': ['B', 'a', 'c']
-		}
+		},
+		'rooms_to_delete': []
 	}
 
 	result_map = {}
 
 	# Get room quotas
 	room_quota_map = {}
+	room_quota_overflow = {}
 	for key, value in ITEM_VARIATIONS_MAP['bed_in_room'].items():
 		capacity = ROOM_CAPACITY_MAP[key] if key in ROOM_CAPACITY_MAP else 1
 		room_quota_map[value] = math.ceil((len(list(filter(lambda y: y.bed_in_room == value, orders.values())))) / capacity)
+		current_quota = len(list(filter(lambda y: y.bed_in_room == value and y.room_owner == True, orders.values())))
+		room_quota_overflow[value] = current_quota - (room_quota_map[value] if value in room_quota_map else 0)
 
-	print('RMQ = ')
-	print(room_quota_map)
+	# Init rooms to remove
+	result_map["void"] = []
+
+	# Dismember rooms that are over quota
+	for room_type, overflow_qty in {key:value for key,value in room_quota_overflow.items() if value > 0}.items():
+		sorted_rooms = sorted(incomplete_orders.values(), key=lambda r: len(r.room_members))
+		for room_to_remove in sorted_rooms[:overflow_qty]:
+			# Room codes to remove
+			result_map["void"].append(room_to_remove.code)
+			# Move room members to the roomless list
+			for member_code in room_to_remove.room_members:
+				roomless_orders[member_code] = all_orders[member_code]
+			del incomplete_orders[room_to_remove.code]
 
 	# Fill already existing rooms
 	for room_order in incomplete_orders.items():
diff --git a/ext.py b/ext.py
index 8db2788..2748003 100644
--- a/ext.py
+++ b/ext.py
@@ -247,6 +247,18 @@ class Order:
 		
 	def get_language(self):
 		return self.country.lower() if self.country.lower() in AVAILABLE_LOCALES else 'en'
+	
+	def __str__(self):
+		to_return = f"{'Room' if self.room_owner else 'Order'} {self.code}"
+		if self.room_owner == True:
+			to_return = f"{to_return} [ members = {self.room_members} ]"
+		return to_return
+	
+	def __repr__(self):
+		to_return = f"{'Room' if self.room_owner == True else 'Order'} {self.code}"
+		if self.room_owner == True:
+			to_return = f"{to_return} [ members = {self.room_members} ]"
+		return to_return
 
 @dataclass
 class Quotas: