From cd87f247d352e2f87bd172e26e2bb0e42168f805 Mon Sep 17 00:00:00 2001 From: Rapptz Date: Fri, 8 Jul 2016 04:24:49 -0400 Subject: [PATCH] [commands] Make Bot.check decorator an actual decorator. --- discord/ext/commands/bot.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/discord/ext/commands/bot.py b/discord/ext/commands/bot.py index 38260a50b..7250f3a68 100644 --- a/discord/ext/commands/bot.py +++ b/discord/ext/commands/bot.py @@ -446,7 +446,7 @@ class Bot(GroupMixin, discord.Client): # global check registration - def check(self): + def check(self, func): """A decorator that adds a global check to the bot. A global check is similar to a :func:`check` that is applied @@ -466,15 +466,13 @@ class Bot(GroupMixin, discord.Client): .. code-block:: python - @bot.check() + @bot.check def whitelist(ctx): return ctx.message.author.id in my_whitelist """ - def decorator(func): - self.add_check(func) - return func - return decorator + self.add_check(func) + return func def add_check(self, func): """Adds a global check to the bot.