mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-07 11:48:55 -05:00
0870403299
API changes: - Cogs must now inherit from `commands.Cog` (see #2151 for discussion and more details) - All functions which are not decorators in the `redbot.core.checks` module are now deprecated in favour of their counterparts in `redbot.core.utils.mod`. This is to make this module more consistent and end the confusing naming convention. - `redbot.core.checks.check_overrides` function is now gone, overrideable checks can now be created with the `@commands.permissions_check` decorator - Command, Group, Cog and Context have some new attributes and methods, but they are for internal use so shouldn't concern cog creators (unless they're making a permissions cog!). - `__permissions_check_before` and `__permissions_check_after` have been replaced: A cog method named `__permissions_hook` will be evaluated as permissions hooks in the same way `__permissions_check_before` previously was. Permissions hooks can also be added/removed/verified through the new `*_permissions_hook()` methods on the bot object, and they will be verified even when permissions is unloaded. - New utility method `redbot.core.utils.chat_formatting.humanize_list` - New dependency [`schema`](https://github.com/keleshev/schema) User-facing changes: - When a `@bot_has_permissions` check fails, the bot will respond saying what permissions were actually missing. - All YAML-related `[p]permissions` subcommands now reside under the `[p]permissions acl` sub-group (tbh I still think the whole cog has too many top-level commands) - The YAML schema for these commands has been changed - A rule cannot be set as allow and deny at the same time (previously this would just default to allow) Documentation: - New documentation for `redbot.core.commands.requires` and `redbot.core.checks` modules - Renewed documentation for the permissions cog - `sphinx.ext.doctest` is now enabled Note: standard discord.py checks will still behave exactly the same way, in fact they are checked before `Requires` is looked at, so they are not overrideable. Signed-off-by: Toby Harradine <tobyharradine@gmail.com>
68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
from redbot.core import commands
|
|
import discord
|
|
|
|
|
|
def mod_or_voice_permissions(**perms):
|
|
async def pred(ctx: commands.Context):
|
|
author = ctx.author
|
|
guild = ctx.guild
|
|
if await ctx.bot.is_owner(author) or guild.owner == author:
|
|
# Author is bot owner or guild owner
|
|
return True
|
|
|
|
admin_role = guild.get_role(await ctx.bot.db.guild(guild).admin_role())
|
|
mod_role = guild.get_role(await ctx.bot.db.guild(guild).mod_role())
|
|
|
|
if admin_role in author.roles or mod_role in author.roles:
|
|
return True
|
|
|
|
for vc in guild.voice_channels:
|
|
resolved = vc.permissions_for(author)
|
|
good = resolved.administrator or all(
|
|
getattr(resolved, name, None) == value for name, value in perms.items()
|
|
)
|
|
if not good:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
return commands.permissions_check(pred)
|
|
|
|
|
|
def admin_or_voice_permissions(**perms):
|
|
async def pred(ctx: commands.Context):
|
|
author = ctx.author
|
|
guild = ctx.guild
|
|
if await ctx.bot.is_owner(author) or guild.owner == author:
|
|
return True
|
|
admin_role = guild.get_role(await ctx.bot.db.guild(guild).admin_role())
|
|
if admin_role in author.roles:
|
|
return True
|
|
for vc in guild.voice_channels:
|
|
resolved = vc.permissions_for(author)
|
|
good = resolved.administrator or all(
|
|
getattr(resolved, name, None) == value for name, value in perms.items()
|
|
)
|
|
if not good:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
return commands.permissions_check(pred)
|
|
|
|
|
|
def bot_has_voice_permissions(**perms):
|
|
async def pred(ctx: commands.Context):
|
|
guild = ctx.guild
|
|
for vc in guild.voice_channels:
|
|
resolved = vc.permissions_for(guild.me)
|
|
good = resolved.administrator or all(
|
|
getattr(resolved, name, None) == value for name, value in perms.items()
|
|
)
|
|
if not good:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
return commands.check(pred)
|