mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 11:18:54 -05:00
d69fd63da7
Ya'll are gonna hate me. * Initial modifications * Add initial setup.py * working setup py help * Modify setup file to package stuff * Move a bunch of shit and fix imports * Fix or skip tests * Must add init files for find_packages to work * Move main to scripts folder and rename * Add shebangs * Copy over translation files * WORKING PIP INSTALL * add dependency information * Hardcoded version for now, will need to figure out a better way to do this * OKAY ITS FINALLY FUCKING WORKING * Add this guy * Fix stuff * Change readme to rst * Remove double sentry opt in * Oopsie * Fix this thing * Aaaand fix test * Aaaand fix test * Fix core cog importing and default cog install path * Adjust readme * change instance name from optional to required * Ayyy let's do more dependency injection
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
"""The checks in this module run on every command."""
|
|
from discord.ext import commands
|
|
|
|
|
|
def init_global_checks(bot):
|
|
|
|
@bot.check
|
|
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.db.whitelist()
|
|
if whitelist:
|
|
return ctx.author.id in whitelist
|
|
|
|
return ctx.author.id not in await bot.db.blacklist()
|
|
|
|
@bot.check
|
|
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.db.guild(ctx.guild)
|
|
local_blacklist = await guild_settings.blacklist()
|
|
local_whitelist = await guild_settings.whitelist()
|
|
|
|
if local_whitelist:
|
|
return ctx.author.id in local_whitelist
|
|
|
|
return ctx.author.id not in local_blacklist
|
|
|
|
@bot.check
|
|
async def bots(ctx):
|
|
"""Check the user is not another bot."""
|
|
return not ctx.author.bot
|