Как использовать команду через @?

Question

[kreyzi]

у меня есть код рп команды для сообщества вк. но он работает ток через ответ на сообщение нужного человека.
нужно что бы он так же мог работать через упоминание пользователя через @ тоесть Рп погладить [id513340695|@vredinka_panda] помогите прошу

@bot.on.message(text=["Призми погладить", "Рп погладить", "[id{user_id}|Фурина]"])
async def pat_user(message: Message):
    # Проверяем, было ли упоминание пользователя через @
    mentioned_user_id = None
    mentioned_users = re.findall(r'\[id(\d+)\|.*?\]', message.text)
    if mentioned_users:
        mentioned_user_id = mentioned_users[0]

    # Получаем информацию об отправителе и упомянутом пользователе
    user_id = message.from_id
    user_info = await bot.api.users.get(user_ids=user_id)

    replied_user_id = mentioned_user_id or message.reply_message.from_id
    replied_user_info = await bot.api.users.get(user_ids=replied_user_id)

    user_name = user_info[0].first_name
    replied_user_name = replied_user_info[0].first_name

    try:
        target_name_out = morph.parse(replied_user_name)[0]
        gent = target_name_out.inflect({'accs'})
        replied_user_name_inflected = morph.parse(replied_user_name)[0].inflect({'accs'}).word.capitalize()

        owner_id = -193467820  # ID сообщества, у которого нужно получить фотографии
        album_id = 299924575  # ID альбома, из которого нужно получить фотографии

        response = await api.photos.get(owner_id=owner_id, album_id=album_id, count=100)
        photos = response.items

        random_photo = random.choice(photos)

        photo_attachment = f"photo{owner_id}_{random_photo.id}"

        await message.answer(
            message=f"@id{user_id} ({user_name}) погладил(а) @id{replied_user_id} ({replied_user_name_inflected})",
            attachment=photo_attachment
        )
    except Exception as e:
        print(f"An error occurred: {e}")

Comments

[Евгений Мартынов]

Поместите свой код в тег

[kreyzi]

Евгений Мартынов, как?

[dim5x]

kreyzi,

[Евгений Мартынов]

dim5x, да

[kreyzi]

Евгений Мартынов, исправил

[o5a]

kreyzi, не совсем понятно, в чем именно вопрос? Помочь правильно написать regexp, который бы из текста вида "что-то там @id12345 и так далее" вытащил "12345", я так понимаю это пытались сделать? Если так, то достаточно "@id(\d+)"

[kreyzi]

o5a, да, но и цыфры может скрыто за именим пользователя (такая особность вк)

[kreyzi]

o5a, не сработало. хотя может я не правильно сделал

@bot.on.message(text=["Призми погладить", "Рп погладить", "[id{user_id}|Фурина]"])
async def pat_user(message: Message):
    # Проверяем, было ли упоминание пользователя через @
    mentioned_user_id = None
    mentioned_users = re.findall(r'\[id(\d+)]', message.text)
    if mentioned_users:
        mentioned_user_id = mentioned_users[0]

    # Получаем информацию об отправителе и упомянутом пользователе
    user_id = message.from_id
    user_info = await bot.api.users.get(user_ids=user_id)

    replied_user_id = mentioned_user_id or message.reply_message.from_id
    replied_user_info = await bot.api.users.get(user_ids=replied_user_id)

    user_name = user_info[0].first_name
    replied_user_name = replied_user_info[0].first_name

    try:
        target_name_out = morph.parse(replied_user_name)[0]
        gent = target_name_out.inflect({'accs'})
        replied_user_name_inflected = morph.parse(replied_user_name)[0].inflect({'accs'}).word.capitalize()

        owner_id = -193467820  # ID сообщества, у которого нужно получить фотографии
        album_id = 299924575  # ID альбома, из которого нужно получить фотографии

        response = await api.photos.get(owner_id=owner_id, album_id=album_id, count=100)
        photos = response.items

        random_photo = random.choice(photos)

        photo_attachment = f"photo{owner_id}_{random_photo.id}"

        await message.answer(
            message=f"@id{user_id} ({user_name}) погладил(а) @id{replied_user_id} ({replied_user_name_inflected})",
            attachment=photo_attachment
        )
    except Exception as e:
        print(f"An error occurred: {e}")

[o5a]

