24 Python的sqlite3模块

概述

        在上一节，我们介绍了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()