mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-11-16 11:02:56 +00:00
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
from pyrogram.enums import ChatMemberStatus
|
|
from pyrogram.errors import ChatAdminRequired, FloodWait, PeerIdInvalid, UsernameInvalid, UserNotParticipant
|
|
from pyrogram.types import Chat
|
|
|
|
from pagermaid import bot
|
|
from pagermaid.listener import listener
|
|
from pagermaid.single_utils import Message
|
|
from pagermaid.utils import lang
|
|
|
|
|
|
async def check_uid(chat: Chat, uid: str):
|
|
member = None
|
|
try:
|
|
uid = int(uid)
|
|
member = await bot.get_chat_member(chat.id, uid)
|
|
except ValueError:
|
|
try:
|
|
chat = await bot.get_chat(uid)
|
|
uid = chat.id
|
|
except PeerIdInvalid:
|
|
member = await bot.get_chat_member(chat.id, uid)
|
|
uid = member.user.id
|
|
if not member:
|
|
member = await bot.get_chat_member(chat.id, uid)
|
|
return uid, member
|
|
|
|
|
|
async def get_uid(chat: Chat, message: Message):
|
|
uid = None
|
|
member = None
|
|
if reply := message.reply_to_message:
|
|
if sender := reply.from_user:
|
|
uid = sender.id
|
|
if sender := reply.sender_chat:
|
|
uid = sender.id
|
|
member = await bot.get_chat_member(chat.id, uid)
|
|
elif len(message.parameter) == 1:
|
|
uid, member = await check_uid(chat, message.arguments)
|
|
return uid, member
|
|
|
|
|
|
@listener(command="unban",
|
|
description="解除封禁一位用户",
|
|
need_admin=True,
|
|
groups_only=True,
|
|
parameters="<reply|id|username>")
|
|
async def unban(message: Message):
|
|
chat = message.chat
|
|
try:
|
|
uid, member = await get_uid(chat, message)
|
|
except (ValueError, PeerIdInvalid, UsernameInvalid, FloodWait):
|
|
return await message.edit(lang("arg_error"))
|
|
except UserNotParticipant:
|
|
return await message.edit("此用户未被限制。")
|
|
if not uid:
|
|
return await message.edit(lang("arg_error"))
|
|
if uid == chat.id:
|
|
return await message.edit(lang("arg_error"))
|
|
try:
|
|
if not member:
|
|
member = await bot.get_chat_member(chat.id, uid)
|
|
if member.status in [ChatMemberStatus.RESTRICTED, ChatMemberStatus.BANNED]:
|
|
await message.bot.unban_chat_member(chat.id, uid)
|
|
else:
|
|
return await message.edit("此用户未被限制。")
|
|
except UserNotParticipant:
|
|
return await message.edit("此用户未被限制。")
|
|
except ChatAdminRequired:
|
|
return await message.edit(lang("sb_no_per"))
|
|
except Exception as e:
|
|
return await message.edit(f"出现错误：{e}")
|
|
await message.edit("已解封此用户。")
|