Как проверить, есть ли у автора роль x на сервере x? - PullRequest
Новая
       30

Как проверить, есть ли у автора роль x на сервере x?

2 голосов
/ 25 мая 2019

Я пытаюсь проверить, имеет ли автор роль x на x сервере, чтобы иметь возможность использовать команду, но не может заставить ее работать. Я могу только проверить, имеет ли Author роль x на этом сервере.

Идея: если у пользователя есть роль x в сервере x, он сможет использовать специальную команду на любых серверах. 

 role_id = 569712344226201600
 author = ctx.message.author

 if role_id in [role.id for role in author.roles]:
     await ctx.send("message")

Ответы [ 2 ]

0 голосов
/ 16 июля 2019

Вы можете просто получить объект роли. 

role = discord.utils.get(ctx.guild.roles, id=role_id_here)

Тогда просто: 

if role in ctx.author.roles:
...

Ваш бот должен быть на этих серверах.

0 голосов
/ 25 мая 2019

Встроенная проверка has_role не будет работать здесь, потому что она проверяет только роли контекста, в котором была вызвана команда. Достаточно просто написать проверку, которая проверяет роли из конкретный сервер независимо от контекста: 

from discord.utils import get
from discord.ext.commands import check, MissingRole, CommandError

class GuildNotFound(CommandError):
    def __init__(self, guild):
        self.missing_guild = guild
        message = f"I could not resolve guild: {guild}"
        super().__init__(message)


class IsNotMember(CommandError):
    def __init__(self, user, guild):
        self.user = user
        self.guild = guild
        message = f"{user} is not a member of guild {guild}"
        super().__init__(message)

def has_role_elsewhere(guild, role):
    def predicate(ctx):
        # Resolve the guild
        if isinstance(guild, int):
            resolved_guild = get(ctx.bot.guilds, id=guild)
        else:
            resolved_guild = get(ctx.bot.guilds, name=guild)
        if resolved_guild is None:
            raise GuildNotFound(guild)

        # Resolve the member
        member = resolved_guild.get_member(ctx.author.id)
        if member is None:
            raise IsNotMember(ctx.author, resolved_guild)

        # Check for the role
        if isinstance(role, int):
            resolved_role = get(member.roles, id=role)
        else:
            resolved_role = get(member.roles, name=role)
        if resolved_role is None:
            raise MissingRole(role)

        return True
    return check(predicate)

Это много кода, но для нашего реального бота нам просто нужно сделать: 

@bot.command()
@has_role_elsewhere(guild=123456, role=569712344226201600)
async def mycommand(ctx):
    ...
...