コード例 #1
0
ファイル: test_dev_PBS_1675.py プロジェクト: ropable/pbs
 def test_report_ignitions_has_dates(self):
     url = self.url + "?report=ignitions"
     req = self.rf.get(url)
     data = {
         'toDate': None,
         'fromDate': None
     }
     response = self.client.post(url, data=data)
     form = response.context['form']
     assert form.is_valid()
     eq_(response.context['title'], 'Ignitions summary')
     eq_(form.cleaned_data.get('toDate'), date.today())
コード例 #2
0
ファイル: sites.py プロジェクト: parksandwildlife/pbs
    def endorse_authorise_summary(self, request, extra_context=None):
        """
        Display summaries of prescriptions, approvals and ignitions.

        DEV:

        """
        form = EndorseAuthoriseSummaryForm(request.GET)

        report_set = {'summary', 'approvals', 'ignitions'}
        report = request.GET.get('report', 'summary')
        if report not in report_set:
            report = 'summary'

        export_csv = True if request.GET.get('Export_CSV') == 'export_csv' else False

        if request.GET.get('fromDate'):
            fromDate = request.GET.get('fromDate')
            fromDate = datetime.datetime.strptime(fromDate, '%d-%m-%Y').date()
        else:
            # default - beginning of financial year
            yr = datetime.date.today().year
            fromDate = datetime.date(yr, 7, 1)

        if request.GET.get('toDate'):
            toDate = request.GET.get('toDate')
            toDate = datetime.datetime.strptime(toDate, '%d-%m-%Y').date()
        else:
            toDate = datetime.date.today()

        burns = []
        if report == 'summary':
            title = _("Endorsements summary")
            queryset = Prescription.objects.filter(
                endorsement_status=Prescription.ENDORSEMENT_SUBMITTED)
        elif report == 'approvals':
            title = _("Approvals summary")
            queryset = Prescription.objects.filter(
                approval_status=Prescription.APPROVAL_SUBMITTED)
        elif report == 'ignitions':
            title = _("Ignitions summary")
            queryset = None
            burns = self.get_burns(fromDate, toDate)
        else:
            raise ValueError("Report {} must be in {}".format(report, report_set))

        if export_csv:
            return self.export_to_csv(request, fromDate, toDate, burns)
        if queryset:
            queryset.prefetch_related('endorsing_roles')

        if form.is_valid():
            region = form.cleaned_data.get('region', None)
            district = form.cleaned_data.get('district', None)

            if region:
                queryset = queryset.filter(region=region)

            if district:
                queryset = queryset.filter(district=district)

        context = {
            'title': title,
            'prescriptions': queryset,
            'form': form,
            'report': report,
            'burns': burns,
            'fromDate': fromDate,
            'toDate': toDate,
        }
        context.update(extra_context or {})
        return TemplateResponse(request, "admin/endorse_authorise_summary.html", context,
                                current_app=self.name)
コード例 #3
0
    def endorse_authorise_summary(self, request, extra_context=None):
        """
        Display summaries of prescriptions, approvals and ignitions.

        DEV:

        """
        form = EndorseAuthoriseSummaryForm(request.GET)

        report_set = {'summary', 'approvals', 'ignitions'}
        report = request.GET.get('report', 'summary')
        if report not in report_set:
            report = 'summary'

        export_csv = True if request.GET.get(
            'Export_CSV') == 'export_csv' else False

        if request.GET.get('fromDate'):
            fromDate = request.GET.get('fromDate')
            fromDate = datetime.datetime.strptime(fromDate, '%d-%m-%Y').date()
        else:
            # default - beginning of current financial year
            today = datetime.date.today()
            financial_year = datetime.date(today.year, 7, 1)
            if today < financial_year:
                #today is early than the first day of this year's financial year, set fromDate to last year's financial year
                fromDate = datetime.date(today.year - 1, 7, 1)
            else:
                #today is later than the first day of this year's financial year, set fromDate to this year's financial year
                fromDate = financial_year

        if request.GET.get('toDate'):
            toDate = request.GET.get('toDate')
            toDate = datetime.datetime.strptime(toDate, '%d-%m-%Y').date()
        else:
            toDate = datetime.date.today()

        burns = []
        if report == 'summary':
            title = _("Endorsements summary")
            queryset = Prescription.objects.filter(
                endorsement_status=Prescription.ENDORSEMENT_SUBMITTED)
        elif report == 'approvals':
            title = _("Approvals summary")
            queryset = Prescription.objects.filter(
                approval_status=Prescription.APPROVAL_SUBMITTED)
        elif report == 'ignitions':
            title = _("Ignitions summary")
            queryset = None
            burns = self.get_burns(fromDate, toDate)
        else:
            raise ValueError("Report {} must be in {}".format(
                report, report_set))

        if export_csv:
            return self.export_to_csv(request, fromDate, toDate, burns)
        if queryset:
            queryset.prefetch_related('endorsing_roles')

        if form.is_valid():
            region = form.cleaned_data.get('region', None)
            district = form.cleaned_data.get('district', None)

            if region:
                queryset = queryset.filter(region=region)

            if district:
                queryset = queryset.filter(district=district)

        context = {
            'title': title,
            'prescriptions': queryset,
            'form': form,
            'report': report,
            'burns': burns,
            'fromDate': fromDate,
            'toDate': toDate,
        }
        context.update(extra_context or {})
        return TemplateResponse(request,
                                "admin/endorse_authorise_summary.html",
                                context,
                                current_app=self.name)