Changeset 8342 for main/waeup.kofa/trunk/src/waeup/kofa/students/export.py

Timestamp:

4 May 2012, 11:52:58 (13 years ago)

Author:

uli

Message:

Add exporter for course tickets.

File:

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

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

@@ -24 +24 @@
 from waeup.kofa.interfaces import MessageFactory as _
 from waeup.kofa.students.interfaces import (
-    IStudent, IStudentStudyCourse, IStudentStudyLevel,)
+    IStudent, IStudentStudyCourse, IStudentStudyLevel, ICourseTicket)
 from waeup.kofa.utils.batching import ExporterBase
 from waeup.kofa.utils.helpers import iface_names
@@ -186 +186 @@
                 levels.append(level)
         return self.export(levels, filepath)
+
+class CourseTicketExporter(grok.GlobalUtility, ExporterBase):
+    """Exporter for CourseTickets.
+    """
+    grok.implements(ICSVExporter)
+    grok.name('coursetickets')
+
+    #: Fieldnames considered by this exporter
+    fields = tuple(sorted(iface_names(ICourseTicket) + [
+        'reg_number', 'matric_number', 'level', 'code', 'title', 'credits',
+        'passmark', 'semester', 'fcode', 'dcode']))
+
+    #: The title under which this exporter will be displayed
+    title = _(u'Course Tickets')
+
+    def mangle_value(self, value, name, context=None):
+        """Treat location values special.
+        """
+        if context is not None:
+            student = context.getStudent()
+            if name in ['reg_number', 'matric_number'] and student is not None:
+                value = getattr(student, name, None)
+            if name == 'level':
+                value = context.getLevel()
+        return super(
+            CourseTicketExporter, self).mangle_value(
+            value, name, context=context)
+
+    def export(self, course_tickets, filepath=None):
+        """Export `course_tickets`, 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 ticket in course_tickets:
+            self.write_item(ticket, writer)
+        return self.close_outfile(filepath, outfile)
+
+    def export_all(self, site, filepath=None):
+        """Export course tickets into filepath as CSV data.
+
+        If `filepath` is ``None``, a raw string with CSV data is returned.
+        """
+        catalog = queryUtility(
+            ICatalog, context=site, name='students_catalog', default=None)
+        if catalog is None:
+            return self.export([], filepath)
+        students = catalog.searchResults(
+            student_id=(None, None))
+        levels = []
+        study_courses = [x.get('studycourse', None) for x in students
+                         if x is not None]
+        for course in study_courses:
+            for level in course.values():
+                levels.append(level)
+
+        tickets = []
+        for level in levels:
+            for ticket in level.values():
+                tickets.append(ticket)
+        return self.export(tickets, filepath)
+