Changeset 12006 for main/waeup.ikoba/trunk/src/waeup/ikoba/customers/export.py

Timestamp:

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

Author:

Henrik Bettermann

Message:

Add customer documents exporter.

File:

• main/waeup.ikoba/trunk/src/waeup/ikoba/customers/export.py (modified) (5 diffs)

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)