Python - sqlite3 轻量数据库使用

2022-08-04 15:42:24 浏览数 (1)

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

简介

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

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

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

使用方法

导入模块

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

代码语言:javascript复制
import sqlite3

创建与SQLite数据库的连接

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

代码语言:javascript复制
# 创建与数据库的连接
conn = sqlite3.connect('test.db')

  • 还可以在内存中创建数据库,只要输入特殊参数值:memory:即可,该数据库只存在于内存中,不会生成本地数据库文件。
代码语言:javascript复制
conn = sqlite3.connect(':memory:')

  • 建立与数据库的连接后,需要创建一个游标cursor对象,该对象的.execute()方法可以执行sql命令,让我们能够进行数据操作。
代码语言:javascript复制
#创建一个游标 cursor
cur = conn.cursor()

在SQLite数据库中创建表

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

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

  • 执行以下语句实现:
代码语言:javascript复制
# 建表的sql语句
sql_text_1 = '''CREATE TABLE scores
           (姓名 TEXT,
            班级 TEXT,
            性别 TEXT,
            语文 NUMBER,
            数学 NUMBER,
            英语 NUMBER);'''
# 执行sql语句
cur.execute(sql_text_1)

向表中插入数据

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

  • 执行以下语句插入单条数据:
代码语言:javascript复制
# 插入单条数据
sql_text_2 = "INSERT INTO scores VALUES('A', '一班', '男', 96, 94, 98)"
cur.execute(sql_text_2)

  • 执行以下语句插入多条数据:
代码语言:javascript复制
data = [('B', '一班', '女', 78, 87, 85),
        ('C', '一班', '男', 98, 84, 90),
        ]
cur.executemany('INSERT INTO scores VALUES (?,?,?,?,?,?)', data)

查询数据

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

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

  • 返回:

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

汇总 sqlite 操作
代码语言:javascript复制
* 创建表
    ```
    # 插入user表
    # id int型 主键自增
    # name varchar型 最大长度20 不能为空
    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 你的改动
代码语言:javascript复制
conn.commit()

关闭连接
代码语言:javascript复制
# 关闭游标
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()

该例程获取查询结果集中所有(剩余)的行,返回一个列表。当没有可用的行时,则返回一个空的列表。

参考源码

代码语言:javascript复制
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()

参考资料

  • https://zhuanlan.zhihu.com/p/216285195
  • http://www.sqlite.org/
  • https://www.jianshu.com/p/3769bb47b499

0 人点赞