add get filtered

This commit is contained in:
IluaAir
2025-08-10 20:51:25 +03:00
parent d6646716b8
commit 644d5614b9

View File

@@ -1,3 +1,5 @@
from typing import Any
from pydantic import BaseModel
from sqlalchemy import delete, insert, select
@@ -10,6 +12,12 @@ class BaseRepo:
def __init__(self, session):
self.session = session
async def get_filtered(self, *filter, **filter_by) -> list[BaseModel | Any]:
query = select(self.model).filter(*filter).filter_by(**filter_by)
result = await self.session.execute(query)
model = result.scalars().one_or_none()
return model
async def create_one(self, data: BaseModel):
statement = insert(self.model).values(data.model_dump()).returning(self.model)
result = await self.session.execute(statement)
@@ -22,11 +30,9 @@ class BaseRepo:
model = result.scalars().one_or_none()
return model
async def get_all(self) -> list[model]:
query = select(self.model)
result = await self.session.execute(query)
models = result.scalars().all()
return models
async def get_all(self, *args, **kwargs) -> list[model]:
result = await self.get_filtered()
return result
async def delete_one(self, **filter_by) -> None:
await self.session.execute(delete(self.model).filter_by(**filter_by))