requests接口自动化-数据库参数化

  • 数据库参数化的场景
    部分接口,参数其他方式获取不到,可以去数据库去获取,如手机验证码

  • 安装pymysql

  • 配置文件

def sql_conf():
    # 定义数据库的配置
    host='127.0.0.1'
    user='root'
    password='123456'
    database='employees'
    port=3306
    charset='utf8'
    return host,user,password,database,port,charset
  • 定义get_mysql方法
import pymysql
from config.conf import sql_conf  #导入参数配置定义的函数
def get_sql(sql):
    # 建立一个连接对象
    host,user,password,database,port,charset=sql_conf() #引用参数配置的值
    db=pymysql.connect(host=host,user=user,database=database,port=port,charset=charset)
    # 建立一个游标
    cursor=db.cursor()
    # 运行sql语句
    cursor.execute(sql)
    # 保存查询的结果
    data=cursor.fetchall()

    # 关闭游标
    cursor.close()
    # 关闭数据库连接
    db.close()
    # 返回数据
    return data
  • 调用get_mysql方法获取id
import requests
from common.get_mysql import get_sql
def test_get_mysql():
    url='http//:127.0.0.1:8080'
    id=get_sql("SELECT dept_no FROM departments WHERE dept_name ='Finance'") # 从数据库获取数据
    data={'name':'Finance','id':id}
    r=requests.post(url=url,data=data)
    print(r.status_code)
    print(r.text)

猜你喜欢

转载自www.cnblogs.com/Uni-Hoang/p/13173160.html