Python接口自动化浅析pymysql数据库操作流程

本文主要介绍pymysql安装、操作流程、语法基础及封装操作数据库类,需要的朋友可以参考下，希望能对大家有所帮助，每日提升一点点，欢迎大家多多交流讨论 

在自动化过程中，我们需要查询数据库，校验结果是否正确，比如充值完成之后，需要查询数据库，查看充值是否成功。

以下主要介绍，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 代码示例

1. import pymysql

2. # 连接数据库

3. conn = pymysql.connect(host='127.0.0.1', user='ITester', password='123456',

4. database='ITester', charset='utf8')

5. # 创建游标

6. cursor = conn.cursor()

7. # 执行sql语句

8. sql = 'select * from user limit 3;'

9. res = cursor.execute(sql)

10. # 获取查询结果的1条数据

11. data = cursor.fetchone()

12. print(data)

13. # 关闭游标连接

14. cursor.close()

15. # 关闭数据库连接

16. 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

1. import pymysql

2. class DBHandler:

3. def __init__(self,host,port,user,password,

4. database,charset,**kwargs):

5. # 连接数据库服务器

6. self.conn = pymysql.connect(host=host, port=port, user=user,password=password,

7. database=database,cursorclass=pymysql.cursors.DictCursor,

8. charset=charset,**kwargs)

9. # 获取游标

10. self.cursor = self.conn.cursor()

12. def query(self, sql, args=None,one=True):

13. self.cursor.execute(sql, args)

14. # 提交事务

15. self.conn.commit()

16. if one:

17. return self.cursor.fetchone()

18. else:

19. return self.cursor.fetchall()

20. def close(self):

21. self.cursor.close()

22. self.conn.close()

23. if __name__ == "__main__":

24. db = DBHandler(host='127.0.0.1', port=3306,

25. user='ITester', password='123456',

26. database='ITester', charset='utf8')

27. sql = 'select * from user limit 1;'

28. data = db.query(sql)

29. print(data)

以上就是Python接口自动化浅析pymysql数据库操作流程的详细内容。

总结：

感谢每一个认真阅读我文章的人！！！

作为一位过来人也是希望大家少走一些弯路，如果你不想再体验一次学习时找不到资料，没人解答问题，坚持几天便放弃的感受的话，在这里我给大家分享一些自动化测试的学习资源，希望能给你前进的路上带来帮助。

软件测试面试文档

我们学习必然是为了找到高薪的工作，下面这些面试题是来自阿里、腾讯、字节等一线互联网大厂最新的面试资料，并且有字节大佬给出了权威的解答，刷完这一套面试资料相信大家都能找到满意的工作。

 

          视频文档获取方式：
这份文档和视频资料，对于想从事【软件测试】的朋友来说应该是最全面最完整的备战仓库，这个仓库也陪伴我走过了最艰难的路程，希望也能帮助到你！以上均可以分享，点下方小卡片即可自行领取。