Added SMS functions to buses and travellers

busManager/coord/admin.py

@@ -26,6 +26,16 @@ class BusRollMixin:
     def show_emergency_contacts(self, request, queryset):
         return render_to_pdf('reports/emergency_contacts.html', emergency_contacts_context(queryset))
 
+    def sms_traveller_contacts(self, request, queryset):
+        travellers = None
+        for bus in queryset:
+            query = Traveller.objects.filter(bus_stops__bus=bus).filter(is_active=True).distinct()
+            if travellers is None:
+                travellers = query
+            else:
+                travellers.union(query)
+        return send_sms(self, request, travellers)
+
     def email_bus_roll(self, request, queryset):
         return email_companies_bus_roll(request, queryset)
 
@@ -137,7 +147,7 @@ class BusesAdmin(MyImportExportModelAdmin, admin.ModelAdmin, BusRollMixin):
     list_filter = ["company"]
     list_display = ["route_name", "company", "contract_number", "seating_capacity", "route_travellers"]
     readonly_fields = ["traveller_count"]
-    actions = ["show_bus_roll", "show_emergency_contacts", "email_bus_roll", "email_emergency_contacts"]
+    actions = ["show_bus_roll", "show_emergency_contacts", "sms_traveller_contacts", "email_bus_roll", "email_emergency_contacts"]
     inlines = [DriverInline, BusStopInline]
     fieldsets = [
         (None, {'fields': [