## $Id: export.py 7863 2012-03-13 02:58:45Z uli $ ## ## Copyright (C) 2012 Uli Fouquet & Henrik Bettermann ## This program is free software; you can redistribute it and/or modify ## it under the terms of the GNU General Public License as published by ## the Free Software Foundation; either version 2 of the License, or ## (at your option) any later version. ## ## This program is distributed in the hope that it will be useful, ## but WITHOUT ANY WARRANTY; without even the implied warranty of ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the ## GNU General Public License for more details. ## ## You should have received a copy of the GNU General Public License ## along with this program; if not, write to the Free Software ## Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA ## """Exporters for faculties, departments, and other academics components. """ import grok from waeup.kofa.interfaces import ICSVExporter from waeup.kofa.utils.batching import ExporterBase class FacultyExporter(grok.GlobalUtility, ExporterBase): """Exporter for faculties. """ grok.implements(ICSVExporter) grok.name('faculties') #: Fieldnames considered by this exporter fields = ('code', 'title', 'title_prefix') 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) for faculty in faculties: self.write_item(faculty, writer) return self.close_outfile(filepath, outfile) 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', {}) return self.export(faculties.values(), filepath) class DepartmentExporter(FacultyExporter, grok.GlobalUtility): """Exporter for departments. """ grok.implements(ICSVExporter) grok.name('departments') #: Fieldnames considered by this exporter fields = ('code', 'faculty_code', 'title', 'title_prefix') def mangle_value(self, value, name, context=None): """Hook for mangling values in derived classes """ if name == 'faculty_code': 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) class CourseExporter(FacultyExporter, grok.GlobalUtility): """Exporter for courses. """ grok.implements(ICSVExporter) grok.name('courses') #: Fieldnames considered by this exporter fields = ('code', 'faculty_code', 'department_code', 'title', 'credits', 'passmark', 'semester') def mangle_value(self, value, name, context=None): """Hook for mangling values in derived classes """ if name == 'faculty_code': try: value = context.__parent__.__parent__.__parent__.code except AttributeError: value = None elif name == 'department_code': try: value = context.__parent__.__parent__.code except AttributeError: value = None return super(CourseExporter, 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(): for course in department.courses.values(): self.write_item(course, writer) return self.close_outfile(filepath, outfile) class CertificateExporter(CourseExporter, grok.GlobalUtility): """Exporter for courses. """ grok.implements(ICSVExporter) grok.name('certificates') #: Fieldnames considered by this exporter fields = ('code', 'faculty_code', 'department_code', 'title', 'study_mode', 'start_level', 'end_level', 'application_category') 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(): for cert in department.certificates.values(): self.write_item(cert, writer) return self.close_outfile(filepath, outfile) class CertificateCourseExporter(CourseExporter, grok.GlobalUtility): """Exporter for courses. """ grok.implements(ICSVExporter) grok.name('certificate_courses') #: Fieldnames considered by this exporter fields = ('course', 'faculty_code', 'department_code', 'certificate_code', 'level', 'mandatory') def mangle_value(self, value, name, context=None): """Hook for mangling values in derived classes """ if name == 'faculty_code': try: value = context.__parent__.__parent__.__parent__.__parent__.code except AttributeError: value = None elif name == 'department_code': try: value = context.__parent__.__parent__.__parent__.code except AttributeError: value = None elif name == 'certificate_code': value = getattr(context, '__parent__', None) value = getattr(value, 'code', None) if name == 'course': value = getattr(value, 'code', None) return super(CourseExporter, 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(): for cert in department.certificates.values(): for certref in cert.values(): self.write_item(certref, writer) return self.close_outfile(filepath, outfile)