Почему я не могу сделать запись в базу данных SQlite?

Question

[char906]

код:

def add_randcode(self, code):
        with self.connection:
            code = self.cursor.execute("SELECT * FROM `subscriptions` WHERE `code` = ?", (code, )).fetchone()
            if code == None:
                code = random.randint(1000,9999)
                self.cursor.execute("INSERT INTO subscriptions VALUES(?)", code,)
            else:
                return code

SQlite выглядит так:

PRAGMA foreign_keys = 0;

CREATE TABLE sqlitestudio_temp_table AS SELECT *
                                          FROM subscriptions;

DROP TABLE subscriptions;

CREATE TABLE subscriptions (
    id         INTEGER   PRIMARY KEY AUTOINCREMENT,
    user_id    INT (255),
    first_name TEXT,
    money      INT       DEFAULT (50),
    status     INT       DEFAULT (5),
    code       INTEGER
);

INSERT INTO subscriptions (
                              id,
                              user_id,
                              first_name,
                              money,
                              status,
                              code
                          )
                          SELECT id,
                                 user_id,
                                 first_name,
                                 money,
                                 status,
                                 code
                            FROM sqlitestudio_temp_table;

DROP TABLE sqlitestudio_temp_table;

PRAGMA foreign_keys = 1;

Вывожу как сообщение вот так вот:

r = db.add_randcode(message.from_user.id)

bot.send_message(message.chat.id, '✅ Добро пожаловать '+str(r)+')

В итоге всегда получаю ✅ Добро пожаловать и 4 рандомные цифры, которые не записываются в БД. Как сделать так чтобы эти 4 цифры записались в базу?

Answer 1

[Ярослав]

Упростите код, уберите все ненужные для отладки строчки (даже если они нужны в работе), чтобы четко было видно, где происходит чудо и почему. Добавьте print'ов чтобы посмотреть, как проходит код и распечатывайте переменные. Дальше, скорее всего, вы сами увидите, где у вас проблема, а если нет - то сможете ее свести до кода с 2-3 строчками.

Возможно, забыли сделать commit() ?

Comments

[char906]

так нечего же упрощать, одна функция на 4 строчки и вывод того что получилось в сообщение, как я понял не работают оба SQL синтаксиса

code = self.cursor.execute("SELECT * FROM `subscriptions` WHERE `code` = ?", (code, )).fetchone()

self.cursor.execute("""INSERT INTO subscriptions(code) 
                VALUES (?)""", (code, ))

почему так я понять пока не могу

[Ярослав]

может быть надо сделать:

self.connection.commit()

[char906]

Ярослав, попробовал, ничего не изменилось, всё так же

code = self.cursor.execute("SELECT * FROM `subscriptions` WHERE `code` = ?", (code, )).fetchone()

выдаёт None, а

self.cursor.execute("""INSERT INTO subscriptions(code) 
                VALUES (?)""", (code, ))

Ничего не записывает в базу

[Ярослав]

И правильно делает, что не записывает.
Я же говорю вам - сделайте commit() - данные сохранятся на диск.

[char906]

Ярослав, я же правильно понимаю, self.connection.commit() ставиться в конце?

def add_randcode(self, code):
        with self.connection:

            code = self.cursor.execute("SELECT * FROM `subscriptions` WHERE `code` = ?", (code, )).fetchone()
            if code == None:
                code = random.randint(1000,9999)
                self.cursor.execute("""INSERT INTO subscriptions(code) 
                VALUES (?)""", (code, ))
                return code
            else:
                return code
        self.connection.commit()

Если это так, то это не помогает

[Александр]

char906, у вас return происходит commit не отрабатывает

[char906]

Александр, а куда тогда впихнуть commit, чтобы он обработался?