mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-16 20:10:36 +00:00
38 lines
1.8 KiB
Python
38 lines
1.8 KiB
Python
|
from telegram import Update
|
||
|
|
from telegram.ext import CallbackContext, filters
|
||
|
|
|
||
|
|
from core.admin import BotAdminService
|
||
|
|
from core.plugin import Plugin, handler
|
||
|
|
from utils.log import logger
|
||
|
|
|
||
|
|
|
||
|
|
class BotJoiningGroupsVerification(Plugin):
|
||
|
|
|
||
|
|
def __init__(self, bot_admin_service: BotAdminService = None):
|
||
|
|
self.bot_admin_service = bot_admin_service
|
||
|
|
|
||
|
|
@handler.message.new_chat_members(priority=1)
|
||
|
|
async def new_member(self, update: Update, context: CallbackContext) -> None:
|
||
|
|
message = update.effective_message
|
||
|
|
chat = message.chat
|
||
|
|
from_user = message.from_user
|
||
|
|
quit_status = False
|
||
|
|
if filters.ChatType.GROUPS.filter(message):
|
||
|
|
for user in message.new_chat_members:
|
||
|
|
if user.id == context.bot.id:
|
||
|
|
if from_user is not None:
|
||
|
|
logger.info(f"用户 {from_user.full_name}[{from_user.id}] 在群 {chat.title}[{chat.id}] 邀请BOT")
|
||
|
|
admin_list = await self.bot_admin_service.get_admin_list()
|
||
|
|
if from_user.id in admin_list:
|
||
|
|
await context.bot.send_message(message.chat_id,
|
||
|
|
'感谢邀请小派蒙到本群！请使用 /help 查看咱已经学会的功能。')
|
||
|
|
else:
|
||
|
|
quit_status = True
|
||
|
|
else:
|
||
|
|
logger.info(f"未知用户 在群 {chat.title}[{chat.id}] 邀请BOT")
|
||
|
|
quit_status = True
|
||
|
|
if quit_status:
|
||
|
|
logger.warning("不是管理员邀请！退出群聊。")
|
||
|
|
await context.bot.send_message(message.chat_id, "派蒙不想进去！不是旅行者的邀请！")
|
||
|
|
await context.bot.leave_chat(chat.id)
|