做自动化测试的时候，注册了一个新用户，产生了多余的数据，下次用同一账号就无法注册了，这种情况该怎么办呢？
自动化测试都有个数据准备和数据清理的操作，如果因为此用例产生了多余的数据，就需要数据清理，可以用python连接mysql直接删除多余的数据就可以了。
那么python3如何连接mysql呢？PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库，Python2中则使用mysqldb。

环境准备：
python3.6
PyMySQL 0.9.2

环境准备

使用pip安装PyMySQL

pip install PyMySQL

image

先使用第三方工具连接mysql数据库，比如navicat连接mysql，参考这篇navicat的安装 https://www.jianshu.com/p/7e88913590e3

如下图连接测试通过，那么连接数据库至少应该知道这些信息，用户名和密码是授权远程客户端的用户名和密码，授权相关参考这篇MySql的安装 https://www.jianshu.com/p/51ed914ce7d0

连接名： 自己定义，这个无所谓
主机名或ip地址：mysql服务的ip地址
端口：3306 （端口号，默认一般是3306）
用户名：root （授权远程登录的用户名）
密码：123456 （授权远程登录的密码）

image

查询操作

如果新建了一个test数据库，里面有一张user表，有name和psw两个字段，使用sql查询结果如下

select name,psw from user

image

接下来使用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'

image

查询的结果是(('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)