python数据库接口设置_基于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()

这样封装后后续接口测试用例需要操作数据库时，就可以引入该模块，实例化对象调用该模块下的方法。