Changeset 7914 for main/waeup.kofa/trunk/src/waeup/kofa

Timestamp:

19 Mar 2012, 01:57:34 (13 years ago)

Author:

uli

Message:

Add applicants exporter (first draft).

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 zope.catalog.interfaces import ICatalog
+from zope.component import queryUtility
+from waeup.kofa.applicants.interfaces import IApplicantBaseData
 from waeup.kofa.interfaces import ICSVExporter
 from waeup.kofa.interfaces import MessageFactory as _
@@ -62 +65 @@
         containers = site.get('applicants', {})
         return self.export(containers.values(), filepath)
+
+class ApplicantsExporter(grok.GlobalUtility, ExporterBase):
+    """Exporter for Applicants.
+    """
+    grok.implements(ICSVExporter)
+    grok.name('applicants')
+
+    #: Fieldnames considered by this exporter
+    fields = tuple(IApplicantBaseData.names())
+
+    #: The title under which this exporter will be displayed
+    title = _(u'Applicants')
+
+    def NOmangle_value(self, value, name, context=None):
+        if name == 'provider' and isinstance(value, tuple):
+            value = value[0]
+        return super(
+            ApplicantsContainerExporter, self).mangle_value(
+            value, name, context=context)
+
+    def export(self, applicants, filepath=None):
+        """Export `applicants`, an iterable, as CSV file.
+
+        If `filepath` is ``None``, a raw string with CSV data is returned.
+        """
+        writer, outfile = self.get_csv_writer(filepath)
+        for applicant in applicants:
+            self.write_item(applicant, writer)
+        return self.close_outfile(filepath, outfile)
+
+    def export_all(self, site, filepath=None):
+        """Export applicants into filepath as CSV data.
+
+        If `filepath` is ``None``, a raw string with CSV data is returned.
+        """
+        catalog = queryUtility(
+            ICatalog, context=site, name='applicants_catalog', default=None)
+        if catalog is None:
+            return self.export([], filepath)
+        applicants = catalog.searchResults(
+            reg_number=(None, None))
+        return self.export(applicants, filepath)