33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
from pyrogram.types import ChatPermissions, User
|
|
|
|
from app import BOT, Message, bot
|
|
|
|
|
|
@bot.add_cmd(cmd=["mute", "unmute"])
|
|
async def mute_or_unmute(bot: BOT, message: Message):
|
|
user, reason = await message.extract_user_n_reason()
|
|
if not isinstance(user, User):
|
|
await message.reply(user, del_in=10)
|
|
return
|
|
perms = message.chat.permissions
|
|
if message.cmd == "mute":
|
|
perms = ChatPermissions(
|
|
can_send_messages=False,
|
|
can_pin_messages=False,
|
|
can_invite_users=False,
|
|
can_change_info=False,
|
|
can_send_media_messages=False,
|
|
can_send_polls=False,
|
|
can_send_other_messages=False,
|
|
can_add_web_page_previews=False,
|
|
)
|
|
try:
|
|
await bot.restrict_chat_member(
|
|
chat_id=message.chat.id, user_id=user.id, permissions=perms
|
|
)
|
|
await message.reply(
|
|
text=f"{message.cmd.capitalize()}d: {user.mention}\nReason: {reason}."
|
|
)
|
|
except Exception as e:
|
|
await message.reply(text=e, del_in=10)
|