Changeset 13457 for main/waeup.kofa/trunk/src/waeup/kofa/students/utils.py

Timestamp:

16 Nov 2015, 09:05:30 (9 years ago)

Author:

Henrik Bettermann

Message:

Add option which allows students to select a desired hostel before booking accommodation.

File:

• main/waeup.kofa/trunk/src/waeup/kofa/students/utils.py (modified) (1 diff)

main/waeup.kofa/trunk/src/waeup/kofa/students/utils.py

@@ -627 +627 @@
         return
 
-    def selectBed(self, available_beds):
-        """Select a bed from a list of available beds.
-        In the base configuration we select the first bed found,
-        but can also randomize the selection if we like.
-        """
-        return available_beds[0]
+    def selectBed(self, available_beds, desired_hostel=None):
+        """Select a bed from a filtered list of available beds.
+        In the base configuration beds are sorted by the sort id
+        of the hostel and the bed number. The first bed found in
+        this sorted list is taken.
+        """
+        sorted_beds = sorted(available_beds,
+                key=lambda bed: 1000 * bed.__parent__.sort_id + bed.bed_number)
+        if desired_hostel:
+            # Filter desired hostel beds
+            filtered_beds = [bed for bed in sorted_beds
+                             if bed.bed_id.startswith(desired_hostel)]
+            if not filtered_beds:
+                return
+            return filtered_beds[0]
+        return sorted_beds[0]
 
     def _admissionText(self, student, portal_language):