mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-17 20:33:41 +00:00
98 lines
4.3 KiB
Python
98 lines
4.3 KiB
Python
import asyncio
|
|
import datetime
|
|
from typing import List, Tuple, Callable
|
|
|
|
from telegram import Update, ReplyKeyboardRemove
|
|
from telegram.error import BadRequest
|
|
from telegram.ext import CallbackContext, ConversationHandler, filters
|
|
|
|
from core.admin.services import BotAdminService
|
|
from logger import Log
|
|
from utils.service.inject import inject
|
|
|
|
|
|
async def clean_message(context: CallbackContext, chat_id: int, message_id: int) -> bool:
|
|
try:
|
|
await context.bot.delete_message(chat_id=chat_id, message_id=message_id)
|
|
return True
|
|
except BadRequest as error:
|
|
if "not found" in str(error):
|
|
Log.warning(f"定时删除消息 chat_id[{chat_id}] message_id[{message_id}]失败 消息不存在")
|
|
elif "Message can't be deleted" in str(error):
|
|
Log.warning(f"定时删除消息 chat_id[{chat_id}] message_id[{message_id}]失败 消息无法删除 可能是没有授权")
|
|
else:
|
|
Log.warning(f"定时删除消息 chat_id[{chat_id}] message_id[{message_id}]失败 \n", error)
|
|
return False
|
|
|
|
|
|
def add_delete_message_job(context: CallbackContext, chat_id: int, message_id: int,
|
|
delete_seconds: int = 60):
|
|
context.job_queue.scheduler.add_job(clean_message, "date",
|
|
id=f"{chat_id}|{message_id}|auto_clean_message",
|
|
name=f"{chat_id}|{message_id}|auto_clean_message",
|
|
args=[context, chat_id, message_id],
|
|
run_date=context.job_queue._tz_now() + datetime.timedelta(
|
|
seconds=delete_seconds), replace_existing=True)
|
|
|
|
|
|
class BasePlugins:
|
|
|
|
@staticmethod
|
|
async def cancel(update: Update, _: CallbackContext) -> int:
|
|
await update.message.reply_text("退出命令", reply_markup=ReplyKeyboardRemove())
|
|
return ConversationHandler.END
|
|
|
|
@staticmethod
|
|
async def _clean(context: CallbackContext, chat_id: int, message_id: int) -> bool:
|
|
return await clean_message(context, chat_id, message_id)
|
|
|
|
@staticmethod
|
|
def _add_delete_message_job(context: CallbackContext, chat_id: int, message_id: int,
|
|
delete_seconds: int = 60):
|
|
return add_delete_message_job(context, chat_id, message_id, delete_seconds)
|
|
|
|
|
|
class NewChatMembersHandler:
|
|
|
|
@inject
|
|
def __init__(self, bot_admin_service: BotAdminService = None):
|
|
self.bot_admin_service = bot_admin_service
|
|
self.callback: List[Tuple[Callable, int]] = []
|
|
|
|
def add_callback(self, callback, chat_id: int):
|
|
if chat_id >= 0:
|
|
raise ValueError
|
|
self.callback.append((callback, chat_id))
|
|
|
|
async def new_member(self, update: Update, context: CallbackContext) -> None:
|
|
message = update.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:
|
|
Log.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:
|
|
Log.info(f"未知用户 在群 {chat.title}[{chat.id}] 邀请BOT")
|
|
quit_status = True
|
|
if quit_status:
|
|
Log.warning("不是管理员邀请！退出群聊。")
|
|
await context.bot.send_message(message.chat_id, "派蒙不想进去！不是旅行者的邀请！")
|
|
await context.bot.leave_chat(chat.id)
|
|
else:
|
|
tasks = []
|
|
for callback, chat_id in self.callback:
|
|
if chat.id == chat_id:
|
|
task = asyncio.create_task(callback(update, context))
|
|
tasks.append(task)
|
|
if len(tasks) >= 1:
|
|
await asyncio.gather(*tasks)
|