欢迎访问宙启技术站
智能推送

使用Python的FileWriter()将数据写入SQLite数据库的实例

发布时间:2024-01-05 05:23:53

在Python中,我们可以使用SQLite3模块来连接和操作SQLite数据库。要将数据写入SQLite数据库,我们需要使用sqlite3模块中的connect()函数连接到数据库,然后使用execute()方法来执行SQL语句。

下面是使用Python的FileWriter()将数据写入SQLite数据库的示例代码:

import sqlite3

def create_table():
    # 创建数据库连接
    conn = sqlite3.connect('example.db')

    # 创建游标对象
    cursor = conn.cursor()

    # 创建表格
    cursor.execute('''CREATE TABLE IF NOT EXISTS data
                  (id INTEGER PRIMARY KEY AUTOINCREMENT,
                   name TEXT,
                   age INTEGER,
                   email TEXT)''')

    # 提交更改
    conn.commit()

    # 关闭连接
    conn.close()


def insert_data(data):
    # 创建数据库连接
    conn = sqlite3.connect('example.db')

    # 创建游标对象
    cursor = conn.cursor()
    
    # 插入数据
    cursor.execute("INSERT INTO data (name, age, email) VALUES (?, ?, ?)", data)

    # 提交更改
    conn.commit()

    # 关闭连接
    conn.close()


if __name__ == '__main__':
    # 创建数据表
    create_table()

    # 用于存储数据的列表
    data_list = [('John', 25, 'john@example.com'),
                 ('Jane', 30, 'jane@example.com'),
                 ('Mike', 35, 'mike@example.com')]

    # 将数据插入数据库
    for data in data_list:
        insert_data(data)

    print("数据成功写入数据库")

在上面的示例代码中,我们首先创建了一个create_table()函数,用于创建一个名为data的数据表。表格有4个列:id(作为主键)、nameageemail。然后,我们定义了一个insert_data()函数,用于将数据插入数据库。在主程序中,我们定义了一个包含数据的列表,然后使用insert_data()函数将每个元组数据插入数据库中。

最后,我们通过打印消息来确认数据已成功写入数据库。

运行这个示例代码后,你将在当前目录下看到一个名为example.db的SQLite数据库文件。可以使用SQLite管理工具(例如SQLiteStudio)来查看和操作数据库。