Возможно ли уменьшить данный код?

Question

[SergeyYatsenko]

@dp.message_handler(text_startswith=['text', 'Text', 'TEXT'])
async def file1(message: types.Message):
    for ret in cur.execute("SELECT id FROM names WHERE name = 'test1'").fetchall():
        await bot.send_photo(message.from_user.id, ret[0],\
                             reply_markup=InlineKeyboardMarkup(row_width=1).add(InlineKeyboardButton(text='ТЕКСТ НА КНОПКЕ 1', callback_data='test1'),\
                                                                                InlineKeyboardButton(text='ЧТЕКСТ НА КНОПКЕ 2', callback_data='test2'),\
                                                                                InlineKeyboardButton(text='ТЕКСТ НА КНОПКЕ 3', callback_data='test3')))
@dp.callback_query_handler(text_contains="test")
async def test(callback: types.CallbackQuery):
    if callback.data == 'test1':
        for ret in cur.execute("SELECT id, title, score FROM names WHERE name = 'test_1'").fetchall():
            await bot.send_video(callback.from_user.id, ret[0], caption=f'Текст под постом1: {ret[1]}\n Текст под постом2: {ret[-1]}\n\n Текст под постом3 \n\nТекст под постом4, reply_markup=underpost_butt)
    elif callback.data == 'test2':
        for ret in cur.execute("SELECT id, title, score FROM names WHERE name = 'test_2'").fetchall():
            await bot.send_video(callback.from_user.id, ret[0], caption=f'Текст под постом1: {ret[1]}\n Текст под постом2: {ret[-1]}\n\n Текст под постом3 \n\nТекст под постом4, reply_markup=underpost_butt)
    elif callback.data == 'test3':
        for ret in cur.execute("SELECT id, title, score FROM names WHERE name = 'test_3'").fetchall():
            await bot.send_video(callback.from_user.id, ret[0], caption=f'Текст под постом1: {ret[1]}\n Текст под постом2: {ret[-1]}\n\n Текст под постом3 \n\nТекст под постом4, reply_markup=underpost_butt)

Comments

[Roman K]

зачем?

[Roman K]

я спрашиваю, потому что если ты человек "далёкий от программирования", то какая разница, сколько занимает буковок код, если он рабочий?

[Ivan Yakushenko]

Roman Kitaev, для красоты же...

Answer 1

[Vindicar]

if callback.data in ('test1', 'test2', 'test3'):
    for ret in cur.execute("SELECT id, title, score FROM names WHERE name = ?", (callback.data, )):
        await bot.send_video(callback.from_user.id, ret[0], caption=f'Текст под постом1: {ret[1]}\n Текст под постом2: {ret[-1]}\n\n Текст под постом3 \n\nТекст под постом4, reply_markup=underpost_butt)