Как принять аргумент в команду после её использования в discord.py?

Question

[Telmor]

У меня есть код (ниже), и я хочу чтобы человек писать команду (допустим !test1) без аргументов. И после нажатия кнопки бот предлагал ему ввести id или упоминание человека (отдельным сообщением). И приняв этот id он делал с этим человеком те же действия если я бы написал команду: !test1 .

Код:

@bot.command()
async def test1(ctx):
    emb = discord.Embed(title='Принятие', descriptione='Здесь принимается id учасника')
    emb.set_thumbnail(url=ctx.author.avatar_url)
    msg = await ctx.send(embed = emb, 
        components = [
            Button(style = ButtonStyle.gray, label='Выдать Роль'),
            Button(style = ButtonStyle.gray, label='Убрать Роль')
        ])
    responce = await bot.wait_for('button_click', check=lambda message: message.author == ctx.author)
    if responce.component.label == 'Выдать Роль':
        emb = discord.Embed(title='Принятие', descriptione='Здесь принимается id роли')
        emb.set_thumbnail(url=ctx.author.avatar_url)
    if responce.component.label == 'Убрать Роль':
        return

Можете помочь с моим вопросом на примере моего или другого кода?

Answer 1

[Vindicar]

Это непрактично.

Тебе нужно для каждого пользователя хранить команду, которую он сейчас выполняет, хранить шаг (номер аргумента), на котором он сейчас находится, и хранить ранее введённые данные.

И тогда при получении сообщения:
1. проверять, не является ли сообщение командой, если является - обрабатывать отдельно.
2. определить, какой шаг обработки какой команды сейчас активен для этого пользователя
3. интерпретировать введённое сообщение соответственно.

Comments

[Telmor]

А если на практике в коде?

[Vindicar]

Telmor, за кодом туда.
Разве что ты сможешь показать какой-то свой код, тогда его и будем обсуждать.

[Telmor]

Vindicar, Я в вопросе сразу накинул примерный код

[Telmor]

import discord
from discord.embeds import Embed
from discord.ext import commands
from discord.ext.commands import Bot
from discord import member
import asyncio
import os
import string
import json
from discord import Activity, ActivityType
from discord.utils import *
import io
import time
import string
from discord_components import *
from dislash import *
from dislash import InteractionClient, ActionRow, Button, ButtonStyle
import requests
import random
import re

Все мои импорты

[Vindicar]

Telmor, твой код только обрабатывает команду.
Там нет ни хранилища состояния (СУБД или хотя бы тупо словаря), ни диспетчера обработки сообщений - кода, который по id пользователя узнаёт в хранилище состояний, какой аргумент для какой команды сейчас этим пользователем вводится, а потом передаёт управление соответствующему обработчику. В целом, гугли в сторону "конечный автомат", он же "state machine".

А вообще, возвращаюсь в первому пункту: имеет ли это смысл для решаемой задачи? Добавить роль вручную почти наверняка окажется проще и быстрее, чем использовать команду бота.

[Telmor]

Vindicar, Может быть и да. Но я поставил себе задачу сделать команду такую хотя бы 1

Answer 2

[Максим Невзоров]

Так же, как вы получаете действие с кнопки:

message = await bot.wait_for('message', check=lambda message: message.author == ctx.author and message.channel == ctx.channel) # получаем сообщение от автора команды в исходном канале
role = await commands.RoleConverter().convert(ctx, message.content)  # получаем роль через конвертер
await ctx.send(role.name)  # показываем имя роли в чате

Comments

[Telmor]

@bot.command(aliases=['t1'])
async def test1(ctx):
    emb = discord.Embed(title='Роли', descriptione='выберите что хотите сделать со своими ролями')
    emb.set_thumbnail(url=ctx.author.avatar_url)
    msg = await ctx.send(embed = emb, 
        components = [
            Button(style = ButtonStyle.gray, label='Выдать Роль'),
            Button(style = ButtonStyle.gray, label='Убрать Роль')
        ])
    responce = await bot.wait_for('button_click', check=lambda message: message.author == ctx.author)
    if responce.component.label == 'Выдать Роль':
        emb = discord.Embed(title='Роль', descriptione='Укажите **id** или **упоминание** роли которую хотите **выдать** себе.')
        emb.set_thumbnail(url=ctx.author.avatar_url)
        await msg.edit(embed = emb, components=[])
        message = await bot.wait_for('message', check=lambda message: message.author == ctx.author and message.channel == ctx.channel) # получаем сообщение от автора команды в исходном канале
        role = await commands.RoleConverter().convert(ctx, message.content)  # получаем роль через конвертер
        await ctx.author.add_roles(role)  # даём роль
    if responce.component.label == 'Убрать Роль':
        return

Данный код хорошо выдаёт роль которую я напишу в чате. Только вопросы:
1) почему я не могу в этом коде отсылать текст, или редактировать отосланый эмбед.
2) Как удалить упоминание роли(или если переделать человека) после принятия ботом.
3) Как сделать что бы эмбед автоматически менялся после принятия роли.