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


@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)

    def register_student(self, student: Student) -> None:
        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)