mirror of
https://github.com/Xtao-Labs/sticker-captcha-bot.git
synced 2024-11-22 07:07:51 +00:00
45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
import contextlib
|
|
|
|
from cashews import cache
|
|
from pyrogram.types import ChatJoinRequest
|
|
from pyrogram import filters
|
|
|
|
from sticker.single_utils import Client
|
|
from sticker.scheduler import add_decline_request_job, rem_decline_request_job
|
|
from sticker import bot, log
|
|
|
|
from pyromod.utils.errors import TimeoutConversationError
|
|
|
|
|
|
MSG = """您好，群组 %s 开启了验证功能。
|
|
|
|
您需要在 30 秒内发送任意一个 贴纸 来完成验证。"""
|
|
MSG_SUCCESS = """验证成功，您已经成为群组的一员了！"""
|
|
MSG_FAILURE = """验证失败，请重试。"""
|
|
|
|
|
|
@bot.on_chat_join_request()
|
|
async def new_member(client: Client, chat_join_request: ChatJoinRequest):
|
|
chat = chat_join_request.chat
|
|
await cache.set(f"cid:{chat.id}", "True", expire=3600, exist=True)
|
|
user = chat_join_request.from_user
|
|
add_decline_request_job(chat_join_request)
|
|
try:
|
|
with contextlib.suppress(Exception):
|
|
await log(chat, user, "REQUEST")
|
|
await client.ask(user.id, MSG % chat.title, filters=filters.sticker, timeout=30)
|
|
with contextlib.suppress(Exception):
|
|
await client.send_message(user.id, MSG_SUCCESS)
|
|
await chat_join_request.approve()
|
|
with contextlib.suppress(Exception):
|
|
await log(chat, user, "ACCEPT")
|
|
with contextlib.suppress(Exception):
|
|
rem_decline_request_job(chat_join_request)
|
|
except TimeoutConversationError:
|
|
with contextlib.suppress(Exception):
|
|
await client.send_message(user.id, MSG_FAILURE)
|
|
with contextlib.suppress(Exception):
|
|
await chat_join_request.decline()
|
|
with contextlib.suppress(Exception):
|
|
await log(chat, user, "FAIL_TIMEOUT")
|