32 lines
1.4 KiB
Python
32 lines
1.4 KiB
Python
from core.builtins import Bot
|
|
from core.component import module
|
|
|
|
p = module('prefix', required_admin=True, base=True)
|
|
|
|
|
|
@p.command('add <prefix> {设置自定义机器人命令前缀}', 'remove <prefix> {移除自定义机器人命令前缀}',
|
|
'reset {重置自定义机器人命令前缀}')
|
|
async def set_prefix(msg: Bot.MessageSession):
|
|
prefixes = msg.options.get('command_prefix')
|
|
arg1 = msg.parsed_msg.get('<prefix>', False)
|
|
if prefixes is None:
|
|
prefixes = []
|
|
if 'add' in msg.parsed_msg:
|
|
if arg1:
|
|
if arg1 not in prefixes:
|
|
prefixes.append(arg1)
|
|
msg.data.edit_option('command_prefix', prefixes)
|
|
await msg.sendMessage(f'已添加自定义命令前缀:{arg1}\n帮助文档将默认使用该前缀进行展示。')
|
|
else:
|
|
await msg.sendMessage(f'此命令前缀已存在于自定义前缀列表。')
|
|
elif 'remove' in msg.parsed_msg:
|
|
if arg1:
|
|
if arg1 in prefixes:
|
|
prefixes.remove(arg1)
|
|
msg.data.edit_option('command_prefix', prefixes)
|
|
await msg.sendMessage(f'已移除自定义命令前缀:{arg1}')
|
|
else:
|
|
await msg.sendMessage(f'此命令前缀不存在于自定义前缀列表。')
|
|
elif 'reset' in msg.parsed_msg:
|
|
msg.data.edit_option('command_prefix', [])
|
|
await msg.sendMessage('已重置自定义命令前缀列表。')
|