MySQL数据库连接的Python脚本

在接口测试的过程中,需要将响应中的数据与数据库中的数据,进行断言,所以单独写了MySQL数据库连接模块。目前根据项目需求,目前实现了查询一条或多条数据,插入数据的功能。

具体实现分2个部分:

  1. 采用YAML的形式描述数据库配置
  2. 基于pymysql模块实现数据库的连接、数据查询和数据插入

数据库配置

采用YAML描述数据库配置,保存在本地

读取数据库配置,连接数据库

定义类MySQLOperation处理数据库连接、查询和插入。在__init__()函数中读取数据库配置,并连接数据库pymysql.connect

前提是安装pymysql,安装命令如pip install pymysql

读取数据

连接数据库成功后,查询返回单条数据记录,或者多条数据记录

cursor.fetchone()将返回第一条数据记录,一维元组。可多次使用cursor.fetchone(),依次取得下一条数据,直到为空。

cursor.fetchall() :将返回所有结果,也可自定义返回的数据记录的条数,二维元组。

插入数据

连接数据库成功后,目前只实现了插入单条数据记录。

断开数据库连接

《MySQL数据库连接的Python脚本》有一个想法

发表评论

电子邮件地址不会被公开。 必填项已用*标注