39 lines
1.7 KiB
Python
39 lines
1.7 KiB
Python
from core.builtins import Bot
|
|
from core.component import module
|
|
|
|
p = module('prefix', required_admin=True, base=True)
|
|
|
|
|
|
@p.command('add <prefix> {{core.help.prefix.add}}',
|
|
'remove <prefix> {{core.help.prefix.remove}}',
|
|
'reset {{core.help.prefix.reset}}',
|
|
'list {{core.help.prefix.list}}')
|
|
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.send_message(msg.locale.t("core.message.prefix.add.success", prefix=arg1))
|
|
else:
|
|
await msg.send_message(msg.locale.t("core.message.prefix.add.already"))
|
|
elif 'remove' in msg.parsed_msg:
|
|
if arg1:
|
|
if arg1 in prefixes:
|
|
prefixes.remove(arg1)
|
|
msg.data.edit_option('command_prefix', prefixes)
|
|
await msg.send_message(msg.locale.t("core.message.prefix.remove.success") + arg1)
|
|
else:
|
|
await msg.send_message(msg.locale.t("core.message.prefix.remove.not_found"))
|
|
elif 'reset' in msg.parsed_msg:
|
|
msg.data.edit_option('command_prefix', [])
|
|
await msg.send_message(msg.locale.t("core.message.prefix.reset"))
|
|
elif 'list' in msg.parsed_msg:
|
|
if len(prefixes) == 0:
|
|
await msg.send_message(msg.locale.t("core.message.prefix.list.none"))
|
|
else:
|
|
await msg.finish(msg.locale.t('core.message.prefix.list', prefixes=', '.join(prefixes)))
|