mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 11:18:54 -05:00
6075c5bde0
* Rename `bot.db` as `bot._config` - Continues work towards strong version guarantees - Added methods for cog use for a few things which were previously only accessible via direct access. - Retained private use in a few internal use locations, though most methods were updated away from this. - Updated documentation for shared api token users * changelog * more detail * docstring fixes * Apparently, I forgot to commit something I had locally - + a copy/paste failue in the changelog * *sigh*: * *sigh*
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
"""The checks in this module run on every command."""
|
|
from . import commands
|
|
|
|
|
|
def init_global_checks(bot):
|
|
@bot.check_once
|
|
async def global_perms(ctx):
|
|
"""Check the user is/isn't globally whitelisted/blacklisted."""
|
|
if await bot.is_owner(ctx.author):
|
|
return True
|
|
|
|
whitelist = await bot._config.whitelist()
|
|
if whitelist:
|
|
return ctx.author.id in whitelist
|
|
|
|
return ctx.author.id not in await bot._config.blacklist()
|
|
|
|
@bot.check_once
|
|
async def local_perms(ctx: commands.Context):
|
|
"""Check the user is/isn't locally whitelisted/blacklisted."""
|
|
if await bot.is_owner(ctx.author):
|
|
return True
|
|
elif ctx.guild is None:
|
|
return True
|
|
guild_settings = bot._config.guild(ctx.guild)
|
|
local_blacklist = await guild_settings.blacklist()
|
|
local_whitelist = await guild_settings.whitelist()
|
|
|
|
_ids = [r.id for r in ctx.author.roles if not r.is_default()]
|
|
_ids.append(ctx.author.id)
|
|
if local_whitelist:
|
|
return any(i in local_whitelist for i in _ids)
|
|
|
|
return not any(i in local_blacklist for i in _ids)
|
|
|
|
@bot.check_once
|
|
async def bots(ctx):
|
|
"""Check the user is not another bot."""
|
|
return not ctx.author.bot
|