mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-16 12:51:35 +00:00
45 lines
1.8 KiB
Python
45 lines
1.8 KiB
Python
from telegram import Update
|
|
from telegram.ext import CallbackContext
|
|
|
|
from plugins.base import BasePlugins
|
|
from service import BaseService
|
|
|
|
|
|
class Admin(BasePlugins):
|
|
def __init__(self, service: BaseService):
|
|
super().__init__(service)
|
|
|
|
async def add_admin(self, update: Update, context: CallbackContext):
|
|
message = update.message
|
|
user = message.from_user
|
|
reply_to_message = message.reply_to_message
|
|
admin_list = await self.service.admin.get_admin_list()
|
|
if user.id in admin_list:
|
|
if reply_to_message is None:
|
|
await message.reply_text("请回复对应消息")
|
|
else:
|
|
if reply_to_message.from_user.id in admin_list:
|
|
await message.reply_text("该用户已经存在管理员列表")
|
|
else:
|
|
await self.service.admin.add_admin(reply_to_message.from_user.id)
|
|
await message.reply_text("添加成功")
|
|
else:
|
|
await message.reply_text("权限不足")
|
|
|
|
async def del_admin(self, update: Update, context: CallbackContext):
|
|
message = update.message
|
|
user = message.from_user
|
|
reply_to_message = message.reply_to_message
|
|
admin_list = await self.service.admin.get_admin_list()
|
|
if user.id in admin_list:
|
|
if reply_to_message is None:
|
|
await message.reply_text("请回复对应消息")
|
|
else:
|
|
if reply_to_message.from_user.id in admin_list:
|
|
await self.service.admin.delete_admin(reply_to_message.from_user.id)
|
|
await message.reply_text("删除成功")
|
|
else:
|
|
await message.reply_text("该用户不存在管理员列表")
|
|
else:
|
|
await message.reply_text("权限不足")
|