Как вставить значения "NULL" в базу данных PostgreSQL с помощью Python?

Есть ли хорошая практика для ввода значений ключа NULL в базу данных PostgreSQL, когда переменная None в Python?

Запуск этого запроса:

mycursor.execute('INSERT INTO products (user_id, city_id, product_id, quantity, price) VALUES (%i, %i, %i, %i, %f)' %(user_id, city_id, product_id, quantity, price))

приводит к исключению a TypeError, если user_id равно None.

Как можно вставить NULL в базу данных, когда значение None, используя драйвер psycopg2?

Ответы

Ответ 1

Чтобы вставить нулевые значения в базу данных, у вас есть два варианта:

  • опустить это поле из инструкции INSERT или
  • использовать None

Кроме того: для защиты от SQL-инъекции вы не должны использовать обычную интерполяцию строк для своих запросов.

Вы должны передать два (2) аргумента execute(), например:

mycursor.execute("""INSERT INTO products 
                    (city_id, product_id, quantity, price) 
                    VALUES (%s, %s, %s, %s)""", 
                 (city_id, product_id, quantity, price))

Альтернатива № 2:

user_id = None
mycursor.execute("""INSERT INTO products 
                    (user_id, city_id, product_id, quantity, price) 
                    VALUES (%s, %s, %s, %s, %s)""", 
                 (user_id, city_id, product_id, quantity, price))

Ответ 2

С текущим psycopg вместо None используйте переменную, установленную в 'NULL'.

variable = 'NULL'
insert_query = """insert into my_table values(date'{}',{},{})"""
format_query = insert_query.format('9999-12-31', variable, variable)
curr.execute(format_query)
conn.commit()

>> insert into my_table values(date'9999-12-31',NULL,NULL)

Ответ 3

Более простой подход, который также практичен с большим количеством столбцов:

Пусть row будет список значений для вставки, которые могут содержать None. Чтобы вставить его в PostgreSQL, мы делаем следующее:

values = ','.join(["'" + str(i) + "'" if i else 'NULL' for i in row])
cursor.execute('insert into myTable VALUES ({});'.format(values))
conn.commit()