Как сделать временный мут Telegram Aiogram?

Question

[Alexndrev]

В общем-то делаю бота-модератора для телеги. Появился вопрос, насчёт команды /mute. Просто мут выдаёт, а вот с временным траблы
Вот часть код команды:

# Mute
@dp.message_handler(commands = "mute")
async def mute(message: types.Message):
    if check_sub_channel(await bot.get_chat_member(chat_id = cfg.channel_id, user_id = message.from_user.id)):
        if message.from_user.id == cfg.admin1 or cfg.admin2 or cfg.admin3:
            await message.bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, time.time()+600)
            await message.answer(f'*Пользователю запрещено право писать. Решение было принято @{message.from_user.username}*.\n_Функция размута ещё тестируется_')
        else:
            await message.reply("Ей, ей, ей. Ты походу что-то попутал. Тебе нельзя использовать эту команду ")
    else:
        await bot.send_message(message.from_user.id, cfg.not_sub_message)

Answer 1

[Alexndrev]

Ответ такой

@dp.message_handler(commands=['мут', 'mute'], commands_prefix='./', is_chat_admin=True)
async def mute(message):
      name1 = message.from_user.get_mention(as_html=True)
      if not message.reply_to_message:
         await message.reply("Эта команда должна быть ответом на сообщение!")
         return
      try:
         muteint = int(message.text.split()[1])
         mutetype = message.text.split()[2]
         comment = " ".join(message.text.split()[3:])
      except IndexError:
         await message.reply('Не хватает аргументов!\nПример:\n`/мут 1 ч причина`')
         return
      if mutetype == "ч" or mutetype == "часов" or mutetype == "час":
         dt = datetime.now() + timedelta(hours=muteint)
         timestamp = dt.timestamp()
         await bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, types.ChatPermissions(False), until_date = timestamp)
         await message.reply(f' | <b>Решение было принято:</b> {name1}\n | <b>Нарушитель:</b> <a href="tg://user?id={message.reply_to_message.from_user.id}">{message.reply_to_message.from_user.first_name}</a>\n⏰ | <b>Срок наказания:</b> {muteint} {mutetype}\n | <b>Причина:</b> {comment}',  parse_mode='html')
      elif mutetype == "м" or mutetype == "минут" or mutetype == "минуты":
         dt = datetime.now() + timedelta(minutes=muteint)
         timestamp = dt.timestamp()
         await bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, types.ChatPermissions(False), until_date = timestamp)
         await message.reply(f' | <b>Решение было принято:</b> {name1}\n | <b>Нарушитель:</b> <a href="tg://user?id={message.reply_to_message.from_user.id}">{message.reply_to_message.from_user.first_name}</a>\n⏰ | <b>Срок наказания:</b> {muteint} {mutetype}\n | <b>Причина:</b> {comment}',  parse_mode='html')
      elif mutetype == "д" or mutetype == "дней" or mutetype == "день":
         dt = datetime.now() + timedelta(days=muteint)
         timestamp = dt.timestamp()
         await bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, types.ChatPermissions(False), until_date = timestamp)
         await message.reply(f' | <b>Решение было принято:</b> {name1}\n | <b>Нарушитель:</b> <a href="tg://user?id={message.reply_to_message.from_user.id}">{message.reply_to_message.from_user.first_name}</a>\n⏰ | <b>Срок наказания:</b> {muteint} {mutetype}\n | <b>Причина:</b> {comment}',  parse_mode='html')

Comments

[Artemider]

Немного оптимизировал, его можно пилить ещё, он од части на украинском)

@dp.message_handler(commands=['mute'], is_chat_admin=True)
@dp.message_handler(lambda message: message.text.startswith("мут"), is_chat_admin=True)
async def mute(message):
      name1 = message.from_user.get_mention(as_html=True)
      if not message.reply_to_message:
         await message.reply("Команда має бути відповідю!")
         return
      try:
         muteint = int(message.text.split()[1])
         mutetype = message.text.split()[2]
         comment = " ".join(message.text.split()[3:])
      except:
         await message.reply('Не правильно написано!\nПриклад:\n`/мут 1 ч причина`')
         return
      if mutetype == "г" or mutetype == "годин" or mutetype == "h":

         await bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, until_date=int(time()) + muteint*3600)
         await message.answer(f' | <b>Решение было принято:</b> {name1}\n | <b>Нарушитель:</b> <a href="tg://user?id={message.reply_to_message.from_user.id}">{message.reply_to_message.from_user.first_name}</a>\n⏰ | <b>Срок наказания:</b> {muteint} {mutetype}\n | <b>Причина:</b> {comment}',  parse_mode='html')
      elif mutetype == "хв" or mutetype == "хвилин" or mutetype == "m":

         await bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, until_date=int(time()) + muteint*60)
         await message.answer(f' | <b>Решение было принято:</b> {name1}\n | <b>Нарушитель:</b> <a href="tg://user?id={message.reply_to_message.from_user.id}">{message.reply_to_message.from_user.first_name}</a>\n⏰ | <b>Срок наказания:</b> {muteint} {mutetype}\n | <b>Причина:</b> {comment}',  parse_mode='html')
      elif mutetype == "д" or mutetype == "дней" or mutetype == "день":

         await bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, until_date=int(time()) + muteint*86400)
         await message.answer(f' | <b>Решение было принято:</b> {name1}\n | <b>Нарушитель:</b> <a href="tg://user?id={message.reply_to_message.from_user.id}">{message.reply_to_message.from_user.first_name}</a>\n⏰ | <b>Срок наказания:</b> {muteint} {mutetype}\n | <b>Причина:</b> {comment}',  parse_mode='html')

