fix typing
This commit is contained in:
@@ -1,5 +1,6 @@
|
||||
from sqlalchemy import delete, select, update
|
||||
|
||||
from schemas.users import User
|
||||
from src.models import UsersORM
|
||||
from src.repository.base import BaseRepo
|
||||
|
||||
@@ -7,7 +8,7 @@ from src.repository.base import BaseRepo
|
||||
class UsersRepo(BaseRepo):
|
||||
model = UsersORM
|
||||
|
||||
async def get_all_users(self):
|
||||
async def get_all_users(self) -> list[UsersORM]:
|
||||
query = select(self.model)
|
||||
result = await self.session.execute(query)
|
||||
models = result.scalars().all()
|
||||
@@ -16,7 +17,7 @@ class UsersRepo(BaseRepo):
|
||||
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) -> None:
|
||||
async def update_one(self, id: int, data: dict) -> UsersORM:
|
||||
stmt = update(self.model).where(self.model.id == id).values(data.model_dump(exclude_unset=True)).returning(self.model)
|
||||
result = await self.session.execute(stmt)
|
||||
model = result.scalar_one()
|
||||
|
||||
Reference in New Issue
Block a user