基于Python的接口自动化-pymysql模块操作数据库

目录

引言

一、PyMySQL安装

二、Python操作数据库

---

引言

        在进行功能或者接口测试时常常需要通过连接数据库，操作和查看相关的数据表数据，用于构建测试数据、核对功能、验证数据一致性，接口的数据库操作是否正确等。因此，在进行接口自动化测试时，我们一样绕不开接口和数据库的交互，我们需要用代码连接数据库，通过操作数据库完成数据的准备、环境检查以及数据库断言的功能。在python3中，使用python操作MySQL数据库需要使用到第三方库：pymysql，该模块本质上就是一个套接字的客户端软件包，它提供了诸多连接数据库、操作数据库表等一系列的方法。

一、PyMySQL安装

1.在windows环境下安装

由于python3.6及以上版本安装python后就自带了pip3，python版本低于3.6的，手动安装下pip即可，因此可以直接使用pip安装该模块

1 pip3 install pymysql pip3 install pymysql

2.在linux环境下安装

下载安装pymysql的tar包，解压后，进入解压的目录下，按如下安装即可：

[root@localhost opt]#tar -xzvf PyMySQL-0.7.11.tar.gz

[root@localhost opt]#cd PyMySQL-0.7.11

[root@localhost PyMySQL-0.7.11]#python36 setup.py install

3.在PyCharm中安装

在PyCharm中直接检索该模块，并安装，步骤如下：

二、Python操作数据库

因为方便测试，我们首先在mysql数据库创建测试表：userinfo，表信息如下：

 

有了数据库和数据表后，我们就可以导入pymysql模块，使用该模块下封装的方法实现数据库操作

数据库连接

pymysql提供的方法如下：

1. 建立数据库连接 conn = pymysql.connect()
2. 从连接建立操作游标 cur = conn.cursor()
3. 使用游标执行sql（读/写） cur.execute(sql)
4. 获取结果（读）/ 提交更改（写） cur.fetchall() / conn.commit()
5. 关闭游标及连接 cur.close();conn.close()

代码示例：

import pymysql

# 建立连接

connection = pymysql.connect(host='119.29.78.234', port=3306, user='root', password='dhcc@2020', db='test123')

cursor = connection.cursor()       # 创建游标

cursor.execute("SELECT * FROM userinfo")   #使用execute()方法执行SQL语句

data = cursor.fetchall()  #使用fetall()获取全部数据

print(data)

cursor.close()   #关闭游标和数据库的连接

connection.close()<br><br>#运行结果<br>((1, '艾佛森', '123