如何使用 Python 连接 SQLite 数据库？

SQLite是一种轻量级的嵌入式数据库，广泛应用于各种应用程序中。

Python提供了内置的sqlite3模块，使得连接和操作SQLite数据库变得非常简单。

下面我将详细介绍如何使用sqlite3模块来连接SQLite数据库，并提供一些实际开发中的建议和注意事项。

1. 使用 `sqlite3` 连接SQLite数据库

sqlite3 是Python标准库中的一个模块，无需额外安装即可使用。

连接数据库

下面是一个简单的示例，展示如何连接到SQLite数据库：

import sqlite3# 创建数据库连接
conn = sqlite3.connect('example.db')# 创建游标对象
cur = conn.cursor()# 创建表
cur.execute('''CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY,name TEXT NOT NULL,email TEXT NOT NULL)''')# 插入数据
cur.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Alice', 'alice@example.com'))
cur.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Bob', 'bob@example.com'))# 提交事务
conn.commit()# 查询数据
cur.execute("SELECT * FROM users")
rows = cur.fetchall()for row in rows:print(row)# 关闭游标和连接
cur.close()
conn.close()

注意事项

错误处理：在实际开发中，应该添加错误处理机制，以防止数据库操作失败或其他异常情况。

import sqlite3try:conn = sqlite3.connect('example.db')cur = conn.cursor()cur.execute('''CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY,name TEXT NOT NULL,email TEXT NOT NULL)''')cur.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Alice', 'alice@example.com'))cur.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Bob', 'bob@example.com'))conn.commit()cur.execute("SELECT * FROM users")rows = cur.fetchall()for row in rows:print(row)
except sqlite3.Error as e:print(f"Error: {e}")
finally:if cur:cur.close()if conn:conn.close()

上下文管理器：为了确保资源被正确释放，可以使用上下文管理器（with语句）来管理数据库连接和游标。

import sqlite3try:with sqlite3.connect('example.db') as conn:with conn.cursor() as cur:cur.execute('''CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY,name TEXT NOT NULL,email TEXT NOT NULL)''')cur.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Alice', 'alice@example.com'))cur.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Bob', 'bob@example.com'))conn.commit()cur.execute("SELECT * FROM users")rows = cur.fetchall()for row in rows:print(row)
except sqlite3.Error as e:print(f"Error: {e}")

2. 使用参数化查询

为了避免SQL注入攻击，应该使用参数化查询。

import sqlite3try:with sqlite3.connect('example.db') as conn:with conn.cursor() as cur:cur.execute('''CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY,name TEXT NOT NULL,email TEXT NOT NULL)''')cur.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Alice', 'alice@example.com'))cur.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Bob', 'bob@example.com'))conn.commit()cur.execute("SELECT * FROM users WHERE name = ?", ('Alice',))rows = cur.fetchall()for row in rows:print(row)
except sqlite3.Error as e:print(f"Error: {e}")

3. 使用事务

SQLite支持事务操作，可以通过事务来确保数据的一致性和完整性。

import sqlite3try:with sqlite3.connect('example.db') as conn:with conn.cursor() as cur:cur.execute('''CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY,name TEXT NOT NULL,email TEXT NOT NULL)''')conn.execute("BEGIN TRANSACTION")try:cur.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Alice', 'alice@example.com'))cur.execute("INSERT INTO users (name, email) VALUES (?, ?)", ('Bob', 'bob@example.com'))conn.commit()except sqlite3.Error as e:conn.rollback()print(f"Error: {e}")cur.execute("SELECT * FROM users")rows = cur.fetchall()for row in rows:print(row)
except sqlite3.Error as e:print(f"Error: {e}")