Python - sqlite3 轻量数据库使用

为为为什么

发布于 2022-08-04 15:42:24

1.6K0

发布于 2022-08-04 15:42:24

文章被收录于专栏：又见苍岚

SQLite是python自带的数据库，不需要任何配置，使用sqlite3模块就可以驱动，本文记录使用方法。

简介

sqlite3模块不同于PyMySQL模块，PyMySQL是一个python与mysql的沟通管道，需要你在本地安装配置好mysql才能使用，SQLite是python自带的数据库，不需要任何配置。

官网：http://www.sqlite.org/

本文我们将进行连接 SQLite数据库、创建表、插入数据、读取数据、修改数据等操作。

使用方法

导入模块

sqlite3是内置模块，所以不需要安装的，直接import导入即可：

import sqlite3

创建与SQLite数据库的连接

使用sqlite3.connect()函数连接数据库，返回一个Connection对象，我们就是通过这个对象与数据库进行交互。数据库文件的格式是filename.db，如果该数据库文件不存在，那么它会被自动创建。该数据库文件是放在电脑硬盘里的，你可以自定义路径，后续操作产生的所有数据都会保存在该文件中。

# 创建与数据库的连接
conn = sqlite3.connect('test.db')

还可以在内存中创建数据库，只要输入特殊参数值:memory:即可，该数据库只存在于内存中，不会生成本地数据库文件。

conn = sqlite3.connect(':memory:')

建立与数据库的连接后，需要创建一个游标cursor对象，该对象的.execute()方法可以执行sql命令，让我们能够进行数据操作。

#创建一个游标 cursor
cur = conn.cursor()

在SQLite数据库中创建表

这里就要执行sql的建表语句了，我们先创建一张如下的学生成绩表-scores：

该表目前只有字段名和数据类型，没有数据

执行以下语句实现：

# 建表的sql语句
sql_text_1 = '''CREATE TABLE scores
           (姓名 TEXT,
            班级 TEXT,
            性别 TEXT,
            语文 NUMBER,
            数学 NUMBER,
            英语 NUMBER);'''
# 执行sql语句
cur.execute(sql_text_1)

向表中插入数据

建完表-scores之后，只有表的骨架，这时候需要向表中插入数据

执行以下语句插入单条数据：

# 插入单条数据
sql_text_2 = "INSERT INTO scores VALUES('A', '一班', '男', 96, 94, 98)"
cur.execute(sql_text_2)

执行以下语句插入多条数据：

data = [('B', '一班', '女', 78, 87, 85),
        ('C', '一班', '男', 98, 84, 90),
        ]
cur.executemany('INSERT INTO scores VALUES (?,?,?,?,?,?)', data)

查询数据

我们已经建好表，并且插入了三条数据，现在来查询特定条件下的数据：

# 查询数学成绩大于90分的学生
sql_text_3 = "SELECT * FROM scores WHERE 数学>90"
cur.execute(sql_text_3)
# 获取查询结果
cur.fetchall()

备注：获取查询结果一般可用.fetchone()方法（获取第一条），或者用.fetchall()方法（获取所有条）。

汇总 sqlite 操作

* 创建表
    ```
    # 插入user表
    # id int型　主键自增
    # name varchar型　最大长度２０　不能为空
    cursor.execute('''create table user(id integer primary key autoincrement,name varchar(20) not null)''')
    ```
* 插入记录
    ```
    # 插入一条id=1 name='xiaoqiang'的记录
    cursor.execute('''insert into user(id,name) values(1,'xiaoqiang')''')
    ```
* 查找记录
    ```
    # 查找user表中id=1的记录
    cursor.execute('''select * from user where id=1''')
    # 获得结果
    values = cursor.fetchall()
    values
    [(u'1', u'Michael')]
    ```
* 删除记录
    ```
    # 删除id=1的记录
    sursor.excute('''delete from user where id=1''')
    ```
* 修改记录
    ```
    # 修改id=1记录中的name为xiaoming
    sursor.excute('''update user set name='xiaoming' where id=1''')
    ```

提交事务

连接完数据库并不会自动提交，所以需要手动 commit 你的改动

conn.commit()

关闭连接

# 关闭游标
cur.close()
# 关闭连接
conn.close()

模块 API

以下是重要的 sqlite3 模块程序，可以满足您在 Python 程序中使用 SQLite 数据库的需求。如果您需要了解更多细节，请查看 Python sqlite3 模块的官方文档。

