First commit

cogs/utils/checks.py

@@ -0,0 +1,69 @@
+from discord.ext import commands
+import discord.utils
+import os.path
+import json
+import __main__
+
+#
+# This is a modified version of checks.py, originally made by Rapptz
+#
+#                 https://github.com/Rapptz
+#          https://github.com/Rapptz/RoboDanny/tree/async
+#
+
+main_path = os.path.dirname(os.path.realpath(__main__.__file__))
+
+try:
+    with open(main_path + "/data/red/settings.json", "r") as f:
+        settings = json.loads(f.read())
+except:
+    settings = {"OWNER" : False, "ADMIN_ROLE" : False, "MOD_ROLE" : False}
+
+def is_owner_check(ctx):
+    return ctx.message.author.id == owner
+
+def is_owner():
+    return commands.check(is_owner_check)
+
+# The permission system of the bot is based on a "just works" basis
+# You have permissions and the bot has permissions. If you meet the permissions
+# required to execute the command (and the bot does as well) then it goes through
+# and you can execute the command.
+# If these checks fail, then there are two fallbacks.
+# A role with the name of Bot Mod and a role with the name of Bot Admin.
+# Having these roles provides you access to certain commands without actually having
+# the permissions required for them.
+# Of course, the owner will always be able to execute commands.
+
+def check_permissions(ctx, perms):
+    if is_owner_check(ctx):
+        return True
+
+    ch = ctx.message.channel
+    author = ctx.message.author
+    resolved = ch.permissions_for(author)
+    return all(getattr(resolved, name, None) == value for name, value in perms.items())
+
+def role_or_permissions(ctx, check, **perms):
+    if check_permissions(ctx, perms):
+        return True
+
+    ch = ctx.message.channel
+    author = ctx.message.author
+    if ch.is_private:
+        return False # can't have roles in PMs
+
+    role = discord.utils.find(check, author.roles)
+    return role is not None
+
+def mod_or_permissions(**perms):
+    def predicate(ctx):
+        return role_or_permissions(ctx, lambda r: r.name in (settings["MOD_ROLE"], settings["ADMIN_ROLE"]), **perms)
+
+    return commands.check(predicate)
+
+def admin_or_permissions(**perms):
+    def predicate(ctx):
+        return role_or_permissions(ctx, lambda r: r.name == settings["ADMIN_ROLE"], **perms)
+
+    return commands.check(predicate)