gitark/taskncoffee
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix typing

Browse Source
This commit is contained in:
IluaAir
2025-08-04 10:59:23 +03:00
parent a6c00e20c3
commit acb3eefcbe
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/repository/users.py
View File

@@ -1,5 +1,6 @@
from sqlalchemy import delete, select, update from sqlalchemy import delete, select, update
from schemas.users import User
from src.models import UsersORM from src.models import UsersORM
from src.repository.base import BaseRepo from src.repository.base import BaseRepo
@@ -7,7 +8,7 @@ from src.repository.base import BaseRepo
class UsersRepo(BaseRepo): class UsersRepo(BaseRepo):
model = UsersORM model = UsersORM
async def get_all_users(self): async def get_all_users(self) -> list[UsersORM]:
query = select(self.model) query = select(self.model)
result = await self.session.execute(query) result = await self.session.execute(query)
models = result.scalars().all() models = result.scalars().all()
@@ -16,7 +17,7 @@ class UsersRepo(BaseRepo):
async def delete_one(self, id: int) -> None: async def delete_one(self, id: int) -> None:
await self.session.execute(delete(self.model).where(self.model.id == id)) 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) 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) result = await self.session.execute(stmt)
model = result.scalar_one() model = result.scalar_one()