86 lines
1.9 KiB
Python
86 lines
1.9 KiB
Python
from typing import Awaitable, Dict
|
|
|
|
import discord
|
|
from discord.ext import commands
|
|
from discord.ext.commands import (
|
|
bot_has_permissions,
|
|
has_permissions,
|
|
is_owner,
|
|
)
|
|
|
|
from tuxbot.core.utils.functions.extra import ContextPlus
|
|
|
|
__all__ = [
|
|
"bot_has_permissions",
|
|
"has_permissions",
|
|
"is_owner",
|
|
"is_mod",
|
|
"is_admin",
|
|
"check_permissions",
|
|
"guild_owner_or_permissions",
|
|
]
|
|
|
|
|
|
def is_mod():
|
|
"""Is the user a moderator ?
|
|
|
|
"""
|
|
|
|
async def pred(ctx):
|
|
if await ctx.bot.is_owner(ctx.author):
|
|
return True
|
|
permissions: discord.Permissions = ctx.channel.permissions_for(ctx.author)
|
|
return permissions.manage_messages
|
|
|
|
return commands.check(pred)
|
|
|
|
|
|
def is_admin():
|
|
"""Is the user admin ?
|
|
|
|
"""
|
|
|
|
async def pred(ctx):
|
|
if await ctx.bot.is_owner(ctx.author):
|
|
return True
|
|
permissions: discord.Permissions = ctx.channel.permissions_for(ctx.author)
|
|
return permissions.administrator
|
|
|
|
return commands.check(pred)
|
|
|
|
|
|
async def check_permissions(ctx: "ContextPlus", **perms: Dict[str, bool]):
|
|
"""Does a user have any perms ?
|
|
|
|
Parameters
|
|
----------
|
|
ctx:ContextPlus
|
|
Command context.
|
|
**perms:dict
|
|
Perms to verify.
|
|
"""
|
|
if await ctx.bot.is_owner(ctx.author):
|
|
return True
|
|
|
|
elif not perms:
|
|
return False
|
|
resolved = ctx.channel.permissions_for(ctx.author)
|
|
|
|
return all(getattr(resolved, name, None) == value for name, value in perms.items())
|
|
|
|
|
|
def guild_owner_or_permissions(**perms: Dict[str, bool]):
|
|
"""Is a user the guild's owner or does this user have any perms ?
|
|
|
|
Parameters
|
|
----------
|
|
**perms:dict
|
|
Perms to verify.
|
|
"""
|
|
|
|
async def pred(ctx):
|
|
if ctx.author is ctx.guild.owner:
|
|
return True
|
|
return await check_permissions(ctx, **perms)
|
|
|
|
return commands.check(pred)
|