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

Timestamp:

26 Oct 2012, 17:57:26 (12 years ago)

Author:

Henrik Bettermann

Message:

Implement BedTicketsExporter?.

File:

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

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

@@ -24 +24 @@
 from waeup.kofa.students.interfaces import (
     IStudent, IStudentStudyCourse, IStudentStudyLevel, ICourseTicket,
-    IStudentOnlinePayment, ICSVStudentExporter)
+    IStudentOnlinePayment, ICSVStudentExporter, IBedTicket)
 from waeup.kofa.utils.batching import ExporterBase
 from waeup.kofa.utils.helpers import iface_names
@@ -76 +76 @@
     return payments
 
+def get_bedtickets(students):
+    """Get all bedtickets of `students`.
+    """
+    tickets = []
+    for student in students:
+        for ticket in student.get('accommodation', {}).values():
+            tickets.append(ticket)
+    return tickets
 
 class StudentExporterBase(ExporterBase):
@@ -273 +281 @@
     def export_student(self, student, filepath=None):
         return self.export(get_payments([student]), filepath)
+
+class BedTicketsExporter(grok.GlobalUtility, StudentExporterBase):
+    """Exporter for BedTicket instances.
+    """
+    grok.name('bedtickets')
+
+    #: Fieldnames considered by this exporter
+    fields = tuple(
+        sorted(iface_names(
+            IBedTicket, exclude_attribs=False))) + (
+                'student_id', 'actual_bed_type')
+
+    #: The title under which this exporter will be displayed
+    title = _(u'Bed Tickets')
+
+    def mangle_value(self, value, name, context=None):
+        """Treat location values and others special.
+        """
+        if context is not None:
+            student = context.student
+            if name in ['student_id'] and student is not None:
+                value = getattr(student, name, None)
+        if name == 'bed' and value is not None:
+            value = getattr(value, 'bed_id', None)
+        if name == 'actual_bed_type':
+            value = getattr(getattr(context, 'bed', None), 'bed_type')
+        return super(
+            BedTicketsExporter, self).mangle_value(
+            value, name, context=context)
+
+    def export_all(self, site, filepath=None):
+        """Export payments into filepath as CSV data.
+
+        If `filepath` is ``None``, a raw string with CSV data is returned.
+        """
+        return self.export(get_bedtickets(get_students(site)), filepath)
+
+    def export_student(self, student, filepath=None):
+        return self.export(get_bedtickets([student]), filepath)