74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
from pyrogram import Client
|
|
from pyrogram.types import Message
|
|
from configparser import RawConfigParser
|
|
from os import getcwd, sep
|
|
from defs.mys import draw_pic
|
|
from defs.redis_load import redis, redis_status
|
|
|
|
|
|
async def mys_msg(client: Client, message: Message):
|
|
# 权限检查
|
|
uid = str(message.from_user.id)
|
|
config = RawConfigParser()
|
|
config.read(f"{getcwd()}{sep}config.ini")
|
|
admin_str: str = "777000"
|
|
admin_str = config.get("basic", "admin", fallback=admin_str)
|
|
admins = admin_str.split('|')
|
|
if redis_status():
|
|
admin = redis.get('mys')
|
|
if admin:
|
|
admins.extend(admin.split('|'))
|
|
if uid not in admins:
|
|
return
|
|
# 生成用户信息
|
|
uid = message.text.replace('账号信息', '').replace('用户信息', '').strip()
|
|
try:
|
|
path = await draw_pic(uid)
|
|
except ConnectionRefusedError:
|
|
await message.reply('派蒙与提瓦特大陆的连接遇到了一点问题：<code>ConnectionRefusedError</code>', quote=True)
|
|
return
|
|
except ConnectionError:
|
|
await message.reply('派蒙与提瓦特大陆的连接遇到了一点问题：<code>ConnectionError</code>', quote=True)
|
|
return
|
|
if path:
|
|
# 上传图片
|
|
await message.reply_photo(photo=path, quote=True)
|
|
else:
|
|
await message.reply('派蒙与提瓦特大陆的连接遇到了一点问题：<code>ConnectionError</code>', quote=True)
|
|
|
|
|
|
async def promote_command(client: Client, message: Message):
|
|
if not message.reply_to_message:
|
|
await message.reply('你需要回复一个用户才能授权 Ta 管理员权限。', quote=True)
|
|
return
|
|
uid = str(message.reply_to_message.from_user.id)
|
|
if redis_status():
|
|
# 读取数据
|
|
admin = redis.get('mys').decode()
|
|
if admin:
|
|
admins = admin.split('|')
|
|
else:
|
|
admins = []
|
|
if len(message.text.split(' ')) >= 2:
|
|
# 列出所有管理员
|
|
if len(admins) > 0:
|
|
await message.reply('管理员列表：\n{" ".join(admins)}', quote=True)
|
|
return
|
|
else:
|
|
await message.reply('当前暂无管理员。', quote=True)
|
|
return
|
|
if uid in admins:
|
|
admins.remove(uid)
|
|
mode = False
|
|
else:
|
|
admins.append(uid)
|
|
mode = True
|
|
# 写入数据
|
|
admin = '|'.join(admins)
|
|
redis.set('mys', admin)
|
|
text = f'移除管理员 {uid} 成功。'
|
|
if mode:
|
|
text = f'添加管理员 {uid} 成功。'
|
|
await message.reply(text, quote=True)
|
|
return
|