fix typing

This commit is contained in:
IluaAir
2025-08-04 10:59:23 +03:00
parent a6c00e20c3
commit acb3eefcbe

View File

@@ -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()