python每日习题：读取excel，根据excel内容实现定制化需求

Posted 2021-10-15 七月的小尾巴

需求

假设上方excel中的内容是我们的测试用例，需要实现需求如下：

1、读取excel中的内容，转换成指定数据类型如下：

2、现在excel中的case_id 是乱序的，判断根据case_id 进行升序排序

3、前两部操作封装成一个函数，再不修改原先函数内容的情况下，将所有的请求方式都改成 GET 请求（建议封装装饰器实现）

代码实现

import xlrd


def up_data(func):
    def swapper(*args, **kwargs):
        res = func(*args, **kwargs)
        for i in res:
            i['methond'] = 'GET'
        return res

    return swapper


@up_data
def get_data(filename):

    # todo 打开excle
    xl = xlrd.open_workbook(filename)

    # 通过索引获取工作表
    table = xl.sheets()[0]
    # 获取一共多少行
    rows = table.nrows
    data_list = []
    for i in range(1, rows):

        a = dict(zip(table.row_values(0), table.row_values(i)))
        a['case_id'] = int(a['case_id'])
        data_list.append(a)

    # 判断根据case_id升序排序
    return sorted(data_list, key=lambda x: x['case_id'], reverse=False)