Changeset 16317 for main/waeup.uniben/trunk/src/waeup/uniben/applicants/export.py

Timestamp:

15 Nov 2020, 19:41:33 (4 years ago)

Author:

Henrik Bettermann

Message:

Implement PGApplicantExporter.

File:

• main/waeup.uniben/trunk/src/waeup/uniben/applicants/export.py (modified) (1 diff)

main/waeup.uniben/trunk/src/waeup/uniben/applicants/export.py

@@ -51 +51 @@
         return super(CustomApplicantPaymentExporter, self).fields + (
             'r_pay_reference',)
+
+class PGApplicantExporter(NigeriaApplicantExporter):
+    """Exports pg applicants only.
+    """
+
+    grok.name('pg_applicants')
+    title = 'PG Applicants'
+
+    def is_postgrad(self, applicant):
+        course_studied = getattr(applicant, 'course_studied', None)
+        if course_studied and (course_studied.study_mode.startswith('pg')
+              or course_studied.study_mode.startswith('special_pg')):
+            return True
+        return False
+
+    def export(self, applicants, filepath=None):
+        """
+        """
+        writer, outfile = self.get_csv_writer(filepath)
+        for applicant in applicants:
+            if self.is_postgrad(applicant):
+                self.write_item(applicant, writer)
+        return self.close_outfile(filepath, outfile)