John Mullins
24 lines
797 B
Python
24 lines
797 B
Python
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})
|