Sobes.tech
Middle
45

Izpildiet SQLAlchemy modeļa increment metodi: palieliniet lietotāja bilanci par diff ar maksimālā bilances pārbaudi. Atgrieziet True, ja bilance ir mainījusies, citādi False.

Uzņēmumi, kur jautāja
Сбер.clouds

Atbilde no AI

sobes.tech AI

python from sqlalchemy import Column, Integer from sqlalchemy.orm import Session

class User: id = Column(Integer, primary_key=True) balance = Column(Integer)

def increment(self, session: Session, diff: int, max_balance: int) -> bool:
    new_balance = self.balance + diff
    if new_balance > max_balance:
        return False
    self.balance = new_balance
    session.add(self)
    session.commit()
    return True

В этом методе мы сначала вычисляем потенциальный новый баланс. Если он превышает максимальный, возвращаем False. Иначе обновляем баланс, сохраняем изменения в сессии и возвращаем True.