Python接口自动化中pymysql数据库操作流程详解

Elena

发布时间：2026-10-12| 3人浏览过

原创

本文详解Python接口自动化测试中pymysql的完整操作流程。涵盖pymysql的安装方式、连接参数配置、游标对象使用、SQL执行与结果获取方法（fetchone/fetchall），并重点介绍如何封装DBHandler类以简化数据库交互，提升测试代码复用性与安全性。

在上一篇关于Python接口自动化测试的系列文章中，我们深入探讨了yaml配置文件的原理及其实际应用，主要涵盖了yaml语法规范、数据存储方式以及如何封装类来读写yaml配置文件。

在自动化测试的实际执行过程中，经常需要直接查询数据库以校验业务逻辑是否正确。例如，在模拟用户充值操作完成后，必须通过查询数据库来确认充值记录是否已成功写入，从而验证接口返回结果与数据库状态的一致性。

接下来，本文将重点介绍pymysql库的安装方法、标准操作流程、基础语法以及一个通用的数据库操作封装类，帮助读者更高效地处理数据库交互。

一、pymysql介绍及安装

01 pymysql介绍

MySQL是目前业界使用最为广泛的数据库系统，而Python作为当前最流行的编程语言之一，与MySQL的集成至关重要。在众多Python MySQL客户端库中，PyMySQL因其纯Python实现且无需编译C扩展，成为使用最为广泛的工具库之一。

PyMySQL是一个纯Python编写的MySQL客户端库，具有良好的兼容性，可以在CPython、PyPy、IronPython和Jython等多种Python运行环境下稳定运行。
在性能方面，PyMySQL与传统的MySQLdb库几乎相当。对于大多数对极致性能要求不高的自动化测试场景，使用PyMySQL能够带来更便捷的体验和更低的配置成本。
此外，PyMySQL的API设计与MySQLdb高度相似，熟悉MySQLdb的开发者可以无缝迁移至PyMySQL，降低了学习成本。

02 pymysql安装

方式一：使用命令行工具安装

pip install pymysql

方式二：在PyCharm集成开发环境内部安装

导入模块：

import pymysql

二、pymysql流程及模块说明

01 pymysql操作流程

首先，导入pymysql模块；
其次，建立数据库连接：调用pymysql的connect()方法传入必要参数，该方法返回一个连接对象（Connection）；
接着，使用连接对象创建游标对象（Cursor），游标用于执行SQL语句；
然后，准备需要执行的SQL语句，例如：select * from student；
使用游标对象调用execute()方法执行SQL语句；
如果是查询操作，使用游标提供的方法获取查询结果；
最后，先关闭游标，再关闭数据库连接，以释放资源。

02pymysql模块说明

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)

以上就是Python接口自动化浅析pymysql数据库操作流程的详细内容，更多关于Python接口自动化pymysql数据库操作的资料请关注本站其它相关文章！