使用Python中的sqlite3库进行数据库插入操作
发布时间:2023-12-25 17:20:32
Python中的sqlite3库是一个轻量级的数据库,可以在程序中使用它进行数据库的操作。使用sqlite3库可以进行数据库的创建、连接、表的创建、数据的插入、查询、更新和删除等操作。
下面是一个使用Python中的sqlite3库进行数据库插入操作的例子:
import sqlite3
# 连接到数据库
conn = sqlite3.connect('test.db')
# 创建一个游标对象
cursor = conn.cursor()
# 创建一个表
cursor.execute('''CREATE TABLE IF NOT EXISTS student
(id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
age INTEGER NOT NULL)''')
# 提示用户输入学生信息
name = input('请输入学生姓名:')
age = int(input('请输入学生年龄:'))
# 将学生信息插入到表中
cursor.execute("INSERT INTO student (name, age) VALUES (?, ?)", (name, age))
# 提交事务
conn.commit()
# 打印插入成功的消息
print("学生信息插入成功!")
# 关闭游标和连接
cursor.close()
conn.close()
该例子的功能是创建一个名为student的表,包含id、name和age三个字段。然后提示用户输入学生姓名和年龄,将用户输入的学生信息插入到student表中。
首先,我们需要调用sqlite3.connect()方法创建一个数据库连接对象,指定数据库文件名(例如这里的test.db)。然后,使用cursor()方法创建一个游标对象,用于执行数据库的操作。接着,我们使用cursor.execute()方法创建一个名为student的表,包含id、name和age三个字段。该语句使用IF NOT EXISTS表示如果表不存在才创建。然后,我们使用input()函数提示用户输入学生姓名和年龄,并通过int()函数将输入的年龄转换为整型。接下来,使用execute()方法插入学生信息到student表中,并使用?占位符绑定参数(这样可以防止SQL注入攻击)。最后,使用commit()方法提交事务(确认插入操作)。关闭游标和连接,程序结束。
运行以上代码,将会得到如下结果:
请输入学生姓名:John 请输入学生年龄:20 学生信息插入成功!
这样,我们就成功地将学生信息插入到数据库中了。可以使用SQLite Browser等工具查看test.db数据库文件中的数据。
