mirror of
https://github.com/PaiGramTeam/MibooGram.git
synced 2024-11-16 12:51:45 +00:00
58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
from telegram import Update, ReplyKeyboardRemove
|
|
from telegram.error import BadRequest, Forbidden
|
|
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 exc:
|
|
if "not found" in str(exc):
|
|
logger.warning(f"删除消息 chat_id[{job.chat_id}] message_id[{job.data}]失败 消息不存在")
|
|
elif "Message can't be deleted" in str(exc):
|
|
logger.warning(f"删除消息 chat_id[{job.chat_id}] message_id[{job.data}]失败 消息无法删除 可能是没有授权")
|
|
else:
|
|
logger.error(f"删除消息 chat_id[{job.chat_id}] message_id[{job.data}]失败")
|
|
logger.exception(exc)
|
|
except Forbidden as exc:
|
|
if "bot was kicked" in str(exc):
|
|
logger.warning(f"删除消息 chat_id[{job.chat_id}] message_id[{job.data}]失败 已经被踢出群")
|
|
else:
|
|
logger.error(f"删除消息 chat_id[{job.chat_id}] message_id[{job.data}]失败")
|
|
logger.exception(exc)
|
|
|
|
|
|
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
|