from datetime import datetime

from django.shortcuts import render
from django import forms

from .context_helpers import bus_roll_context
from .email_helpers import render_to_pdf


class RollDateSelector(forms.Form):
    _selected_action = forms.CharField(widget=forms.MultipleHiddenInput)


def roll_date_selector(mixin, request, queryset):
    if 'generate' in request.POST:
        date = request.POST.get("date")
        if date:
            date = datetime.strptime(date, '%Y-%m-%d')
        else:
            date = None
        return render_to_pdf('reports/bus_roll.html', bus_roll_context(queryset, date=date))
    form = RollDateSelector(initial={'_selected_action': queryset.values_list('id', flat=True)})
    return render(request, 'admin/date_selector.html', context={'form': form})