Как в aiogram сделать ограниченное число ввода определенной команды?

Question

[Алексей Мартьянов]

Пишу бота на aiogram с привязкой к БД PostgreSQL.

Суть вопроса:

В боте есть 2 команды: ("/start", "/menu")
При нажатии на /start пользователь добавляется в БД PostgreSQL. (Таблица имеет вид: ID, full_name, user_name, telegram_id, quantity). quantity = 10 по дефолту.
При нажатии на /menu - выводится обычный текст.
Как сделать так, чтобы при нажатии на /test - quantity уменьшался каждый раз на 1? Т.е. 9,8,7... А когда quantity становился 0, то выводилось сообщение о том, что лимит исчерпан.

Answer 1

[Vindicar]

В обработчике команды /test делать SELECT quantity для отправителя.
Если ничего не найдено, сообщить о необходимости регистрации через /start.
Если вернулось положительное значение, выполнить запрос вида UPDATE, а потом сделать что требуется.
Если вернулся ноль, сообщить об исчерпании лимита.
В чём конкретно проблемы?

Comments

[Алексей Мартьянов]

Проблема в том, что, ошибки не выдает, но и не отнимает.





Вот как я это реализовал.

[Vindicar]

1. не кидай код скриншотами, есть же кнопка </>
2. Не вполне ясно, зачем тебе передавать quantity в запрос? Ведь можно выполнить UPDATE users SET quantity = quantity - 1 WHERE telegram_id = $1
3. Ну и как следствие предыдущего... у тебя в определении minus_quantity в параметрах идёт сначала quantity, потом id, а в вызове - наоборот.

[Алексей Мартьянов]

Vindicar,
1. Прошу прощения за скрины.

2.

async def minus_quantity(self, telegram_id):
        sql = "UPDATE users SET quantity=quantity-1 WHERE telegram_id=$1"
        return await self.execute(sql, telegram_id)

3.

@dp.message_handler(Command("test"))
async def bot_echo(message: types.Message):
    await message.answer("текст")

    telegram_id = message.from_user.id
    quantity = await db.quantity(telegram_id)
    print(quantity)

    await db.minus_quantity(telegram_id)
    await message.answer(f"Число попыток: {quantity}")
    print(quantity)

[Алексей Мартьянов]

Решено:

async def minus_quantity(self, telegram_id):
        sql = "UPDATE users SET quantity=quantity-1 WHERE telegram_id=$1"
        return await self.execute(sql, telegram_id)