Changeset 12803 for main/ikobacustom.pcn/trunk/src/ikobacustom/pcn/customers/browser.py

Timestamp:

20 Mar 2015, 18:51:29 (10 years ago)

Author:

Henrik Bettermann

Message:

Prepare products and contracts for annual payments.

File:

• main/ikobacustom.pcn/trunk/src/ikobacustom/pcn/customers/browser.py (modified) (2 diffs)

main/ikobacustom.pcn/trunk/src/ikobacustom/pcn/customers/browser.py

@@ -20 +20 @@
 from zope.component import getUtility
 from waeup.ikoba.interfaces import IIkobaUtils, IExtFileStore
+from waeup.ikoba.browser.layout import action
 from waeup.ikoba.customers.browser import (
-    PDFContractSlipPage, CustomerBaseEditFormPage)
+    PDFContractSlipPage, CustomerBaseEditFormPage, ContractSelectProductPage, msave)
 from ikobacustom.pcn.interfaces import MessageFactory as _
 from ikobacustom.pcn.customers.interfaces import (
@@ -88 +89 @@
             return error
         return
+
+
+class ContractSelectProductPage(ContractSelectProductPage):
+    """ Page to select a contract product
+
+    This page is for both customers and officers.
+    """
+
+    @action(_('Save and proceed'), style='primary')
+    def save(self, **data):
+        msave(self, **data)
+        self.context.contract_year = self.context.product_object.contract_year
+        self.context.title = self.context.product_object.contract_autotitle
+        self.context.tc_dict = self.context.product_object.tc_dict
+        self.context.valid_from = self.context.product_object.valid_from
+        self.context.valid_to = self.context.product_object.valid_to
+        isCustomer = getattr(
+            self.request.principal, 'user_type', None) == 'customer'
+        if isCustomer:
+            self.redirect(self.url(self.context, 'edit'))
+        else:
+            self.redirect(self.url(self.context, 'manage'))
+        return