## $Id: utils.py 17481 2023-07-11 09:43:01Z henrik $
##
## Copyright (C) 2011 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
##
import grok
import os
import csv
from time import time
from zope.component import createObject, queryUtility
from zope.catalog.interfaces import ICatalog
from waeup.kofa.interfaces import (
    ADMITTED, CLEARANCE, REQUESTED, CLEARED, RETURNING, PAID,
    REGISTERED, VALIDATED, academic_sessions_vocab)
from kofacustom.nigeria.students.utils import NigeriaStudentsUtils
from waeup.kofa.accesscodes import create_accesscode
from waeup.kofa.students.utils import trans
from waeup.aaue.interfaces import MessageFactory as _

MINIMUM_UNITS_THRESHOLD = 15

SCHOOLFEES = dict()
SFEECHANGES = (12, 13, 14, 15, 20, 21, 22)

for year in SFEECHANGES:
    schoolfees_path = os.path.join(
        os.path.dirname(__file__), 'schoolfees_%s.csv' %year)
    reader = csv.DictReader(open(schoolfees_path, 'rb'))
    SCHOOLFEES[year] = {item['code']:(item['tuition'], item.values()) for item in reader}

acceptancefees_path = os.path.join(
    os.path.dirname(__file__), 'acceptancefees.csv')
reader = csv.DictReader(open(acceptancefees_path, 'rb'))
ACCEPTANCEFEES = {item['code']:(item['acceptance'], item.values()) for item in reader}

