简介
SQLite 是一个开源的轻型关系型数据库,广泛应用于桌面、移动端和 Web 应用。SQLite 数据库可以直接存储在本地文件中。Python 自带的 sqlite3
模块让操作 SQLite 数据库变得非常简便。
本文将详细介绍如何使用 Python 操作 SQLite 数据库,包括数据库的创建、表的操作、数据的增删改查及一些常见用法。
一、连接数据库
可以直接通过文件路径创建数据库连接(如果文件不存在会自动创建):
Python
import sqlite3
# 创建一个数据库连接(example.db为数据库文件名)
conn = sqlite3.connect('example.db')
如果只想使用内存数据库:
Python
conn = sqlite3.connect(':memory:')
二、创建数据表
使用游标(Cursor)对象执行 SQL 语句:
Python
cur = conn.cursor()
cur.execute('''
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
age INTEGER
);
''')
conn.commit()
三、插入数据
Python
cur.execute('INSERT INTO users (name, age) VALUES (?, ?)', ('Alice', 23))
cur.execute('INSERT INTO users (name, age) VALUES (?, ?)', ('Bob', 30))
conn.commit()
四、查询数据
Python
cur.execute('SELECT * FROM users')
rows = cur.fetchall()
for row in rows:
print(row)
也可以逐行取:
Python
cur.execute('SELECT id, name FROM users')
for id, name in cur:
print(f'User ID: {id}, Name: {name}')
五、更新和删除数据
更新:
Python
cur.execute('UPDATE users SET age = ? WHERE name = ?', (24, 'Alice'))
conn.commit()
删除:
Python
cur.execute('DELETE FROM users WHERE name = ?', ('Bob',))
conn.commit()
六、使用 with 上下文管理
推荐使用 with 语法自动管理数据库连接:
Python
with sqlite3.connect('example.db') as conn:
cur = conn.cursor()
cur.execute('SELECT * FROM users')
print(cur.fetchall())
七、取字典型结果
默认情况下 fetchone()
等得到的是元组。如果想用字段名读取:
Python
conn.row_factory = sqlite3.Row
cur = conn.cursor()
cur.execute('SELECT * FROM users')
row = cur.fetchone()
print(row['name'], row['age'])
八、关闭连接
完成所有操作后,记得关闭游标和连接:
Python
cur.close()
conn.close()
九、全部流程完整代码
汇总一份完整示例脚本:
Python
import sqlite3
# 1. 连接数据库
conn = sqlite3.connect('example.db')
cur = conn.cursor()
# 2. 创建表
cur.execute('''
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
name TEXT NOT NULL,
age INTEGER
)
''')
conn.commit()
# 3. 插入数据
cur.execute('INSERT INTO users (name, age) VALUES (?, ?)', ('Alice', 22))
cur.execute('INSERT INTO users (name, age) VALUES (?, ?)', ('Bob', 28))
conn.commit()
# 4. 查询数据
print('初始数据:')
cur.execute('SELECT * FROM users')
for row in cur.fetchall():
print(row)
# 5. 更新数据
cur.execute('UPDATE users SET age=? WHERE name=?', (23, 'Alice'))
conn.commit()
# 6. 删除数据
cur.execute('DELETE FROM users WHERE name=?', ('Bob',))
conn.commit()
# 7. 再次查询数据
print('更新和删除后的数据:')
cur.execute('SELECT * FROM users')
for row in cur.fetchall():
print(row)
# 8. 使用字典方式获取结果
conn.row_factory = sqlite3.Row
cur = conn.cursor()
cur.execute('SELECT * FROM users')
row = cur.fetchone()
if row:
print('字典方式取数据:', dict(row))
# 9. 关闭连接
cur.close()
conn.close()
说明与进阶
1. 使用 executemany 实现批量操作
Python
users = [('Charlie', 20), ('David', 21), ('Eve', 22)]
cur.executemany('INSERT INTO users (name, age) VALUES (?, ?)', users)
conn.commit()
2. 事务管理
默认情况下,sqlite3
会自动处理事务。对于多步写操作,也可用 with conn:
自动提交或回滚。
3. 防止 SQL 注入
始终使用 参数化查询(如 ?
占位符)传入数据,避免 SQL 注入风险。
结语
本文介绍了 Python 操作 SQLite 数据库的基本方法和完整实用代码,包括数据库连接、表的创建、数据增删改查及进阶用法。SQLite 适合中小型项目及原型开发,非常轻便易用。