python笔记33-python3连mysql增删改查
前言
做自动化测试的时候，注册了一个新用户，产生了多余的数据，下次用同一账号就无法注册了，这种情况该怎么办呢？
自动化测试都有个数据准备和数据清理的操作，如果因为此用例产生了多余的数据，就需要数据清理，可以用python连接mysql直接删除多余的数据就可以了。
那么python3如何连接mysql呢？PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库，Python2中则使用mysqldb。
环境准备：
python3.6
PyMySQL 0.9.2
环境准备
使用pip安装PyMySQL
pip install PyMySQL
先使用第三方工具连接mysql数据库，比如navicat连接mysql，参考这篇https://www.cnblogs.com/yoyoketang/p/9994078.html
如下图连接测试通过，那么连接数据库至少应该知道这些信息，用户名和密码是授权远程客户端的用户名和密码，授权相关参考这篇https://www.cnblogs.com/yoyoketang/p/10268896.html
连接名： 自己定义，这个无所谓
主机名或ip地址：mysql服务的ip地址
端口：3306 （端口号，默认一般是3306）
用户名：root （授权远程登录的用户名）
密码：123456 （授权远程登录的密码）
查询操作
如果新建了一个test数据库，里面有一张user表，有name和psw两个字段，使用sql查询结果如下
select name,psw from user
接下来使用python转换成对应的代码查询
import pymysql
# 打开数据库连接
db = pymysql.connect(host='47.104.x.x',
port=3306,
user='root',
passwd='123456',
db='test')
# 使用 cursor() 方法创建一个游标对象cur
cur = db.cursor()
# 使用 execute() 方法执行 SQL 查询
cur.execute("select name, psw from user")
# 使用 fetchall() 方法获取查询结果
data = cur.fetchall()
print(data)
# 关闭数据库连接
db.close()
运行结果：
(('yoyo_1', '111111'), ('yoyo', '123456'), ('yoyo_2', '111111'), ('yoyo_3', '222222'), ('yoyo_4', '444444'))
如果想查询结果以字典格式输出，可以用db.cursor(cursor=pymysql.cursors.DictCursor)推荐！！！
有时候我们只想查询某个字段对应的值，比如查询yoyo_1账号对应的psw值，并且取出来
select psw from user where name='yoyo_1'
查询的结果是(('111111',),) 元组嵌套元组，取值的话用下标取出来就可以了
import pymysql
def select_db(sql):
'''查询数据库'''
# 打开数据库连接
db = pymysql.connect(host='47.104.x.x',
port=3306,
user='root',
passwd='123456',
db='test')
# 使用 cursor() 方法创建一个游标对象cur
cur = db.cursor()
# 使用 execute() 方法执行 SQL 查询
cur.execute(sql)
# 使用 fetchall() 方法获取查询结果
data = cur.fetchall()
# print(data) # 取出对应的psw值
# 关闭数据库连接
db.close()
return data
if __name__ == "__main__":
sql = "select psw from user where name='yoyo_3'"
a = select_db(sql)[0][0]
print("查询结果：%s" %str(a))
删除操作
使用python删除一条数据，比如我要删除yoyo_1这条记录
delete from user where name='yoyo_1'
import pymysql
def delete_db(sql_delete):
'''删除操作'''
# 打开数据库连接
db = pymysql.connect(host='47.104.x.x',
port=3306,
user='root',
passwd='123456',
db='test')
# 使用cursor()方法获取操作游标
cur = db.cursor()
try:
cur.execute(sql_delete) # 执行
# 提交
db.commit()
except Exception as e:
print("操作异常：%s" % str(e))
# 错误回滚
db.rollback()
finally:
db.close()
if __name__ == '__main__':
sql_delete ="delete from user where name='yoyo_1' "
delete_db(sql_delete)
更新操作
更新name用户名是yoyo的用户，把psw改成666666
update user set psw='666666' where name='yoyo'
import pymysql
def update_db(sql_update):
'''3.更新操作'''
db = pymysql.connect(host='47.104.x.x',
port=3306,
user='root',
passwd='123456',
db='test')
# 使用cursor()方法获取操作游标
cur = db.cursor()
try:
cur.execute(sql_update) # 执行sql
# 提交
db.commit()
except Exception as e:
# 错误回滚
print("错误信息：%s" % str(e))
db.rollback()
finally:
db.close()
if __name__ == '__main__':
sql_update ="update user set psw='666666' where name='yoyo'"
update_db(sql_update)
新增数据
往数据库里面插入一条数据，比如在user表里面新增一个用户信息yoyo_10,123456
insert into user(id, name, psw) values(10, 'yoyo_10', '123456')
import pymysql
def insert_db(sql_insert):
'''插入操作'''
db = pymysql.connect(host='47.104.x.x',
port=3306,
user='root',
passwd='123456',
db='test')
# 使用cursor()方法获取操作游标
cur = db.cursor()
try:
cur.execute(sql_insert)
# 提交
db.commit()
except Exception as e:
print("错误信息：%s" % str(e))
# 错误回滚
db.rollback()
finally:
db.close()
if __name__ == "__main__":
sql_insert = "insert into user(id, name, psw) values(10, 'yoyo_10', '123456')"
insert_db(sql_insert)
从上面的代码可以看出，除了查询的代码不一样，新增、删除、更新数据库操作代码都一样，只是执行的sql不一样
python自动化交流 QQ群：779429633
