一、操作数据库的必要性
1、校验数据的正确性。
2、接口测试中常常需要通过数据库的操作,来准备数据、检测环境及核对功能、以此来验证数据是否正确。
3、在自动化测试中,需要我们用代码连接数据库自动完成数据准备、环境检查及数据库断言的功能。
二、安装pyMysql
使用Python操作MySQL数据库这里我们需要用到三方库PyMySQl
安装方法:pip install pymysql
也可通过pycharm的setting--->project:process--->project interpreter----+导入包pymsql
三、操作数据库
1、建立数据库连接 conn = pymysql.connect()
2、从连接建立操作游标 cur = conn.cursor()
3、使用游标执行sql(读/写) cur.execute(sql)
4、获取结果(读)/ 提交更改(写) cur.fetchall() / conn.commit()
5、关闭游标及连接 cur.close();conn.close()
具体如下图:
查询操作通过cur.execute()来查询,获取数据通常有3种方式:
1>cur.fetchone():获取一条数据,返回的是一个元组
2>cur.fetchmany():获取多条数据,返回一个嵌套的元组
3>cur.fetchall():获取所有数据,返回一个嵌套元组
!!!注意:获取完数据后,数据会从数据集中删除,再次获取获取不到,所以需要重复查询时,我们应该将查询结果赋给某个变量
四、封装数据库操作
由于经常使用数据库,我们可以将数据库封装成公用的数据库模块
1、简单封装便于理解,新建一个.py文件,代码如下:
2、x新建一个文件,调用封装的方法
从上面的封装可以看到,每一次查询,都建立了一次数据库连接,这样会导致执行效率比较低,所以我们可以采用面向对象的封装方法
3、面向对象的封装方法:
调用方法:
以上就是完成了python连接数据库的操作,一般数据库连接信息建议写到配置文件里面,从配置文件中读取。通过封装各种sql可以完成各种业务操作,但更改数据库会有一定的风险,操作时需要谨慎处理。