unban 解除封禁一位用户

list.json

@@ -449,6 +449,16 @@
             "supported": true,
             "des-short": "查询天气。",
             "des": "查询天气。指令：,weather_lite"
+        },
+        {
+            "name": "unban",
+            "version": "1.0",
+            "section": "chat",
+            "maintainer": "xtaodada",
+            "size": "2.5 kb",
+            "supported": true,
+            "des-short": "解除封禁一位用户。",
+            "des": "解除封禁一位用户。指令：,unban"
         }
     ]
 }

unban/main.py

@@ -0,0 +1,72 @@
+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("已解封此用户。")