kreyzi, зачем вы строку поиска regexp (шаблон) в квадратные скобки заключаете? У них другое назначение. Указанное у вас '\[id(\d+)]' будет искать строку вида ".... [id12345]"
В тексте вопросе же вроде речь шла про "... @id12345". Какого вида текст тогда все-таки использовать планируется?

[kreyzi]

o5a, [id513340695|@vredinka_panda] вот пример использования

[o5a]

kreyzi, как именно это в тексте будет выглядеть? Так в скобках и будут писать что ли? Или все-таки текст будет в таком виде: "Погладить id1235", "Погладить @vredinka_panda"?

[kreyzi]

o5a,


Вот, хотелось бы что бы бот мог воспринимать 2 варианта и где просто id и тот где айди перекрыт именем

[o5a]

kreyzi, Т.е. все-таки в самом тексте квадратных скобок нет. Я не особо в курсе, как в vk отображается "перекрытие именем", если там как раз и идет id12345, то можно так:

mentioned_users = re.findall(r'(?:id|@)(.+?)\b', message.text)
mentioned_users[0]

[kreyzi]

o5a, внес в код. не работает, перекрытие работает так: 'text': 'Рп погладить [id559313892|@valia_frosya] сначала айди, потом каким текстом хочешь перекрыть.

[kreyzi]

kreyzi, это из логов

[o5a]

kreyzi, т.е. получается, что в тексте с упоминанием всегда будет использован idxxx ? Тогда достаточно только его и искать, вообще не важно, кто там в подмене:
mentioned_users = re.findall(r'\[id(\d+)', text)

[o5a]

kreyzi, если все-таки хотите из текста 'Рп погладить [id559313892|@valia_frosya]' доставать именно 'valia_frosya', то так:
mentioned_users = re.findall(r'\|@(.+?)\b', text)

[kreyzi]

o5a, ERROR:vkbottle:name 'text' is not defined
Traceback (most recent call last):
File "C:\Users\samar\AppData\Local\Programs\Python\Python39\lib\site-packages\vkbottle\dispatch\base.py", line 22, in route
await view.handle_event(event, ctx_api, self.state_dispenser)
File "C:\Users\samar\AppData\Local\Programs\Python\Python39\lib\site-packages\vkbottle\dispatch\views\abc\message.py", line 73, in handle_event
handler_response = await handler.handle(message, **context_variables)
File "C:\Users\samar\AppData\Local\Programs\Python\Python39\lib\site-packages\vkbottle\dispatch\handlers\from_func_handler.py", line 32, in handle
return await self.handler(event, **acceptable_context)
File "c:\bot\2.py", line 555, in pat_user
mentioned_users = re.findall(r'\[id(\d+)', text)
NameError: name 'text' is not defined

[o5a]

kreyzi, я думал, поймете сами, что там не просто text, а message.text, я же просто показывал шаблон

mentioned_users = re.findall(r'\|@(.+?)\b', message.text)

[kreyzi]

o5a, ERROR:vkbottle:list index out of range
Traceback (most recent call last):
File "C:\Users\samar\AppData\Local\Programs\Python\Python39\lib\site-packages\vkbottle\dispatch\base.py", line 22, in route
await view.handle_event(event, ctx_api, self.state_dispenser)
File "C:\Users\samar\AppData\Local\Programs\Python\Python39\lib\site-packages\vkbottle\dispatch\views\abc\message.py", line 73, in handle_event
handler_response = await handler.handle(message, **context_variables)
File "C:\Users\samar\AppData\Local\Programs\Python\Python39\lib\site-packages\vkbottle\dispatch\handlers\from_func_handler.py", line 32, in handle
return await self.handler(event, **acceptable_context)
File "c:\bot\2.py", line 556, in pat_user
mentioned_users[0]
IndexError: list index out of range

[kreyzi]

o5a, извините за неудобство, но сам я правда не разберусь

[o5a]

kreyzi, проверку наличия найденного значения пропускать тоже не надо, которую до этого делали. Без нее конечно будет в ошибку уходить, если присылать сообщение, в котором вообще нет такого кода

mentioned_users = re.findall(r'\|@(.+?)\b', message.text)
if mentioned_users:
    mentioned_user_id = mentioned_users[0]

