from django.db.models import Q from coord.models import Bus, BusStop, TravellerRoute, Driver, Traveller, Shuttle def bus_roll_context(queryset=None): table_header_size = 4 page_max_size = 45 bus_routes = [] if queryset is None: buses = Bus.objects.all() else: buses = queryset for bus in buses: route_stops = [] page_size = 3 # Account for traveller numbers at the top of the first page for bus_stop in BusStop.objects.filter(bus=bus): traveller_routes = TravellerRoute.objects.filter(busStop=bus_stop) traveller_list = [] for trav_route in traveller_routes: 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_size = len(traveller_list) page_break = False page_size += table_header_size + stop_size if page_size > page_max_size: if len(route_stops) > 0: # Don't break the page if it's the first stop page_break = True page_size = table_header_size + stop_size route_stops.append({ 'stop_num': bus_stop.get_stop_number(), 'name': bus_stop.address, 'am': bus_stop.am_time, 'pm': bus_stop.pm_time, 'travellers': traveller_list, 'page_break': page_break }) shuttle_routes = [] for shuttle in Shuttle.objects.filter(bus=bus): shuttle_travellers = [] for traveller in Traveller.objects.filter(shuttle=shuttle): if traveller.is_active(): shuttle_travellers.append({ 'display': f"{traveller} ({traveller.get_year_level_display()}, {traveller.school})", }) shuttle_routes.append({'shuttle': shuttle, 'shuttle_travellers': shuttle_travellers}) bus_routes.append({'bus': bus, 'route_stops': route_stops, 'shuttle_routes': shuttle_routes}) return {'routes': bus_routes} def school_roll_context(queryset): 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(): 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_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 school_travellerRoute_context(school): travellers = [] for travellerRoute in TravellerRoute.objects.filter(traveller__school=school): traveller = travellerRoute.traveller if not traveller.is_active(): continue travellers.append(traveller_route_context(travellerRoute)) return travellers def traveller_route_context(traveller_route): traveller = traveller_route.traveller return { 'first_name': traveller.first_name, 'last_name': traveller.last_name, 'school': traveller.school, 'dob': traveller.dob, 'year_level': traveller.year_level, 'address': f"{traveller.residential_address} {traveller.residential_suburb}", '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"#{traveller_route.busStop.get_stop_number()} - {traveller_route.busStop.address}" } 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}