Как повторить код из команды в обработчике нажатий кнопки?

Question

[szjyakgf]

Если при нажатии кнопки нужно выполнить тоже самое что и в команде. Нужно снова прописывать это или есть что-то проще?

@dp.message_handler(commands=['test'])
async def test(message):
    user_id = message.from_user.id
    keyboard = types.InlineKeyboardMarkup()
    callback_button1 = types.InlineKeyboardButton(text="повторить", callback_data="test1")
    keyboard.add(callback_button1)
    rrr = str(random.randint(1, 51))
    await message.reply(f'Твое число{rrr}', reply_markup=keyboard)

@dp.callback_query_handler(lambda call: True)
async def callback_inline1(call):
    user_id = call.from_user.id
    result = cur.execute("SELECT * FROM users WHERE user_id = ?", (user_id, )).fetchone()
    with suppress(MessageNotModified):
        if call.message:
            if call.data == "test1":
                #тут я должен повторить все как в команде /test

P.S .Код приведен для примера, в моем проекте нужно повторять 200+ строчек

Answer 1

[Vindicar]

А вызвать await test(call.message) что мешает?

Comments

[szjyakgf]

Мне нужно повторять все кроме

await message.reply(f'Твое число{rrr}', reply_markup=keyboard)

заменить на

await bot.edit_message_text(chat_id=call.message.chat.id, message_id=call.message.message_id, text=f'Твое число{rrr}', reply_markup=keyboard)

[Vindicar]

szjyakgf, тогда вынеси общую часть выше в отдельную функцию, и вызывай её и в test(), и в callback(). Нужные данные передавай через параметры, результаты возвращай return'ом. В чем сложность-то возникла?

[szjyakgf]

Vindicar, так?

async def test111():
    user_id = message.from_user.id
    keyboard = types.InlineKeyboardMarkup()
    callback_button1 = types.InlineKeyboardButton(text="повторить", callback_data="test1")
    keyboard.add(callback_button1)
    rrr = str(random.randint(1, 51))
    return rrr, keyboard

@dp.message_handler(commands=['test'])
async def test(message, rrr, keyboard):
    await message.reply(f'Твое число{rrr}', reply_markup=keyboard)

[Vindicar]

szjyakgf, всё ты перепутал.

def test111(message):  # мы не делаем тут await-операций, так что и async не нужен.
    user_id = message.from_user.id
    keyboard = types.InlineKeyboardMarkup()
    callback_button1 = types.InlineKeyboardButton(text="повторить", callback_data="test1")
    keyboard.add(callback_button1)
    rrr = str(random.randint(1, 51))
    return rrr, keyboard

async def test(message):
    rrr, keyboard = test111(message)
    await message.reply(f'Твое число{rrr}', reply_markup=keyboard)