Moved emergency contacts view to pdf view

busManager/coord/context_helpers.py

@@ -1,4 +1,4 @@
-from coord.models import Bus, BusStop, TravellerRoute
+from coord.models import Bus, BusStop, TravellerRoute, Driver
 
 
 def bus_roll_context(queryset=None):
@@ -10,7 +10,6 @@ def bus_roll_context(queryset=None):
 
     for bus in buses:
         bus_route = []
-
         for bus_stop in BusStop.objects.filter(bus=bus):
             traveller_list = []
             for trav_route in TravellerRoute.objects.filter(busStop=bus_stop):
@@ -38,3 +37,44 @@ def bus_roll_context(queryset=None):
 
         bus_routes.append({'bus': bus, 'stops': bus_route})
     return {'routes': bus_routes}
+
+
+def emergency_contacts_context(queryset=None):
+    if queryset is None:
+        buses = Bus.objects.all()
+    else:
+        buses = queryset
+
+    bus_routes = []
+    for bus in buses:
+        drivers = []
+        for driver in Driver.objects.filter(bus=bus):
+            drivers.append(driver)
+        traveller_list = []
+        for travellerRoute in TravellerRoute.objects.filter(busStop__bus=bus):
+            traveller = travellerRoute.traveller
+            if not traveller.is_active():
+                continue
+            parent_a = ""
+            if travellerRoute.traveller.parent_A_firstname:
+                parent_a = f"{traveller.parent_A_firstname} {traveller.parent_A_lastname} ({traveller.parent_A_phone})"
+            parent_b = ""
+            if travellerRoute.traveller.parent_B_firstname:
+                parent_b = f"{traveller.parent_B_firstname} {traveller.parent_B_lastname} ({traveller.parent_B_phone})"
+            contact_a = ""
+            if travellerRoute.traveller.emergency_contact_A_firstname:
+                contact_a = f"{traveller.emergency_contact_A_firstname} {traveller.emergency_contact_A_lastname} ({traveller.emergency_contact_A_phone})"
+            contact_b = ""
+            if travellerRoute.traveller.emergency_contact_B_firstname:
+                contact_b = f"{traveller.emergency_contact_B_firstname} {traveller.emergency_contact_B_lastname} ({traveller.emergency_contact_B_phone})"
+            traveller_list.append({
+                'traveller': traveller,
+                'parent_a': parent_a,
+                'parent_b': parent_b,
+                'contact_a': contact_a,
+                'contact_b': contact_b,
+                'note': travellerRoute.notes
+            })
+        bus_routes.append({'bus': bus, 'drivers': drivers, 'travellers': traveller_list})
+
+    return {'routes': bus_routes}