Как автоматизировать запись в sqlite на Python, регулируя количество перезаписываемых переменных?

Question

[DRON_161Rus]

Суть в том, чтобы каждый раз не писать conn cur cur.execute и т.д., а просто вводить произвольное количество параметров, которые нужно передать в функцию для добавления в базу данных.

Я не понимаю, как на входе показать, что, например, date не нужно уже записывать, чтобы при этом всё работало (если пробовать None, то это обнуляет запись в базе).

def UPD_in_db(name, date, ank, idd):
    conn = sqlite3.connect('baza.sql')
    cur = conn.cursor()

    cur.execute("UPDATE users SET name = ? date = ? ank = ? WHERE gi = ?", (name, date, ank, idd))
    conn.commit()

    cur.close()
    conn.close()

UPD_in_db(Anton, 05 mar, Olala lalala, id99938)

Answer 1

[Михаил Р.]

import sqlite3

def update_db(**params):
    # Если 'idd' нет в параметрах
    if 'idd' not in params:
        raise ValueError("Ты забыл передать 'idd'! Это обязательно.")

    # Достаем 'idd' из параметров
    user_id = params.pop('idd')

    # Создаем строку для SQL запроса
    set_clause = ', '.join(f"{key} = ?" for key in params.keys())
    query = f"UPDATE users SET {set_clause} WHERE gi = ?"

    # Подготавливаем значения для SQL запроса
    values = list(params.values()) + [user_id]

    # Подключаемся к БД
    with sqlite3.connect('baza.sql') as conn:
        cur = conn.cursor()
        cur.execute(query, values)
        conn.commit()

update_db(name="Антон", date="05 марта", ank="Ой всё!", idd="id99938")
update_db(name="Сергей", idd="id99939")

Comments

[DRON_161Rus]

спасибо за код!
Ковыряю третий час его, перепробовал другие варианты реализации - не идёт запись. Код срабатывает без ошибок, но не записывает ничего.
Такое ошушение, что дело в values, поскольку его принт выглядит так:
['name', 'idd'], хотя должен быть без кавычек, вроде бы

раскавычить пробовал, пока не работает ничего

[DRON_161Rus]

сам затупил.
не надо было мне брать в кавычки переменные))
спасибо ещё раз

Answer 2

[Сергей Горностаев]

Например так как-то

def UPD_in_db(condition, **kwargs):
    with sqlite3.connect('baza.sql') as conn:
        with conn.cursor() as cur:
            items = kwargs.items()  # Для совместимости со старыми версия Python, у которых несортированный словарь
            statement = 'UPDATE users SET ' + ', '.join(k + ' = ?' for k, v in items) + ' WHERE gi = ?'
            params = tuple(v for k,v in items) + (condition,)
            cur.execute(statement, params)
            conn.commit()


UPD_in_db(99938, name='John', ank='lalala')

Comments

[Ипатьев]

я бы только cur тоже передавал как параметр.
коннектиться прямо в функции как-то не алё

[Ипатьев]

и да - commit жи

[Сергей Горностаев]

Ипатьев, про коммит исправил, спасибо. А вот всё остальное - это уже не в контексте этого вопроса. Если делать по уму, то такие функции вообще не пригодятся.

[DRON_161Rus]

Сергей Горностаев, почему не пригодятся? Как было бы по уму? И где почитать об этом?

[o5a]

DRON_161Rus, Потому что в реальности если у вас есть какая-то форма параметров (по сути сама таблица или же ее представление, view, с измененным набором полей), то при ее изменении вы напрямую и записываете полностью все те поля, что в ней и были, не разбираясь, какое изменилось, а какое нет. Если значение не меняли, после update оно просто и будет заполнено старым значением.
Это если вы планируете обновлять именно всю строку данных. Как пример: пользователь меняет свою страницу профиля: какие-то поля изменил, какие-то нет, но в таблицу профиля просто пишете все поля, не разбираясь, какие именно из них изменились.
Если же вы хотите свою функцию использовать как некий универсальный обработчик, т.е. меняется какой-то параметр записи - вызывается эту UPD_in_db с изменением данного параметра, то в таком случае это как правило только 1 параметр и будет, а не какой-то нефиксированный список разных полей.

[DRON_161Rus]

o5a, если параметров много, зачем их все вкидывать? думалось, это оптимизация получится)

[o5a]

DRON_161Rus, может зависеть от базы данных. Но например в Postgres update по сути создает новую запись, так что выгоды от попытки обновить только определенные поля, не будет.