Changeset 18021 for main/waeup.kofa/trunk/src/waeup/kofa/applicants/export.py

Timestamp:

13 Feb 2025, 11:36:26 (2 days ago)

Author:

Henrik Bettermann

Message:

Implement payment ticket exporter with date range filter.

File:

• main/waeup.kofa/trunk/src/waeup/kofa/applicants/export.py (modified) (2 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/applicants/export.py

@@ -19 +19 @@
 """
 import grok
+from datetime import datetime, timedelta
 from zope.catalog.interfaces import ICatalog
-from zope.component import queryUtility
+from zope.component import queryUtility, getUtility
 from waeup.kofa.applicants.interfaces import (
     IApplicantBaseData, IApplicantsContainer, IApplicantOnlinePayment,
     IApplicantRefereeReport, IApplicant)
-from waeup.kofa.interfaces import ICSVExporter
+from waeup.kofa.interfaces import ICSVExporter, IKofaUtils
 from waeup.kofa.interfaces import MessageFactory as _
 from waeup.kofa.utils.batching import ExporterBase
-from waeup.kofa.utils.helpers import iface_names
+from waeup.kofa.utils.helpers import iface_names, to_timezone
 
 class ApplicantsContainerExporter(grok.GlobalUtility, ExporterBase):
@@ -213 +214 @@
         """
         container = grok.getSite()['applicants'][kw['container']]
+        p_start = kw.get('p_start')
+        p_end = kw.get('p_end')
         container_values = container.values()
         used = [value for value in container_values
                 if value.container_code.endswith('+')]
         payments = []
+        date_format = '%d/%m/%Y'
         for applicant in used:
             for payment in applicant.payments:
+                if p_start and p_end:
+                    if not payment.payment_date:
+                        continue
+                    payments_start = datetime.strptime(p_start, date_format)
+                    payments_end = datetime.strptime(p_end, date_format)
+                    tz = getUtility(IKofaUtils).tzinfo
+                    payments_start = tz.localize(payments_start)
+                    payments_end = tz.localize(payments_end) + timedelta(days=1)
+                    payment_date = to_timezone(payment.payment_date, tz)
+                    if payment_date < payments_start or payment_date > payments_end:
+                        continue
                 if payment.p_state == 'paid':
                     payments.append(payment)