本文将详细介绍如何使用sqlite3模块在Python中进行SQLite数据库的基本操作，包括创建数据库、创建表、插入数据、查询数据、更新数据和删除数据等。
一、创建数据库
在Python中使用sqlite3模块创建SQLite数据库非常简单。首先，你需要导入sqlite3模块，然后使用connect()方法连接到数据库。如果数据库文件不存在，sqlite3会自动创建它。
import sqlite3
# 连接到SQLite数据库，如果数据库不存在，则会自动创建
conn = sqlite3.connect('example.db')
# 创建一个Cursor对象，你将使用它来执行所有的SQL命令
cursor = conn.cursor()
# 关闭到数据库的连接
conn.close()
二、创建表
创建表需要使用SQL的CREATE TABLE语句。你可以通过Cursor对象的execute()方法执行SQL命令。
import sqlite3
# 连接到SQLite数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()
# 创建一个名为users的表
cursor.execute('''CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY,
name TEXT NOT NULL,
email TEXT NOT NULL UNIQUE
)''')
# 提交（保存）更改
conn.commit()
# 关闭连接
conn.close()
三、插入数据
插入数据通常使用SQL的INSERT INTO语句。
import sqlite3
# 连接到SQLite数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()
# 插入一条记录
cursor.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Alice', 'alice@example.com'))
# 提交更改
conn.commit()
# 关闭连接
conn.close()
在上述代码中，我们使用了参数化查询（使用问号占位符），这是一种推荐的做法，因为它可以帮助防止SQL注入攻击。
四、查询数据
查询数据使用SQL的SELECT语句。
import sqlite3
# 连接到SQLite数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()
# 执行查询
cursor.execute("SELECT * FROM users")
# 获取查询结果
rows = cursor.fetchall()
for row in rows:
print(row)
# 关闭连接
conn.close()
fetchall()方法返回查询结果的所有行。你也可以使用fetchone()方法获取结果集的下一行，或者使用fetchmany(size)获取指定数量的行。
五、更新数据
更新数据使用SQL的UPDATE语句。
import sqlite3
# 连接到SQLite数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()
# 更新一条记录
cursor.execute("UPDATE users SET email=? WHERE id=?", ('alice@newemail.com', 1))
# 提交更改
conn.commit()
# 关闭连接
conn.close()
六、删除数据
删除数据使用SQL的DELETE语句。
import sqlite3
# 连接到SQLite数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()
# 删除一条记录
cursor.execute("DELETE FROM users WHERE id=?", (1,))
# 提交更改
conn.commit()
# 关闭连接
conn.close()
七、事务处理
在数据库操作中，事务是一个非常重要的概念。事务是一个作为单个逻辑单元执行的工作单位。事务保证了一组操作的原子性：要么完全执行，要么完全不执行。在sqlite3中，你可以使用commit()方法提交事务，使用rollback()方法回滚事务。
import sqlite3
# 连接到SQLite数据库
conn = sqlite3.connect('example.db')
cursor = conn.cursor()
try:
# 开始一系列数据库操作
cursor.execute("UPDATE users SET email=? WHERE id=?", ('alice@newemail.com', 1))
# ... 其他操作 ...
# 如果所有操作都成功，则提交事务
conn.commit()
except Exception as e:
# 如果在操作过程中发生任何异常，回滚事务
print(f"An error occurred: {e}")
conn.rollback()
finally:
# 关闭连接
conn.close()
在上述代码中，我们使用try-except-finally结构来确保事务的完整性。如果在执行数据库操作期间发生任何异常，我们将回滚事务以撤销所有更改，并打印错误信息。如果所有操作都成功完成，则提交事务以保存更改。
八、使用sqlite3的Row工厂
默认情况下，当你从sqlite3查询中检索数据时，结果将被返回为元组。然而，有时你可能希望将结果作为字典，其中列名是键，对应的值是值。为此，你可以使用sqlite3的Row工厂。
import sqlite3
# 连接到SQLite数据库
conn = sqlite3.connect('example.db')
# 设置Row工厂，以便查询结果返回字典
conn.row_factory = sqlite3.Row
cursor = conn.cursor()
# 执行查询
cursor.execute("SELECT * FROM users")
# 获取查询结果，现在每一行都是一个字典
rows = cursor.fetchall()
for row in rows:
print(row['name'], row['email'])
# 关闭连接
conn.close()
总结
sqlite3模块为Python提供了强大的SQLite数据库操作能力。通过简单的API调用，你可以执行创建数据库、表、插入、查询、更新和删除数据等操作。结合异常处理和事务管理，你可以确保数据库操作的健壮性和数据完整性。在小型应用和原型设计中，SQLite通常是一个很好的选择，因为它轻量级、易于使用，并且不需要单独配置和管理数据库服务器。
