bus-manager/busManager/coord/context_helpers.py

import datetime

from django.db.models import Q

from .models import Bus, BusStop, TravellerRoute, Driver, Traveller, Shuttle


def bus_roll_context(queryset=None, include_bus_stops=True, date=None):
    bus_routes = []
    if queryset is None:
        buses = Bus.objects.all()
    else:
        buses = queryset

    for bus in buses:
        route_stops = []
        if include_bus_stops:
            route_stops = route_paged_context(bus=bus, date=date)

        shuttle_routes = []
        for shuttle in Shuttle.objects.filter(bus=bus):
            shuttle_routes.append(shuttle_route_context(shuttle, date))

        bus_routes.append({'name': bus.route_name, 'traveller_count': bus.traveller_count(date), 'seating_capacity': bus.seating_capacity, 'bus': bus, 'route_stops': route_stops, 'shuttle_routes': shuttle_routes})
    if date is None:
        date = datetime.date.today()
    return {'routes': bus_routes, 'date': date.strftime('%Y-%m-%d')}


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_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 route_paged_context(bus, date=None):
    table_header_size = 5
    page_max_size = 45
    page_size = 3  # Account for traveller numbers at the top of the first page
    route_stops = []
    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(date):
                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
        })
    return route_stops


def shuttle_route_context(shuttle, date=None):
    shuttle_travellers = []
    for traveller in Traveller.objects.filter(shuttle=shuttle):
        if traveller._is_active(date):
            shuttle_travellers.append({
                'display': f"{traveller} ({traveller.get_year_level_display()}, {traveller.school})",
            })
    return {'shuttle': shuttle, 'shuttle_travellers': shuttle_travellers}


def school_travellerRoute_context(school, date=None):
    travellers = []
    for travellerRoute in TravellerRoute.objects.filter(traveller__school=school):
        traveller = travellerRoute.traveller
        if not traveller._is_active(date):
            continue
        travellers.append(traveller_route_context(travellerRoute))
    return travellers


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 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
            for family in traveller.get_families():
                parent_a = ""
                if family.parent_A_firstname:
                    parent_a = f"{family.parent_A_firstname} {family.parent_A_lastname} ({family.parent_A_phone})"
                parent_b = ""
                if family.parent_B_firstname:
                    parent_b = f"{family.parent_B_firstname} {family.parent_B_lastname} ({family.parent_B_phone})"
                contact_a = ""
                if family.emergency_contact_A_firstname:
                    contact_a = f"{family.emergency_contact_A_firstname} {family.emergency_contact_A_lastname} ({family.emergency_contact_A_phone})"
                contact_b = ""
                if family.emergency_contact_B_firstname:
                    contact_b = f"{family.emergency_contact_B_firstname} {family.emergency_contact_B_lastname} ({family.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}


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}