Feat: add TribeSQLAlchemyRepository
This commit is contained in:
23
backend/repository/abstract_repository.py
Normal file
23
backend/repository/abstract_repository.py
Normal file
@@ -0,0 +1,23 @@
|
||||
import abc
|
||||
|
||||
|
||||
class AbstractRepository(abc.ABC):
|
||||
@abc.abstractmethod
|
||||
def add(self, element):
|
||||
raise NotImplementedError
|
||||
|
||||
@abc.abstractmethod
|
||||
def update(self, element):
|
||||
raise NotImplementedError
|
||||
|
||||
@abc.abstractmethod
|
||||
def list(self):
|
||||
raise NotImplementedError
|
||||
|
||||
@abc.abstractmethod
|
||||
def get(self, reference):
|
||||
raise NotImplementedError
|
||||
|
||||
@abc.abstractmethod
|
||||
def delete(self, reference):
|
||||
raise NotImplementedError
|
||||
23
backend/repository/tribe_sqlalchemy_repository.py
Normal file
23
backend/repository/tribe_sqlalchemy_repository.py
Normal file
@@ -0,0 +1,23 @@
|
||||
from backend.model.tribe import Tribe
|
||||
from backend.repository.abstract_repository import AbstractRepository
|
||||
|
||||
|
||||
class TribeSQLAlchemyRepository(AbstractRepository):
|
||||
def __init__(self, session) -> None:
|
||||
self.session = session
|
||||
|
||||
def add(self, tribe: Tribe) -> None:
|
||||
self.session.add(tribe)
|
||||
|
||||
def update(self, name: str, tribe: Tribe) -> None:
|
||||
self.session.query(Tribe).filter_by(name=name).update(tribe.to_dict())
|
||||
|
||||
def get(self, name: str) -> Tribe:
|
||||
return self.session.query(Tribe).filter_by(name=name).one()
|
||||
|
||||
def list(self) -> list[Tribe]:
|
||||
return self.session.query(Tribe).all()
|
||||
|
||||
def delete(self, tribe: Tribe) -> None:
|
||||
the_tribe = self.get(tribe.name)
|
||||
self.session.delete(the_tribe)
|
||||
Reference in New Issue
Block a user