VOL 120
06
2020-05
今天距2021年239天
这是ITester软件测试小栈第120次推文
点击上方蓝字“ITester软件测试小栈“关注我，每周一、三、五早上 07:30准时推送。
微信公众号后台回复“资源”、“测试工具包”、“简历模板”领取测试资源,回复“21天打卡”一起学习成长，打怪升级。
本文3518字，阅读约需9分钟
在上一篇Python接口自动化测试系列文章：Python接口自动化之yaml配置文件，主要介绍主要介绍yaml语法、yaml存储数据，封装类读写yaml配置文件。
在自动化过程中，我们需要查询数据库，校验结果是否正确，比如充值完成之后，需要查询数据库，查看充值是否成功。
以下主要介绍，pymysql安装、操作流程、语法基础及封装操作数据库类。
一
pymysql介绍及安装
01 pymysql介绍
MySQL应该说是如今使用最为普遍的数据库了，没有之一，而Python作为最为流行的语言之一，自然少不了与MySQL打交道，其中PyMySQL就是使用最多的工具库。
PyMySQL是一个纯Python写的MySQL客户端，可以在CPython、PyPy、IronPython和Jython环境下运行；
PyMySQL的性能和MySQLdb几乎相当，如果对性能要求 不是特别的强，使用PyMySQL将更加方便；
PyMySQL的使用方法和MySQLdb几乎一样；
02 pymysql安装
方式一：使用命令安装
pip install pymysql
方式二：PyCharm内部安装
导入模块：
import pymysql
二
pymysql流程及模块说明
01 pymysql操作流程
导入pymysql；
建立数据库连接：使用pymysql的connect()方法连接数据，返回连接对象；
使用连接对象创建游标对象（用于操作sql）;
准备写sql语句（select * from student）;
使用游标对象执行sql;
查询数据使用游标获取;
关闭游标（先）和数据库连接（后）。
02 pymysql模块说明
▌Connection对象
表示：conn=connect(参数列表)
作用：用于建立与数据库的连接；
创建对象：调用connect()方法；
参数列表：
host：连接的mysql主机，如本机是'localhost'；
port：连接的mysql主机的端口，默认是3306；
database：数据库的名称；
user：连接的用户名；
password：连接的密码；
charset：通信采用的编码方式，推荐使用utf8；
▌对象的方法
对象方法如下：
close()：关闭连接；
commit()：提交；
cursor()：返回Cursor对象，用于执行sql语句并获得结果；
execute(operation [, parameters ])：执行语句，返回受影响的行数，主要用于执行insert、update、delete语句，也可以执行create、alter、drop等语句；
fetchone()：执行查询语句时，获取查询结果集的第一个行数据，返回一个元组；
fetchall()：执行查询时，获取结果集的所有行，一行构成一个元组，再将这些元组装入一个元组返回；
关于pymysql防注入，字符串拼接查询，容易造成注入，为了避免注入，使用pymysql提供的参数化语句;
▌Cursor对象
游标（cursor）就是游动的标识,通俗的说，一条sql取出对应n条结果资源的接口/句柄，就是游标，沿着游标可以一次取出一行。
用于执行sql语句，使用频度最高的语句为select、insert、update、delete；
获取Cursor对象：调用Connection对象的cursor()方法:
cs1=conn.cursor()
▌对象的属性
rowcount只读属性，表示最近一次execute()执行后受影响的行数；
connection获得当前连接对象；
三
pymysql语法基础
01 代码示例
import pymysql
# 连接数据库
conn = pymysql.connect(host='127.0.0.1', user='ITester', password='123456',
database='ITester', charset='utf8')
# 创建游标
cursor = conn.cursor()
# 执行sql语句
sql = 'select * from user limit 3;'
res = cursor.execute(sql)
# 获取查询结果的1条数据
data = cursor.fetchone()
print(data)
# 关闭游标连接
cursor.close()
# 关闭数据库连接
conn.close()
02 语法总结
1.连接数据库，需要host、user、password、database、charset等信息；
2.操作数据库先创建游标；
3.执行指定的sql语句，如果涉及到增、删、改数据库必须要conn.commit()，提交事务
4.查询获取数据条数有三种方法fetchone、fetchmany、fetchall。
cursor.fetchone() ：默认获取查询结果的第一条数据；
cursor.fetchmany(2) ：获取查询结果的指定条数，比如获取2条数据；
cursor.fetchall() ：获取查询结果的所有数据；
5.需要注意的是，fetch获取的数据默认是元组，如果想要字典类型，
cursor=pymysql.cursors.DictCursor；
6.先关闭游标，后关闭数据库连接；
四
封装数据库类
01 封装说明
在实际项目中，很多地方都有用到数据库的操作，所以需要将数据库相关操作进行封装，方便其他模块调用。
如下，在common目录下，新建文件db_handler.py 用于封装数据库操作。
db_handler.py
import pymysql
class DBHandler:
def __init__(self,host,port,user,password,
database,charset,**kwargs):
# 连接数据库服务器
self.conn = pymysql.connect(host=host, port=port, user=user,password=password,
database=database,cursorclass=pymysql.cursors.DictCursor,
charset=charset,**kwargs)
# 获取游标
self.cursor = self.conn.cursor()
def query(self, sql, args=None,one=True):
self.cursor.execute(sql, args)
# 提交事务
self.conn.commit()
if one:
return self.cursor.fetchone()
else:
return self.cursor.fetchall()
def close(self):
self.cursor.close()
self.conn.close()
if __name__ == "__main__":
db = DBHandler(host='127.0.0.1', port=3306,
user='ITester', password='123456',
database='ITester', charset='utf8')
sql = 'select * from user limit 1;'
data = db.query(sql)
print(data)
总结：本文主要介绍pymysql安装、操作流程、语法基础及封装操作数据库类。
下一篇：Python接口自动化之logger日志
以上
That‘s all
更多系列文章
敬请期待
ITester软件测试小栈
往期内容宠幸
2.Python接口自动化-接口基础(二)
3.Python接口自动化-requests模块之get请求
4.Python接口自动化-requests模块之post请求
5.Python接口自动化之cookie、session应用
快来星标 置顶 关注我
<< 滑动查看下一张图片 >>
今日问题：
做接口自动化测试中，你的项目用的是什么数据库，用什么来操作的这个数据库?
（欢迎在下方留言区发表你的看法）
后台 回复"资源"取干货
回复"21天打卡"一起打怪升级
测试交流Q群：727998947
点亮一下在看，你更好看