Пытаюсь сделать подсчет количество сообщений но не выходит discord.py?

Question

[MIHUTKA]

@commands.Cog.listener()
    async def on_message(self, message):
        if not message.author.bot:
            if not message.content.startswith('*'):
                base = sqlite3.connect('cogs/Warn.db')
                cur = base.cursor()
                cur.execute("""CREATE TABLE IF NOT EXISTS users(
                    guildid INT,
                    id INT,
                    messages INT)
                """)
                base.commit()
                messages = cur.execute(f"SELECT id FROM users WHERE guildid = {message.guild.id} and id={message.author.id}")
                if messages == None:
                    cur.execute(f"INSERT INTO users VALUES ({guild.id}, {author.id}, 0)")
                    base.commit()
                    cur.execute(f"UPDATE users SET messages + 1 WHERE guildid = {message.guild.id} and id={message.author.id}")
                    base.commit()
                else:
                    cur.execute(f"SELECT id FROM users where guildid = {message.guild.id} and id={message.author.id}")
                    if len(message.content) >=  1:
                        cur.execute(f"UPDATE users SET messages = messages + 1 WHERE guildid = {message.guild.id} and id={message.author.id}")
                        base.commit()

Должно заносить все в дб, но туда ничего не пишет

Answer 1

[Vindicar]

1. Запрос CREATE TABLE не нужно делать при каждом сообщении. Перенеси его в код, который вполняется при запуске бота.
2. Не используй f-строки для формирования запросов. Почитай про подстановку параметров запроса.
3. cur.execute() не вернёт тебе None или данные. Вызови метод fetchone(), чтобы получить одну строку данных, или None если данных нет.
4. Если ты используешь sqlite, почитай про синтаксис INSERT ON CONFLICT DO UPDATE. Позволяет выполнить операцию в духе "создай запись, а если такая уже есть - обнови её", и тем самым свернуть твой if в один запрос. В других СУБД есть аналогичная возможность.

Comments

[MIHUTKA]

а можно просто кодом?

[Vindicar]

MIHUTKA, тыща баксов, деньги вперёд.

Кроме шуток.

Если тебе нужно, чтобы кто-то писал код за тебя - тебе на фриланс.
Здесь могут помочь советом, при этом подразумевается что ты хочешь научиться, чтобы потом сделать сам.

[MIHUTKA]

Vindicar, ну просто у меня не получается, хотелось бы помощи твоей

[Vindicar]

MIHUTKA, тогда задавай конкретные вопросы, а не "сделай за меня".
И приводи свои результаты, пусть даже неправильные, или сообщения об ошибках. А то не получаться может очень по-разному.
Для начала исправь косяк с формированием запросов. Я тебе привёл ссылку, там есть примеры, как надо.

[mayton2019]

А что делает эта строка? Не видно полезного эффекта.

cur.execute(f"SELECT id FROM users where guildid = {message.guild.id} and id={message.author.id}")

[Vindicar]

mayton2019, проверяет, есть ли уже такой пользователь. В зависимости от этого делается INSERT или UPDATE. Но это можно свернуть в один запрос UPSERT, так как других действий тут не выполняется.

[mayton2019]

Vindicar, ты результат Select-а куда сохранил?

[Vindicar]

mayton2019, я не автор. И да, я не сразу понял - думал ты про первый запрос, который сохраняется в messages. А там и впрямь дубль.
Туплю.

[MIHUTKA]

Может кому надо, у меня все получилось вот код

@commands.Cog.listener()
    async def on_message(self, message):
        if not message.author.bot:
            if not message.content.startswith('*'):
                base = sqlite3.connect('cogs/Warn.db')
                cur = base.cursor()
                base.execute('CREATE TABLE IF NOT EXISTS users(userid INT, count INT)')
                base.commit()

                messages = cur.execute(f'SELECT userid FROM users WHERE userid == {message.author.id}').fetchone()

                if messages == None:
                    cur.execute(f"INSERT INTO users VALUES ({message.author.id}, 1)")
                    base.commit()
                else:
                    cur.execute(f"SELECT * FROM users WHERE userid={message.author.id}")
                    if len(message.content) >=  1:
                        cur.execute(f"UPDATE users SET count = count + 1 WHERE userid={message.author.id}")
                        base.commit()

[Vindicar]

MIHUTKA, неплохо, но серьёзно, избавься от f-строк в запросах. Вот так правильнее.

messages = cur.execute('SELECT userid FROM users WHERE userid == ?', (message.author.id, )).fetchone()

С числами-то проблем не будет, но если ты так попытаешься строки вставлять, ничего хорошего не выйдет.

[MIHUTKA]

Vindicar, ну теперь не робит

[Vindicar]

MIHUTKA, "не работать" программа может многими способами. Как именно?