from django.contrib.admin.views.decorators import staff_member_required
from django.http import HttpResponse
from django.shortcuts import render
from django.views.generic import ListView

from .context_helpers import bus_roll_context
from .models import Company, Bus, Traveller, BusStop, TravellerRoute, Shuttle, Driver


# def index(request):
#    return HttpResponse("Hello, world. You're at the coord index.")


@staff_member_required
def bus_numbers(request):
    buses = []
    for bus in Bus.objects.all():

        drivers = ""
        for driver in Driver.objects.filter(bus=bus):
            driver_name = f"{driver.first_name} {driver.last_name}"
            if drivers == "":
                drivers = driver_name
            else:
                drivers += f", {driver_name}"

        route_travellers = Traveller.objects.filter(bus_stops__bus=bus).count()
        over_capacity = route_travellers > bus.seating_capacity
        shuttle = Shuttle.objects.filter(bus=bus).first()
        if shuttle:
            shuttle = shuttle.school.shortName
            shuttle_travellers = Traveller.objects.filter(shuttle__bus=bus).count()
            over_capacity = over_capacity or int(shuttle_travellers) > bus.seating_capacity
        else:
            shuttle = ""
            shuttle_travellers = ""
        buses.append({
            'route_name': bus.route_name,
            'drivers': drivers,
            'route_travellers': route_travellers,
            'shuttle': shuttle,
            'shuttle_travellers': shuttle_travellers,
            'seating_capacity': bus.seating_capacity,
            'over_capacity': over_capacity,
        })
    return render(request, 'reports/bus_numbers.html', {'buses': buses})


@staff_member_required
def bus_roll(request, queryset=None):
    return render(request, 'reports/bus_roll.html', bus_roll_context(queryset))


@staff_member_required
def bus_summary(request, queryset=None):
    bus_routes = []
    if queryset is None:
        buses = Bus.objects.all()
    else:
        buses = queryset

    for bus in buses:

        drivers = ""
        for driver in Driver.objects.filter(bus=bus):
            driver_name = f"{driver.first_name} {driver.last_name}"
            if drivers == "":
                drivers = driver_name
            else:
                drivers += f", {driver_name}"

        stops = []
        for bus_stop in BusStop.objects.filter(bus=bus):
            stops.append(bus_stop)

        bus_routes.append({
            'bus': bus,
            'drivers': drivers,
            'stops': stops
        })
    return render(request, 'reports/bus_summary.html', {'routes': bus_routes})


@staff_member_required
def emergency_contacts(request, queryset=None):
    if queryset is None:
        buses = Bus.objects.all()
    else:
        buses = queryset

    bus_routes = []
    for bus in buses:
        traveller_list = []
        for travellerRoute in TravellerRoute.objects.filter(busStop__bus=bus):
            traveller = travellerRoute.traveller
            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, 'travellers': traveller_list})

    return render(request, 'reports/emergency_contacts.html', {'routes': bus_routes})


@staff_member_required
class CompanyList(ListView):
    model = Company
    template_name = "companies.html"