Files
bus-manager/busManager/coord/views.py
T
John Mullins 0c8de0285a Changed bus roll to be a pdf view. Small changes needed to be able to email to companies.
Dropdown selection in admin > buses > Email Bus Roll To Company
Will email bus roll only to the company the route is a member of. Will show all to the admin in a pdf file.

Recipients are currently hardcoded for testing purposes.
2023-08-29 12:40:34 +10:00

124 lines
4.4 KiB
Python

from django.contrib.admin.views.decorators import staff_member_required
from django.http import HttpResponse
from django.shortcuts import render
from django.views.generic import ListView
from .context_helpers import bus_roll_context
from .models import Company, Bus, Traveller, BusStop, TravellerRoute, Shuttle, Driver
# def index(request):
# return HttpResponse("Hello, world. You're at the coord index.")
@staff_member_required
def bus_numbers(request):
buses = []
for bus in Bus.objects.all():
drivers = ""
for driver in Driver.objects.filter(bus=bus):
driver_name = f"{driver.first_name} {driver.last_name}"
if drivers == "":
drivers = driver_name
else:
drivers += f", {driver_name}"
route_travellers = Traveller.objects.filter(bus_stops__bus=bus).count()
over_capacity = route_travellers > bus.seating_capacity
shuttle = Shuttle.objects.filter(bus=bus).first()
if shuttle:
shuttle = shuttle.school.shortName
shuttle_travellers = Traveller.objects.filter(shuttle__bus=bus).count()
over_capacity = over_capacity or int(shuttle_travellers) > bus.seating_capacity
else:
shuttle = ""
shuttle_travellers = ""
buses.append({
'route_name': bus.route_name,
'drivers': drivers,
'route_travellers': route_travellers,
'shuttle': shuttle,
'shuttle_travellers': shuttle_travellers,
'seating_capacity': bus.seating_capacity,
'over_capacity': over_capacity,
})
return render(request, 'reports/bus_numbers.html', {'buses': buses})
@staff_member_required
def bus_roll(request, queryset=None):
return render(request, 'reports/bus_roll.html', bus_roll_context(queryset))
@staff_member_required
def bus_summary(request, queryset=None):
bus_routes = []
if queryset is None:
buses = Bus.objects.all()
else:
buses = queryset
for bus in buses:
drivers = ""
for driver in Driver.objects.filter(bus=bus):
driver_name = f"{driver.first_name} {driver.last_name}"
if drivers == "":
drivers = driver_name
else:
drivers += f", {driver_name}"
stops = []
for bus_stop in BusStop.objects.filter(bus=bus):
stops.append(bus_stop)
bus_routes.append({
'bus': bus,
'drivers': drivers,
'stops': stops
})
return render(request, 'reports/bus_summary.html', {'routes': bus_routes})
@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})
@staff_member_required
class CompanyList(ListView):
model = Company
template_name = "companies.html"