## $Id: export.py 16653 2021-09-29 10:21:24Z henrik $
##
## Copyright (C) 2014 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 applicant-related stuff.
"""
import grok
from waeup.kofa.applicants.interfaces import IApplicantBaseData
from waeup.kofa.utils.helpers import iface_names
from waeup.kofa.applicants.export import ApplicantRefereeReportExporter
from kofacustom.nigeria.applicants.export import NigeriaApplicantExporter
from kofacustom.nigeria.applicants.interfaces import (
    INigeriaUGApplicant, INigeriaPGApplicant)
from kofacustom.iuokada.applicants.interfaces import (
    ICustomUGApplicant, ICustomPGApplicant,
    ICustomApplicantRefereeReport)

class CustomApplicantExporter(NigeriaApplicantExporter):
    """Exporter for Custom Applicants.
    """

    fields = tuple(sorted(set(
        iface_names(ICustomUGApplicant) +
        iface_names(ICustomPGApplicant) +
        iface_names(INigeriaUGApplicant) +
        iface_names(INigeriaPGApplicant) +
        iface_names(IApplicantBaseData)
        ))) + (
        'password', 'state', 'history', 'container_code', 'application_number',
        'display_fullname', 'application_date', 'department', 'college',)

    def mangle_value(self, value, name, context=None):
        if name in ('department', 'college') and context is not None:
            course_admitted = getattr(context, 'course_admitted', None)
            if course_admitted is not None:
                try:
                    if name == 'department':
                        value = course_admitted.__parent__.__parent__.title
                    else:
                        value = course_admitted.__parent__.__parent__.__parent__.title
                except AttributeError:
                    value = 'N/A'
        return super(
            CustomApplicantExporter, self).mangle_value(
            value, name, context=context)

class CustomApplicantRefereeReportExporter(ApplicantRefereeReportExporter):

    fields = tuple(sorted(iface_names(
        ICustomApplicantRefereeReport,
        exclude_attribs=False))) + (
              'applicant_id',
              'reg_number',
              'display_fullname',)