|
|
@@ -1,12 +1,96 @@
|
|
|
-from pyrogram import filters, emoji
|
|
|
-from pyrogram.types import Message
|
|
|
-from wbb import app
|
|
|
+"""
|
|
|
+ Thanks to @dashezup for this module
|
|
|
+ Repo - https://github.com/dashezup/tgbot
|
|
|
|
|
|
-MESSAGE = "{} Welcome {}!"
|
|
|
+A Pyrogram Smart Plugin to verify if new members are human
|
|
|
+# the bot must be admin in the chat
|
|
|
+# should not be lesser than 30 or the user may be banned if he didn't
|
|
|
+# press the verification button
|
|
|
+"""
|
|
|
+
|
|
|
+
|
|
|
+import asyncio
|
|
|
+from datetime import datetime
|
|
|
+from pyrogram import filters
|
|
|
+from wbb import app, WELCOME_DELAY_KICK_SEC
|
|
|
+from pyrogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton
|
|
|
+from pyrogram.types import ChatPermissions
|
|
|
+from pyrogram.errors.exceptions.bad_request_400 import UserNotParticipant
|
|
|
|
|
|
|
|
|
@app.on_message(filters.new_chat_members)
|
|
|
async def welcome(_, message: Message):
|
|
|
- new_members = [i.mention for i in message.new_chat_members]
|
|
|
- text = MESSAGE.format(emoji.CROWN, ", ".join(new_members))
|
|
|
- await message.reply_text(text, disable_web_page_preview=True)
|
|
|
+ """Mute new member and send message with button"""
|
|
|
+ new_members = [f"{u.mention}" for u in message.new_chat_members]
|
|
|
+ text = (f"Welcome, {', '.join(new_members)}\n**Are you human?**\n"
|
|
|
+ "You will be removed from this chat if you are not verified "
|
|
|
+ f"in {WELCOME_DELAY_KICK_SEC} seconds")
|
|
|
+ await message.chat.restrict_member(message.from_user.id, ChatPermissions())
|
|
|
+ button_message = await message.reply(
|
|
|
+ text,
|
|
|
+ reply_markup=InlineKeyboardMarkup(
|
|
|
+ [
|
|
|
+ [
|
|
|
+ InlineKeyboardButton(
|
|
|
+ "Press Here to Verify",
|
|
|
+ callback_data="pressed_button"
|
|
|
+ )
|
|
|
+ ]
|
|
|
+ ]
|
|
|
+ ),
|
|
|
+ quote=True
|
|
|
+ )
|
|
|
+ await kick_restricted_after_delay(WELCOME_DELAY_KICK_SEC, button_message)
|
|
|
+
|
|
|
+
|
|
|
+@app.on_callback_query(filters.regex("pressed_button"))
|
|
|
+async def callback_query_welcome_button(_, callback_query):
|
|
|
+ """After the new member presses the button, set his permissions to
|
|
|
+ chat permissions, delete button message and join message
|
|
|
+ """
|
|
|
+ button_message = callback_query.message
|
|
|
+ join_message = button_message.reply_to_message
|
|
|
+ pending_user = join_message.from_user
|
|
|
+ pending_user_id = pending_user.id
|
|
|
+ pressed_user_id = callback_query.from_user.id
|
|
|
+ if pending_user_id == pressed_user_id:
|
|
|
+ await callback_query.answer("Congrats, captcha passed!")
|
|
|
+ await button_message.chat.unban_member(pending_user_id)
|
|
|
+ await button_message.delete()
|
|
|
+ await join_message.delete()
|
|
|
+ else:
|
|
|
+ await callback_query.answer("This is not for you")
|
|
|
+
|
|
|
+
|
|
|
+async def kick_restricted_after_delay(delay, button_message: Message):
|
|
|
+ """If the new member is still restricted after the delay, delete
|
|
|
+ button message and join message and then kick him
|
|
|
+ """
|
|
|
+ await asyncio.sleep(delay)
|
|
|
+ join_message = button_message.reply_to_message
|
|
|
+ group_chat = button_message.chat
|
|
|
+ user_id = join_message.from_user.id
|
|
|
+ await join_message.delete()
|
|
|
+ await button_message.delete()
|
|
|
+ await _ban_restricted_user_until_date(group_chat, user_id, duration=delay)
|
|
|
+
|
|
|
+
|
|
|
+@app.on_message(filters.left_chat_member)
|
|
|
+async def left_chat_member(_, message: Message):
|
|
|
+ """When a restricted member left the chat, ban him for a while"""
|
|
|
+ group_chat = message.chat
|
|
|
+ user_id = message.left_chat_member.id
|
|
|
+ await _ban_restricted_user_until_date(group_chat, user_id,
|
|
|
+ duration=WELCOME_DELAY_KICK_SEC)
|
|
|
+
|
|
|
+
|
|
|
+async def _ban_restricted_user_until_date(group_chat,
|
|
|
+ user_id: int,
|
|
|
+ duration: int):
|
|
|
+ try:
|
|
|
+ member = await group_chat.get_member(user_id)
|
|
|
+ if member.status == "restricted":
|
|
|
+ until_date = int(datetime.utcnow().timestamp() + duration)
|
|
|
+ await group_chat.kick_member(user_id, until_date=until_date)
|
|
|
+ except UserNotParticipant:
|
|
|
+ pass
|
TheHamkerCat