class CustomStudentsUtils(NigeriaStudentsUtils):
    """A collection of customized methods.

    """

    # PORTRAIT_CHANGE_STATES = (ADMITTED, CLEARANCE, REQUESTED, CLEARED)

    def allowPortraitChange(self, student):
        if student.is_fresh:
            return True
        return False

    def GPABoundaries(self, faccode=None, depcode=None,
                            certcode=None, student=None):
        if student and student.current_mode.startswith('dp'):
            return ((1.5, 'IRNS / NER / NYV'),
                    (2.4, 'Pass'),
                    (3.5, 'Merit'),
                    (4.5, 'Credit'),
                    (5, 'Distinction'))
        elif student:
            return ((1, 'FRNS / NER / NYV'),
                    (1.5, 'Pass'),
                    (2.4, '3rd Class Honours'),
                    (3.5, '2nd Class Honours Lower Division'),
                    (4.5, '2nd Class Honours Upper Division'),
                    (5, '1st Class Honours'))
        # Session Results Presentations depend on certificate
        results = None
        if certcode:
            cat = queryUtility(ICatalog, name='certificates_catalog')
            results = list(
                cat.searchResults(code=(certcode, certcode)))
        if results and results[0].study_mode.startswith('dp'):
            return ((1.5, 'IRNS / NER / NYV'),
                    (2.4, 'Pass'),
                    (3.5, 'Merit'),
                    (4.5, 'Credit'),
                    (5, 'Distinction'))
        else:
            return ((1, 'FRNS / NER / NYV'),
                    (1.5, 'Pass'),
                    (2.4, '3rd Class Honours'),
                    (3.5, '2nd Class Honours Lower Division'),
                    (4.5, '2nd Class Honours Upper Division'),
                    (5, '1st Class Honours'))

    def getClassFromCGPA(self, gpa, student):
        gpa_boundaries = self.GPABoundaries(student=student)

        try:
            certificate = getattr(student['studycourse'],'certificate',None)
            end_level = getattr(certificate, 'end_level', None)
            final_level = max([studylevel.level for studylevel in student['studycourse'].values()])
            if end_level and final_level >= end_level:
                dummy, repeat = divmod(final_level, 100)
                if gpa <= 5.1 and repeat == 20:
                    # Irrespective of the CGPA of a student, if the He/She has
                    # 3rd Extension, such student will be graduated with a "Pass".
                    return 1, gpa_boundaries[1][1]
        except ValueError:
            pass

        if gpa < gpa_boundaries[0][0]:
            # FRNS / Fail
            return 0, gpa_boundaries[0][1]
        if student.entry_session < 2013 or \
            student.current_mode.startswith('dp'):
            if gpa < gpa_boundaries[1][0]:
                # Pass
                return 1, gpa_boundaries[1][1]
        else:
            if gpa < gpa_boundaries[1][0]:
                # FRNS
                # Pass degree has been phased out in 2013 for non-diploma
                # students
                return 0, gpa_boundaries[0][1]
        if gpa < gpa_boundaries[2][0]:
            # 3rd / Merit
            return 2, gpa_boundaries[2][1]
        if gpa < gpa_boundaries[3][0]:
            # 2nd L / Credit
            return 3, gpa_boundaries[3][1]
        if gpa < gpa_boundaries[4][0]:
            # 2nd U / Distinction
            return 4, gpa_boundaries[4][1]
        if gpa <= gpa_boundaries[5][0]:
            # 1st
            return 5, gpa_boundaries[5][1]
        return

    def getDegreeClassNumber(self, level_obj):
        """Get degree class number (used for SessionResultsPresentation
        reports).
        """
        certificate = getattr(level_obj.__parent__,'certificate', None)
        end_level = getattr(certificate, 'end_level', None)
        if level_obj.level_verdict in ('FRNS', 'NER', 'NYV'):
            return 0
        if end_level and level_obj.level >= end_level:
            if level_obj.level > end_level:
                # spill-over level
                if level_obj.gpa_params[1] == 0:
                    # no credits taken
                    return 0
            elif level_obj.gpa_params[1] < MINIMUM_UNITS_THRESHOLD:
                # credits taken below limit
                return 0
            failed_courses = level_obj.passed_params[4]
            not_taken_courses = level_obj.passed_params[5]
            if '_m' in failed_courses:
                return 0
            if len(not_taken_courses) \
                and not not_taken_courses == 'Nil':
                return 0
        elif level_obj.gpa_params[1] < MINIMUM_UNITS_THRESHOLD:
            # credits taken below limit
            return 0
        # use gpa_boundaries above
        return self.getClassFromCGPA(
            level_obj.cumulative_params[0], level_obj.student)[0]

    def increaseMatricInteger(self, student):
        """Increase counter for matric numbers.
        This counter can be a centrally stored attribute or an attribute of
        faculties, departments or certificates. In the base package the counter
        is as an attribute of the site configuration container.
        """
        if student.current_mode in ('ug_pt', 'de_pt', 'ug_dsh', 'de_dsh'):
            grok.getSite()['configuration'].next_matric_integer += 1
            return
        elif student.is_postgrad:
            grok.getSite()['configuration'].next_matric_integer_3 += 1
            return
        elif student.current_mode in ('dp_ft',):
            grok.getSite()['configuration'].next_matric_integer_4 += 1
            return
        grok.getSite()['configuration'].next_matric_integer_2 += 1
        return

    def _concessionalPaymentMade(self, student):
        if len(student['payments']):
            for ticket in student['payments'].values():
                if ticket.p_state == 'paid' and \
                    ticket.p_category == 'concessional':
                    return True
        return False

    def constructMatricNumber(self, student):
        faccode = student.faccode
        depcode = student.depcode
        certcode = student.certcode
        degree = getattr(
            getattr(student.get('studycourse', None), 'certificate', None),
                'degree', None)
        year = unicode(student.entry_session)[2:]
        if not student.state in (PAID, ) or not student.is_fresh or \
            student.current_mode in ('found', 'ijmbe'):
            return _('Matriculation number cannot be set.'), None
        #if student.current_mode not in ('mug_ft', 'mde_ft') and \
        #    not self._concessionalPaymentMade(student):
        #    return _('Matriculation number cannot be set.'), None
        if student.is_postgrad:
            next_integer = grok.getSite()['configuration'].next_matric_integer_3
            if not degree or next_integer == 0:
                return _('Matriculation number cannot be set.'), None
            if student.faccode in ('IOE'):
                return None, "AAU/SPS/%s/%s/%s/%05d" % (
                    faccode, year, degree, next_integer)
            return None, "AAU/SPS/%s/%s/%s/%s/%05d" % (
                faccode, depcode, year, degree, next_integer)
        if student.current_mode in ('ug_dsh', 'de_dsh'):
            next_integer = grok.getSite()['configuration'].next_matric_integer
            if next_integer == 0:
                return _('Matriculation number cannot be set.'), None
            return None, "DSH/%s/%s/%s/%05d" % (
                faccode, depcode, year, next_integer)
        if student.current_mode in ('ug_pt', 'de_pt'):
            next_integer = grok.getSite()['configuration'].next_matric_integer
            if next_integer == 0:
                return _('Matriculation number cannot be set.'), None
            return None, "PTP/%s/%s/%s/%05d" % (
                faccode, depcode, year, next_integer)
        if student.current_mode in ('dp_ft',):
            next_integer = grok.getSite()['configuration'].next_matric_integer_4
            if next_integer == 0:
                return _('Matriculation number cannot be set.'), None
            return None, "IOE/DIP/%s/%05d" % (year, next_integer)
        next_integer = grok.getSite()['configuration'].next_matric_integer_2
        if next_integer == 0:
            return _('Matriculation number cannot be set.'), None
        if student.faccode in ('FBM', 'FCS', 'FMLS'):
            return None, "CMS/%s/%s/%s/%05d" % (
                faccode, depcode, year, next_integer)
        return None, "%s/%s/%s/%05d" % (faccode, depcode, year, next_integer)

    def getReturningData(self, student):
        """ This method defines what happens after school fee payment
        of returning students depending on the student's senate verdict.
        """
        prev_level = student['studycourse'].current_level
        cur_verdict = student['studycourse'].current_verdict
        if cur_verdict in ('A','B','L','M','N','Z',):
            # Successful student
            new_level = divmod(int(prev_level),100)[0]*100 + 100
        elif cur_verdict == 'C':
            # Student on probation
            new_level = int(prev_level) + 10
        else:
            # Student is somehow in an undefined state.
            # Level has to be set manually.
            new_level = prev_level
        new_session = student['studycourse'].current_session + 1
        return new_session, new_level

    def _isPaymentDisabled(self, p_session, category, student):
        academic_session = self._getSessionConfiguration(p_session)
        if category.startswith('schoolfee'):
            if 'sf_all' in academic_session.payment_disabled:
                return True
            if 'sf_pg' in academic_session.payment_disabled and \
                student.is_postgrad:
                return True
            if 'sf_ug_pt' in academic_session.payment_disabled and \
                student.current_mode in ('ug_pt', 'de_pt'):
                return True
            if 'sf_found' in academic_session.payment_disabled and \
                student.current_mode == 'found':
                return True
        if category.startswith('clearance') and \
            'cl_regular' in academic_session.payment_disabled and \
            student.current_mode in ('ug_ft', 'de_ft', 'mug_ft', 'mde_ft'):
            return True
        if category == 'hostel_maintenance' and \
            'maint_all' in academic_session.payment_disabled:
            return True
        return False

    def setPaymentDetails(self, category, student,
            previous_session=None, previous_level=None, combi=[]):
        """Create Payment object and set the payment data of a student for
        the payment category specified.

        """
        details = {}
        p_item = u''
        amount = 0.0
        error = u''
        if previous_session:
            if previous_session < student['studycourse'].entry_session:
                return _('The previous session must not fall below '
                         'your entry session.'), None
            if category == 'schoolfee':
                # School fee is always paid for the following session
                if previous_session > student['studycourse'].current_session:
                    return _('This is not a previous session.'), None
            else:
                if previous_session > student['studycourse'].current_session - 1:
                    return _('This is not a previous session.'), None
            p_session = previous_session
            p_level = previous_level
            p_current = False
        else:
            p_session = student['studycourse'].current_session
            p_level = student['studycourse'].current_level
            p_current = True
        academic_session = self._getSessionConfiguration(p_session)
        if academic_session == None:
            return _(u'Session configuration object is not available.'), None
        # Determine fee.
        if category == 'transfer':
            amount = academic_session.transfer_fee
        elif category == 'transcript_local':
            amount = academic_session.transcript_fee_local
        elif category == 'transcript_inter':
            amount = academic_session.transcript_fee_inter
        elif category == 'bed_allocation':
            acco_details = self.getAccommodationDetails(student)
            p_session = acco_details['booking_session']
            p_item = acco_details['bt']
            amount = academic_session.booking_fee
        elif category == 'hostel_maintenance':
            amount = 0.0
            booking_session = grok.getSite()['hostels'].accommodation_session
            bedticket = student['accommodation'].get(str(booking_session), None)
            if bedticket is not None and bedticket.bed is not None:
                p_session = booking_session
                p_item = bedticket.display_coordinates
                if bedticket.bed.__parent__.maint_fee > 0:
                    amount = bedticket.bed.__parent__.maint_fee
                else:
                    # fallback
                    amount = academic_session.maint_fee
            else:
                return _(u'No bed allocated.'), None
        elif student.current_mode == 'found' and category not in (
            'schoolfee', 'clearance', 'late_registration'):
            return _('Not allowed.'), None
        elif category.startswith('clearance'):
            if student.state not in (ADMITTED, CLEARANCE, REQUESTED, CLEARED):
                return _(u'Acceptance Fee payments not allowed.'), None
            try:
                certificate = student['studycourse'].certificate
                p_item = certificate.code
            except (AttributeError, TypeError):
                return _('Study course data are incomplete.'), None
            if student.current_mode in (
                'ug_ft', 'ug_pt', 'de_ft', 'de_pt',
                'transfer', 'mug_ft', 'mde_ft') \
                and category != 'clearance_incl':
                    return _("Additional fees must be included."), None
            elif student.current_mode == 'special_pg_ft':
                if category != 'clearance':
                    return _("No additional fees required."), None
            try:
                acceptancefees = ACCEPTANCEFEES[student.certcode]
            except KeyError:
                return _('Acceptance fees not yet fixed.'), None
            if category == 'clearance_incl':
                for item in acceptancefees[1]:
                    try:
                        amount += int(item)
                    except:
                        pass
            else:
                amount = float(acceptancefees[0])
        elif category == 'late_registration':
            if student.is_postgrad:
                amount = academic_session.late_pg_registration_fee
            else:
                amount = academic_session.late_registration_fee
        elif category == 'ict':
            if student.is_fresh:
                amount = 2200
            else:
                amount = 1200
        elif category.startswith('schoolfee'):
            try:
                certificate = student['studycourse'].certificate
                p_item = certificate.code
            except (AttributeError, TypeError):
                return _('Study course data are incomplete.'), None
            try:
                if student.entry_session < 2013:
                    schoolfees = SCHOOLFEES[12][student.certcode]
                elif student.entry_session < 2014:
                    schoolfees = SCHOOLFEES[13][student.certcode]
                elif student.entry_session < 2015:
                    schoolfees = SCHOOLFEES[14][student.certcode]
                elif student.entry_session < 2020:
                    schoolfees = SCHOOLFEES[15][student.certcode]
                elif student.entry_session < 2021:
                    schoolfees = SCHOOLFEES[20][student.certcode]
                elif student.entry_session < 2022:
                    schoolfees = SCHOOLFEES[21][student.certcode]
                else:
                    schoolfees = SCHOOLFEES[22][student.certcode]
            except KeyError:
                return _('School fees not yet fixed.'), None
            if student.is_postgrad and category != 'schoolfee':
                return _("No additional fees required."), None
            if not previous_session and student.current_mode in (
                'ug_ft', 'ug_pt', 'de_ft', 'de_pt',
                'transfer', 'mug_ft', 'mde_ft') \
                and not category in (
                'schoolfee_incl', 'schoolfee_1', 'schoolfee_2'):
                    return _("You must choose a payment which includes "
                             "additional fees."), None
            if category in ('schoolfee_1', 'schoolfee_2'):
                if student.current_mode == 'ug_pt':
                    return _("Part-time students are not allowed "
                             "to pay by instalments."), None
                if student.entry_session < 2015:
                    return _("You are not allowed "
                             "to pay by instalments."), None
            additional = 0.0
            for item in schoolfees[1]:
                try:
                    additional += int(item)
                except:
                    pass
            amount = float(schoolfees[0])
            additional -= amount
            if previous_session:
                # Students can pay for previous sessions in all
                # workflow states.  Fresh students are excluded by the
                # update method of the PreviousPaymentAddFormPage.
               # Cut school fee by 50%
                if category in ('schoolfee_1', 'schoolfee_2') and amount:
                    amount /= 2
                pass
            elif student.state == CLEARED:
                # Cut school fee by 50%
                if category in ('schoolfee_1', 'schoolfee_2') and amount:
                    amount /= 2
            elif student.state == RETURNING and category != 'schoolfee_2':
                if not student.father_name:
                    return _("Personal data form is not properly filled."), None
                # In case of returning school fee payment the payment session
                # and level contain the values of the session the student
                # has paid for.
                p_session, p_level = self.getReturningData(student)
                try:
                    academic_session = grok.getSite()[
                        'configuration'][str(p_session)]
                except KeyError:
                    return _(u'Session configuration object is not available.'), None
                # Cut school fee by 50%
                if category == 'schoolfee_1' and amount:
                    amount /= 2
            elif category == 'schoolfee_2' and amount:
                amount /= 2
            else:
                return _('Wrong state.'), None
            if amount in (0.0, None):
                return _(u'Amount could not be determined.'), None
            # Add additional fees
            if category in ('schoolfee_incl', 'schoolfee_1'):
                amount += additional
            # Add non-indigenous fee and session specific penalty fees
            if student.is_postgrad:
                amount += academic_session.penalty_pg
                if student.lga and not student.lga.startswith('edo'):
                    amount += 20000.0
            else:
                amount += academic_session.penalty_ug
        elif not student.is_postgrad:
            fee_name = category + '_fee'
            amount = getattr(academic_session, fee_name, 0.0)
        if amount in (0.0, None):
            return _(u'Amount could not be determined.'), None
        # Create ticket.
        for key in student['payments'].keys():
            ticket = student['payments'][key]
            if ticket.p_state == 'paid' and\
               ticket.p_category == category and \
               not ticket.p_category.startswith('transcript') and \
               ticket.p_item == p_item and \
               ticket.p_session == p_session:
                  return _('This type of payment has already been made.'), None
            # Additional condition in AAUE
            if category in ('schoolfee', 'schoolfee_incl', 'schoolfee_1'):
                if ticket.p_state == 'paid' and \
                   ticket.p_category in ('schoolfee',
                                         'schoolfee_incl',
                                         'schoolfee_1') and \
                   ticket.p_item == p_item and \
                   ticket.p_session == p_session:
                      return _(
                          'Another school fee payment for this '
                          'session has already been made.'), None

        if self._isPaymentDisabled(p_session, category, student):
            return _('This category of payments has been disabled.'), None
        payment = createObject(u'waeup.StudentOnlinePayment')
        timestamp = ("%d" % int(time()*10000))[1:]
        payment.p_id = "p%s" % timestamp
        payment.p_category = category
        payment.p_item = p_item
        payment.p_session = p_session
        payment.p_level = p_level
        payment.p_current = p_current
        payment.amount_auth = amount
        return None, payment

    def _admissionText(self, student, portal_language):
        inst_name = grok.getSite()['configuration'].name
        entry_session = student['studycourse'].entry_session
        entry_session = academic_sessions_vocab.getTerm(entry_session).title
        text = trans(_(
            'This is to inform you that you have been offered provisional'
            ' admission into ${a} for the ${b} academic session as follows:',
            mapping = {'a': inst_name, 'b': entry_session}),
            portal_language)
        return text

    def warnCreditsOOR(self, studylevel, course=None):
        studycourse = studylevel.__parent__
        certificate = getattr(studycourse,'certificate', None)
        current_level = studycourse.current_level
        if None in (current_level, certificate):
            return
        end_level = certificate.end_level
        if current_level >= end_level:
            limit = 52
        else:
            limit = 48
        if course and studylevel.total_credits + course.credits > limit:
            return  _('Maximum credits exceeded.')
        elif studylevel.total_credits > limit:
            return _('Maximum credits exceeded.')
        return

    def getBedCoordinates(self, bedticket):
        """Return descriptive bed coordinates.
        This method can be used to customize the `display_coordinates`
        property method in order to  display a
        customary description of the bed space.
        """
        bc = bedticket.bed_coordinates.split(',')
        if len(bc) == 4:
            return bc[0]
        return bedticket.bed_coordinates

    def getAccommodationDetails(self, student):
        """Determine the accommodation data of a student.
        """
        d = {}
        d['error'] = u''
        hostels = grok.getSite()['hostels']
        d['booking_session'] = hostels.accommodation_session
        d['allowed_states'] = hostels.accommodation_states
        d['startdate'] = hostels.startdate
        d['enddate'] = hostels.enddate
        d['expired'] = hostels.expired
        # Determine bed type
        bt = 'all'
        if student.sex == 'f':
            sex = 'female'
        else:
            sex = 'male'
        special_handling = 'regular'
        d['bt'] = u'%s_%s_%s' % (special_handling,sex,bt)
        return d

    def checkAccommodationRequirements(self, student, acc_details):
        msg = super(CustomStudentsUtils, self).checkAccommodationRequirements(
            student, acc_details)
        if msg:
            return msg
        if student.current_mode not in ('ug_ft', 'de_ft', 'mug_ft', 'mde_ft'):
            return _('You are not eligible to book accommodation.')
        return

    # AAUE prefix
    STUDENT_ID_PREFIX = u'E'

    STUDENT_EXPORTER_NAMES = (
            'students',
            'studentstudycourses',
            'studentstudycourses_1',
            'studentstudylevels',
            #'studentstudylevels_1',
            'coursetickets',
            #'coursetickets_1',
            'studentpayments',
            'bedtickets',
            'unpaidpayments',
            'sfpaymentsoverview',
            'studylevelsoverview',
            'combocard',
            'bursary',
            'levelreportdata',
            'outstandingcourses',
            'sessionpaymentsoverview',
            'accommodationpayments',
            'transcriptdata',
            'trimmedpayments',
            'trimmed',
            'outstandingcourses_2'
            )

    # Maximum size of upload files in kB
    MAX_KB = 500