mirror of
https://github.com/Xtao-Labs/sticker-captcha-bot.git
synced 2024-11-23 16:27:51 +00:00
100 lines
3.3 KiB
Python
100 lines
3.3 KiB
Python
import contextlib
|
|
|
|
from cashews import cache
|
|
from pyrogram import filters
|
|
from pyrogram.enums import ChatMemberStatus
|
|
from pyrogram.types import ChatMemberUpdated
|
|
|
|
from pyromod.utils.errors import TimeoutConversationError
|
|
from sticker.languages import MSG_PUBLIC, ADMIN_MSG, MSG, VERIFY_TIME
|
|
from sticker.scheduler import add_ban_chat_member_job
|
|
from sticker.service_message import ServiceMessage
|
|
from sticker.single_utils import Client, Message
|
|
from sticker import bot, log, LogAction
|
|
|
|
|
|
async def start_verify(client: "Client", chat, user):
|
|
key = f"wait:{chat.id}:{user.id}"
|
|
await cache.set(key, "True", expire=VERIFY_TIME + 5)
|
|
try:
|
|
msg: "Message" = await client.send_message(
|
|
chat.id, MSG % (user.mention, user.mention)
|
|
)
|
|
except Exception:
|
|
return
|
|
await log(chat, user, LogAction.REQUEST)
|
|
try:
|
|
msg_: "Message" = await client.listen(
|
|
chat.id, filters=filters.user(user.id), timeout=VERIFY_TIME
|
|
)
|
|
await msg.delay_delete(1)
|
|
await msg_.delay_delete(1)
|
|
if not msg_.sticker:
|
|
add_ban_chat_member_job(chat.id, user.id)
|
|
await log(chat, user, LogAction.FAIL_ERROR)
|
|
await ServiceMessage.try_delete(user.id, chat.id)
|
|
else:
|
|
await cache.delete(key)
|
|
await log(chat, user, LogAction.ACCEPT)
|
|
except TimeoutConversationError:
|
|
await msg.delay_delete(1)
|
|
add_ban_chat_member_job(chat.id, user.id)
|
|
await log(chat, user, LogAction.FAIL_TIMEOUT)
|
|
await ServiceMessage.try_delete(user.id, chat.id)
|
|
|
|
|
|
@bot.on_chat_member_updated()
|
|
async def invite(client: Client, chat_member_updated: ChatMemberUpdated):
|
|
chat = chat_member_updated.chat
|
|
member = chat_member_updated.new_chat_member
|
|
old_member = chat_member_updated.old_chat_member
|
|
if not member:
|
|
return
|
|
if not member.user:
|
|
return
|
|
user = member.user
|
|
old_user = old_member.user if old_member else None
|
|
if user.is_self:
|
|
if member.status not in {
|
|
ChatMemberStatus.ADMINISTRATOR,
|
|
ChatMemberStatus.MEMBER,
|
|
}:
|
|
return
|
|
await log(chat, chat_member_updated.from_user, LogAction.NEW_GROUP)
|
|
if chat.username:
|
|
with contextlib.suppress(Exception):
|
|
await client.send_message(chat.id, MSG_PUBLIC)
|
|
return
|
|
if user.is_verified or user.is_bot or user.is_deleted or user.is_support:
|
|
return
|
|
if member.status not in {ChatMemberStatus.MEMBER}:
|
|
return
|
|
if (
|
|
old_user
|
|
and old_user.id == user.id
|
|
and old_member.status
|
|
in {
|
|
ChatMemberStatus.ADMINISTRATOR,
|
|
ChatMemberStatus.OWNER,
|
|
ChatMemberStatus.MEMBER,
|
|
ChatMemberStatus.RESTRICTED,
|
|
}
|
|
):
|
|
return
|
|
from_user = chat_member_updated.from_user
|
|
if from_user and from_user.id == user.id:
|
|
from_user = None
|
|
if from_user and from_user.is_self:
|
|
return
|
|
if (
|
|
user
|
|
and from_user
|
|
and (await bot.get_chat_member(chat.id, from_user.id)).status
|
|
in {ChatMemberStatus.ADMINISTRATOR, ChatMemberStatus.OWNER}
|
|
):
|
|
with contextlib.suppress(Exception):
|
|
msg: "Message" = await client.send_message(chat.id, ADMIN_MSG)
|
|
await msg.delay_delete()
|
|
return
|
|
await start_verify(client, chat, user)
|