Python接口自动化浅析pymysql数据库操作流程
作者:软件测试自动化测试 发布时间:2024-01-16 16:33:02
在上一篇Python接口自动化测试系列文章:Python接口自动化浅析yaml配置文件原理及用法,主要介绍主要介绍yaml语法、yaml存储数据,封装类读写yaml配置文件。
在自动化过程中,我们需要查询数据库,校验结果是否正确,比如充值完成之后,需要查询数据库,查看充值是否成功。
以下主要介绍,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 代码示例
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)
来源:https://blog.csdn.net/ZangKang1/article/details/119645893
猜你喜欢
- python中shape()函数shape函数是numpy.core.fromnumeric中的函数,它的功能是读取矩阵的长度。1、shap
- 前言其实我觉得接口测试很简单,比一般的功能测试还简单(这话我先这样说,以后可能会删O(∩_∩)O哈!),现在
- 这是毕业校招二面时遇到的手写编程题,当时刚刚开始学习python,整个栈写下来也是费了不少时间。毕竟语言只是工具,只要想清楚实现,使用任何语
- 随机显示图片代码:<div class="sidebarcen"><script language=&
- 1. 安装 xlrd 库Python 读取 Excel 中的数据主要用到 xlrd 第三方库。xlrd 其实就是两个单词的简化拼接,我们可以
- 本文实例为大家分享了Python曲线拟合的最小二乘法,供大家参考,具体内容如下模块导入import numpy as npimport ga
- print函数的功能是打印图窗或保存为特定文件格式。语法print(filename,formattype)print(filename,f
- 在安装数据库的时候出现了如下错误:解决办法如下:1.在bin目录下 输入:kill -s 9 9907 再输入:ps
- 前言:Python 面向对象的高级编程语言,以其语法简单、免费开源、免编译扩展性高,同时也可以嵌入到C/C++程序和丰富的第三方
- 由于DOM(文档对象模型)概念的推出,这个API使HTML如虎添翼,但是有些学DHTML的朋友还是有些困挠,只是因为目前的手册的书写不太科学
- 在头条看了一篇文章,说五行代码实现人脸识别,一时感兴趣了,来搞搞先是按照文章说的 操作了几步,到后面虽然,import dlib 不报错,但
- 花了两个多钟在看 ThinkPHP 框架,不想太过深入的知道它的所有高深理论。单纯想知道怎么可以用起来,可以快捷的搭建一个网站。所以是有选择
- s={ x1,x2,x3.....};集合有自动去重的功能,而且可以进行交并补运算,而且集合是无序的,每次打印的结果不一样,故不可以用元素下
- 有时候,女神发来一条消息,说约你看电影,她考虑了一下,又撤回了,不约你了…而你又想知道她究竟发了什么,该怎么办?微信防撤回了解一下。环境要求
- 从视频中提取音频安装 moviepypip install moviepy相关代码:audio_file = work_path +
- 使用到的函数是curl_init, curl_setopt, curl_exec,curl_close。默认是GET方法,可以选择是否使用H
- 1 简介DataFrame是Python中Pandas库中的一种数据结构,它类似excel,是一种二维表。或许说它可能有点像matlab的矩
- 开发动机:最近用手机QQ浏览器下载了一些视频,视频越来越多,占用了手机内存,于是想把下载的视频传到电脑上保存,可后来发现这些视频都是m3u8
- 本文实例讲述了PHP+redis实现添加处理投票的方法。分享给大家供大家参考,具体如下:<?php header("Cont
- 前言大家好,我是空空star,本篇给大家分享一下通过Python的pyttsx3库将文字转为音频。一、pyttsx3是什么?pyttsx3是