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}