Добавление проблемы с проверкой - PullRequest
0 голосов
/ 23 апреля 2019

Привет. Я пытаюсь проверить, набрал ли только конкретный пользователь yes или no, затем примените эту роль.

В тот момент, когда добавляется упоминаемая роль, сначала возвращается сообщение с вопросом, хотят ли они добавить эту роль.Однако эта строка вызывает конфликт, скажем, когда другой пользователь печатает да, он применяет роль исходного пользователя.

Это не тот результат, который я ожидал.Итак, как мне добавить чек?

Вот код, с которым я работаю, и строка, в которой у меня возникла проблема.

  if msg is None or msg.content.lower().strip() in ("yes", "y"):
            await author.add_roles(role)
            message = '{} added the role **{}**.'.format(author.display_name, role.name)
            embed = discord.Embed(description=message.format(author.display_name, role.name), colour=0x56e011)
            await ctx.send(embed=embed)
        else:
            embed = discord.Embed(description='Okay I won\'t add the role **{}**.'.format(role.name))
            await ctx.send(embed=embed)
            return

Полный код:

async def add(self, ctx, *, rolename):
    author = ctx.message.author
    role_dict = {
        "blue":556228119565041726,
        "green":556228124719710261,
        "orange":556228127567904790,
        "yellow":556228225320222757}
    role_id = role_dict.get(rolename.lower())
    if not role_id:
        message = 'I cannot find the role **{}**.'
        embed = discord.Embed(description=message.format(rolename))
        await ctx.send(embed=embed)
        return
    role = discord.utils.get(ctx.message.guild.roles, id = role_id)
    if role in author.roles:
        message = 'It looks like you already have the role **{}**.'
        embed = discord.Embed(description=message.format(role.name))
        await ctx.send(embed=embed)
        return

    if role.mentionable:
        message = '**@mention** notifications are enabled for the role **{}**. If you still want to add this role type **Yes** otherwise **No**.'.format(role.name)
        embed = discord.Embed(description=message.format(author.display_name, role.name), colour=0xff8100)
        await ctx.send(embed=embed)
        try:
            msg = await self.bot.wait_for("message", timeout=20.0)
        except asyncio.TimeoutError:
            await ctx.send('Sorry, you took too long. Try again.')
            return 

        if msg is None or msg.content.lower().strip() in ("yes", "y"):
            await author.add_roles(role)
            message = '{} added the role **{}**.'.format(author.display_name, role.name)
            embed = discord.Embed(description=message.format(author.display_name, role.name), colour=0x56e011)
            await ctx.send(embed=embed)
        else:
            embed = discord.Embed(description='Okay I won\'t add the role **{}**.'.format(role.name))
            await ctx.send(embed=embed)
            return
    else:
        await author.add_roles(role)
        message = '{} added the role **{}**.'.format(author.display_name, role.name)
        embed = discord.Embed(description=message.format(author.display_name, role.name), colour=0x56e011)
        await ctx.send(embed=embed)

1 Ответ

1 голос
/ 23 апреля 2019

У меня есть код, который я использую для создания функций проверки для wait_for. Вот что я использую для ожидания сообщений

from collections.abc import Sequence

def make_sequence(seq):
    if seq is None:
        return ()
    if isinstance(seq, Sequence) and not isinstance(seq, str):
        return seq
    else:
        return (seq,)

def message_check(channel=None, author=None, content=None, ignore_bot=True, lower=True):
    channel = make_sequence(channel)
    author = make_sequence(author)
    content = make_sequence(content)
    if lower:
        content = tuple(c.lower() for c in content)
    def check(message):
        if ignore_bot and message.author.bot:
            return False
        if channel and message.channel not in channel:
            return False
        if author and message.author not in author:
            return False
        if content and message.content not in content:
            return False
        return True
    return check

После этого вы можете легко передать требования сообщения, которое хотите получить, на wait_for

check = message_check(author=ctx.author, channel=ctx.channel, content=('y', 'yes'))
msg = await self.bot.wait_for("message", timeout=20.0, check=check)
...