mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-05-12 08:49:48 +00:00
Code Issues Projects Releases Wiki Activity

[commands] Special handling for when the converter is bool.

Browse Source
This commit is contained in:
Rapptz 2016-01-07 17:31:56 -05:00
parent 542ddc4938
commit b79a4549ab
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
discord/ext/commands/core.py
View File

@ -36,6 +36,15 @@ from .view import quoted_word
__all__ = [ 'Command', 'Group', 'GroupMixin', 'command', 'group', __all__ = [ 'Command', 'Group', 'GroupMixin', 'command', 'group',
'has_role', 'has_permissions', 'has_any_role', 'check' ] 'has_role', 'has_permissions', 'has_any_role', 'check' ]
def _convert_to_bool(argument):
lowered = argument.lower()
if lowered in ('yes', 'y', 'true', 't', '1', 'enable', 'on'):
return True
elif lowered in ('no', 'n', 'false', 'f', '0', 'disable', 'off'):
return False
else:
raise BadArgument(lowered + ' is not a recognised boolean option')
class Command: class Command:
"""A class that implements the protocol for a bot text command. """A class that implements the protocol for a bot text command.
@ -90,6 +99,9 @@ class Command:
return result return result
def do_conversion(self, bot, message, converter, argument): def do_conversion(self, bot, message, converter, argument):
if converter is bool:
return _convert_to_bool(argument)
if converter.__module__.split('.')[0] != 'discord': if converter.__module__.split('.')[0] != 'discord':
return converter(argument) return converter(argument)