Feat: build tribe and student model

This commit is contained in:
Bertrand Benjamin 2022-12-20 06:59:35 +01:00
parent a2a0269f39
commit c4fcb6a0ef
2 changed files with 37 additions and 0 deletions

28
backend/model/student.py Normal file
View File

@ -0,0 +1,28 @@
from __future__ import annotations
from dataclasses import dataclass, field
@dataclass
class Tribe:
name: str
level: str
students: list[Student] = field(default_factory=list)
def register_student(self, student: Student):
self.students.append(student)
@dataclass
class Student:
def __init__(self, id: str, name: str, tribe: Tribe) -> None:
self.id = id
self.name = name
self.tribe = tribe
self.tribe.register_student(self)
def __eq__(self, other: object) -> bool:
if isinstance(other, Student):
return self.id == other.id
return False

View File

@ -0,0 +1,9 @@
from backend.model.student import Student, Tribe
def test_tribe_register_student():
tribe = Tribe("foo", "2nd")
student = Student("1", "Bob", tribe)
assert len(tribe.students) == 1
assert tribe.students[0] == student