mirror of
https://github.com/PaiGramTeam/MibooGram.git
synced 2024-11-27 10:26:26 +00:00
43 lines
1.8 KiB
Python
43 lines
1.8 KiB
Python
|
import contextlib
|
||
|
|
|
||
|
|
from telegram import Update, ChatMemberAdministrator
|
||
|
|
from telegram.ext import CallbackContext, filters
|
||
|
|
|
||
|
|
from core.plugin import Plugin, handler
|
||
|
|
from utils.log import logger
|
||
|
|
|
||
|
|
|
||
|
|
class TitlePlugin(Plugin):
|
||
|
|
@handler.command("title", filters=filters.ChatType.SUPERGROUP, block=False)
|
||
|
|
async def start(self, update: Update, context: CallbackContext) -> None:
|
||
|
|
user = update.effective_user
|
||
|
|
message = update.effective_message
|
||
|
|
args = self.get_args(context)
|
||
|
|
title = args[0].strip() if args else ""
|
||
|
|
logger.info("用户 %s[%s] 发出 title 命令", user.full_name, user.id)
|
||
|
|
can_edit = False
|
||
|
|
with contextlib.suppress(Exception):
|
||
|
|
member = await context.bot.get_chat_member(message.chat.id, user.id)
|
||
|
|
if isinstance(member, ChatMemberAdministrator):
|
||
|
|
can_edit = member.can_be_edited
|
||
|
|
if not can_edit:
|
||
|
|
reply = await message.reply_text("你没有权限使用此命令。")
|
||
|
|
self.add_delete_message_job(message)
|
||
|
|
self.add_delete_message_job(reply)
|
||
|
|
return
|
||
|
|
if not title:
|
||
|
|
reply = await message.reply_text("参数不能为空。")
|
||
|
|
self.add_delete_message_job(message)
|
||
|
|
self.add_delete_message_job(reply)
|
||
|
|
return
|
||
|
|
try:
|
||
|
|
await context.bot.set_chat_administrator_custom_title(message.chat.id, user.id, title)
|
||
|
|
except Exception:
|
||
|
|
reply = await message.reply_text("设置失败，可能是参数不合法。")
|
||
|
|
self.add_delete_message_job(message)
|
||
|
|
self.add_delete_message_job(reply)
|
||
|
|
return
|
||
|
|
reply = await message.reply_text("设置成功。")
|
||
|
|
self.add_delete_message_job(message)
|
||
|
|
self.add_delete_message_job(reply)
|