edan_f0/oscar/apps/reports/views.py

from django.http import HttpResponse, HttpResponseForbidden, Http404
from django.template.response import TemplateResponse
from django.views.generic import TemplateView

from oscar.core.loading import import_module
report_forms = import_module('reports.forms', ['ReportForm'])
report_utils = import_module('reports.utils', ['GeneratorRepository'])


class DashboardView(TemplateView):
    template_name = 'reports/dashboard.html'
    
    def get(self, request, *args, **kwargs):
        if 'report_type' in request.GET:
            form = report_forms.ReportForm(request.GET)
            if form.is_valid():
                generator = _get_generator(form)
                if not generator.is_available_to(request.user):
                    return HttpResponseForbidden("You do not have access to this report")
                
                response = HttpResponse(mimetype=generator.mimetype)
                response['Content-Disposition'] = 'attachment; filename=%s' % generator.filename()
                generator.generate(response)
                return response
        else:
            form = report_forms.ReportForm()
        return TemplateResponse(request, self.template_name, {'form': form})


def _get_generator(form):
    code = form.cleaned_data['report_type']

    repo = report_utils.GeneratorRepository()
    generator_cls = repo.get_generator(code)
    if not generator_cls:
        raise Http404
    return generator_cls(start_date=form.cleaned_data['start_date'], 
                         end_date=form.cleaned_data['end_date'])