python接口自动化学习（四）数据库操作

给脚本添加操作数据库功能

configDb.py代码：

import pymysql
from FlychordInterfaceTest.readConfig import get_mysql


# 获取连接方法
def get_db_conn():
    conn = pymysql.connect(host=get_mysql('host'),
                           port=int(get_mysql('port')),
                           user=get_mysql('username'),
                           passwd=get_mysql('password'),
                           db=get_mysql('database'),
                           charset='utf8')  # 如果查询有中文，需要指定测试集编码

    return conn


# 封装数据库查询操作
def query_db(sql):
    conn = get_db_conn()  # 获取连接
    cur = conn.cursor()  # 建立游标
    cur.execute(sql)  # 执行sql
    result = cur.fetchall()  # 获取所有查询结果
    cur.close()  # 关闭游标
    conn.close()  # 关闭连接
    return result  # 返回结果


# 封装更改数据库操作
def change_db(sql):
    conn = get_db_conn()  # 获取连接
    cur = conn.cursor()  # 建立游标
    try:
        cur.execute(sql)  # 执行sql
        conn.commit()  # 提交更改
    except Exception as e:
        conn.rollback()  # 回滚
    finally:
        cur.close()  # 关闭游标
        conn.close()  # 关闭连接


# 封装常用数据库操作
def check_user(name):
    # 注意sql中''号嵌套的问题
    sql = "select * from company where company_name = '{}'".format(name)
    result = query_db(sql)
    return result
    # return True if result else False


def add_user(name, password):
    sql = "insert into user (name, passwd) values ('{}','{}')".format(name, password)
    change_db(sql)


def del_user(name):
    sql = "delete from user where name='{}'".format(name)
    change_db(sql)


if __name__ == '__main__':
    print(check_user('测试'))

这个比较简单，不多做解释，注意数据类型就好