Как купить роль по «айди»?

Question

[LxneSamurai]

Взялся за магазин ролей для дискорд бота, пишу через БД SQLite. Никак в голову не приходит,
как реализовать идею с покупкой роли через её айди (порядковый номер), а не через пинг самой роли.
Вот сам код + команда на добавление ролей в магазин.

connection = sqlite3.connect('server.db')
cursor=connection.cursor()

@bot.event
async def on_ready():
    print('Ready!`')
    cursor.execute("""CREATE TABLE IF NOT EXISTS shop(
            
            role_id INT,  #айди роли
            id INT, #айди сервера
            cost BIGINT #цена роли
            )""")

#Добавление роли
@bot.command(aliases=['ar'])
@commands.has_role(955395334358577162)#РОЛЬ#РОЛЬ#РОЛЬ#РОЛЬ#РОЛЬ#РОЛЬ#РОЛЬ#РОЛЬ#РОЛЬ#РОЛЬ
async def addrole(ctx, role : discord.Role = None, cost: int= None):
        logs=bot.get_channel(955754477300904026) #LOGSLOGSLOGSLOGSLOGSLOGSLOGSLOGS
        if role is None:
                await ctx.send(embed=discord.Embed(
                        timestamp=ctx.message.created_at,
                        colour=discord.Color.dark_red()
                        ).add_field(name=f'<a:noentr:965583104360198205> | Некорректный ввод команды.', 
                        value=f'Укажите роль.'
                        ).set_footer(text="{}".format(ctx.author.name),
                        icon_url=ctx.author.avatar_url),
                        delete_after=5
                        ) 
        else:
                if cost is None:
                        await ctx.send(embed=discord.Embed(
                        timestamp=ctx.message.created_at,
                        colour=discord.Color.dark_red()
                        ).add_field(name=f'<a:noentr:965583104360198205> | Некорректный ввод команды.', 
                        value=f'Укажите стоимость роли.'
                        ).set_footer(text="{}".format(ctx.author.name),
                        icon_url=ctx.author.avatar_url),
                        delete_after=5
                        )
                elif cost <1:
                        await ctx.send(embed=discord.Embed(
                        timestamp=ctx.message.created_at,
                        colour=discord.Color.dark_red()
                        ).add_field(name=f'<a:noentr:965583104360198205> | Некорректный ввод команды.', 
                        value=f'Сумма не может быть меньше **1** <:servvalute:967795640425930822>.'
                        ).set_footer(text="{}".format(ctx.author.name),
                        icon_url=ctx.author.avatar_url),
                        delete_after=5
                        )
                else:
                        cursor.execute("INSERT INTO shop VALUES ({}, {}, {})".format(role.id, ctx.guild.id, cost))
                        connection.commit()

                        await ctx.send(embed=discord.Embed(
                        timestamp=ctx.message.created_at,
                        colour=discord.Color.from_rgb(27, 227, 124)
                        ).add_field(name=f'<a:Yes1:965581422121353216> | Роль успешно добавлена в магазин.', 
                        value=f'Стоимость {role.name} = {cost} <:servvalute:967795640425930822>.'
                        ).set_footer(text="{}".format(ctx.author.name),
                        icon_url=ctx.author.avatar_url),
                        delete_after=5)

                        await logs.send(embed=discord.Embed(
                        timestamp=ctx.message.created_at,
                        colour=discord.Color.from_rgb(27, 227, 124)
                        ).add_field(name=f'<a:Yes1:965581422121353216> | Роль успешно добавлена в магазин.', 
                        value=f'Стоимость {role.name} = {cost} <:servvalute:967795640425930822>.'
                        ).set_footer(text="{}".format(ctx.author.name),
                        icon_url=ctx.author.avatar_url))

