From acb3eefcbec6724adf7bb1484a2b8687a185b972 Mon Sep 17 00:00:00 2001 From: IluaAir Date: Mon, 4 Aug 2025 10:59:23 +0300 Subject: [PATCH] fix typing --- src/repository/users.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/repository/users.py b/src/repository/users.py index 3ada358..def44c9 100644 --- a/src/repository/users.py +++ b/src/repository/users.py @@ -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()