Files
bus-manager/busManager/traveller/context_helpers.py
T
st01765 b695dd8054 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

90 lines
3.4 KiB
Python

from django.db.models import Q
from transport.models import Bus
from traveller.models import TravellerRoute
def school_roll_context(queryset, date=None):
school_list = []
for school in queryset:
school_routes = []
query = Q(traveller__school=school) | Q(traveller__shuttle__school=school)
for bus in Bus.objects.all():
travellers = []
for trav_route in TravellerRoute.objects.filter(query).filter(busStop__bus=bus).order_by('busStop__am_time'):
traveller = trav_route.traveller
if not traveller._is_active(date):
continue
bus_stop = trav_route.busStop
display_name = f"{traveller} ({traveller.get_year_level_display()})"
if traveller.school != school:
display_name = f"{traveller} ({traveller.get_year_level_display()}, {traveller.school.shortName})"
is_fared = "---"
if traveller.eligibility_status == "2":
is_fared = "Y"
shuttle_name = " "
if traveller.shuttle:
shuttle_name = traveller.shuttle.bus
travellers.append({
'display': display_name,
'isFared': is_fared,
'shuttle': shuttle_name,
'stop': f"#{bus_stop.get_stop_number()} - {bus_stop.address}",
'am_time': bus_stop.am_time,
'pm_time': bus_stop.pm_time
})
if travellers:
school_routes.append({
'bus': bus,
'travellers': travellers
})
school_list.append({"name": school.name, "routes": school_routes})
return {"schools": school_list}
def traveller_route_context(traveller_route):
traveller = traveller_route.traveller
bus_stop = traveller_route.busStop
families = traveller.get_families()
address = ""
for family in families:
if address:
address += ";"
address = address + f"{family.residential_address} {family.residential_suburb}"
return {
'first_name': traveller.first_name,
'last_name': traveller.last_name,
'active': traveller.is_active,
'school': traveller.school,
'dob': traveller.dob,
'year_level': traveller.year_level,
'address': address,
'start_date': traveller.travel_start_date,
'end_date': traveller.travel_end_date,
'eligibility': traveller.get_eligibility_status_display(),
'shuttle': traveller.shuttle,
'route': traveller_route.busStop.bus,
'stop': f"#{bus_stop.get_stop_number()} - {bus_stop.address}",
'pickup': bus_stop.am_time,
'drop-off': bus_stop.pm_time
}
def traveller_roll_context(queryset):
travellers = []
for traveller in queryset:
for traveller_route in TravellerRoute.objects.filter(traveller=traveller):
travellers.append(traveller_route_context(traveller_route))
return travellers
def confirmation_letter_context(queryset):
travellers = []
for traveller in queryset:
for travellerRoute in TravellerRoute.objects.filter(traveller=traveller):
travellers.append({
'traveller': traveller,
'stop': travellerRoute.busStop,
'shuttle': traveller.shuttle,
})
return {'travellers': travellers}