#Вызов магазина
@bot.command()
async def shop(ctx):
        emb=discord.Embed(title="Магазин ролей", timestamp=ctx.message.created_at, colour=discord.Color.from_rgb(255, 255, 255))
        emb.set_footer(text="{}".format(ctx.author.name), icon_url=ctx.author.avatar_url)
        counter=0
        if ctx.channel.id==955186277270048808: 
                for row in cursor.execute("SELECT role_id, cost FROM shop WHERE id = {}".format(ctx.guild.id)):
                        if ctx.guild.get_role(row[0]) != None:
                                counter+=1
                                emb.add_field(
                                        name=f"{counter}. Стоимость - {row[1]} <:servvalute:967795640425930822>",
                                        value=f"Роль {ctx.guild.get_role(row[0]).mention}",
                                        inline=False
                                )
                        else:
                                pass

                await ctx.send(embed=emb)

#Покупка роли
@bot.command()
async def buy(ctx, role: discord.Role = None):
        if ctx.channel.id == 955186277270048808: 
                if role is None:
                        await ctx.reply(embed=discord.Embed(
                                timestamp=ctx.message.created_at,
                                colour=discord.Color.dark_red()
                                ).add_field(name=f'<a:noentr:965583104360198205> | Некорректный ввод команды.', 
                                value=f'Укажите роль, которую хотите приобрести.'
                                ).set_footer(text="{}".format(ctx.author.name),
                                icon_url=ctx.author.avatar_url),
                                mention_author=False)
                else:
                        if role in ctx.author.roles:
                                await ctx.reply(embed=discord.Embed(
                                timestamp=ctx.message.created_at,
                                colour=discord.Color.dark_red()
                                ).add_field(name=f'<a:noentr:965583104360198205> | У вас уже имеется такая роль.', 
                                value=f'Попробуйте выбрать другую.'
                                ).set_footer(text="{}".format(ctx.author.name),
                                icon_url=ctx.author.avatar_url),
                                mention_author=False)
                        elif cursor.execute("SELECT cost FROM shop WHERE role_id = {}".format(role.id)).fetchone()[0] > cursor.execute("SELECT cash FROM users WHERE id = {}".format(ctx.author.id)).fetchone()[0]:
                                await ctx.reply(embed=discord.Embed(
                                timestamp=ctx.message.created_at,
                                colour=discord.Color.dark_red()
                                ).add_field(name=f'<a:noentr:965583104360198205> | Недостаточно средств.', 
                                value=f'Накопите немного деньжат и возвращайтесь!'
                                ).set_footer(text="{}".format(ctx.author.name),
                                icon_url=ctx.author.avatar_url),
                                mention_author=False)

                        else:
                                await ctx.author.add_roles(role)
                                cursor.execute("UPDATE users SET cash = cash - {} WHERE id = {}".format(cursor.execute("SELECT cost FROM shop WHERE role_id = {}".format(role.id)).fetchone()[0], ctx.author.id))
                                connection.commit()
                                await ctx.reply(embed=discord.Embed(
                                        timestamp=ctx.message.created_at,
                                        colour=discord.Color.from_rgb(27, 227, 124)
                                        ).add_field(name=f'<a:Yes1:965581422121353216> | Роль успешно приобретена.', 
                                        value=f'Она уже находится в вашем профиле.'
                                        ).set_footer(text="{}".format(ctx.author.name),
                                        icon_url=ctx.author.avatar_url),
                                        mention_author=False)
                                await asyncio.sleep(5)
                                await ctx.message.delete()

Comments

[Dream]

Сделай id у роли и по ней покупку. Код для айди: id INTEGER PRIMARY KEY, и чтобы купить, выводишь в магазине этот айди, а для покупки пользователь должен ввести buy

[LxneSamurai]

Dream, окей, а как этот же параметр ввести в команду по добавлению роли в магазин?

[Dream]

LxneSamurai, id в таблицу бд я имел ввиду, и делать обычные действия с бд

[LxneSamurai]

Dream, ага, понял, сейчас попробую

[LxneSamurai]

Dream, с покупкой проблема. Всё ещё не понимаю, как правильно будет
(rid - порядковый номер роли)

