Created new models to replicate out of and split coord app

Moved helpers and views to their respective new apps
2026-02-05 11:18:08 +11:00
parent ad4fd19cc7
commit b695dd8054
26 changed files with 1536 additions and 10 deletions
@@ -1,3 +1,79 @@
 from django.contrib import admin
+from django.urls import reverse
+from django.utils.html import format_html
+from django.utils.http import urlencode

-# Register your models here.
+from common.admin import MyImportExportModelAdmin
+from transport.admin_mixins import BusRollMixin, ShuttleRollMixin
+from transport.models import Driver, BusStop, Company, Bus, Shuttle
+
+
+@admin.register(Company)
+class CompanyAdmin(MyImportExportModelAdmin, admin.ModelAdmin):
+    list_display = ["name", "contact_name", "email", "buses"]
+
+    def buses(self, obj):
+        count = obj.bus_set.count()
+        url = (
+                reverse("admin:coord_bus_changelist")
+                + "?"
+                + urlencode({"company__id__exact": f"{obj.id}"})
+        )
+        return format_html('<a href="{}">{} Buses</a>', url, count)
+
+    def email(self, obj):
+        return format_html('<a href="mailto:{}">{}</a>', obj.contact_email, obj.contact_email)
+
+
+class DriverInline(admin.StackedInline):
+    model = Driver
+    extra = 0
+
+
+class BusStopInline(admin.TabularInline):
+    model = BusStop
+    extra = 0
+    ordering = ("am_time",)
+
+
+@admin.register(Bus)
+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_bus_roll_on_date", "show_emergency_contacts", "sms_traveller_contacts", "email_bus_roll", "email_emergency_contacts"]
+    inlines = [DriverInline, BusStopInline]
+    fieldsets = [
+        (None, {'fields': [
+            "company", "route_name", "contract_number", "registration",
+            "traveller_count", "seating_capacity", "make", "model", "notes"
+        ]})
+    ]
+
+    def route_travellers(self, obj):
+        url = (
+                reverse("admin:coord_traveller_changelist")
+                + "?"
+                + urlencode({"bus_stops__bus__id__exact": f"{obj.id}"})
+        )
+        return format_html('<a href="{}">{} Travellers</a>', url, obj.traveller_count())
+
+
+# @admin.register(BusStop)
+class BusStopAdmin(MyImportExportModelAdmin, admin.ModelAdmin):
+    list_filter = ["bus__company", "bus__route_name"]
+    list_display = ["__str__", "am_time", "pm_time", "address"]
+    search_fields = ["bus__route_name", "address"]
+
+@admin.register(Shuttle)
+class ShuttleAdmin(MyImportExportModelAdmin, admin.ModelAdmin, ShuttleRollMixin):
+    list_display = ["__str__", "school", "bus", "shuttle_travellers"]
+    actions = ["show_shuttle_roll", "email_shuttle_roll"]
+
+    def shuttle_travellers(self, obj):
+        url = (
+                reverse("admin:coord_traveller_changelist")
+                + "?"
+                + urlencode({"shuttle__id__exact": f"{obj.id}"})
+        )
+        return format_html('<a href="{}">{} Travellers</a>', url, obj.traveller_count())