from datetime import date
from io import BytesIO

from django.core.mail import EmailMessage
from django.http import HttpResponse
from django.template.loader import get_template
from xhtml2pdf import pisa

from coord.context_helpers import bus_roll_context, school_roll_context
from coord.models import Company, School


def render_to_pdf(template, context):
    html = get_template(template).render(context)
    result = BytesIO()
    pdf = pisa.pisaDocument(BytesIO(html.encode("ISO-8859-1")), result)
    if pdf.err:
        return HttpResponse("Invalid PDF", status_code=400, content_type='text/plan')
    return HttpResponse(result.getvalue(), content_type='application/pdf')


def email_companies_bus_roll(request, query_set=None):
    html_template = 'reports/bus_roll.html'
    context = bus_roll_context(query_set)

    for company in Company.objects.all():
        if not company.contact_email:
            continue
        company_route = []

        for route in context.get("routes"):
            if route.get("bus").company == company:
                company_route.append(route)
        if not company_route:
            continue
        company_context = {'routes': company_route}
        pdf = render_to_pdf(html_template, company_context)

        subject = "Echuca Schools Bus Roll"
        message = f"A new bus roll for {company.name} has been generated"
        email_from = "bus.manager@education.vic.gov.au"
        recipient = [company.contact_email]
        email = EmailMessage(subject, message, email_from, recipient)
        email.attach(f"school_bus_roll_{date.today()}.pdf", pdf.content)
        email.send(fail_silently=True)

    return render_to_pdf(html_template, context)


def email_school_roll(request, query_set):
    html_template = 'reports/school_roll.html'
    context = school_roll_context(query_set)

    for school in School.objects.all():
        if not school.email:
            continue
        school_route = []
        for school_context in context.get("schools"):
            if school_context.get("name") == school.name:
                school_route.append(school_context)
        if not school_route:
            continue
        school_context = {'schools': school_route}
        pdf = render_to_pdf(html_template, school_context)

        subject = "Echuca Schools Bus Roll"
        message = f"A new bus roll for {school.name} has been generated"
        email_from = "bus.manager@education.vic.gov.au"
        recipient = [school.email]
        email = EmailMessage(subject, message, email_from, recipient)
        email.attach(f"school_bus_roll_{date.today()}.pdf", pdf.content)
        email.send(fail_silently=True)

    return render_to_pdf(html_template, context)