设置一个开关来控制执行用例的数量

 1     def get_data(self, button="all"):
 2         """ button控制是否执行所有用例，默认值为all
 3                     如果不为all就进行分值判断
 4                     button的值只能输入all，或者列表
 5                 """
 6         
 7         wb = load_workbook(self.file_name)
 8         sheet = wb[self.sheet_name]
 9         test_data = []
10 
11         for i in range(2, sheet.max_row+1):
12             sub_data = {}
13             sub_data[‘case_id‘] = sheet.cell(i, 1).value
14             sub_data[‘module‘] = sheet.cell(i, 2).value
15             sub_data[‘title‘] = sheet.cell(i, 3).value
16             sub_data[‘method‘] = sheet.cell(i, 4).value
17             sub_data[‘url‘] = sheet.cell(i, 5).value
18             sub_data[‘data‘] = sheet.cell(i, 6).value
19             sub_data[‘expected‘] = sheet.cell(i, 7).value
20             test_data.append(sub_data)  # 存储了所有的数据
21             final_data = []
22             if button == "all":
23                 final_data = test_data
24             else:
25                 for item in test_data:
26                     if item[‘case_id‘] in button:
27                         final_data.append(item)
28         return final_data