python pandas 按行取excle数据作为接口请求数据

excle文件内容示例:
image.png
获取excle数据源码:
import pandas as pd

def get_all_excle_data(excle_file, sheet_name, row):
    #固定excle某列字段的数据类型,防止excle整数变为浮点数
    converters = {
        'employeeId':str,
        'authCode':str,
        'orderSource':str
    }
   # 将excle的数据提取出来
    data = pd.DataFrame(pd.read_excel(excle_file, sheet_name=sheet_name, converters = converters ,na_values='NULL',))

    # 将excle里面单元格值为空的值都替换为NULL
    data = data.fillna('')
    # excle所有列名字
    all_para_col = list(set(data))

    #生成请求字典信息
    params = {}
    for v in all_para_col:
        params[v] = data.loc[row,v]

    # 请求地址 和 key
    url = data.loc[row, 'host'] + data.loc[row, 'path']
    key = data.loc[row, 'key']  # key的值

    result = {
        'testCases': data.loc[row,'testCases'],
        'host': data.loc[row,'host'],
        'row': data.loc[row,'row'],
        'params': params,
        }
    return result
订单查询接口用excle的数据作为请求数据:
import requests
from a.utils.get_excle_data import get_all_excle_data
from a.utils.md5_python import md5Python as md5
from a.utils.date_random import get_random
from a.utils.tools import del_null
from a.log.logger import Logger
from a.utils.get_excle_path import get_excle_path
import json

def pi_query(excle_file,sheet_name,row):
    # 获取接口的数据
    excle_data = get_all_excle_data(excle_file, sheet_name, row)
    url = excle_data['host'] + "/open/pay/query"
    key = excle_data['key']
    params = excle_data['params']
    random = get_random()

    params = {
        "appId" : params['appId'],
        "merchantCode" :  params['merchantCode'],
        "random" : random,
        "outTradeNo" : params['outTradeNo']
    }

    params = del_null(params)
    params["sign"] = md5.md5_sign(params, key)

    # 接口请求
    r = requests.post(url, data=params)

    # 返回接口的请求和返回信息
    result = {
        'req': params,
        'res': r.json(),
        'url': url
    }
    return result

if __name__ == '__main__':
    log = Logger("info")
    excle_file = get_excle_path('refund_data.xlsx')
    sheet_name = 'pi_test'

    for row in range(3):
        result = pi_query(excle_file, sheet_name, row)
        log.info("支付查询请求地址:" + result['url'])
        log.info("支付查询请求信息:" + str(json.dumps(result['req'], ensure_ascii=False)))
        log.info("支付查询返回信息:" + str(json.dumps(result['res'], ensure_ascii=False)))
        print(result['res']["msg"])
        assert result['res']['code'] == "SUCCESS"
        assert result['res']['msg'] == "订单已支付"

这样做到数据源与源码分开管理,一套代码,只用改excle的请求数据就可以测试不同的场景

你可能感兴趣的:(python pandas 按行取excle数据作为接口请求数据)