mirror of
https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git
synced 2024-11-22 08:27:39 +00:00
81 lines
2.5 KiB
Python
81 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.enums import Client, 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(client: Client, 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 client.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("已解封此用户。")
|