import csv
from datetime import date

from django.contrib.admin import DateFieldListFilter, BooleanFieldListFilter
from django.forms import widgets
from django.contrib import admin
from django.http import HttpResponse
from django.urls import reverse
from django.utils.html import format_html
from django.utils.http import urlencode
from django.utils.translation import gettext_lazy
from import_export.admin import ImportExportModelAdmin
from rangefilter.filters import DateRangeFilterBuilder

from .adminClone import CloneModelAdmin
from .context_helpers import *
from .email_helpers import email_companies_bus_roll, render_to_pdf, email_school_roll
from .models import *
from .utils.send_sms import send_sms


class BusRollMixin:

    def show_bus_roll(self, request, queryset):
        return render_to_pdf('reports/bus_roll.html', bus_roll_context(queryset))

    def show_emergency_contacts(self, request, queryset):
        return render_to_pdf('reports/emergency_contacts.html', emergency_contacts_context(queryset))

    def email_company(self, request, queryset):
        return email_companies_bus_roll(request, queryset)

    email_company.short_description = "Email Bus Roll to Company"


class ShuttleRollMixin:

    def show_shuttle_roll(self, request, queryset):
        if queryset is None:
            buses = None
        else:
            buses = []
            for shuttle in queryset:
                if shuttle.bus not in buses:
                    buses.append(shuttle.bus)
        return render_to_pdf('reports/bus_roll.html', bus_roll_context(buses, include_bus_stops=False))


class SchoolRollMixin:

    def email_travellers_to_school(self, request, queryset):
        return email_school_roll(request, queryset)

    def show_school_travellers(self, request, queryset):
        return render_to_pdf('reports/school_roll.html', school_roll_context(queryset))

    def export_travellers_to_csv(self, request, queryset):
        traveller_list = []
        for school in queryset:
            for travellerRoute in TravellerRoute.objects.filter(traveller__school=school):
                if not travellerRoute.traveller._is_active():
                    continue
                traveller_list.append(traveller_route_context(travellerRoute))

        response = HttpResponse(content_type="text/csv")
        response["Content-Disposition"] = f"attachment; filename=traveller_list_{date.today()}.csv"

        writer = csv.DictWriter(response, fieldnames=traveller_list[0].keys())
        writer.writeheader()
        writer.writerows(traveller_list)

        return response


class TravellerRollMixin:

    def confirmation_letter(self, request, queryset):
        return render_to_pdf('mail/confirmation_letter.html', confirmation_letter_context(queryset))

    def send_sms(self, request, queryset):
        return send_sms(self, request, queryset)

    def export_to_csv(self, request, queryset):
        traveller_list = traveller_roll_context(queryset)
        response = HttpResponse(content_type="text/csv")
        response["Content-Disposition"] = f"attachment; filename=traveller_list_{date.today()}.csv"

        writer = csv.DictWriter(response, fieldnames=traveller_list[0].keys())
        writer.writeheader()
        writer.writerows(traveller_list)

        return response


class HiddenNowTime(widgets.TimeInput):
    pass


class MyImportExportModelAdmin(ImportExportModelAdmin):
    def has_import_permission(self, request):
        return request.user.is_superuser


@admin.register(Company)
class CompanyAdmin(MyImportExportModelAdmin, admin.ModelAdmin):
    list_display = ["name", "contact_name", "email", "buses"]

    def buses(self, obj):
        count = obj.bus_set.count()
        url = (
                reverse("admin:coord_bus_changelist")
                + "?"
                + urlencode({"company__id__exact": f"{obj.id}"})
        )
        return format_html('<a href="{}">{} Buses</a>', url, count)

    def email(self, obj):
        return format_html('<a href="mailto:{}">{}</a>', obj.contact_email, obj.contact_email)


class DriverInline(admin.StackedInline):
    model = Driver
    extra = 0


class BusStopInline(admin.TabularInline):
    model = BusStop
    extra = 0
    ordering = ("am_time",)


@admin.register(Bus)
class BusesAdmin(MyImportExportModelAdmin, admin.ModelAdmin, BusRollMixin):
    list_filter = ["company"]
    list_display = ["route_name", "company", "contract_number", "seating_capacity", "route_travellers"]
    readonly_fields = ["traveller_count"]
    actions = ["email_company", "show_bus_roll", "show_emergency_contacts"]
    inlines = [DriverInline, BusStopInline]
    fieldsets = [
        (None, {'fields': [
            "company", "route_name", "contract_number", "registration",
            "traveller_count", "seating_capacity", "make", "model", "notes"
        ]})
    ]

    def route_travellers(self, obj):
        url = (
                reverse("admin:coord_traveller_changelist")
                + "?"
                + urlencode({"bus_stops__bus__id__exact": f"{obj.id}"})
        )
        return format_html('<a href="{}">{} Travellers</a>', url, obj.traveller_count())


# @admin.register(BusStop)
class BusStopAdmin(MyImportExportModelAdmin, admin.ModelAdmin):
    list_filter = ["bus__company", "bus__route_name"]
    list_display = ["__str__", "am_time", "pm_time", "address"]
    search_fields = ["bus__route_name", "address"]


# @admin.register(Suburb)
class SuburbsAdmin(MyImportExportModelAdmin, admin.ModelAdmin):
    list_filter = ["state"]


