python接口自动化之pymsql的应用

慢慢来，谁还没有一个努力的过程，好运是藏在努力里面的！

---

前言

在python接口自动化测试时，我们需要对数据库的数据进行校验，利用pymysql连接数据库，完成与数据库的交互。

---

提示：以下是本篇文章正文内容，下面案例可供参考

一、什么是pymysql？

当python需要连接数据库时，需要用到驱动，用于与数据库交互的库。
pymysql是纯python库

二、安装pymysql

pip install pymysql

三、pymysql的应用

## 1.连接数据库

import  pymysql
#1、建立连接:数据库名，用户名，密码，端口号，ip地址，编码格式
conn=pymysql.connect(
    host="主机名",
    port=3306,
    user="用户名",
    password='密码',
    database='数据库名',
    charset='utf8',
    #将元组换成字典格式
    cursorclass=pymysql.cursors.DictCursor
)
#2、创建游标
cur=conn.cursor()
#3、执行sql语句
sql=‘sql语句’
#执行后的结果返回的是条数
count=cur.execute(sql)
#4、获取结果内容,  不可反复取
onedata=cur.fetchone()#获取结果中一条数据
# print(onedata)
alldata=cur.fetchall()#获取结果中所有数据
#5、关闭游标，关闭数据库连接
cur.close()
conn.close()

四、执行事务

事务机制可以确保数据一致性,对于支持事务的数据库， 在Python数据库编程中，当游标建立之时，就自动开始了一个隐形的数据库事务。
commit()方法游标的所有更新操作，rollback（）方法回滚当前游标的所有操作。每一个方法都开始了一个新的事务。

# SQL删除记录语句
sql = "DELETE FROM EMPLOYEE WHERE AGE > %s" % (20)
try:
   # 执行SQL语句
   cursor.execute(sql)
   # 向数据库提交
   db.commit()
except:
   # 发生错误时回滚
   db.rollback()

---

总结

以上就是今天要讲的内容，本文仅仅简单介绍了pymysql的使用:
1、pymysql安装
2、连接数据库
3、创建游标
4、执行语句
5、获取数据
6、更新数据库
7、关闭游标
8、关闭连接
9、异常回滚