[Security] Prevent commands in DM - Fixes #60

python-discord · Aug 30, 2018 · 1de1e8d · 1de1e8d
1 parent 0ec3d2b
commit 1de1e8d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/bot/cogs/security.py b/bot/cogs/security.py
@@ -13,10 +13,14 @@ class Security:
     def __init__(self, bot: Bot):
         self.bot = bot
         self.bot.check(self.check_not_bot)  # Global commands check - no bots can run any commands at all
+        self.bot.check(self.check_on_guild)  # Global commands check - commands can't be run in a DM
 
     def check_not_bot(self, ctx: Context):
         return not ctx.author.bot
 
+    def check_on_guild(self, ctx: Context):
+        return ctx.guild is not None
+
 
 def setup(bot):
     bot.add_cog(Security(bot))