uclouvain/osis-dissertation

##############################################################################
#
#    OSIS stands for Open Student Information System. It's an application
#    designed to manage the core business of higher education institutions,
#    such as universities, faculties, institutes and professional schools.
#    The core business involves the administration of students, teachers,
#    courses, programs and so on.
#
#    Copyright (C) 2015-2019 Université catholique de Louvain (http://www.uclouvain.be)
#
#    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 3 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.
#
#    A copy of this license - GNU General Public License - is available
#    at the root of the source code of this program.  If not,
#    see http://www.gnu.org/licenses/.
#
##############################################################################
from django.core.exceptions import ObjectDoesNotExist, ValidationError
from django.db import models

from osis_common.models.serializable_model import SerializableModel, SerializableModelAdmin
from .enums.dissertation_role_status import DissertationRoleStatus

MAX_PROPOSITION_ROLE = 4


class PropositionRoleAdmin(SerializableModelAdmin):
    list_display = ('adviser', 'status', 'proposition_dissertation')
    raw_id_fields = ('adviser', 'proposition_dissertation')
    search_fields = ('uuid', 'proposition_dissertation__author__person__last_name',
                     'proposition_dissertation__author__person__first_name',
                     'proposition_dissertation__title',
                     'adviser__person__last_name',
                     'adviser__person__first_name')


class PropositionRole(SerializableModel):
    status = models.CharField(
        max_length=12,
        choices=DissertationRoleStatus.choices(),
        default=DissertationRoleStatus.PROMOTEUR.name
    )
    adviser = models.ForeignKey('Adviser', on_delete=models.CASCADE)
    proposition_dissertation = models.ForeignKey('PropositionDissertation', on_delete=models.CASCADE)

    def __str__(self):
        return u"%s %s" % (self.status if self.status else "",
                           self.adviser if self.adviser else "")

    def clean(self):
        if PropositionRole.objects.filter(
                proposition_dissertation=self.proposition_dissertation
        ).count() >= MAX_PROPOSITION_ROLE:
            raise ValidationError("Number of maximum jury reached")
        return super().clean()


def count_by_dissertation(dissertation):
    return PropositionRole.objects.filter(proposition_dissertation=dissertation.proposition_dissertation) \
        .count()


def search_by_dissertation(dissertation):
    return PropositionRole.objects.filter(proposition_dissertation=dissertation.proposition_dissertation)


def count_by_proposition(prop_dissert):
    return PropositionRole.objects.filter(proposition_dissertation=prop_dissert) \
        .count()


def search_by_proposition(prop_dissert):
    return PropositionRole.objects.filter(proposition_dissertation=prop_dissert)


def add(status, adviser, proposition_dissertation):
    if count_by_status_adviser_proposition(status, adviser, proposition_dissertation) == 0:
        role = PropositionRole(status=status, adviser=adviser, proposition_dissertation=proposition_dissertation)
        role.save()


def delete(status, proposition_dissertation):
    roles = PropositionRole.objects.filter(proposition_dissertation=proposition_dissertation) \
        .filter(status=status)

    for role in roles:
        role.delete()


def count_by_status_adviser_proposition(status, adviser, proposition_dissertation):
    return PropositionRole.objects.filter(proposition_dissertation=proposition_dissertation) \
        .filter(status=status) \
        .filter(adviser=adviser) \
        .count()


def get_by_id(prop_role_id):
    try:
        return PropositionRole.objects.get(pk=prop_role_id)
    except ObjectDoesNotExist:
        return None