206 lines
8.9 KiB
Python
206 lines
8.9 KiB
Python
import asyncio
|
|
|
|
from graia.application import Group, MessageChain, Member, Friend
|
|
from graia.application.message.elements.internal import Plain
|
|
|
|
import database
|
|
from core.template import sendMessage, check_permission, revokeMessage
|
|
from .admin import *
|
|
|
|
|
|
async def enable_modules(kwargs: dict):
|
|
"""
|
|
~enable [self] <modules/all>"""
|
|
command = kwargs['trigger_msg'].split(' ')
|
|
function_list = kwargs['function_list']
|
|
friend_function_list = kwargs['friend_function_list']
|
|
if not len(command) > 1:
|
|
await sendMessage(kwargs, '命令格式错误。' + enable_modules.__doc__)
|
|
return
|
|
command_second_word = command[1]
|
|
if Group in kwargs:
|
|
if command_second_word == 'self':
|
|
if not len(command) > 2:
|
|
await sendMessage(kwargs, '命令格式错误。' + enable_modules.__doc__)
|
|
return
|
|
command_third_word = command[2]
|
|
if command_third_word in function_list:
|
|
msg = database.update_modules_self('add', kwargs[Member].id, command_third_word)
|
|
await sendMessage(kwargs, msg)
|
|
else:
|
|
await sendMessage(kwargs, '此模块不存在。')
|
|
elif command_second_word == 'all':
|
|
if check_permission(kwargs):
|
|
msglist = []
|
|
for function in function_list:
|
|
msg = database.update_modules('add', kwargs[Group].id, function)
|
|
msglist.append(msg)
|
|
await sendMessage(kwargs, '\n'.join(msglist))
|
|
else:
|
|
await sendMessage(kwargs, '你没有使用该命令的权限。')
|
|
elif command_second_word in function_list:
|
|
if check_permission(kwargs):
|
|
msg = database.update_modules('add', kwargs[Group].id, command_second_word)
|
|
await sendMessage(kwargs, msg)
|
|
else:
|
|
await sendMessage(kwargs, '你没有使用该命令的权限。')
|
|
else:
|
|
msgchain = MessageChain.create([Plain('此模块不存在。')])
|
|
await sendMessage(kwargs, msgchain)
|
|
elif Friend in kwargs:
|
|
if command_second_word == 'self':
|
|
if not len(command) > 2:
|
|
await sendMessage(kwargs, '命令格式错误。' + enable_modules.__doc__)
|
|
return
|
|
command_second_word = command[2]
|
|
do = 'add'
|
|
if command_second_word == 'all':
|
|
msglist = []
|
|
for function in function_list:
|
|
msg = database.update_modules_self(do, kwargs[Friend].id, function)
|
|
msglist.append(msg)
|
|
await sendMessage(kwargs, '\n'.join(msglist))
|
|
elif command_second_word in function_list:
|
|
msg = database.update_modules_self(do, kwargs[Friend].id, command_second_word)
|
|
await sendMessage(kwargs, msg)
|
|
elif command_second_word in friend_function_list:
|
|
msg = database.update_modules(do, kwargs[Friend].id, command_second_word, table='friend_permission')
|
|
await sendMessage(kwargs, msg)
|
|
else:
|
|
await sendMessage(kwargs, '此模块不存在。')
|
|
|
|
|
|
async def disable_modules(kwargs: dict):
|
|
"""
|
|
~disable [self] <modules/all>"""
|
|
command = kwargs['trigger_msg'].split(' ')
|
|
if not len(command) > 1:
|
|
await sendMessage(kwargs, '命令格式错误。' + disable_modules.__doc__)
|
|
return
|
|
function_list = kwargs['function_list']
|
|
friend_function_list = kwargs['friend_function_list']
|
|
command_second_word = command[1]
|
|
if Group in kwargs:
|
|
if command_second_word == 'self':
|
|
if not len(command) > 2:
|
|
await sendMessage(kwargs, '命令格式错误。' + disable_modules.__doc__)
|
|
return
|
|
command_third_word = command[2]
|
|
if command_third_word in function_list:
|
|
msg = database.update_modules_self('del', kwargs[Member].id, command_third_word)
|
|
await sendMessage(kwargs, msg)
|
|
else:
|
|
await sendMessage(kwargs, '此模块不存在。')
|
|
elif command_second_word == 'all':
|
|
if check_permission(kwargs):
|
|
msglist = []
|
|
for function in function_list:
|
|
msg = database.update_modules('del', kwargs[Group].id, function)
|
|
msglist.append(msg)
|
|
await sendMessage(kwargs, '\n'.join(msglist))
|
|
else:
|
|
await sendMessage(kwargs, '你没有使用该命令的权限。')
|
|
elif command_second_word in function_list:
|
|
if check_permission(kwargs):
|
|
msg = database.update_modules('del', kwargs[Group].id, command_second_word)
|
|
await sendMessage(kwargs, msg)
|
|
else:
|
|
await sendMessage(kwargs, '你没有使用该命令的权限。')
|
|
else:
|
|
await sendMessage(kwargs, '此模块不存在。')
|
|
elif Friend in kwargs:
|
|
if command_second_word == 'self':
|
|
if not len(command) > 2:
|
|
await sendMessage(kwargs, '命令格式错误。' + disable_modules.__doc__)
|
|
return
|
|
command_second_word = command[2]
|
|
do = 'del'
|
|
if command_second_word == 'all':
|
|
msglist = []
|
|
for function in function_list:
|
|
msg = database.update_modules_self(do, kwargs[Friend].id, function)
|
|
msglist.append(msg)
|
|
await sendMessage(kwargs, '\n'.join(msglist))
|
|
elif command_second_word in function_list:
|
|
msg = database.update_modules_self(do, kwargs[Friend].id, command_second_word)
|
|
await sendMessage(kwargs, msg)
|
|
elif command_second_word in friend_function_list:
|
|
msg = database.update_modules(do, kwargs[Friend].id, command_second_word, table='friend_permission')
|
|
await sendMessage(kwargs, msg)
|
|
else:
|
|
await sendMessage(kwargs, '此模块不存在。')
|
|
|
|
|
|
async def bot_help(kwargs: dict):
|
|
help_list = kwargs['help_list']
|
|
command = kwargs['trigger_msg'].split(' ')
|
|
try:
|
|
msg = []
|
|
if command[1] in help_list:
|
|
msg.append(help_list[command[1]]['help'])
|
|
for x in help_list:
|
|
if 'depend' in help_list[x]:
|
|
if help_list[x]['depend'] == command[1]:
|
|
msg.append(help_list[x]['help'])
|
|
await sendMessage(kwargs, '\n'.join(msg))
|
|
except:
|
|
print(help_list)
|
|
help_msg = []
|
|
help_msg.append('基础命令:')
|
|
essential = []
|
|
for x in help_list:
|
|
if 'essential' in help_list[x]:
|
|
essential.append(x)
|
|
help_msg.append(' | '.join(essential))
|
|
help_msg.append('模块扩展命令:')
|
|
module = []
|
|
for x in help_list:
|
|
if Group in kwargs:
|
|
if database.check_enable_modules(kwargs, x):
|
|
if 'help' in help_list[x]:
|
|
module.append(x)
|
|
elif Friend in kwargs:
|
|
if 'help' in help_list[x]:
|
|
module.append(x)
|
|
help_msg.append(' | '.join(module))
|
|
print(help_msg)
|
|
help_msg.append('使用~help <对应模块名>查看详细信息。\n你也可以通过查阅文档获取帮助:\nhttps://bot.teahou.se/modules/')
|
|
if Group in kwargs:
|
|
help_msg.append('[本消息将在一分钟后撤回]')
|
|
send = await sendMessage(kwargs, '\n'.join(help_msg))
|
|
if Group in kwargs:
|
|
await asyncio.sleep(60)
|
|
await revokeMessage(send)
|
|
|
|
|
|
async def modules_help(kwargs: dict):
|
|
help_list = kwargs['help_list']
|
|
help_msg = []
|
|
help_msg.append('当前可用的模块有:')
|
|
module = []
|
|
for x in help_list:
|
|
if 'help' in help_list[x]:
|
|
module.append(x)
|
|
help_msg.append(' | '.join(module))
|
|
help_msg.append('使用~help <模块名>查看详细信息。\n你也可以通过查阅文档获取帮助:\nhttps://bot.teahou.se/modules/')
|
|
if Group in kwargs:
|
|
help_msg.append('[本消息将在一分钟后撤回]')
|
|
send = await sendMessage(kwargs, '\n'.join(help_msg))
|
|
if Group in kwargs:
|
|
await asyncio.sleep(60)
|
|
await revokeMessage(send)
|
|
|
|
|
|
async def bot_version(kwargs: dict):
|
|
git_commit_version = os.popen('git rev-parse HEAD', 'r')
|
|
msg = '当前运行的代码版本号为:' + git_commit_version.read()[0:7]
|
|
await sendMessage(kwargs, msg)
|
|
|
|
|
|
essential = {'enable': enable_modules, 'disable': disable_modules, 'add_base_su': add_base_su, 'help': bot_help,
|
|
'modules': modules_help, 'version': bot_version}
|
|
|
|
admin = {'add_su': add_su, 'del_su': del_su, 'set': set_modules, 'restart': restart_bot, 'update': update_bot}
|
|
help = {'enable': {'help': '~enable <模块名> - 开启一个模块', 'essential': True},
|
|
'disable': {'help': '~disable <模块名> - 关闭一个模块', 'essential': True},
|
|
'module': {'help': '~modules - 查询所有可用模块。'}}
|