概述

在上一节,我们介绍了Python的shutil模块,包括:shutil模块中一些常用的函数。在这一节,我们将介绍Python的sqlite3模块。sqlite3模块是Python中的内置模块,用于与SQLite数据库交互。SQLite是一个轻量级的磁盘数据库,不需要单独的服务器进程。你可以在多个线程和进程之间共享SQLite数据库,并且它支持事务处理、零配置以及多种数据类型。

下面,我们将逐一介绍sqlite3模块中一些常用的函数和类。

connect()函数

connect()函数用于建立与SQLite数据库的连接。该函数接受一个数据库文件名作为参数,并返回一个连接对象,该对象可用于执行SQL命令和操作数据库。如果指定的数据库文件不存在,connect()函数将创建一个新的数据库文件。

import sqlite3# 连接到数据库文件conn = sqlite3.connect('test.db')

close()函数

close()函数用于关闭数据库连接。一旦完成了与数据库的交互,使用close()函数是一个推荐的做法,因为它可以释放资源并确保连接被正确关闭。

import sqlite3# 连接到数据库文件conn = sqlite3.connect('test.db')# 关闭数据库连接conn.close()

cursor()函数

cursor()函数用于创建一个游标对象,以执行SQL命令和处理结果。

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命令,并不会自动提交更改。

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()函数更有效。

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()函数之前发生了错误,可以选择回滚事务以撤销所有更改。

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()函数来分批获取查询结果。

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。

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’) 来重置游标位置。

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()