recopytex/backend/model/tribe.py

from __future__ import annotations

from dataclasses import dataclass, field
from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from backend.model.assessment import Assessment
    from backend.model.student import Student


class TribeError(Exception):
    pass


@dataclass
class Tribe:
    name: str
    level: str
    students: list[Student] = field(default_factory=list)
    assessments: list[Assessment] = field(default_factory=list)

    def register_assessment(self, assessment: Assessment) -> None:
        self.assessments.append(assessment)

    @property
    def students_id(self) -> list[str]:
        return [s.id for s in self.students]

    def register_student(self, student: Student) -> None:
        """Register a student

        If the student is already registered, it is modified.
        """
        try:
            old_student = next(filter(lambda s: s.id == student.id, self.students))
        except StopIteration:
            pass
        else:
            self.students.remove(old_student)

        self.students.append(student)

    def __eq__(self, other: object) -> bool:
        if isinstance(other, Tribe):
            return self.name == other.name
        return False

    def to_dict(self) -> dict:
        return {"name": self.name, "level": self.level}

    def to_tuple(self) -> tuple:
        return (self.name, self.level)