Email roll to schools function

busManager/coord/admin.py

@@ -13,8 +13,8 @@ from import_export.admin import ImportExportModelAdmin
 
 from .adminClone import CloneModelAdmin
 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
+    traveller_route_context, school_roll_context
+from .email_helpers import email_companies_bus_roll, render_to_pdf, email_school_roll
 from .models import *
 
 
@@ -34,8 +34,13 @@ class BusRollMixin:
 
 class SchoolRollMixin:
 
-    def export_travellers_to_csv(self, request, queryset):
+    def email_travellers_to_school(self, request, queryset):
+        return email_school_roll(request, queryset)
 
+    def show_school_travellers(self, request, queryset):
+        return render_to_pdf('reports/school_roll.html', school_roll_context(queryset))
+
+    def export_travellers_to_csv(self, request, queryset):
         traveller_list = []
         for school in queryset:
             for travellerRoute in TravellerRoute.objects.filter(traveller__school=school):
@@ -270,7 +275,7 @@ class TravellerRouteAdmin(MyImportExportModelAdmin, admin.ModelAdmin):
 @admin.register(School)
 class SchoolAdmin(MyImportExportModelAdmin, admin.ModelAdmin, SchoolRollMixin):
     list_display = ["__str__", "address", "suburb", "school_email", "phone"]
-    actions = ["export_travellers_to_csv"]
+    actions = ["email_travellers_to_school", "show_school_travellers", "export_travellers_to_csv"]
 
     def school_email(self, obj):
         return format_html('<a href="mailto:{}">{}</a>', obj.email, obj.email)