python将数据插入sqlite3表

Question

我正在使用 python 3.7 和 sqlite3 将数据写入我的数据库。我有这个功能

def add_item(self, item):
    stmt = "INSERT INTO users (id) VALUES (?)"
    args = (item,)
    self.conn.execute(stmt, args)
    self.conn.commit()

它工作正常，但它只允许我将数据写入 id 列。我怎样才能使这个函数动态化，以便我可以将任何数据添加到任何列？

# new variable to specify column to write to

def add_item(self, item, column):
    stmt = "INSERT INTO users (column) VALUES (?)"
    args = (item,)
    self.conn.execute(stmt, args)
    self.conn.commit()

Answer 1

您需要根据所提供的列和值列表的长度动态构建列名和值占位符列表。

这可以使用str.join 完成，但是您可能需要处理调用者将单个字符串作为列名和单个值传递的情况。

def add_item(self, values, columns):
    # Check for single column insert
    if isinstance(columns, str):
        values, columns = (values,), (columns,)
    assert len(values) == len(columns), 'Mismatch between values and columns'

    template = """INSERT INTO users ({}) VALUES ({})"""

    cols = ','.join([f'"{col}"' for col in columns])
    placeholders = ','.join(['?'] * len(values))
    stmt = template.format(cols, placeholders)

    self.conn.execute(stmt, values)
    self.conn.commit()
    return