序号	API	描述
1	sqlite3.connect(database [,timeout ,other optional arguments])	该 API 打开一个到 SQLite 数据库文件 database 的链接。您可以使用 “:memory:” 来在 RAM 中打开一个到 database 的数据库连接，而不是在磁盘上打开。如果数据库成功打开，则返回一个连接对象。当一个数据库被多个连接访问，且其中一个修改了数据库，此时 SQLite 数据库被锁定，直到事务提交。timeout 参数表示连接等待锁定的持续时间，直到发生异常断开连接。timeout 参数默认是 5.0（5 秒）。如果给定的数据库名称 filename 不存在，则该调用将创建一个数据库。如果您不想在当前目录中创建数据库，那么您可以指定带有路径的文件名，这样您就能在任意地方创建数据库。
2	connection.cursor([cursorClass])	该例程创建一个 cursor，将在 Python 数据库编程中用到。该方法接受一个单一的可选的参数 cursorClass。如果提供了该参数，则它必须是一个扩展自 sqlite3.Cursor 的自定义的 cursor 类。
3	cursor.execute(sql [, optional parameters])	该例程执行一个 SQL 语句。该 SQL 语句可以被参数化（即使用占位符代替 SQL 文本）。sqlite3 模块支持两种类型的占位符：问号和命名占位符（命名样式）。例如：cursor.execute(“insert into people values (?, ?)”, (who, age))
4	connection.execute(sql [, optional parameters])	该例程是上面执行的由光标（cursor）对象提供的方法的快捷方式，它通过调用光标（cursor）方法创建了一个中间的光标对象，然后通过给定的参数调用光标的 execute 方法。
5	cursor.executemany(sql, seq_of_parameters)	该例程对 seq_of_parameters 中的所有参数或映射执行一个 SQL 命令。
6	connection.executemany(sql[, parameters])	该例程是一个由调用光标（cursor）方法创建的中间的光标对象的快捷方式，然后通过给定的参数调用光标的 executemany 方法。
7	cursor.executescript(sql_script)	该例程一旦接收到脚本，会执行多个 SQL 语句。它首先执行 COMMIT 语句，然后执行作为参数传入的 SQL 脚本。所有的 SQL 语句应该用分号 ; 分隔。
8	connection.executescript(sql_script)	该例程是一个由调用光标（cursor）方法创建的中间的光标对象的快捷方式，然后通过给定的参数调用光标的 executescript 方法。
9	connection.total_changes()	该例程返回自数据库连接打开以来被修改、插入或删除的数据库总行数。
10	connection.commit()	该方法提交当前的事务。如果您未调用该方法，那么自您上一次调用 commit() 以来所做的任何动作对其他数据库连接来说是不可见的。
11	connection.rollback()	该方法回滚自上一次调用 commit() 以来对数据库所做的更改。
12	connection.close()	该方法关闭数据库连接。请注意，这不会自动调用 commit()。如果您之前未调用 commit() 方法，就直接关闭数据库连接，您所做的所有更改将全部丢失！
13	cursor.fetchone()	该方法获取查询结果集中的下一行，返回一个单一的序列，当没有更多可用的数据时，则返回 None。
14	cursor.fetchmany([size=cursor.arraysize])	该方法获取查询结果集中的下一行组，返回一个列表。当没有更多的可用的行时，则返回一个空的列表。该方法尝试获取由 size 参数指定的尽可能多的行。
15	cursor.fetchall()	该例程获取查询结果集中所有（剩余）的行，返回一个列表。当没有可用的行时，则返回一个空的列表。

参考源码

import sqlite3


if __name__ == '__main__':
    # 创建 / 加载硬盘数据库链接
    conn = sqlite3.connect('test.db')

    # 内存数据库
    # conn = sqlite3.connect(':memory:')

    #创建一个游标 cursor
    cur = conn.cursor()

    # 建表的sql语句
    sql_text_1 = '''CREATE TABLE scores
            (姓名 TEXT,
                班级 TEXT,
                性别 TEXT,
                语文 NUMBER,
                数学 NUMBER,
                英语 NUMBER);'''
    # 执行sql语句
    cur.execute(sql_text_1)

    # 插入数据
    # 插入单条数据
    sql_text_2 = "INSERT INTO scores VALUES('A', '一班', '男', 96, 94, 98)"
    cur.execute(sql_text_2)

    # 插入多条数据
    data = [('B', '一班', '女', 78, 87, 85),
        ('C', '一班', '男', 98, 84, 90),
        ]
    cur.executemany('INSERT INTO scores VALUES (?,?,?,?,?,?)', data)

    # 手动 commit 改动
    conn.commit()

    # 查询数据
    # 查询数学成绩大于90分的学生
    sql_text_3 = "SELECT * FROM scores WHERE 数学>90"
    cur.execute(sql_text_3)

    # 获取查询结果
    cur.fetchall()

    # 关闭游标
    cur.close()

    # 关闭连接
    conn.close()