mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-11-24 19:27:36 +00:00
93 lines
4.0 KiB
Python
93 lines
4.0 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from pyrogram.enums import ChatType, ParseMode
|
|
from pyrogram.errors import UserAdminInvalid, BadRequest, ChatAdminRequired
|
|
from pyrogram.types import ChatPermissions
|
|
|
|
from pagermaid.listener import listener
|
|
from pagermaid.enums import Client, Message
|
|
|
|
|
|
@listener(
|
|
command="portball",
|
|
is_plugin=True,
|
|
outgoing=True,
|
|
need_admin=True,
|
|
description="回复你要临时禁言的人的消息来实现XX秒的禁言",
|
|
parameters="[理由]|<时间/秒>",
|
|
)
|
|
async def portball(bot: Client, message: Message):
|
|
if message.chat.type in (ChatType.GROUP, ChatType.SUPERGROUP):
|
|
reply_to_message = message.reply_to_message
|
|
if reply_to_message is not None:
|
|
from_user = reply_to_message.from_user
|
|
chat = reply_to_message.chat
|
|
if from_user is None:
|
|
return
|
|
if from_user.is_self:
|
|
edit_message: Message = await message.edit_text("无法禁言自己。")
|
|
await edit_message.delay_delete()
|
|
return
|
|
seconds: int = -1
|
|
reason: str = ""
|
|
if len(message.parameter) == 1:
|
|
try:
|
|
seconds = int(message.parameter[0])
|
|
except ValueError:
|
|
edit_message: Message = await message.edit_text("出错了呜呜呜 ~ 无效的参数。")
|
|
await edit_message.delay_delete()
|
|
return
|
|
elif len(message.parameter) == 2:
|
|
try:
|
|
reason = message.parameter[0]
|
|
seconds = int(message.parameter[1])
|
|
except ValueError:
|
|
edit_message: Message = await message.edit_text("出错了呜呜呜 ~ 无效的参数。")
|
|
await edit_message.delay_delete()
|
|
return
|
|
else:
|
|
edit_message: Message = await message.edit_text("出错了呜呜呜 ~ 无效的参数。")
|
|
await edit_message.delay_delete()
|
|
return
|
|
if seconds < 60:
|
|
edit_message: Message = await message.edit_text("诶呀不要小于60秒啦")
|
|
await edit_message.delay_delete()
|
|
return
|
|
try:
|
|
await bot.restrict_chat_member(
|
|
chat.id,
|
|
from_user.id,
|
|
ChatPermissions(),
|
|
datetime.now() + timedelta(seconds=seconds),
|
|
)
|
|
except (UserAdminInvalid, ChatAdminRequired):
|
|
await bot.send_message(chat.id, "错误：该操作需要管理员权限")
|
|
await message.delay_delete()
|
|
return
|
|
except BadRequest:
|
|
await message.edit_text("出错了呜呜呜 ~ 执行封禁时出错")
|
|
await message.delay_delete()
|
|
return
|
|
else:
|
|
if from_user.first_name is None:
|
|
full_name = f"{from_user.id}"
|
|
elif from_user.last_name is None:
|
|
full_name = from_user.first_name
|
|
else:
|
|
full_name = f"{from_user.first_name} {from_user.last_name}"
|
|
text = f"[{full_name}](tg://user?id={from_user.id}) "
|
|
if reason != "":
|
|
text += f"由于 {reason} "
|
|
text += f"被塞了{seconds}秒口球.\n"
|
|
text += "到期自动拔出,无后遗症."
|
|
await bot.send_message(chat.id, text)
|
|
await message.safe_delete()
|
|
else:
|
|
edit_message: Message = await message.edit_text("你好蠢诶，都没有回复人，我哪知道你要搞谁的事情……")
|
|
await edit_message.delay_delete()
|
|
await message.delay_delete()
|
|
else:
|
|
edit_message: Message = await message.edit_text("你好蠢诶，又不是群组，怎么禁言啦！")
|
|
await edit_message.delay_delete()
|
|
await message.delay_delete()
|