前言

数据库是软件系统重要的 “测量点” 之一，而 PyMySQL 则很好的让我们在 Python 代码中去掌控 MySQL，测试数据库。

数据库操作场景

校验接口测试

接口发送请求后，会明确对数据库中的某个字段进行修改，但是响应结果没有该字段的时候

构造测试数据

• 测试数据使用一次就失效
• 测试前，无法保证测试数据是否存在

PyMySQL 操作数据库

安装 PyMySQL

pip install pymysql

操作步骤

1.导包

import pymysql

2.创建连接

conn = pymysql.connect( host , port , user , password , database , charset)

3.创建游标

cursor = conn.cursor()

4.执行 SQL

cursor.execute("SQL语句")

5.关闭游标

cursor.close()

6.关闭连接

conn.close()

事务的概念

事务

是数据库操作的逻辑单元，是一组不可分割的操作集合，要么全部成功提交，要么全部失败回滚。

事务操作

• 提交：conn.commit()

• 回滚：conn.rollback()

建立连接方法

conn = pymysql.connect(host = “”,port = 0,
user = “”,password = “”,database = “”,charset = “”)

• host： 数据库所在的主机，IP 地址

• port：数据库使用的端口

• user：数据库使用的用户

• password：数据库使用的密码

• database：数据库使用的名字

• charset：字符集，常用 UTF-8

• conn：连接数据库的对象

数据库查询

查询操作流程

常用方法

• fetchone()：从结果集中，提取一行

• fetchmany(size)： 从结果集中，提取 size 行

• fetchall()： 从结果集中，提取所有（注意：依旧和游标位置相关，只提取游标位置以下的数据）

• scroll()：游标位置控制

示例：

# 移动到结果集开头 (绝对位置)
cursor.scroll(0, mode='absolute')

# 向前移动3行 (相对位置)
cursor.scroll(3, mode='relative')

数据库增删改

后记

PyMySQL 作为 Python 的库体现了其简洁优雅的特性，使用起来也是异常的舒服便捷。