from sqlalchemy import update

from src.models import UsersORM
from src.repository.base import BaseRepo


class UsersRepo(BaseRepo):
    model: type[UsersORM] = UsersORM

    async def update_one(self, id: int, data: dict) -> UsersORM:
        stmt = (
            update(self.model)
            .where(self.model.id == id)
            .values(data)
            .returning(self.model)
        )
        result = await self.session.execute(stmt)
        model = result.scalar_one()
        return model