Moved emergency contacts view to pdf view

2023-08-29 16:17:04 +10:00
parent eb03670290
commit ab22868584
3 changed files with 47 additions and 37 deletions
@@ -2,7 +2,8 @@ from django.contrib.admin.views.decorators import staff_member_required
 from django.shortcuts import render
 from django.views.generic import ListView

-from .context_helpers import bus_roll_context
+from .context_helpers import bus_roll_context, emergency_contacts_context
+from .email_helpers import render_to_pdf
 from .models import Company, Bus, Traveller, BusStop, TravellerRoute, Shuttle, Driver


@@ -73,39 +74,8 @@ def bus_summary(request, queryset=None):

@staff_member_required
 def emergency_contacts(request, queryset=None):
-    if queryset is None:
-        buses = Bus.objects.all()
-    else:
-        buses = queryset
-
-    bus_routes = []
-    for bus in buses:
-        traveller_list = []
-        for travellerRoute in TravellerRoute.objects.filter(busStop__bus=bus):
-            traveller = travellerRoute.traveller
-            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, 'travellers': traveller_list})
-
-    return render(request, 'reports/emergency_contacts.html', {'routes': bus_routes})
+    pdf = render_to_pdf('reports/emergency_contacts.html', emergency_contacts_context(queryset))
+    return pdf


@staff_member_required