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