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


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 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}