python笔记33-python3连mysql增删改查

前言

做自动化测试的时候，注册了一个新用户，产生了多余的数据，下次用同一账号就无法注册了，这种情况该怎么办呢？

自动化测试都有个数据准备和数据清理的操作，如果因为此用例产生了多余的数据，就需要数据清理，可以用python连接mysql直接删除多余的数据就可以了。

那么python3如何连接mysql呢？PyMySQL 是在 Python3.x 版本中用于连接 MySQL 服务器的一个库，Python2中则使用mysqldb。

环境准备：

python3.6

PyMySQL 0.9.2

环境准备

使用pip安装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查询结果如下

接下来使用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值，并且取出来

查询的结果是(('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这条记录

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

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

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