Changeset 12006 for main/waeup.ikoba

Timestamp:

20 Nov 2014, 08:32:17 (10 years ago)

Author:

Henrik Bettermann

Message:

Add customer documents exporter.

Location:

main/waeup.ikoba/trunk/src/waeup/ikoba

Files:

• catalog.py (modified) (1 diff)
• customers/catalog.py (modified) (1 diff)
• customers/export.py (modified) (5 diffs)
• customers/tests/test_batching.py (modified) (2 diffs)
• customers/tests/test_customer.py (modified) (1 diff)

main/waeup.ikoba/trunk/src/waeup/ikoba/catalog.py

@@ -124 +124 @@
             if idx_name == 'catalog':
                 continue
-            if value is not None:
-                if 'session' in idx_name or 'level' in idx_name:
-                    value = int(value)
-                if idx_name in ('level', 'current_level'):
-                    value = int(value)
-                    if value not in (10, 999):
-                        value = (value, value + 90)
             if not isinstance(value, tuple):
                 value = (value, value)

main/waeup.ikoba/trunk/src/waeup/ikoba/customers/catalog.py

@@ -1 @@
-## $Id: catalog.py 10465 2013-08-07 11:18:43Z henrik $
+## $Id$
 ##
 ## Copyright (C) 2011 Uli Fouquet & Henrik Bettermann

main/waeup.ikoba/trunk/src/waeup/ikoba/customers/export.py

@@ -1 @@
-## $Id: export.py 11757 2014-07-10 12:20:30Z henrik $
+## $Id$
 ##
 ## Copyright (C) 2014 Uli Fouquet & Henrik Bettermann
@@ -27 +27 @@
 from waeup.ikoba.customers.catalog import CustomersQuery
 from waeup.ikoba.customers.interfaces import (
-    ICustomer, ICSVCustomerExporter)
+    ICustomer, ICSVCustomerExporter, ICustomerDocument)
 from waeup.ikoba.utils.batching import ExporterBase
 from waeup.ikoba.utils.helpers import iface_names, to_timezone
@@ -33 +33 @@
 #: A tuple containing all exporter names referring to customers or
 #: subobjects thereof.
-EXPORTER_NAMES = ('customers', )
+EXPORTER_NAMES = ('customers', 'customerdocuments')
 
 
@@ -40 +40 @@
     """
     return cust_filter.query()
+
+def get_documents(customers):
+    """Get all documents of `customers`.
+    """
+    documents = []
+    for customer in customers:
+        for document in customer.get('documents', {}).values():
+            documents.append(document)
+    return documents
 
 
@@ -127 +136 @@
             CustomersExporter, self).mangle_value(
             value, name, context=context)
+
+class CustomerDocumentsExporter(grok.GlobalUtility, CustomerExporterBase):
+    """Exporter for CustomerDocument instances.
+    """
+    grok.name('customerdocuments')
+
+    #: Fieldnames considered by this exporter
+    fields = tuple(
+        sorted(iface_names(
+            ICustomerDocument, exclude_attribs=False))) + (
+            'customer_id',)
+
+    #: The title under which this exporter will be displayed
+    title = _(u'Customer Documents')
+
+    def filter_func(self, x, **kw):
+        return get_documents(x, **kw)
+
+    def mangle_value(self, value, name, context=None):
+        """Treat location values special.
+        """
+        if context is not None:
+            customer = context.customer
+            if name in ('customer_id',) and customer is not None:
+                value = getattr(customer, name, None)
+        return super(
+            CustomerDocumentsExporter, self).mangle_value(
+            value, name, context=context)

main/waeup.ikoba/trunk/src/waeup/ikoba/customers/tests/test_batching.py

@@ -38 +38 @@
     CustomerProcessor, CustomerDocumentProcessor)
 from waeup.ikoba.customers.customer import Customer
+from waeup.ikoba.customers.documents import CustomerDocument
 from waeup.ikoba.testing import FunctionalLayer, FunctionalTestCase
 
@@ -116 +117 @@
         customer.nationality = u'NG'
 
+        document = CustomerDocument()
+        document.title = u'My Document'
+        customer['documents'].addDocument(document)
         return customer
 

main/waeup.ikoba/trunk/src/waeup/ikoba/customers/tests/test_customer.py

@@ -1 @@
-## $Id: test_customer.py 10059 2013-04-05 08:11:29Z henrik $
+## $Id$
 ##
 ## Copyright (C) 2014 Uli Fouquet & Henrik Bettermann