基于Python的接口自动化实战-基础篇之pymysql模块操作数据库
引言
在进行功能或者接口测试时常常需要通过连接数据库,操作和查看相关的数据表数据,用于构建测试数据、核对功能、验证数据一致性,接口的数据库操作是否正确等。因此,在进行接口自动化测试时,我们一样绕不开接口和数据库的交互,我们需要用代码连接数据库,通过操作数据库完成数据的准备、环境检查以及数据库断言的功能。在python3中,使用python操作MySQL数据库需要使用到第三方库:pymysql,该模块本质上就是一个套接字的客户端软件包,它提供了诸多连接数据库、操作数据库表等一系列的方法。
一、PyMySQL安装
1.在windows环境下安装
由于python3.6及以上版本安装python后就自带了pip3,python版本低于3.6的,手动安装下pip即可,因此可以直接使用pip安装该模块
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()
#运行结果
((1, '艾佛森', '123'), (2, '科比', '123'), (3, '詹姆斯', '123'), (4, '库里', '123'))
什么是游标? 游标类似文件句柄,可以逐条的访问数据库执行结果集。pymysql中只能通过游标来执行sql和获取结果
以上代码执行后,默认返回的是一个嵌套元组数据类型
数据库增删改查
查询操作:
使用cur.execute(), 执行数据库查询后无返回的是影响的行数,而非查询结果。我们要使用cur.fetchone()/cur.fetchmany()/cur.fetchall()来获取查询结果
cur.fetchone(): 获取一条数据(同时获取的数据会从结果集删除),返回元组
cur.fetchmany(3): 获取多条数据,返回嵌套元组
cur.fetchall(): 获取所有数据,返回嵌套元组
代码示例:
查询单条数据:
import pymysql
db_config = {
"host":"119.29.78.234",
"port":3306,
"user":"root",
"password":"dhcc@2020",
"db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "SELECT * FROM userinfo"
cursor.execute(sql)
res = cursor.fetchone() # fetchone()第一次只能查询表中的首行数据
print(res)
res = cursor.fetchone() # 第二次查询下一行数据
print(res)
cursor.close()
db.close()
# 返回结果
((1, '艾佛森', '123'))
((2, '科比', '123'))
查询多条数据:
import pymysql
db_config = {
"host":"119.29.78.234",
"port":3306,
"user":"root",
"password":"dhcc@2020",
"db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "SELECT * FROM userinfo"
cursor.execute(sql)
res = cursor.fetchmany(3) # 第一次查询表中的前3行数据
print(res)
res = cursor.fetchmany(3) # 第二次查询下一个3行的数据
print(res)
cursor.close()
db.close()
#返回结果
((1, '艾佛森', '123'), (2, '科比', '123'), (3, '詹姆斯', '123'))
((4, '库里', '123'),)
查询所有数据:
import pymysql
db_config = {
"host":"119.29.78.234",
"port":3306,
"user":"root",
"password":"dhcc@2020",
"db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "SELECT * FROM userinfo"
cursor.execute(sql)
res = cursor.fetchall() # 第一次查询表中的所有数据
print(res)
res = cursor.fetchall() # 第二次查询无数据
print(res)
cursor.close()
db.close()
#返回结果
((1, '艾佛森', '123'), (2, '科比', '123'), (3, '詹姆斯', '123'), (4, '库里', '123'))
()
默认都是返回元组的数据类型,看起来不太直观,因此,在实例化时可以将游标设置成如下这样,就可以返回字典类型的数据
cursor = db.cursor(cursor=pymysql.cursors.DictCursor)
#返回结果
[{'username': '艾佛森', 'id': 1, 'passwd': '123'}, {'username': '科比', 'id': 2, 'passwd': '123'}, {'username': '詹姆斯', 'id': 3, 'passwd': '123'}, {'username': '库里', 'id': 4, 'passwd': '123'}]
增删改操作:
在进行增删改,执行修改数据库的操作后不立即生效,使用连接conn.commit()提交后才生效,支持事物及回滚
代码示例:
import pymysql
db_config = {
"host":"119.29.78.234",
"port":3306,
"user":"root",
"password":"dhcc@2020",
"db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "INSERT INTO userinfo(username,passwd) VALUES('克莱','123')"
#sql = "UPDATE userinfo SET username = '奥尼尔' WHERE username = '科比'" # 修改数据
#sql = "DELETE FROM username WHERE username ='奥尼尔'" # 删除数据
try:
cursor.execute(sql)
db.commit()
except Exception as e : # 执行异常回滚
db.rollback()
cursor.close()
db.close()
#或者在execute提供需要插入的数据
import pymysql
db_config = {
"host":"119.29.78.234",
"port":3306,
"user":"root",
"password":"dhcc@2020",
"db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "INSERT INTO userinfo(username,passwd) VALUES(%s,%s)"
try:
cursor.execute(sql,("克莱","123"))
db.commit()
except Exception as e :
db.rollback()
cursor.close()
db.close()
#批量插入数据
import pymysql
db_config = {
"host":"119.29.78.234",
"port":3306,
"user":"root",
"password":"dhcc@2020",
"db":"test123"
}
db = pymysql.connect(**db_config)
cursor = db.cursor()
sql = "INSERT INTO userinfo(username,passwd) VALUES(%s,%s)"
try:
cursor.executemany(sql,[("韦德","123"),("字母哥","123")])
db.commit()
except Exception as e :
db.rollback()
cursor.close()
db.close()
封装数据库操作
由于经常要使用到数据库操作,建议将所有数据库操作封装成公用的数据库模块
封装的代码示例如下:
import pymysql.cursors class Operation_mysql(object):
def __init__(self):
# 建立连接
db_config = {
"host": "119.29.78.234",
"port": 3306,
"user": "root",
"password": "dhcc@2020",
"db": "test123"
}
self.connection = pymysql.connect(**db_config)
# 创建游标
self.cursor = self.connection.cursor() def execute_sql(self, sql):
try:
self.cursor.execute(sql)
self.connection.commit()
except Exception as e: # 执行异常回滚
db.rollback() def get_data(self):
data = self.cursor.fetchone()
#data = self.cursor.fetchall() # 查询所有数据
return data def close_mysql(self):
# 关闭游标
self.cursor.close()
# 关闭数据库连接
self.connection.close()
这样封装后后续接口测试用例需要操作数据库时,就可以引入该模块,实例化对象调用该模块下的方法。
基于Python的接口自动化实战-基础篇之pymysql模块操作数据库的更多相关文章
- 基于Python的接口自动化实战-基础篇之读写配置文件
引言 在编写接口自动化测试脚本时,有时我们需要在代码中定义变量并给变量固定的赋值.为了统一管理和操作这些固定的变量,咱们一般会将这些固定的变量以一定规则配置到指定的配置文件中,后续需要用到这些变量和变 ...
- 基于python的webUI自动化-小白基础篇
最近打算研究一下基于python的webUI自动化,先自学了一下相关基础知识,大概用了一个多月的时间,主要是找视频,一边看视频或者文档一边对照着敲代码运行. 重点强调:一定要一边看一边对照着敲代码运行 ...
- 基于Python的接口自动化-01
为什么要做接口测试 当前互联网产品迭代速度越来越快,由之前的2-3个月到个把月,再到班车制,甚至更短,每次发版之前都需要对所有功能进行回归测试,在人力资源有限的情况下,做自动化测试很有必要.由于UI更 ...
- 基于Python的接口自动化-读写excel文件
引言 使用python进行接口测试时常常需要接口用例测试数据.断言接口功能.验证接口响应状态等,如果大量的接口测试用例脚本都将接口测试用例数据写在脚本文件中,这样写出来整个接口测试用例脚本代码将看起来 ...
- 基于Python的接口自动化
第一步 Python的安装配置 打开官网: https://www.python.org/downloads/ 目前官网上已经更新到3.6.1啦,有两个版本,大家可以按自己喜欢的去下载,我自己选择的是 ...
- 基于Python的接口自动化-unittest测试框架和ddt数据驱动
引言 在编写接口自动化用例时,我们一般针对一个接口建立一个.py文件,一条接口测试用例封装为一个函数(方法),但是在批量执行的过程中,如果其中一条出错,后面的用例就无法执行,还有在运行大量的接口测试用 ...
- 第9期《jmeter接口自动化实战》零基础入门!
2019年 第9期<jmeter接口自动化实战>课程,12月6号开学! 上课方式:QQ群视频在线教学 本期上课时间:12月6号-1月18号,每周五.周六晚上20:00-22:00 报名费: ...
- (数据科学学习手札47)基于Python的网络数据采集实战(2)
一.简介 马上大四了,最近在暑期实习,在数据挖掘的主业之外,也帮助同事做了很多网络数据采集的内容,接下来的数篇文章就将一一罗列出来,来续写几个月前开的这个网络数据采集实战的坑. 二.马蜂窝评论数据采集 ...
- 学习Keras:《Keras快速上手基于Python的深度学习实战》PDF代码+mobi
有一定Python和TensorFlow基础的人看应该很容易,各领域的应用,但比较广泛,不深刻,讲硬件的部分可以作为入门人的参考. <Keras快速上手基于Python的深度学习实战>系统 ...
随机推荐
- Python使用import导入模块时执行了模块的文件但报ModuleNotFoundError错误的愚蠢问题
老猿在学习import导入自定义模块时,搜索路径中sys.path中已经添加对应路径,发现会报ModuleNotFoundError,但对应的模块代码被执行了,代码myfib.py如下: def fi ...
- 建立SQL Server用户登录
第一步:右击服务器名,点击"属性",选择"安全性",选中"Sql server和Windows身份验证模式". 第二步:鼠标右键点击安全性中 ...
- zookeeper未授权访问渗透测试及修复方法
zookeeper未授权访问危害 服务器信息泄露.集群被破坏 一. 四字命令未授权使用 1.1 测试 工具:netcat ,Linux或Windows都可以测 命令行输入echo envi | nc ...
- 如何设置10px
chrome浏览器小于12px字体默认显示12px? font-size:10px: -webkit-transform:scal(0.83):
- CSP-S 2020 题解
赛后我重拳出击,赛场上我却爆零.哎. 题解本人口胡.有错请各位大佬们指出. A. 儒略日 这题是大型模拟题. 介绍两种写法:一种代码量致死(赛 场 自 闭),一种是非常好写的. 写法 1 我在赛场的思 ...
- do{}while(false)的用法
do{}while(false): 在工作中我们能经常发现有人写 do{}while(false) 这样的代码,初看时让人迷惑不解,按照上面的语法 do{}while(false) 这样 do{} ...
- 数据结构与算法——循环链表的算法实现(Joseph 问题)
Joseph 问题: 如果有10 个人,按编号顺序1,2,...,10 顺时针方向围成一圈.从1 号开始顺时针方向1,2,...,9 报数,凡报数9 者出列(显然,第一个出圈为编号9 者). 最后一个 ...
- 女朋友突然问我DNS是个啥....
女朋友突然问我DNS是个啥.... 今天晚上我正在床上躺着刷手机,然后我女朋友突然说她的电脑坏了.说连着WIFi上不了网,让我给她看一下.(这就是有个程序员男朋友的好处) 然后我拿到电脑看了一下发现访 ...
- el-amap 遮罩(带洞多边形)
el-amap 遮罩(带洞多边形) 遮罩(带洞多边形) 效果图 代码 <template> <div> <el-amap vid="amapDemo" ...
- Scala中的IO操作及ArrayBuffer线程安全问题
通过Scala对文件进行读写操作在实际业务中应用也比较多,这里介绍几种常用的方式,直接上代码: 1. 从文件中读取内容 object Main { def loadData(): Array[Stri ...