Как организовать функцию мута в Telegram чате через упомнание юзеров в Aiogram 3?

Question

[Аркадий Гальченко]

Это мог бы быть неплохой код:

Код предпологаемой функции

import asyncio
from aiogram import Router, types, F
from aiogram.filters import Command
from datetime import datetime, timedelta

router = Router()
router.message.filter(F.chat.id == '1234567890')

MUTE_P = {
    'can_send_messages': False,
    'can_send_documents': False,
    'can_send_photos': False,
    'can_send_videos': False,
    'can_send_video_notes': False,
    'can_send_voice_notes': False,
    'can_send_polls': False,
    'can_send_other_messages': False,
    'can_add_web_page_preview': False,
}


@router.message(Command("mute"))
async def c_mute(message: types.Message):
    if message.entities:
        # Собираем юзеров для бана
        users_to_mute = [e.user for e in message.entities if e.type == 'text_mention']
        
        # Если список не пуст
        if users_to_mute:
            arg = message.text.rsplit(maxsplit=1)[-1]  # в конце сообщения время мута
            mute_min = int(arg) if arg.isdigit() else 120  # определяем есть ли цифра
            mute_time = datetime.now() + timedelta(minutes=mute_min)  # определяем время

            targets = []

            # Проходим циклом по списку и даём ограничения
            for usr in users_to_mute:
                await message.chat.restrict(usr.id, types.ChatPermissions(**MUTE_P), until_date=mute_time)
                targets.append(f'<b>{usr.full_name}</b> id:<code>{usr.id}</code>')

            # Собираем и отправляем отчёт в чат
            targets_str = '\n'.join(targets)
            text = f" Юзеры:\n{targets_str}\nВ муте на <b>{mute_min}</b> минут"
            await message.answer(text, parse_mode='html')
        # Если список пуст
        else:
            tmp = await message.reply(' Нет упоминания юзеров')
            await asyncio.sleep(5)
            await tmp.delete()

Но, к сожалению, при упоминании пользователей в чате через собачку @, то имеющие юзернейм пользователи упоминаются не текстовой ссылкой, а так @_username . Это уже другой тип message_entities - mention , в отличие от text_mention не содержит объекта user внутри. Сначала думал выйти из положения тем, чтобы вытащить юзернейм из сообщения, но бот Aiogram (и скорее всего любой бот) работает только с user_id. Метот get_chat_member() тоже работает только с user_id.

Подскажите какие есть варианты выйти из ситуации, чтобы обработать юзернеймы @_username из сообщения?

Comments

[Аркадий Гальченко]

Мне приходит на ум только вытаскивать id по username из базы данных. Но если человек зашёл в чат без никнейма а уже позже его установил, и после этого ничего не писал - такой метод не сработает.

Answer 1

[Everything_is_bad]

Варианты:
1 веди базу данных всех юзеров чата, тогда сопоставишь username и id
2. сделай команду mute как ответ на сообщения юзера, которого хочешь замутить, тогда у тебя будет ссылка на это сообщение, откуда ты достанешь id юзера

Comments

[Аркадий Гальченко]

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

По вытягиванию из базы данных есть момент - если юзер вошёл без никнейма, потом установил никнейм и ничего не писал. Так в базе он не обновится. Естественно потом и забанить не выйдет.

[Everything_is_bad]

Аркадий Гальченко, так надо просто следить за актуальными данными юзера, если у него появился/изменился обновить базу

[Аркадий Гальченко]

Everything_is_bad, как следить?) подробнее)

[Everything_is_bad]

Аркадий Гальченко, короче через bot api такое не сделать, но такое есть в клиентском протоколе, т.е надо использовать pyrogram и подобные

[Everything_is_bad]

Аркадий Гальченко,

как следить?)

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

[Аркадий Гальченко]

Everything_is_bad, У меня так и реализовано с сообщениями. Но знаешь есть ботофермы, которые просто эмодзи спамят на сообщения. Эмодзи на чужих сообщениях через bot API вряд ли приходят в апдейтах :D