在自动化过程中,我们需要查询数据库,校验结果是否正确,比如充值完成之后,需要查询数据库,查看充值是否成功。
以下主要介绍,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;
- 查询数据使用游标获取;
- 关闭游标(先)和数据库连接(后)。
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 代码示例
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
|
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 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 |
|
现在我也找了很多测试的朋友,做了一个分享技术的交流群,共享了很多我们收集的技术文档和视频教程。
如果你不想再体验自学时找不到资源,没人解答问题,坚持几天便放弃的感受
可以加入我们一起交流。而且还有很多在自动化,性能,安全,测试开发等等方面有一定建树的技术大牛
分享他们的经验,还会分享很多直播讲座和技术沙龙
可以免费学习!划重点!开源的!!!
qq群号:485187702【暗号:csdn11】
最后感谢每一个认真阅读我文章的人,看着粉丝一路的上涨和关注,礼尚往来总是要有的,虽然不是什么很值钱的东西,如果你用得到的话可以直接拿走! 希望能帮助到你!【100%无套路免费领取】