Changeset 7732 for main/waeup.sirp

Timestamp:

29 Feb 2012, 18:34:38 (13 years ago)

Author:

uli

Message:

• Add exporter for department.
• Reflect changes in ICSVExporter interface.
• Use utility names that could also be used in export filenames.

File:

• main/waeup.sirp/trunk/src/waeup/sirp/university/export.py (modified) (3 diffs)

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

@@ -10 +10 @@
     """
     grok.implements(ICSVExporter)
-    grok.name('Faculty Exporter')
+    grok.name('faculties')
 
     #: Fieldnames considered by this exporter
@@ -66 +66 @@
         return
 
-    def export(self, faculty, filepath=None):
-        """Export single `faculty` as CSV file.
+    def export(self, faculties, filepath=None):
+        """Export `faculties`, an iterable, as CSV file.
 
         If `filepath` is ``None``, a raw string with CSV data is returned.
         """
         writer, outfile = self.get_csv_writer(filepath)
-        self.write_item(faculty, writer)
+        for faculty in faculties:
+            self.write_item(faculty, writer)
         return self.close_outfile(filepath, outfile)
 
@@ -82 +83 @@
         writer, outfile = self.get_csv_writer(filepath)
         faculties = site.get('faculties', {})
+        return self.export(faculties.values(), filepath)
         for faculty in faculties.values():
             self.write_item(faculty, writer)
         return self.close_outfile(filepath, outfile)
+
+class DepartmentExporter(FacultyExporter, grok.GlobalUtility):
+    """Exporter for departments.
+    """
+    grok.implements(ICSVExporter)
+    grok.name('departments')
+
+    #: Fieldnames considered by this exporter
+    fields = ('code', 'faculty', 'title', 'title_prefix')
+
+    def mangle_value(self, value, name, context=None):
+        """Hook for mangling values in derived classes
+        """
+        if name == 'faculty':
+            value = getattr(
+                getattr(context, '__parent__', None),
+                'code', None)
+        return super(DepartmentExporter, self).mangle_value(
+            value, name, context)
+
+    def export_all(self, site, filepath=None):
+        """Export faculties in facultycontainer into filepath as CSV data.
+
+        If `filepath` is ``None``, a raw string with CSV data is returned.
+        """
+        writer, outfile = self.get_csv_writer(filepath)
+        faculties = site.get('faculties', {})
+        for faculty in faculties.values():
+            for department in faculty.values():
+                self.write_item(department, writer)
+        return self.close_outfile(filepath, outfile)