Если же эту ошибку выдает даже на "правильном" сообщении с указанием пользователя, то приведите полный текст того сообщения, в котором пользователь есть, но не находит.

[kreyzi]

o5a, ошибка пропала. но теперь бот просто молчит

[kreyzi]

o5a,

@bot.on.message(text=["Призми погладить", "Рп погладить"]
async def pat_user(message: Message):
    # Проверяем, было ли упоминание пользователя через @
    mentioned_users = re.findall(r'\|@(.+?)\b', message.text)
    if mentioned_users:
        mentioned_user_id = mentioned_users[0]

    # Получаем информацию об отправителе и упомянутом пользователе
    user_id = message.from_id
    user_info = await bot.api.users.get(user_ids=user_id)

    replied_user_id = mentioned_user_id or message.reply_message.from_id
    replied_user_info = await bot.api.users.get(user_ids=replied_user_id)

    user_name = user_info[0].first_name
    replied_user_name = replied_user_info[0].first_name

    try:
        target_name_out = morph.parse(replied_user_name)[0]
        gent = target_name_out.inflect({'accs'})
        replied_user_name_inflected = morph.parse(replied_user_name)[0].inflect({'accs'}).word.capitalize()

        owner_id = -193467820  # ID сообщества, у которого нужно получить фотографии
        album_id = 299924575  # ID альбома, из которого нужно получить фотографии

        response = await api.photos.get(owner_id=owner_id, album_id=album_id, count=100)
        photos = response.items

        random_photo = random.choice(photos)

        photo_attachment = f"photo{owner_id}_{random_photo.id}"

        await message.answer(
            message=f"@id{user_id} ({user_name}) погладил(а) @id{replied_user_id} ({replied_user_name_inflected})",
            attachment=photo_attachment
        )
    except Exception as e:
        print(f"An error occurred: {e}")

[o5a]

kreyzi, Да, что-то этим упоминанием больше запутали. Т.е. помимо обычных команд просто хотите чтобы срабатывало и от формата с упоминанием. Само упоминание насколько я понимаю в данном случае не важно для обработки, все равно важен только id. В таком случае насколько я понимаю, можно без regexp обойтись, просто методами самого vkbottle.
Вместо всего этого

@bot.on.message(text=["Призми погладить", "Рп погладить", "[id{user_id}|Фурина]"])
async def pat_user(message: Message):
    # Проверяем, было ли упоминание пользователя через @
    mentioned_user_id = None
    mentioned_users = re.findall(r'\[id(\d+)\|.*?\]', message.text)
    if mentioned_users:
        mentioned_user_id = mentioned_users[0]

Просто напрямую прописать этот формат в хэндлере, и затем использовать найденные им значения в функции

@bot.on.message(text=["Рп погладить [id<mentioned_user_id>|<mentioned_user_name>]", "Призми погладить", "Рп погладить"])
async def pat_user(message: Message, mentioned_user_id=None, mentioned_user_name=None):
    mentioned_user_id ... # уже можно использовать напрямую
    # можно и само имя использовать, если вдруг пригодится mentioned_user_name

Так должно работать для "Рп погладить [idxxx|кто-тотам]", для второго аналогично сделать.

[kreyzi]

o5a, сработало! напиши в ответ что бы я его отметил

Answer 1

[o5a]

Можно без regexp обойтись, просто методами самого vkbottle.
Вместо всего этого

@bot.on.message(text=["Призми погладить", "Рп погладить", "[id{user_id}|Фурина]"])
async def pat_user(message: Message):
    # Проверяем, было ли упоминание пользователя через @
    mentioned_user_id = None
    mentioned_users = re.findall(r'\[id(\d+)\|.*?\]', message.text)
    if mentioned_users:
        mentioned_user_id = mentioned_users[0]

Просто напрямую прописать этот формат в хэндлере, и затем использовать найденные им значения в функции

@bot.on.message(text=["Рп погладить [id<mentioned_user_id>|<mentioned_user_name>]", "Призми погладить", "Рп погладить"])
async def pat_user(message: Message, mentioned_user_id=None, mentioned_user_name=None):
    mentioned_user_id ... # уже можно использовать напрямую
    # можно и само имя использовать, если вдруг пригодится mentioned_user_name

Так должно работать для "Рп погладить [idxxx|кто-тотам]", для второго аналогично сделать.