add get filtered
This commit is contained in:
@@ -1,3 +1,5 @@
|
|||||||
|
from typing import Any
|
||||||
|
|
||||||
from pydantic import BaseModel
|
from pydantic import BaseModel
|
||||||
from sqlalchemy import delete, insert, select
|
from sqlalchemy import delete, insert, select
|
||||||
|
|
||||||
@@ -10,6 +12,12 @@ class BaseRepo:
|
|||||||
def __init__(self, session):
|
def __init__(self, session):
|
||||||
self.session = 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):
|
async def create_one(self, data: BaseModel):
|
||||||
statement = insert(self.model).values(data.model_dump()).returning(self.model)
|
statement = insert(self.model).values(data.model_dump()).returning(self.model)
|
||||||
result = await self.session.execute(statement)
|
result = await self.session.execute(statement)
|
||||||
@@ -22,11 +30,9 @@ class BaseRepo:
|
|||||||
model = result.scalars().one_or_none()
|
model = result.scalars().one_or_none()
|
||||||
return model
|
return model
|
||||||
|
|
||||||
async def get_all(self) -> list[model]:
|
async def get_all(self, *args, **kwargs) -> list[model]:
|
||||||
query = select(self.model)
|
result = await self.get_filtered()
|
||||||
result = await self.session.execute(query)
|
return result
|
||||||
models = result.scalars().all()
|
|
||||||
return models
|
|
||||||
|
|
||||||
async def delete_one(self, **filter_by) -> None:
|
async def delete_one(self, **filter_by) -> None:
|
||||||
await self.session.execute(delete(self.model).filter_by(**filter_by))
|
await self.session.execute(delete(self.model).filter_by(**filter_by))
|
||||||
|
|||||||
Reference in New Issue
Block a user