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)