mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-22 14:26:45 +00:00
8f424bf0d4
♻️ 重构插件系统 ⚙️ 重写插件 🎨 改进代码结构 📝 完善文档 Co-authored-by: zhxy-CN <admin@owo.cab> Co-authored-by: 洛水居室 <luoshuijs@outlook.com> Co-authored-by: xtaodada <xtao@xtaolink.cn> Co-authored-by: Li Chuangbo <im@chuangbo.li>
49 lines
2.1 KiB
Python
49 lines
2.1 KiB
Python
from telegram import Update, ReplyKeyboardRemove
|
|
from telegram.error import BadRequest
|
|
from telegram.ext import CallbackContext, ConversationHandler
|
|
|
|
from core.plugin import handler, conversation
|
|
from utils.log import logger
|
|
|
|
|
|
async def clean_message(context: CallbackContext):
|
|
job = context.job
|
|
logger.debug(f"删除消息 chat_id[{job.chat_id}] 的 message_id[{job.data}]")
|
|
try:
|
|
# noinspection PyTypeChecker
|
|
await context.bot.delete_message(chat_id=job.chat_id, message_id=job.data)
|
|
except BadRequest as error:
|
|
if "not found" in str(error):
|
|
logger.warning(f"Auth模块删除消息 chat_id[{job.chat_id}] message_id[{job.data}]失败 消息不存在")
|
|
elif "Message can't be deleted" in str(error):
|
|
logger.warning(
|
|
f"Auth模块删除消息 chat_id[{job.chat_id}] message_id[{job.data}]失败 消息无法删除 可能是没有授权")
|
|
else:
|
|
logger.error(f"Auth模块删除消息 chat_id[{job.chat_id}] message_id[{job.data}]失败", error)
|
|
|
|
|
|
def add_delete_message_job(context: CallbackContext, chat_id: int, message_id: int, delete_seconds: int):
|
|
context.job_queue.run_once(callback=clean_message, when=delete_seconds, data=message_id,
|
|
name=f"{chat_id}|{message_id}|clean_message", chat_id=chat_id,
|
|
job_kwargs={"replace_existing": True,
|
|
"id": f"{chat_id}|{message_id}|clean_message"})
|
|
|
|
|
|
class _BasePlugin:
|
|
@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 _Conversation(_BasePlugin):
|
|
|
|
@conversation.fallback
|
|
@handler.command(command='cancel', block=True)
|
|
async def cancel(self, update: Update, _: CallbackContext) -> int:
|
|
await update.effective_message.reply_text("退出命令", reply_markup=ReplyKeyboardRemove())
|
|
return ConversationHandler.END
|
|
|
|
|
|
class BasePlugin(_BasePlugin):
|
|
Conversation = _Conversation
|