使用 Python 字典向 SQLite 插入数据

news2024/10/1 23:47:36

将Python字典数据插入SQLite数据库中其实有很多好处，例如使用字典的结构可以简化插入数据的代码，使其更加紧凑和易于理解。这降低了代码的复杂性，并使代码更易于维护。那么在我们日常使用中会有哪些问题呢？一起看看我是如何解决的吧。

在这里插入图片描述

问题背景

我正在使用 Python 字典将数据插入到 SQLite 表中。我有一个如下所示的代码段来插入数据，其中 sqlDataDict 是一个字典，其中有16列：

cur.execute(''' INSERT INTO ProductAtt (imgID, productName, col1, col2, col3, col4, col5, col6, 
              col7, col8, col9, col10, col11, col12, col13, col14) 

          VALUES (:imgID, :productName, :col1, :col2, :col3, :col4,:col5, :col6, :col7, :col8, :col9, 
                  :col10, :col11, :col12, :col13, :col14 )''', sqlDataDict)

但在某些情况下，字典中的值可能不是完整的，即有些列可能没有对应的值。当这种情况发生时，我收到错误消息 “You did not supply a value for binding”。

解决方案

要解决这个问题，我们需要一种方法来处理字典中丢失的值。我们可以使用 None 值来表示这些丢失的值，然后在执行 SQL 语句之前将它们添加到字典中。这可以通过以下方式完成：

# 创建一个新的字典，将丢失的值用 None 填充
sqlDataDict = {key: value if value is not None else None for key, value in sqlDataDict.items()}

# 执行 SQL 语句
cur.execute(''' INSERT INTO ProductAtt (imgID, productName, col1, col2, col3, col4, col5, col6, 
              col7, col8, col9, col10, col11, col12, col13, col14) 

          VALUES (:imgID, :productName, :col1, :col2, :col3, :col4,:col5, :col6, :col7, :col8, :col9, 
                  :col10, :col11, :col12, :col13, :col14 )''', sqlDataDict)

现在，当字典中存在丢失的值时，SQL 语句仍然可以正常执行，而不会出现错误。

以下是一个使用此解决方案的完整代码示例：

import sqlite3

conn = sqlite3.connect('database.db')
cur = conn.cursor()

# 创建表
cur.execute('''CREATE TABLE IF NOT EXISTS ProductAtt (
    imgID INTEGER PRIMARY KEY,
    productName TEXT,
    col1 REAL,
    col2 REAL,
    col3 TEXT,
    col4 TEXT,
    col5 TEXT,
    col6 REAL,
    col7 TEXT,
    col8 TEXT,
    col9 TEXT,
    col10 TEXT,
    col11 TEXT,
    col12 TEXT,
    col13 TEXT,
    col14 TEXT
)''')

# 准备数据
data = [
    {'imgID': '1', 'productName': 'Product 1', 'col1': 'Value 1', 'col2': 'Value 2', 'col3': 'Value 3'},
    {'imgID': '2', 'productName': 'Product 2', 'col1': 'Value 4', 'col2': 'Value 5', 'col4': 'Value 6'},
    {'imgID': '3', 'productName': 'Product 3', 'col1': 'Value 7', 'col3': 'Value 8', 'col5': 'Value 9'},
]

# 插入数据
for row in data:
    # 创建一个新的字典，将丢失的值用 None 填充
    sqlDataDict = {key: value if value is not None else None for key, value in row.items()}

    # 执行 SQL 语句
    cur.execute(''' INSERT INTO ProductAtt (imgID, productName, col1, col2, col3, col4, col5, col6, 
                  col7, col8, col9, col10, col11, col12, col13, col14) 

              VALUES (:imgID, :productName, :col1, :col2, :col3, :col4,:col5, :col6, :col7, :col8, :col9, 
                      :col10, :col11, :col12, :col13, :col14 )''', sqlDataDict)

# 提交事务
conn.commit()

# 关闭连接
conn.close()