#PLEASE DO NOT EDIT THIS CODE
#This code was generated using the UMPLE 1.32.1.6535.66c005ced modeling language!
# line 3 "../OneToNTest.ump"

class Mentor():
    #------------------------
    # MEMBER VARIABLES
    #------------------------
    #Mentor Attributes
    #Mentor Associations
    #------------------------
    # CONSTRUCTOR
    #------------------------
    def __init__(self, aName):
        self._students = None
        self._name = None
        self._name = aName
        self._students = []

    #------------------------
    # INTERFACE
    #------------------------
    def setName(self, aName):
        wasSet = False
        self._name = aName
        wasSet = True
        return wasSet

    def getName(self):
        return self._name

    # Code from template association_GetMany 
    def getStudent(self, index):
        aStudent = self._students[index]
        return aStudent

    def getStudents(self):
        newStudents = tuple(self._students)
        return newStudents

    def numberOfStudents(self):
        number = len(self._students)
        return number

    def hasStudents(self):
        has = len(self._students) > 0
        return has

    def indexOfStudent(self, aStudent):
        index = (-1 if not aStudent in self._students else self._students.index(aStudent))
        return index

    # Code from template association_IsNumberOfValidMethod 
    def isNumberOfStudentsValid(self):
        isValid = self.numberOfStudents() >= Mentor.minimumNumberOfStudents() and self.numberOfStudents() <= Mentor.maximumNumberOfStudents()
        return isValid

    # Code from template association_RequiredNumberOfMethod 
    @staticmethod
    def requiredNumberOfStudents():
        return 4

    # Code from template association_MinimumNumberOfMethod 
    @staticmethod
    def minimumNumberOfStudents():
        return 4

    # Code from template association_MaximumNumberOfMethod 
    @staticmethod
    def maximumNumberOfStudents():
        return 4

    # Code from template association_AddMNToOnlyOne 
    def addStudent1(self, aNumber):
        from Student import Student
        if self.numberOfStudents() >= Mentor.maximumNumberOfStudents() :
            return None
        else :
            return Student(aNumber, self)

    def addStudent2(self, aStudent):
        wasAdded = False
        if (aStudent) in self._students :
            return False
        if self.numberOfStudents() >= Mentor.maximumNumberOfStudents() :
            return wasAdded
        existingMentor = aStudent.getMentor()
        isNewMentor = not (existingMentor is None) and not self == existingMentor
        if isNewMentor and existingMentor.numberOfStudents() <= Mentor.minimumNumberOfStudents() :
            return wasAdded
        if isNewMentor :
            aStudent.setMentor(self)
        else :
            self._students.append(aStudent)
        wasAdded = True
        return wasAdded

    def removeStudent(self, aStudent):
        wasRemoved = False
        #Unable to remove aStudent, as it must always have a mentor
        if self == aStudent.getMentor() :
            return wasRemoved
        #mentor already at minimum (4)
        if self.numberOfStudents() <= Mentor.minimumNumberOfStudents() :
            return wasRemoved
        self._students.remove(aStudent)
        wasRemoved = True
        return wasRemoved

    def delete(self):
        i = len(self._students)
        while i > 0 :
            aStudent = self._students[i - 1]
            aStudent.delete()
            i -= 1

    def __str__(self):
        return str(super().__str__()) + "[" + "name" + ":" + str(self.getName()) + "]"

    def addStudent(self, *argv):
        from Student import Student
        if len(argv) == 1 and isinstance(argv[0], str) :
            return self.addStudent1(argv[0])
        if len(argv) == 1 and isinstance(argv[0], Student) :
            return self.addStudent2(argv[0])
        raise TypeError("No method matches provided parameters")

