Python实现接口自动化封装导出excel和读写excel数据

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

class Export:

    """

    导出域

    """

    def __init__(self, token):

        self.token = token

        self.headers = {

                'Authorization': self.token,

                'Content-Type': 'application/json;charset=UTF-8'

        }

        ```

def export_sku_excel(self, payload, path):

    """

            商品：商品明细导出

            """

    url = f'{HOST}/api/v1/commodity/exportSKU'

    res = client.post(url=url, json=payload, verify=False, headers=self.headers)

    resp = res.content

    with open(path, 'wb') as f:  # 第一个参数是保存文件路径，不加路径就是当前路径

        if res.status_code == 200:

            return f.write(resp)

        else:

            return False

1

2

3

4

5

6

7

8

9

10

11

12

13

class ExcelMethod:

    def __init__(self, filename):

        self.filename = filename

    def read_excel(self, row, col):

        """

        读取导出文件的数据

        Returns:excel单元格数据

        """

        wb = xlrd.open_workbook(self.filename)

        sheet_name = wb.sheet_names()[0]

        sheet1 = wb.sheet_by_index(0)

        cellInfo = sheet1.cell_value(row, col)  # 获取文件中某单元格的值

        return cellInfo  # 返回文件单元格数据