概述
在上一节,我们介绍了Python的shutil模块,包括:shutil模块中一些常用的函数。在这一节,我们将介绍Python的sqlite3模块。sqlite3模块是Python中的内置模块,用于与SQLite数据库交互。SQLite是一个轻量级的磁盘数据库,不需要单独的服务器进程。你可以在多个线程和进程之间共享SQLite数据库,并且它支持事务处理、零配置以及多种数据类型。
下面,我们将逐一介绍sqlite3模块中一些常用的函数和类。
connect()函数
connect()函数用于建立与SQLite数据库的连接。该函数接受一个数据库文件名作为参数,并返回一个连接对象,该对象可用于执行SQL命令和操作数据库。如果指定的数据库文件不存在,connect()函数将创建一个新的数据库文件。
python
import sqlite3
# 连接到数据库文件
conn = sqlite3.connect('test.db')
close()函数
close()函数用于关闭数据库连接。一旦完成了与数据库的交互,使用close()函数是一个推荐的做法,因为它可以释放资源并确保连接被正确关闭。
python
import sqlite3
# 连接到数据库文件
conn = sqlite3.connect('test.db')
# 关闭数据库连接
conn.close()
cursor()函数
cursor()函数用于创建一个游标对象,以执行SQL命令和处理结果。
python
import sqlite3
# 连接到数据库文件
conn = sqlite3.connect('test.db')
# 创建游标对象
cursor = conn.cursor()
# 关闭游标
cursor.close()
# 关闭数据库连接
conn.close()
execute()函数
execute()函数是游标对象(Cursor)的一个方法,用于执行SQL命令。使用execute()函数,可以执行各种SQL命令,比如:SELECT、INSERT、UPDATE、DELETE等。注意:execute()函数只是执行SQL命令,并不会自动提交更改。
python
import sqlite3
# 连接到数据库文件
conn = sqlite3.connect('test.db')
# 创建游标对象
cursor = conn.cursor()
# 执行SQL命令
cursor.execute('CREATE TABLE IF NOT EXISTS users(id INTEGER PRIMARY KEY, name TEXT)')
# 关闭游标
cursor.close()
# 关闭数据库连接
conn.close()
executemany()函数
executemany()函数是游标对象(Cursor)的一个方法,用于执行相同的SQL命令多次,每次使用来自序列的不同参数。使用executemany()函数,可以在一次数据库交互中执行多次插入、更新或删除操作,这通常比重复执行execute()函数更有效。
python
import sqlite3
# 连接到数据库文件
conn = sqlite3.connect('test.db')
# 创建游标对象
cursor = conn.cursor()
# 执行SQL命令多次
users = [('Jack',), ('Tom',), ('Mike',)]
cursor.executemany("INSERT INTO users (name) VALUES (?)", users)
# 关闭游标
cursor.close()
# 关闭数据库连接
conn.close()
commit()函数
commit()函数是数据库连接对象(Connection)的一个方法,用于提交事务。在SQLite中,事务是一组数据库操作,要么全部成功执行,要么全部回滚(撤销)。当我们在SQLite中执行多个操作时,可以使用事务来确保数据的一致性。
使用commit()函数,可以提交之前执行的所有数据库操作。一旦调用了commit()函数,所有未提交的更改都会被保存到数据库中。如果在调用commit()函数之前发生了错误,可以选择回滚事务以撤销所有更改。
python
import sqlite3
# 连接到数据库文件
conn = sqlite3.connect('test.db')
# 创建游标对象
cursor = conn.cursor()
# 执行一些数据库操作
cursor.execute('CREATE TABLE IF NOT EXISTS users (id INTEGER PRIMARY KEY, name TEXT)')
cursor.execute("INSERT INTO users (name) VALUES ('Jack')")
cursor.execute("INSERT INTO users (name) VALUES ('Tom')")
# 提交更改
conn.commit()
# 关闭游标
cursor.close()
# 关闭数据库连接
conn.close()
fetchall()函数
fetchall()函数是游标对象(Cursor)的一个方法,用于获取查询结果集中的所有行。使用该函数,可以一次性获取查询结果集中的所有行,并以列表的形式返回。每行数据表示为一个元组或列表,其中包含了该行的各个列的值。
注意:如果查询结果集非常大,使用fetchall()函数可能会消耗较多的内存。在这种情况下,可以考虑使用 fetchone()或fetchmany()函数来分批获取查询结果。
python
import sqlite3
# 连接到数据库文件
conn = sqlite3.connect('test.db')
# 创建游标对象
cursor = conn.cursor()
# 执行查询
cursor.execute('SELECT * FROM users')
# 获取查询结果集中的所有行
rows = cursor.fetchall()
# 输出查询结果
for row in rows:
print(row)
# 关闭游标
cursor.close()
# 关闭数据库连接
conn.close()
fetchone()函数
fetchone()函数是游标对象(Cursor)的一个方法,用于获取查询结果集中的下一行。使用该函数,可以逐行获取查询结果集中的数据。每次调用fetchone()函数,它会返回结果集中的下一行数据。当没有更多的行可用时,它将返回None。
python
import sqlite3
# 连接到数据库文件
conn = sqlite3.connect('test.db')
# 创建游标对象
cursor = conn.cursor()
# 执行查询
cursor.execute('SELECT * FROM users')
# 获取查询结果集中的下一行数据
row = cursor.fetchone()
# 输出查询结果
while row:
print(row)
row = cursor.fetchone()
# 关闭游标
cursor.close()
# 关闭数据库连接
conn.close()
fetchmany()函数
fetchmany(size)函数是游标对象(Cursor)的一个方法,用于获取查询结果集中的多行数据,最多获取size行。注意:每次调用 fetchmany()都会从上次获取数据的地方开始继续获取,也就是说,如果先调用了fetchmany(5),然后再调用 fetchmany(5),那么第二次调用将返回查询结果的第6-10行。如果想要重新从头开始获取数据,需要先调用 cursor.scroll(0, mode='absolute') 来重置游标位置。
python
import sqlite3
# 连接到数据库文件
conn = sqlite3.connect('test.db')
# 创建游标对象
cursor = conn.cursor()
# 执行查询
cursor.execute('SELECT * FROM users')
# 最多获取5行数据
rows = cursor.fetchmany(5)
# 输出查询结果
for row in rows:
print(row)
# 关闭游标
cursor.close()
# 关闭数据库连接
conn.close()