From 542ddc49386dbb103ce8172831c37189acf93017 Mon Sep 17 00:00:00 2001 From: Rapptz Date: Thu, 7 Jan 2016 16:20:33 -0500 Subject: [commands] Allow registration of multiple command prefixes. --- discord/ext/commands/bot.py | 20 +++++++++++++++++--- 1 file changed, 17 insertions(+), 3 deletions(-) (limited to 'discord/ext/commands/bot.py') diff --git a/discord/ext/commands/bot.py b/discord/ext/commands/bot.py index c602a204..9d5a09de 100644 --- a/discord/ext/commands/bot.py +++ b/discord/ext/commands/bot.py @@ -43,7 +43,7 @@ class Bot(GroupMixin, discord.Client): This class also subclasses :class:`GroupMixin` to provide the functionality to manage commands. - Parameters + Attributes ----------- command_prefix The command prefix is what the message content must contain initially @@ -51,6 +51,11 @@ class Bot(GroupMixin, discord.Client): indicate what the prefix should be, or a callable that takes in a :class:`discord.Message` as its first parameter and returns the prefix. This is to facilitate "dynamic" command prefixes. + + The command prefix could also be a list or a tuple indicating that + multiple checks for the prefix should be used and the first one to + match will be the invocation prefix. You can get this prefix via + :attr:`Context.prefix`. """ def __init__(self, command_prefix, **options): super().__init__(**options) @@ -199,8 +204,16 @@ class Bot(GroupMixin, discord.Client): return prefix = self._get_prefix(message) - if not view.skip_string(prefix): - return + invoked_prefix = prefix + + if not isinstance(prefix, (tuple, list)): + if not view.skip_string(prefix): + return + else: + invoked_prefix = discord.utils.find(view.skip_string, prefix) + if invoked_prefix is None: + return + invoker = view.get_word() tmp = { @@ -208,6 +221,7 @@ class Bot(GroupMixin, discord.Client): 'invoked_with': invoker, 'message': message, 'view': view, + 'prefix': invoked_prefix } ctx = Context(**tmp) del tmp -- cgit v1.2.3