John Mullins
126 lines
4.6 KiB
Python
126 lines
4.6 KiB
Python
from coord.models import Bus, BusStop, TravellerRoute, Driver, Traveller, Shuttle
|
|
|
|
|
|
def bus_roll_context(queryset=None):
|
|
bus_routes = []
|
|
if queryset is None:
|
|
buses = Bus.objects.all()
|
|
else:
|
|
buses = queryset
|
|
|
|
for bus in buses:
|
|
bus_route = []
|
|
for bus_stop in BusStop.objects.filter(bus=bus):
|
|
traveller_list = []
|
|
for trav_route in TravellerRoute.objects.filter(busStop=bus_stop):
|
|
traveller = trav_route.traveller
|
|
if not traveller.is_active():
|
|
continue
|
|
is_fared = "---"
|
|
if traveller.eligibility_status == "2":
|
|
is_fared = "Y"
|
|
traveller_list.append({
|
|
'display': f"{traveller} ({traveller.get_year_level_display()}, {traveller.school.shortName})",
|
|
'isFared': is_fared
|
|
})
|
|
stop_result = {
|
|
'stop_num': bus_stop.get_stop_number(),
|
|
'name': bus_stop.address,
|
|
'am': bus_stop.am_time,
|
|
'pm': bus_stop.pm_time,
|
|
'travellers': traveller_list
|
|
}
|
|
# print(traveller_list)
|
|
bus_route.append(stop_result)
|
|
|
|
# Todo Add shuttles
|
|
|
|
bus_routes.append({'bus': bus, 'stops': bus_route})
|
|
return {'routes': bus_routes}
|
|
|
|
|
|
def emergency_contacts_context(queryset=None):
|
|
if queryset is None:
|
|
buses = Bus.objects.all()
|
|
else:
|
|
buses = queryset
|
|
|
|
bus_routes = []
|
|
for bus in buses:
|
|
drivers = []
|
|
for driver in Driver.objects.filter(bus=bus):
|
|
drivers.append(driver)
|
|
traveller_list = []
|
|
for travellerRoute in TravellerRoute.objects.filter(busStop__bus=bus):
|
|
traveller = travellerRoute.traveller
|
|
if not traveller.is_active():
|
|
continue
|
|
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, 'drivers': drivers, 'travellers': traveller_list})
|
|
|
|
return {'routes': bus_routes}
|
|
|
|
|
|
def bus_summary_context():
|
|
bus_routes = []
|
|
for bus in Bus.objects.all():
|
|
|
|
drivers = []
|
|
for driver in Driver.objects.filter(bus=bus):
|
|
drivers.append(driver)
|
|
|
|
stops = []
|
|
for bus_stop in BusStop.objects.filter(bus=bus):
|
|
stops.append(bus_stop)
|
|
|
|
traveller_count = 0
|
|
for travellerRoute in TravellerRoute.objects.filter(busStop__bus=bus):
|
|
if travellerRoute.traveller.is_active():
|
|
traveller_count += 1
|
|
|
|
shuttle_name = ""
|
|
shuttle_count = 0
|
|
for shuttle in Shuttle.objects.filter(bus=bus):
|
|
if shuttle_name == "":
|
|
shuttle_name = shuttle.school.shortName
|
|
else:
|
|
shuttle_name += f", {shuttle.school.shortName}"
|
|
|
|
for traveller in Traveller.objects.filter(shuttle=shuttle):
|
|
if traveller.is_active():
|
|
shuttle_count += 1
|
|
|
|
over_capacity = traveller_count > bus.seating_capacity or shuttle_count > bus.seating_capacity
|
|
if shuttle_count == 0:
|
|
shuttle_count = ""
|
|
|
|
bus_routes.append({
|
|
'bus': bus,
|
|
'drivers': drivers,
|
|
'stops': stops,
|
|
'traveller_count': traveller_count,
|
|
'shuttle_name': shuttle_name,
|
|
'shuttle_count': shuttle_count,
|
|
'over_capacity': over_capacity,
|
|
})
|
|
return {'routes': bus_routes}
|