python操作mysql的增删改查_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

如下图连接测试通过，那么连接数据库至少应该知道这些信息，用户名和密码是授权远程客户端的用户名和密码，授权相关参考这篇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