mirror of
https://github.com/Cog-Creators/Red-DiscordBot.git
synced 2025-11-06 03:08:55 -05:00
b2497386bb
Core dependencies: - discord.py: Rapptz/discord.py@00a659c6 -> Rapptz/discord.py@00a659c6 - multidict: 4.4.0 -> 4.4.2 [docs] - pyparsing: 2.2.0 -> 2.2.1 - sphinx: 1.7.8 -> 1.7.9 [test] - pytest: 3.7.4 -> 3.8.1 [style] - toml: 0.9.4 -> 0.9.6 ---- I've also replaced usages of `discord.utils.get(guild.roles, id=id)` with the new O(1) `guild.get_role(id)` method. Signed-off-by: Toby <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.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.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)
|