@bot.command()
async def buy(ctx, role : discord.Role = None):
        if ctx.channel.id == 955186277270048808: 
                rid=cursor.execute("SELECT role_id, rid FROM shop WHERE id = {}".format(ctx.guild.id))
                if rid and role is None:
                        await ctx.reply(embed=discord.Embed(
                                timestamp=ctx.message.created_at,
                                colour=discord.Color.dark_red()
                                ).add_field(name=f'<a:noentr:965583104360198205> | Некорректный ввод команды.', 
                                value=f'Укажите роль, которую хотите приобрести.'
                                ).set_footer(text="{}".format(ctx.author.name),
                                icon_url=ctx.author.avatar_url),
                                mention_author=False)
                else:   
                        role=cursor.execute("SELECT role_id, rid FROM shop WHERE id = {}".format(ctx.guild.id))
                        if role in ctx.author.roles:
                                await ctx.reply(embed=discord.Embed(
                                timestamp=ctx.message.created_at,
                                colour=discord.Color.dark_red()
                                ).add_field(name=f'<a:noentr:965583104360198205> | У вас уже имеется такая роль.', 
                                value=f'Попробуйте выбрать другую.'
                                ).set_footer(text="{}".format(ctx.author.name),
                                icon_url=ctx.author.avatar_url),
                                mention_author=False)
                        elif cursor.execute("SELECT cost FROM shop WHERE role_id = {}".format(role.id)).fetchone()[0] > cursor.execute("SELECT cash FROM users WHERE id = {}".format(ctx.author.id)).fetchone()[0]:
                                await ctx.reply(embed=discord.Embed(
                                timestamp=ctx.message.created_at,
                                colour=discord.Color.dark_red()
                                ).add_field(name=f'<a:noentr:965583104360198205> | Недостаточно средств.', 
                                value=f'Накопите немного деньжат и возвращайтесь!'
                                ).set_footer(text="{}".format(ctx.author.name),
                                icon_url=ctx.author.avatar_url),
                                mention_author=False)

                        else:
                                await ctx.author.add_roles(role)
                                cursor.execute("UPDATE users SET cash = cash - {} WHERE id = {}".format(cursor.execute("SELECT cost FROM shop WHERE rid = {}".format(rid)).fetchone()[0], ctx.author.id))
                                connection.commit()
                                await ctx.reply(embed=discord.Embed(
                                        timestamp=ctx.message.created_at,
                                        colour=discord.Color.from_rgb(27, 227, 124)
                                        ).add_field(name=f'<a:Yes1:965581422121353216> | Роль успешно приобретена.', 
                                        value=f'Она уже находится в вашем профиле.'
                                        ).set_footer(text="{}".format(ctx.author.name),
                                        icon_url=ctx.author.avatar_url),
                                        mention_author=False)
                                await asyncio.sleep(5)
                                await ctx.message.delete()

Answer 1

[Vindicar]

У тебя противоречия в запросах к БД. Для начала ответь на такие вопросы:
1. Сколько серверов будет обслуживать бот - один или несколько?
2. Сколько ролей можно покупать на сервере?
3. Баланс пользователя будет свой на каждом сервере или общий?
Исходя из этого и нужно проектировать БД.

Comments

[LxneSamurai]

1. Один
2. Всё зависит от фантазии, но в планах до 20-ти
3. Свой

[Vindicar]

LxneSamurai, так, т.е. один сервер.
Но зачем тогда вносить в БД ctx.guild.id?

[LxneSamurai]

Vindicar, уже убрал, но всё ещё не хочет работать покупка по порядковому номеру

[Vindicar]

LxneSamurai, вот. Т.е. и впрямь 1. Теперь подумаем над таблицами.
Есть таблица баланса:

CREATE TABLE Users (
    id INT PRIMARY KEY,
    cash FLOAT NOT NULL);

Есть таблица ролей:

CREATE TABLE Shop (
    role_id INT PRIMARY KEY,
    price FLOAT NOT NULL);

Есть таблица покупок:

CREATE TABLE Purchases( 
    user INT NOT NULL,
    role INT NOT NULL,
    FOREIGN KEY (user) REFERENCES (Users.id),
    FOREIGN KEY (role) REFERENCES (Roles.id),
    PRIMARY KEY (user, role)
);

