#PLEASE DO NOT EDIT THIS CODE
#This code was generated using the UMPLE 1.38.0.8942.247e17422 modeling language!


class Mentor:
    def __init__(self, aName):
        self._name = aName
        self._students = []

    def setName(self, aName):
        self._name = aName
        return True

    def getName(self):
        return self._name

    def getStudent(self, index):
        return self._students[index]

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

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

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

    def indexOfStudent(self, aStudent):
        return next((i for i, x in enumerate(self._students) if x is aStudent), -1)

    def isNumberOfStudentsValid(self):
        return self.numberOfStudents() >= __class__.minimumNumberOfStudents()

    @staticmethod
    def minimumNumberOfStudents():
        return 5

    def addStudent1(self, aNumber):
        from example.Student import Student
        return Student(aNumber, self)

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

    def addStudent(self, *args, **kwargs):
        from example.Student import Student
        bound = __class__._bindArguments(("aStudent",), False, args, kwargs)
        if bound is not None and (bound[0] is None or isinstance(bound[0], Student)):
            return __class__.addStudent2(self, *bound)
        bound = __class__._bindArguments(("aNumber",), False, args, kwargs)
        if bound is not None and (bound[0] is None or isinstance(bound[0], str)):
            return __class__.addStudent1(self, *bound)
        raise TypeError("No method matches provided parameters")

    def removeStudent(self, aStudent):
        if aStudent.getMentor() is self:
            return False
        if self.numberOfStudents() <= self.minimumNumberOfStudents():
            return False
        index = next((i for i, x in enumerate(self._students) if x is aStudent), -1)
        if index >= 0:
            del self._students[index]
        return True

    def addStudentAt(self, aStudent, index):
        if not self.addStudent2(aStudent):
            return False
        if index < 0:
            index = 0
        if index > self.numberOfStudents():
            index = self.numberOfStudents() - 1
        del self._students[next((i for i, x in enumerate(self._students) if x is aStudent), -1)]
        self._students.insert(index, aStudent)
        return True

    def addOrMoveStudentAt(self, aStudent, index):
        if not any(x is aStudent for x in self._students):
            return self.addStudentAt(aStudent, index)
        if index < 0:
            index = 0
        if index > self.numberOfStudents():
            index = self.numberOfStudents() - 1
        del self._students[next((i for i, x in enumerate(self._students) if x is aStudent), -1)]
        self._students.insert(index, aStudent)
        return True

    def delete(self):
        for i in range(len(self._students), 0, -1):
            aStudent = self._students[i - 1]
            aStudent.delete()

    @staticmethod
    def _bindArguments(names, varargs, args, kwargs):
        if len(args) > len(names) and not varargs:
            return None
        bound = list(args[:len(names)])
        for name in names[len(bound):]:
            if name not in kwargs:
                return None
            bound.append(kwargs[name])
        if len(kwargs) != len(names) - min(len(args), len(names)):
            return None
        if varargs:
            bound.append(args[len(names):])
        return bound

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

#PLEASE DO NOT EDIT THIS CODE
#This code was generated using the UMPLE 1.38.0.8942.247e17422 modeling language!


class Student:
    def __init__(self, aNumber, aMentor):
        self._number = aNumber
        self._mentor = None
        if not self.setMentor(aMentor):
            raise RuntimeError("Unable to create student due to mentor. See https://manual.umple.org?RE002ViolationofAssociationMultiplicity.html")

    def setNumber(self, aNumber):
        self._number = aNumber
        return True

    def getNumber(self):
        return self._number

    def getMentor(self):
        return self._mentor

    def setMentor(self, aMentor):
        if aMentor is None:
            return False
        if self._mentor is not None and self._mentor.numberOfStudents() <= self._mentor.minimumNumberOfStudents():
            return False
        existingMentor = self._mentor
        self._mentor = aMentor
        if existingMentor is not None and existingMentor is not aMentor:
            if not existingMentor.removeStudent(self):
                self._mentor = existingMentor
                return False
        self._mentor.addStudent(self)
        return True

    def delete(self):
        placeholderMentor = self._mentor
        self._mentor = None
        if placeholderMentor is not None:
            placeholderMentor.removeStudent(self)

    def __str__(self):
        return (super().__str__() + "["
                + "number:" + str(self.getNumber())
                + "]"
                + "\n  mentor = " + ("null" if self.getMentor() is None else format(id(self.getMentor()), "x")))