Changeset 17091 for main/kofacustom.lpng/trunk/src/kofacustom/lpng/applicants/browser.py

Timestamp:

13 Sep 2022, 07:06:32 (2 years ago)

Author:

Henrik Bettermann

Message:

Implement Polling Unit Locator.

File:

• main/kofacustom.lpng/trunk/src/kofacustom/lpng/applicants/browser.py (modified) (6 diffs)

main/kofacustom.lpng/trunk/src/kofacustom/lpng/applicants/browser.py

@@ -26 +26 @@
     FriendlyDatetimeDisplayWidget)
 from waeup.kofa.applicants.pdf import PDFApplicationSlip
+from waeup.kofa.browser.layout import KofaEditFormPage, action
 from waeup.kofa.applicants.browser import (
     ApplicantRegistrationPage, ApplicantsContainerPage,
@@ -46 +47 @@
     ICustomApplicantOnlinePayment,
     ICustomApplicantEdit,
+    IPollingUnit,
+    STATES, LGAS, WARDS, PUNITS
     )
 from kofacustom.lpng.interfaces import MessageFactory as _
@@ -55 +58 @@
 
     @property
-    def form_fields(self):
-        form_fields = grok.AutoFields(ICustomApplicant)
+    def render_punit(self):
+        if self.context.punit:
+            punit = self.context.punit.split('_')
+            return ' / '.join([STATES[punit[0]], 
+                    LGAS['_'.join(punit[:2])], 
+                    WARDS['_'.join(punit[:3])] , 
+                    PUNITS['_'.join(punit[:4])]
+                   ])
+        return
+
+    @property
+    def form_fields(self):
+        form_fields = grok.AutoFields(
+            ICustomApplicant).omit('locked', 'suspended', 'federalstate', 
+                                   'lga', 'ward', 'punit')
         form_fields['perm_address'].custom_widget = BytesDisplayWidget
-        form_fields['notice'].custom_widget = BytesDisplayWidget
+        #form_fields['notice'].custom_widget = BytesDisplayWidget
         if not getattr(self.context, 'student_id'):
             form_fields = form_fields.omit('student_id')
@@ -91 +107 @@
     @property
     def form_fields(self):
-        form_fields = grok.AutoFields(ICustomApplicant)
+        form_fields = grok.AutoFields(ICustomApplicant).omit( 
+            'federalstate', 'lga', 'ward', 'punit')
         if not getattr(self.context, 'student_id'):
             form_fields = form_fields.omit('student_id')
@@ -136 +153 @@
     def form_fields(self):
         form_fields = grok.AutoFields(ICustomApplicantEdit)
-        form_fields = form_fields.omit('notice', 'locked', 'suspended')
+        form_fields = form_fields.omit('locked', 'suspended', 'federalstate', 
+                                       'lga', 'ward', 'punit')
         if not getattr(self.context, 'student_id'):
             form_fields = form_fields.omit('student_id')
@@ -177 +195 @@
             self.context.__parent__, applicantview, note=self.note,
             omit_fields=())
+
+class PunitFormPage(KofaEditFormPage):
+    """
+    """
+    grok.context(ICustomApplicant)
+    grok.require('waeup.handleApplication')
+    grok.name('punitformpage')
+    form_fields = grok.AutoFields(IPollingUnit)
+    grok.template('puniteditpage')
+    label = _('Locate polling unit')
+    pnav = 3
+
+    @action(_('Save'), style='primary')
+    def save(self, **data):
+        changed_fields = self.applyData(self.context, **data)
+        # Turn list of lists into single list
+        if changed_fields:
+            changed_fields = reduce(lambda x, y: x+y, changed_fields.values())
+        if 'federalstate' in changed_fields:
+            # Clear other form fields
+            self.context.lga = self.context.ward = self.context.punit = None
+            self.flash(_('Federal State has been saved.'))
+            return          
+        if 'lga' in changed_fields:
+            # Clear other form fields
+            self.context.ward = self.context.punit = None
+            self.flash(_('LGA has been saved.'))
+            return          
+        if 'ward' in changed_fields:
+            # Clear other form fields
+            self.context.punit = None
+            self.flash(_('Ward has been saved.'))
+            return
+        if 'punit' in changed_fields and self.context.punit:
+            self.flash(_('Polling Unit has been successfully saved.'))
+            self.redirect(self.url(self.context))
+        return