from backend.model.student import Student from backend.model.tribe import Tribe from backend.repository.abstract_repository import AbstractRepository from backend.repository.student_sqlite_repository import StudentRepositoryError from backend.repository.tribe_sqlite_repository import TribeRepositoryError class TribeExists(Exception): pass class TribeDoesNotExist(Exception): pass class StudentExists(Exception): pass class StudentDoesExist(Exception): pass def add_tribe(name: str, level: str, tribe_repo: AbstractRepository, conn) -> Tribe: tribe = Tribe(name=name, level=level) try: tribe_repo.add(tribe) except TribeRepositoryError: raise TribeExists(f"The tribe {tribe.name} already exists") conn.commit() return tribe def update_tribe(name: str, level: str, tribe_repo: AbstractRepository, conn) -> Tribe: tribe = Tribe(name=name, level=level) try: tribe_repo.update(name=name, tribe=tribe) except TribeRepositoryError: raise TribeDoesNotExist(f"The tribe {name} does not exists you can't update it") conn.commit() return tribe def delete_tribe(name: str, tribe_repo: AbstractRepository, conn) -> None: try: tribe_repo.delete(name=name) except TribeRepositoryError: raise TribeDoesNotExist(f"The tribe {name} does not exists you can't delete it") conn.commit() def add_student( name: str, tribe: str, student_repo: AbstractRepository, tribe_repo: AbstractRepository, conn, ) -> Student: try: _tribe = tribe_repo.get(tribe) except TribeRepositoryError: raise TribeDoesNotExist( f"The tribe {tribe} does not exists. Can't add a student in it" ) student = Student(name=name, tribe=_tribe) try: student_repo.add(student) except StudentRepositoryError: raise StudentExists(f"The student {student.name} already exists. Can't add it.") conn.commit() return student