Changeset 7861

Timestamp:

13 Mar 2012, 01:26:21 (13 years ago)

Author:

uli

Message:

Use the new exporter base for our exporters.

File:

• main/waeup.kofa/trunk/src/waeup/kofa/university/export.py (modified) (2 diffs)

main/waeup.kofa/trunk/src/waeup/kofa/university/export.py

@@ -22 +22 @@
 from cStringIO import StringIO
 from waeup.kofa.interfaces import ICSVExporter
+from waeup.kofa.utils.batching import ExporterBase
 
-class FacultyExporter(grok.GlobalUtility):
+class FacultyExporter(grok.GlobalUtility, ExporterBase):
     """Exporter for faculties.
     """
@@ -31 +32 @@
     #: Fieldnames considered by this exporter
     fields = ('code', 'title', 'title_prefix')
-
-    def mangle_value(self, value, name, context=None):
-        """Hook for mangling values in derived classes
-        """
-        if isinstance(value, bool):
-            value = value and '1' or '0'
-        elif isinstance(value, unicode):
-            # CSV writers like byte streams better than unicode
-            value = value.encode('utf-8')
-        elif value is None:
-            # None is not really representable in CSV files
-            value = ''
-        return value
-
-    def get_csv_writer(self, filepath=None):
-        """Get a CSV dict writer instance open for writing.
-
-        Returns a tuple (<writer>, <outfile>) where ``<writer>`` is a
-        :class:`csv.DictWriter` instance and outfile is the real file
-        which is written to. The latter is important when writing to
-        StringIO and can normally be ignored otherwise.
-
-        The returned file will already be filled with the header row.
-        """
-        if filepath is None:
-            outfile = StringIO()
-        else:
-            outfile = open(filepath, 'wb')
-        writer = csv.DictWriter(outfile, self.fields)
-        writer.writerow(dict(zip(self.fields, self.fields))) # header
-        return writer, outfile
-
-    def write_item(self, faculty, writer):
-        """Write a row into using `writer`.
-        """
-        row = {}
-        for name in self.fields:
-            value = getattr(faculty, name, None)
-            value = self.mangle_value(value, name, faculty)
-            row[name] = value
-        writer.writerow(row)
-        return
-
-    def close_outfile(self, filepath, outfile):
-        """Close outfile.
-
-        If filepath is None, the contents of outfile is returned.
-        """
-        outfile.seek(0)
-        if filepath is None:
-            return outfile.read()
-        outfile.close()
-        return
 
     def export(self, faculties, filepath=None):