matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

[commands] Add support for stripping whitespace after the prefix

Browse Source 
This is configured with the strip_after_prefix option in `Bot.__init__`
This commit is contained in:
Rapptz 2021-03-28 21:21:09 -04:00
parent c4c2cbf6d2
commit 862d509d2e
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
discord/ext/commands/bot.py
View File

@ -108,6 +108,7 @@ class BotBase(GroupMixin):
self.description = inspect.cleandoc(description) if description else '' self.description = inspect.cleandoc(description) if description else ''
self.owner_id = options.get('owner_id') self.owner_id = options.get('owner_id')
self.owner_ids = options.get('owner_ids', set()) self.owner_ids = options.get('owner_ids', set())
self.strip_after_prefix = options.get('strip_after_prefix', False)
if self.owner_id and self.owner_ids: if self.owner_id and self.owner_ids:
raise TypeError('Both owner_id and owner_ids are set.') raise TypeError('Both owner_id and owner_ids are set.')
@ -911,6 +912,9 @@ class BotBase(GroupMixin):
# Getting here shouldn't happen # Getting here shouldn't happen
raise raise
if self.strip_after_prefix:
view.skip_ws()
invoker = view.get_word() invoker = view.get_word()
ctx.invoked_with = invoker ctx.invoked_with = invoker
ctx.prefix = invoked_prefix ctx.prefix = invoked_prefix
@ -1041,6 +1045,12 @@ class Bot(BotBase, discord.Client):
for the collection. You cannot set both ``owner_id`` and ``owner_ids``. for the collection. You cannot set both ``owner_id`` and ``owner_ids``.
.. versionadded:: 1.3 .. versionadded:: 1.3
strip_after_prefix: :class:`bool`
Whether to strip whitespace characters after encountering the command
prefix. This allows for ``! hello`` and ``!hello`` to both work if
the ``command_prefix`` is set to ``!``. Defaults to ``False``.
.. versionadded:: 1.7
""" """
pass pass