python调用接口数据 python调用接口获取数据

读取Excel表中的数据

2.提取测试用例

3.发送接口请求

4.接口响应结果

1 import xlrd
 2 import ast
 3 import json
 4 import requests
 5 
 6 def excel_to_list(data_file, sheet):
 7     '''
 8     一次获取工作表中的所有数据
 9     :param data_file: Excel文件名
10     :param sheet: 工作簿名
11     :return: 数据表
12     '''
13     data_list = []  # 新建个空列表，来乘装所有的数据
14     wb = xlrd.open_workbook(data_file)  # 打开excel
15     sh = wb.sheet_by_name(sheet)  # 获取工作簿
16     header = sh.row_values(0)  # 获取标题行数据
17     for i in range(1, sh.nrows):  # 跳过标题行，从第二行开始取数据
18         d = dict(zip(header, sh.row_values(i)))# 将标题和每行数据组装成字典
19         data_list.append(d)
20         #logger.info(f"\n全部用例数据：{data_list}\n")
21     return data_list  # 列表嵌套字典格式，每个元素是一个字典
22 
23 def get_test_data(data_file, sheet, case_name):
24     '''
25     从用例工作簿数据中提取用例数据
26     :param data_list: 工作簿中表中所有用例数据
27     :param case_name: 用例名字
28     :return: 用例数据
29     '''
30     data_list = excel_to_list(data_file, sheet)
31     for case_data in data_list:
32         if case_name == case_data['case_name']:  # 如果字典数据中case_name与参数一致
33             #logger.info(f"\nexcel中读取用例数据：\n{case_data}")
34             #logger.info(case_data['headers'])
35             return case_data
36 
37 def send_request(case_data, case_name):
38     if not case_data:  # 有可能为None
39         # 从字典中取数据，excel中的标题也必须是小写url
40         print("用例数据不存在")
41     url = case_data.get('url')  # 从字典中取数据，excel中的标题也必须是小写url
42     data = case_data.get('data')  # 注意字符串格式
43     headers = case_data.get('headers')
44     expect_res = case_data.get('expect_res')  # 期望数据
45     method = case_data.get('method')
46     if method == "GET":
47         res = requests.get(url=url, data=None, headers=None)  # 表单请求，数据转为字典格式
48         res_j = res.json()
49         print(f"用例名称：{case_name}\n"
50               f"请求URL：{url}\n"
51               f"请求数据：{data}\n"
52               f"实际结果：{res_j}\n"
53               f"期望结果：{expect_res}\n")
54         return res
55     else:
56         headers = ast.literal_eval（headers)
57         data = json.dumps(ast.literal_eval（data))
58         res = requests.post(url=url, headers=headers, data=data)
59         res_j = res.json()
60         print(f"用例名称：{case_name}\n"
61               f"请求URL：{url}\n"
62               f"请求数据：{data}\n"
63               f"实际结果：{res_j}\n"
64               f"期望结果：{expect_res}\n")
65         return res
66 
67 
68 def get_response(data_file, sheet, case_name):
69     case_data = get_test_data(data_file, sheet, case_name)
70     res = send_request(case_data, case_name)
71     return res
72 
73 if __name__ == '__main__':
74     data_file = "test_user_data.xls"
75     sheet = "TestUserLogin"
76     get_response(data_file, sheet, "test_getstock_normal")
77     get_response(data_file, sheet, "test_user_login_password")