diff --git a/app/telegram/database/requests.py b/app/telegram/database/requests.py index 00b02f7..8e30589 100644 --- a/app/telegram/database/requests.py +++ b/app/telegram/database/requests.py @@ -390,3 +390,15 @@ async def update_user_timer(tg_id, minutes: int): except Exception as e: logging.error(f"Ошибка обновления таймера пользователя {tg_id}: {e}") + +async def get_martingale_step(tg_id): + async with async_session() as session: + result = await session.execute(select(UMS).where(UMS.tg_id == tg_id)) + user_settings = result.scalars().first() + return user_settings.martingale_step + +async def update_martingale_step(tg_id, step): + async with async_session() as session: + await session.execute(update(UMS).where(UMS.tg_id == tg_id).values(martingale_step = step)) + + await session.commit() \ No newline at end of file