У тебя таблицы примерно так создаются?

[LxneSamurai]

Vindicar, нет, таблицы мои намного проще выглядят.

cursor.execute("""CREATE TABLE IF NOT EXISTS users(
        name TEXT,
        id INT,
        cash BIGINT,
        rep INT,
        lvl INT,
        server_id INT
    )""")

    cursor.execute("""CREATE TABLE IF NOT EXISTS shop(
            
            role_id INT,
            id INT,
            cost BIGINT
            )""")

[Vindicar]

LxneSamurai, ну users - допустим. Но опять же, зачем тебе server_id если ты не планируешь поддержку нескольких серверов.
В shop неясно зачем тебе отдельный id, если у тебя есть естественный ключ - dicord id роли уже гарантированно уникален, как и discord id пользователя.
Ограничением первичного ключа PRIMARY KEY лучше не пренебрегать, так как оно гарантирует, что в разных строках будут разные значения этого поля или комбинации полей. Например, без него у тебя может быть несколько записей для одного пользователя, или для одной роли.
Аналогично, FOREIGN KEY указывает, что это поле должно ссылаться на первичный ключ какой-то строки в указанной таблице, и не может иметь произвольное значение.

Без таблицы Purchases можно обойтись, в принципе. Дискорд всё равно хранит роли.
Т.е. у тебя будет что-то типа

CREATE TABLE IF NOT EXISTS users(
        name TEXT,
        id BIGINT PRIMARY KEY,
        cash BIGINT,
        rep INT,
        lvl INT
)
CREATE TABLE IF NOT EXISTS shop(
        id BIGINT PRIMARY KEY,
        cost BIGINT
)

[LxneSamurai]

Vindicar, ок, а как тогда внести данные в БД? В два значения (роль, стоимость) не робит, говорит, что в данной табл. три значения, а я только два заполняю.

+ в моей таблице второй "id" выступал в роли "порядкового номера", чтобы благодаря нему и покупать роль, о чём, собственно, и начался "разговор"

[Vindicar]

LxneSamurai, далее, теперь логика работы. При покупке роли проверяем:
1. Что такая роль существует в магазине, заодно выбираем её стоимость

# обрати внимание, вот как правильно вставлять параметры в запрос
data = cursor.execute("SELECT cost FROM shop WHERE id = ?", (role.id, )).fetchall()
if not data:
    #сообщаем что такая роль не продаётся
    return
cost = data[0][0]  #роль есть, вот её цена

2. Проверяем, что у пользователя такой роли ещё нет:

if role in ctx.member.roles:
    # сообщаем, что роль уже есть
    return

3. Проверяем, что у пользователя хватит денег:

# обрати внимание, вот как правильно вставлять параметры в запрос
data = cursor.execute("SELECT cash FROM users WHERE id = ?", (ctx.member.id, )).fetchall()
if not data:
    #сообщаем что такого пользователя нет
    return
cash = data[0][0]  #пользователь есть, вот его деньги
if cost > cash:
    # сообщаем, что денег мало
    return

4. Проводим продажу

# обрати внимание, вот как правильно вставлять параметры в запрос
cursor.execute("UPDATE users SET cash = cash - ? WHERE id = ?", (cost, ctx.member.id))
# и выдаём роль

[Vindicar]

LxneSamurai, Если ты изменил структуру таблицы, грохни файл с БД. CREATE IF NOT EXIST не отредактирует таблицу, если она уже существует, но имеет не такую структуру.

[LxneSamurai]

Vindicar,
всё та же ошибка, пинговать роль просит

[Vindicar]

LxneSamurai, как называется роль, и какой командой ты пытаешься её купить?
Преобразователь discord.Role принимает пинг, имя роли или discord id роли.

[LxneSamurai]

Vindicar, роль называется «qqq» (тестовый сервер), покупаю через buy(ctx, role: discord.Role = None). Пытался вместо роли поставить что-то типо number : int = None, но тоже не то. Не понимаю, как сделать запрос указывая число (айди/порядковый номер в списке) на нужную роль.

[Vindicar]