[Alexndrev]

Artemider, Україна понад усе, друже)
дякую за оновлення!

[qwwwwwty]

не подскажите, какие библиотеки вы использовали для этого кода?

[Alexndrev]

qwwwwwty, aiogram

сделал версию лучше. если актуально- пиши

[qwwwwwty]

Alexndrev, очень хотелось бы посмотреть на вашу новую версию

[Константин]

Доброго дня,що потрбно import ?

в мене такий код та не працюе мут

import config
import logging

from aiogram import Bot, Dispatcher, executor, types

logging.basicConfig(level=logging.INFO)
#---------------------------------
bot = Bot(token=config.token)
dp = Dispatcher(bot)
#-------------------------------------

@dp.message_handler(commands=['mute'], is_chat_admin=True)
@dp.message_handler(lambda message: message.text.startswith("мут"), is_chat_admin=True)
async def mute(message):
name1 = message.from_user.get_mention(as_html=True)
if not message.reply_to_message:
await message.reply("Команда має бути відповідю!")
return
try:
muteint = int(message.text.split()[1])
mutetype = message.text.split()[2]
comment = " ".join(message.text.split()[3:])
except:
await message.reply('Не правильно написано!\nПриклад:\n`/мут 1 ч причина`')
return
if mutetype == "г" or mutetype == "годин" or mutetype == "h":

await bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, until_date=int(time()) + muteint*3600)
await message.answer(f' | Решение было принято: {name1}\n | Нарушитель: {message.reply_to_message.from_user.first_name}\n⏰ | Срок наказания: {muteint} {mutetype}\n | Причина: {comment}', parse_mode='html')
elif mutetype == "хв" or mutetype == "хвилин" or mutetype == "m":

await bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, until_date=int(time()) + muteint*60)
await message.answer(f' | Решение было принято: {name1}\n | Нарушитель: {message.reply_to_message.from_user.first_name}\n⏰ | Срок наказания: {muteint} {mutetype}\n | Причина: {comment}', parse_mode='html')
elif mutetype == "д" or mutetype == "дней" or mutetype == "день":

await bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, until_date=int(time()) + muteint*86400)
await message.answer(f' | Решение было принято: {name1}\n | Нарушитель: {message.reply_to_message.from_user.first_name}\n⏰ | Срок наказания: {muteint} {mutetype}\n | Причина: {comment}', parse_mode='html')

@dp.message_handler(content_types=["new_chat_members"])
async def on_user_joined(message: types.Message):
await message.delete()

#фильтры--------------------
@dp.message_handler()
async def filter_messages(message: types.Message):
if "дик" in message.text:
await message.delete()
if "/dick" in message.text:
await message.delete()
if "Расти страну" in message.text:
await message.delete()
if "/iq" in message.text:
await message.delete()
if "mq" in message.text:
await message.delete()



if __name__ == "__main__":
executor.start_polling(dp, skip_updates=True)

Видае помилку await bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, until_date=int(time()) + muteint*3600)
NameError: name 'time' is not defined

[motionless123]

Константин, Вам потрібно імпортувати бібліотеку time:
import time

Answer 2

[Elvis]

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

Comments

[Alexndrev]

без бд можно реализовать? чисто через аргументы телеги

[Elvis]

Alexndrev, можно.
Если посмотреть документацию, то в методе restrict_chat_member есть параметры: chat_id, user_id, permissions и until_date. Последний как раз и отвечает когда будет мут снят.
Там же в документации написано что параметр должен быть datetime. и есть ссылка на офф документацию python.
Идем и смотрим туда.
Чтобы указать текущее время нужно использовать datetime.datetime.now()
Получается нам нужно взять время которое сейчас и прибавить к нему, как я понял 10 минут, или 600 секунд.
Гуглим как прибавить время к текущему и получаем что нужно взять текущую дату и прибавить дельту, то есть:

datetime.datetime.now() + datetime.timedelta(seconds=600)

Теперь соединяем всё это в один вызов:

await message.bot.restrict_chat_member(message.chat.id, message.reply_to_message.from_user.id, datetime.datetime.now() + datetime.timedelta(seconds=600))

P.S.: только не забываем сделать в начале файла import datetime если еще не сделано.

[Elvis]

Извиняюсь, забыл указать что еще нужен permissions
Тут нужно указать какие права ты хочешь выставить пользователю.

[Elvis]

Alexndrev, ой... у вас же Aiogram... не заметил. ссылки немного другие будут
restrict_chat_member
permissions

[Константин]

какие библиотеки нужно import?