class TravellerRouteInline(admin.TabularInline):
    model = TravellerRoute
    extra = 0


@admin.register(Traveller)
class TravellerAdmin(MyImportExportModelAdmin, CloneModelAdmin, admin.ModelAdmin, TravellerRollMixin):
    list_display = ["first_name", "last_name", "school", "year_level", "residential_address", "residential_suburb", "stop_route", "shuttle", "travel_start_date", "travel_end_date"]
    list_filter = [
        "is_active", "school", "year_level", "eligibility_status", "bus_stops__bus", "shuttle", "residential_suburb",
        ("travel_start_date", DateRangeFilterBuilder(
            title="Start date"
        )),
        ("travel_end_date", DateRangeFilterBuilder(
            title="End date"
        ))
    ]
    search_fields = ["first_name", "last_name", "residential_address"]
    cloneable_fields = ["last_name", "residential_address", "residential_suburb",
                        "postal_address", "postal_suburb", "eligibility_status", "shuttle",
                        "parent_A_firstname", "parent_A_lastname", "parent_A_phone", "parent_A_email",
                        "parent_B_firstname", "parent_B_lastname", "parent_B_phone", "parent_B_email",
                        "emergency_contact_A_firstname", "emergency_contact_A_lastname", "emergency_contact_A_phone",
                        "emergency_contact_A_relation", "emergency_contact_B_firstname", "emergency_contact_B_lastname",
                        "emergency_contact_B_phone", "emergency_contact_B_relation"]
    inlines = [TravellerRouteInline]
    readonly_fields = ["travel_start_date", "travel_end_date", "fare_paying", "created_on", "last_edit", "is_active"]
    actions = ["export_to_csv", "send_sms", "confirmation_letter", "letter_creator"]
    fieldsets = [
        (None, {
            'fields': [
                "school",
                "first_name",
                "last_name",
                "dob",
                "year_level",
            ]
        }),
        ('Address', {
            'classes': ('collapse',),
            'fields': [
                "residential_address",
                "residential_suburb",
                "postal_address",
                "postal_suburb",
            ]
        }),
        ('Office Use', {
            'classes': ('collapse',),
            'fields': [
                "distance_to_school",
                "travel_start_date",
                "travel_end_date",
                "eligibility_status",
                "fare_paying",
                "term_1_paid",
                "term_2_paid",
                "term_3_paid",
                "term_4_paid",
                "assessment_date",
                "application_form_completed",
                "parent_notified",
                "seat_number",
                "created_on",
                "last_edit",
                "is_active",
            ]
        }),
        ('Adult Contacts', {
            'classes': ('collapse',),
            'fields': [
                "parent_A_firstname",
                "parent_A_lastname",
                "parent_A_phone",
                "parent_A_email",
                "parent_B_firstname",
                "parent_B_lastname",
                "parent_B_phone",
                "parent_B_email",
                "emergency_contact_A_firstname",
                "emergency_contact_A_lastname",
                "emergency_contact_A_phone",
                "emergency_contact_A_relation",
                "emergency_contact_B_firstname",
                "emergency_contact_B_lastname",
                "emergency_contact_B_phone",
                "emergency_contact_B_relation"
            ]
        }),
        (None, {'fields': ["notes", "shuttle"]})
    ]
    # list_display_links = None

    def yearly_rollover(self, request, queryset):
        pass

    def stop_route(self, obj):
        stops = BusStop.objects.filter(traveller__id__exact=obj.id)
        if stops.count() == 0:
            return ""
        if stops.count() == 1:
            return stops.first()
        return "Multiple"

    def save_model(self, request, obj, form, change):
        if obj.is_archived and obj.travel_end_date is None:
            obj.is_archived = False
        super().save_model(request, obj, form, change)


# @admin.register(TravellerRoute)
class TravellerRouteAdmin(MyImportExportModelAdmin, admin.ModelAdmin):
    list_display = ["traveller", "busStop"]


@admin.register(School)
class SchoolAdmin(MyImportExportModelAdmin, admin.ModelAdmin, SchoolRollMixin):
    list_display = ["__str__", "address", "suburb", "school_email", "phone"]
    actions = ["email_travellers_to_school", "show_school_travellers", "export_travellers_to_csv"]

    def school_email(self, obj):
        return format_html('<a href="mailto:{}">{}</a>', obj.email, obj.email)


@admin.register(Setting)
class SettingAdmin(MyImportExportModelAdmin, admin.ModelAdmin):
    list_display = ["name", "value"]


@admin.register(Shuttle)
class ShuttleAdmin(MyImportExportModelAdmin, admin.ModelAdmin, ShuttleRollMixin):
    list_display = ["__str__", "school", "bus", "shuttle_travellers"]
    actions = ["show_shuttle_roll"]

    def shuttle_travellers(self, obj):
        url = (
                reverse("admin:coord_traveller_changelist")
                + "?"
                + urlencode({"shuttle__id__exact": f"{obj.id}"})
        )
        return format_html('<a href="{}">{} Travellers</a>', url, obj.traveller_count())


# @admin.register(Driver)
class DriverAdmin(MyImportExportModelAdmin, admin.ModelAdmin):
    list_display = ["__str__", "route", "phone_number"]

    def route(self, obj):
        url = reverse("admin:coord_bus_change", args=(obj.bus.id,))
        return format_html('<a href="{}">{}</a>', url, obj.bus)