LxneSamurai, я имею ввиду, какую команду вводишь.
!buy qqq должно сработать.
А порядковый номер... не, ну ты можешь попробовать сделать

async def buy(ctx, role: typing.Union[discord.Role, int, None] = None):
    if role is None:
        # роль не указана
    elif isinstance(role, int):
        # роль по индексу
    else:
        # роль - это discord.Role, смотрим по role.id

Но не факт, что это сработает. Уж проще тогда вручную анализировать строку.

async def buy(ctx, *, role: str):
    role = role.strip()
    if not role:
        # роль не указана
        return
    try:
        role_number = int(role)
    except ValueError:
        # ищем роль в списке ролей гильдии по имени
    else:
        # роль по индексу

[LxneSamurai]

Vindicar, а, использовал !buy @пинг роли
Почему не сделаю через название: роли на основном сервере будут различные, с отступами, разными шрифтами и тд, поэтому банально на клавиатуре такое не ввести, а копировать может кому-то будет лень

[LxneSamurai]

Vindicar, через порядковый номер действительно не выходит, поэтому решил сделать через SelectOption под эмбедом магазина. Как правильно "прописать покупку" в bot.event on_select_option?

@bot.command()
async def shop(ctx):
        r1=discord.utils.get(ctx.guild.roles, id=966041637618847794)
        emb=discord.Embed(title="Магазин ролей", timestamp=ctx.message.created_at, colour=discord.Color.from_rgb(255, 255, 255))
        emb.set_footer(text="{}".format(ctx.author.name), icon_url=ctx.author.avatar_url)
        role1=discord.utils.get(ctx.guild.roles, id=966041637618847794)
        role2=discord.utils.get(ctx.guild.roles, id=966046291740217384)
        counter=0
        if ctx.channel.id==955186277270048808:
                for row in cursor.execute("SELECT role_id, cost FROM shop"):
                        if ctx.guild.get_role(row[0]) != None:
                                counter+=1
                                emb.add_field(
                                        name=f"{counter}. Стоимость - {row[1]} <:servvalute:967795640425930822>",
                                        value=f"Роль {ctx.guild.get_role(row[0]).mention}",
                                        inline=False
                                )
                        else:
                                pass
                counter+=1
                await ctx.send(embed=emb, components=[
                        Select(
                                placeholder="Выберите роль, которую хотите приобрести.",
                                options=[
                                        SelectOption(label=f"{role1}", value="r1"),
                                        SelectOption(label=f"{role2}", value="r2")
                                        
                                ]
                                
                        )
                ])

@bot.event
async def on_select_option(ctx, interaction):
        if interaction.channel == ctx.channel:
                if interaction.values[0]=="r1":
                        role=discord.utils.get(ctx.guild.roles, id=966041637618847794)
                        cursor.execute("UPDATE users SET cash = cash - {} WHERE id = {}".format(cursor.execute("SELECT cost FROM shop WHERE role_id = {}".format(role.id)).fetchone()[0], ctx.author.id))
                        connection.commit()
                        await ctx.author.add_roles(role)
                        await ctx.send(f"{role} успешно приборетена!", ephemeral = True)
                elif interaction.values[0]=="r2":
                        role=discord.utils.get(ctx.guild.roles, id=966046291740217384)
                        cursor.execute("UPDATE users SET cash = cash - {} WHERE id = {}".format(cursor.execute("SELECT cost FROM shop WHERE role_id = {}".format(role.id)).fetchone()[0], ctx.author.id))
                        connection.commit()
                        await ctx.author.add_roles(role)
                        await ctx.send(f"{role} успешно приборетена!", ephemeral = True)

[Vindicar]

LxneSamurai, без понятия, никогда эмбеды не использовал. А что у тебя не так с индексом, там же просто?

CREATE TABLE IF NOT EXISTS shop(
        id BIGINT PRIMARY KEY,
        idx INT UNIQUE,
        cost BIGINT
)

async def buy(ctx, index: int):
    ....
    data = cursor.execute("SELECT cost FROM shop WHERE idx = ?", (index, )).fetchall()
    # ну и далее по тексту