sb 优化代码

sb/DES.md

@@ -1,2 +1,3 @@
 在自己拥有管理员权限的共同群组中封禁一位用户。
-指令：,sb
+
+指令： `,sb` 回复用户

sb/main.py

@@ -1,3 +1,4 @@
+import contextlib
 from asyncio import sleep
 
 from pyrogram.enums import ChatType
@@ -36,20 +37,18 @@ async def delete_all_messages(chat: Chat, uid):
 
 async def check_uid(chat: Chat, uid: str):
     channel = False
-    try:
+    with contextlib.suppress(ValueError):
         uid = int(uid)
         if uid < 0:
             channel = True
+    try:
         await bot.get_chat_member(chat.id, uid)
-    except ValueError:
+    except ChatAdminRequired:
-        try:
+        with contextlib.suppress(PeerIdInvalid):
             chat = await bot.get_chat(uid)
             uid = chat.id
             if chat.type in [ChatType.CHANNEL, ChatType.SUPERGROUP, ChatType.GROUP]:
                 channel = True
-        except PeerIdInvalid:
-            member = await bot.get_chat_member(chat.id, uid)
-            uid = member.user.id
     return uid, channel
 
 
@@ -70,7 +69,7 @@ async def get_uid(chat: Chat, message: Message):
         uid, channel = await check_uid(chat, message.parameter[0])
         delete_all = False
     elif len(message.parameter) == 1:
-        uid, channel = await check_uid(chat, message.arguments)
+        uid, channel = await check_uid(chat, message.parameter[0])
     return uid, channel, delete_all, sender
 
 
@@ -85,16 +84,17 @@ async def super_ban(message: Message):
     chat = message.chat
     try:
         uid, channel, delete_all, sender = await get_uid(chat, message)
+        if not uid:
+            raise ValueError
+        if channel and uid == chat.id:
+            raise ValueError
     except (ValueError, PeerIdInvalid, UsernameInvalid, FloodWait):
         await message.edit(lang("arg_error"))
-        return add_delete_message_job(message, 10)
+        return add_delete_message_job(message, 30)
-    if not uid:
+    except Exception as e:
-        await message.edit(lang("arg_error"))
+        await message.edit(f"出现错误：{e}")
-        return add_delete_message_job(message, 10)
+        return add_delete_message_job(message, 30)
     if channel:
-        if uid == chat.id:
-            await message.edit(lang("arg_error"))
-            return add_delete_message_job(message, 10)
         try:
             await ban_one(chat, uid)
         except ChatAdminRequired:
@@ -131,8 +131,7 @@ async def super_ban(message: Message):
         except Exception:  # noqa
             pass
     if not sender:
-        member = await bot.get_chat_member(chat.id, uid)
+        sender = await bot.get_users(uid)
-        sender = member.user
     if count == 0:
         text = f'{lang("sb_no")} {sender.mention}'
     else: