def get_data(file_name):
wb = load_workbook(file_name)
mode = eval(ReadConfig.read_config(test_config_path, 'MODE', 'mode'))
test_data = []
for key in mode: #遍历配置文件中的字典（键）
sheet = wb[key]
if mode[key] == 'all':
for i in range(2, sheet.max_row + 1):
row_data = {}
row_data['case_id'] = sheet.cell(i, 1).value
row_data['method'] = sheet.cell(i, 2).value
row_data['url'] = sheet.cell(i, 3).value
row_data['data'] = sheet.cell(i, 4).value
row_data['expected'] = sheet.cell(i, 5).value
row_data['sheet_name'] = sheet.cell(i, 6).value
test_data.append(row_data)
else:
for case_id in mode[key]:
row_data = {}
row_data['case_id'] = sheet.cell(case_id + 1, 1).value
row_data['method'] = sheet.cell(case_id + 1, 2).value
row_data['url'] = sheet.cell(case_id + 1, 3).value
row_data['data'] = sheet.cell(case_id + 1, 4).value
row_data['expected'] = sheet.cell(case_id + 1, 5).value
row_data['sheet_name'] = sheet.cell(case_id + 1, 6).value
test_data.append(row_data)
return test_data
def do_mysql(self, query, state='all'): #query-->查询语句
config = eval(ReadConfig.read_config(test_config_path, 'DB', 'config'))
#创建一个数据库连接
cnn = mysql.connector.connect(**config)
#游标cursor
cursor = cnn.cursor()
#执行语句
cursor.execute(query)
#获取结果 打印结果
if state == 1:
res = cursor.fetchone() #针对一条数据 返回的是元组
else:
res = cursor.fetchall() #针对多条数据 返回的是列表
#关闭游标
cursor.close()
#关闭连接
cnn.close()
return res
