Email roll to schools function

busManager/coord/admin.py

@@ -12,7 +12,8 @@ from django.utils.translation import gettext_lazy
 from import_export.admin import ImportExportModelAdmin
 
 from .adminClone import CloneModelAdmin
-from .context_helpers import bus_roll_context, emergency_contacts_context, traveller_roll_context
+from .context_helpers import bus_roll_context, emergency_contacts_context, traveller_roll_context, \
+    traveller_route_context
 from .email_helpers import email_companies_bus_roll, render_to_pdf, email_school_roll_csv
 from .models import *
 
@@ -33,11 +34,23 @@ class BusRollMixin:
 
 class SchoolRollMixin:
 
-    def email_school_roll(self, request, queryset):
+    def export_travellers_to_csv(self, request, queryset):
+
+        traveller_list = []
         for school in queryset:
-            if not school.email:
+            for travellerRoute in TravellerRoute.objects.filter(traveller__school=school):
-                return HttpResponse(f"No email is set for {school.name}")
+                if not travellerRoute.traveller.is_active():
-            email_school_roll_csv(request, school)
+                    continue
+                traveller_list.append(traveller_route_context(travellerRoute))
+
+        response = HttpResponse(content_type="text/csv")
+        response["Content-Disposition"] = f"attachment; filename=traveller_list_{date.today()}.csv"
+
+        writer = csv.DictWriter(response, fieldnames=traveller_list[0].keys())
+        writer.writeheader()
+        writer.writerows(traveller_list)
+
+        return response
 
 
 class TravellerRollMixin:
@@ -55,7 +68,6 @@ class TravellerRollMixin:
 
 
 class ArchiveFilter(BooleanFieldListFilter):
-    title = gettext_lazy('Previous Years')
     parameter_name = 'is_archived'
 
     def choices(self, changelist):
@@ -258,7 +270,7 @@ class TravellerRouteAdmin(MyImportExportModelAdmin, admin.ModelAdmin):
 @admin.register(School)
 class SchoolAdmin(MyImportExportModelAdmin, admin.ModelAdmin, SchoolRollMixin):
     list_display = ["__str__", "address", "suburb", "school_email", "phone"]
-    actions = ["email_school_roll"]
+    actions = ["export_travellers_to_csv"]
 
     def school_email(self, obj):
         return format_html('<a href="mailto:{}">{}</a>', obj.email, obj.email)

busManager/coord/context_helpers.py

@@ -1,3 +1,5 @@
+from django.db.models import Q
+
 from coord.models import Bus, BusStop, TravellerRoute, Driver, Traveller, Shuttle, School
 
 
@@ -38,6 +40,48 @@ def bus_roll_context(queryset=None):
     return {'routes': bus_routes}
 
 
+def school_roll_context(queryset=None):
+    school_list = []
+    if queryset is None:
+        schools = School.objects.all()
+    else:
+        schools = queryset
+
+    for school in schools:
+        school_routes = []
+        query = Q(traveller__school=school) | Q(traveller__shuttle__school=school)
+        for bus in Bus.objects.all():
+            travellers = []
+            for trav_route in TravellerRoute.objects.filter(query).filter(busStop__bus=bus).order_by('busStop__am_time'):
+                traveller = trav_route.traveller
+                if not traveller.is_active():
+                    continue
+                bus_stop = trav_route.busStop
+
+                display_name = f"{traveller} ({traveller.get_year_level_display()})"
+                if traveller.school != school:
+                    display_name = f"{traveller} ({traveller.get_year_level_display()}, {traveller.school.shortName})"
+                is_fared = "---"
+                if traveller.eligibility_status == "2":
+                    is_fared = "Y"
+                travellers.append({
+                    'display': display_name,
+                    'isFared': is_fared,
+                    'stop': f"#{bus_stop.get_stop_number()} - {bus_stop.address}",
+                    'am_time': bus_stop.am_time,
+                    'pm_time': bus_stop.pm_time
+                })
+            if travellers:
+                school_routes.append({
+                    'bus': bus,
+                    'travellers': travellers
+                })
+
+
+        school_list.append({"name": school.name, "routes": school_routes})
+    return {"schools": school_list}
+
+
 def traveller_roll_context(queryset):
     travellers = []
     for traveller in queryset:
@@ -46,7 +90,7 @@ def traveller_roll_context(queryset):
     return travellers
 
 
-def school_roll_context(school):
+def school_travellerRoute_context(school):
     travellers = []
     for travellerRoute in TravellerRoute.objects.filter(traveller__school=school):
         traveller = travellerRoute.traveller

busManager/coord/email_helpers.py

@@ -7,7 +7,7 @@ from django.http import HttpResponse
 from django.template.loader import get_template
 from xhtml2pdf import pisa
 
-from coord.context_helpers import bus_roll_context, school_roll_context
+from coord.context_helpers import bus_roll_context, school_travellerRoute_context
 from coord.models import Company
 
 
@@ -49,7 +49,7 @@ def email_companies_bus_roll(request, query_set=None):
 
 def email_school_roll_csv(request, school):
 
-    travellers = school_roll_context(school)
+    travellers = school_travellerRoute_context(school)
     csvFile = StringIO()
     fieldnames = list(travellers[0].keys())
 

busManager/coord/templates/reports/school_roll.html

@@ -0,0 +1,35 @@
+<style>
+
+    td {
+        border: 2px solid #000;
+        padding-top: 3px
+    }
+
+</style>
+
+{% for school in schools %}
+    <h1>{{ school.name }}</h1>
+    {% for route in school.routes %}
+        <br>
+        <h2>{{ route.bus }}</h2>
+        <table>
+            <tr>
+                <th style="width: 60%">Traveller</th>
+                <th style="width: 5%">Fare</th>
+                <th style="width: 40%">Stop</th>
+                <th style="width: 12%">AM Time</th>
+                <th style="width: 12%">PM Time</th>
+            </tr>
+            {% for traveller in route.travellers %}
+                <tr>
+                    <td style="padding-left: 2px">{{ traveller.display }}</td>
+                    <td style="text-align: center">{{ traveller.isFared }}</td>
+                    <td style="padding-left: 2px">{{ traveller.stop }}</td>
+                    <td style="text-align: center">{{ traveller.am_time }}</td>
+                    <td style="text-align: center">{{ traveller.pm_time }}</td>
+                </tr>
+            {% endfor %}
+        </table>
+    {% endfor %}
+    <p style="page-break-after: always">
+{% endfor %}

busManager/coord/urls.py

@@ -7,4 +7,5 @@ urlpatterns = [
     path("contacts", views.emergency_contacts, name="emergency_contacts"),
     path("roll", views.bus_roll, name="bus_roll"),
     path("summary", views.bus_summary, name="bus_summary"),
+    path("school", views.school_roll, name="school_roll"),
 ]

busManager/coord/views.py

@@ -18,3 +18,8 @@ def emergency_contacts(request):
 @staff_member_required
 def bus_roll(request):
     return render_to_pdf('reports/bus_roll.html', bus_roll_context())
+
+
+@staff_member_required
+def school_roll(request):
+    return render_to_pdf('reports/school_roll.html', school_roll_context())