diff --git a/src/repository/base.py b/src/repository/base.py index 6ec7c9c..24ce697 100644 --- a/src/repository/base.py +++ b/src/repository/base.py @@ -1,5 +1,5 @@ from pydantic import BaseModel -from sqlalchemy import insert, select +from sqlalchemy import delete, insert, select from src.core.database import Base @@ -21,3 +21,6 @@ class BaseRepo: result = await self.session.execute(query) model = result.scalars().one_or_none() return model + + async def delete_one(self, **filter_by) -> None: + await self.session.execute(delete(self.model).filter_by(**filter_by)) diff --git a/src/repository/users.py b/src/repository/users.py index 4f11a90..155951b 100644 --- a/src/repository/users.py +++ b/src/repository/users.py @@ -1,4 +1,4 @@ -from sqlalchemy import delete, select, update +from sqlalchemy import select, update from src.models import UsersORM from src.repository.base import BaseRepo @@ -13,9 +13,6 @@ class UsersRepo(BaseRepo): models = result.scalars().all() return models - async def delete_one(self, id: int) -> None: - await self.session.execute(delete(self.model).where(self.model.id == id)) - async def update_one(self, id: int, data: dict) -> UsersORM: stmt = ( update(self.model)