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

[commands] Make Bot.check decorator an actual decorator.

Browse Source
This commit is contained in:
Rapptz 2016-07-08 04:24:49 -04:00
parent 1aaeeff02b
commit cd87f247d3
1 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -446,7 +446,7 @@ class Bot(GroupMixin, discord.Client):
# global check registration # global check registration
def check(self): def check(self, func):
"""A decorator that adds a global check to the bot. """A decorator that adds a global check to the bot.
A global check is similar to a :func:`check` that is applied A global check is similar to a :func:`check` that is applied
@ -466,15 +466,13 @@ class Bot(GroupMixin, discord.Client):
.. code-block:: python .. code-block:: python
@bot.check() @bot.check
def whitelist(ctx): def whitelist(ctx):
return ctx.message.author.id in my_whitelist return ctx.message.author.id in my_whitelist
""" """
def decorator(func): self.add_check(func)
self.add_check(func) return func
return func
return decorator
def add_check(self, func): def add_check(self, func):
"""Adds a global check to the bot. """Adds a global check to the bot.