feat: reverify

plugins/reverify.py

@@ -0,0 +1,61 @@
+import contextlib
+
+from pyrogram import filters
+from pyrogram.enums import ChatMemberStatus
+
+from datetime import datetime, timedelta
+
+from pyromod.utils.errors import TimeoutConversationError
+from sticker.single_utils import Message, Client
+from sticker import bot, log
+
+MSG = """您好 %s ，您被管理员要求重新验证。
+
+您需要在 30 秒内发送任意一个 贴纸 来完成验证。"""
+
+
+@bot.on_message(filters=filters.group & filters.command("reverify"))
+async def re_verify(client: Client, message: Message):
+    if not message.from_user or not message.reply_to_message:
+        return
+    if not message.reply_to_message.from_user:
+        return
+    chat = message.chat
+    user = message.from_user
+    member = await client.get_chat_member(chat.id, user.id)
+    if member.status not in [ChatMemberStatus.OWNER, ChatMemberStatus.ADMINISTRATOR]:
+        return
+
+    user = message.reply_to_message.from_user
+    with contextlib.suppress(Exception):
+        await message.delete()
+    try:
+        msg = await message.reply_to_message.reply(MSG % user.mention)
+    except Exception as e:
+        return
+    try:
+        msg_ = await client.listen(chat.id, filters=filters.user(user.id), timeout=30)
+        with contextlib.suppress(Exception):
+            await msg.delete()
+        if not msg_.sticker:
+            with contextlib.suppress(Exception):
+                await message.reply_to_message.delete()
+            with contextlib.suppress(Exception):
+                await bot.ban_chat_member(chat.id, user.id, datetime.now() + timedelta(minutes=5))
+            with contextlib.suppress(Exception):
+                await log(chat, user, "FAIL_ERROR")
+        else:
+            with contextlib.suppress(Exception):
+                await log(chat, user, "ACCEPT")
+        with contextlib.suppress(Exception):
+            await msg_.delete()
+    except TimeoutConversationError:
+        with contextlib.suppress(Exception):
+            await msg.delete()
+        with contextlib.suppress(Exception):
+            await message.reply_to_message.delete()
+        with contextlib.suppress(Exception):
+            await bot.ban_chat_member(chat.id, user.id, datetime.now() + timedelta(minutes=5))
+        with contextlib.suppress(Exception):
+            await log(chat, user, "FAIL_TIMEOUT")
+

plugins/service.py

@@ -43,10 +43,11 @@ async def chat_members_handle(client: Client, message: Message):
                 await bot.ban_chat_member(chat.id, user.id, datetime.now() + timedelta(minutes=5))
             with contextlib.suppress(Exception):
                 await log(chat, user, "FAIL_ERROR")
-        with contextlib.suppress(Exception):
-            await msg_.delete()
+        else:
             with contextlib.suppress(Exception):
                 await log(chat, user, "ACCEPT")
+        with contextlib.suppress(Exception):
+            await msg_.delete()
     except TimeoutConversationError:
         with contextlib.suppress(Exception):
             await msg.delete()