openpyxl

处理excel数据

import openpyxl
import os
from utils.path_util import Path


class LocalExcelData:

    def __init__(self, files_name, files_path="exceldata"):
        path = os.path.join(Path().get_other_path(files_path), files_name)
        self.wb = openpyxl.load_workbook(path)

    def read_all_row(self):
        all_list = list()
        sheets = self.wb.sheetnames
        for sheet in sheets:
            all_list.extend(self.read_sheet_row(sheet))
        print(len(all_list))
        return all_list

    def read_sheet_row(self, sheet_name, row=None):

        sheet = self.wb[sheet_name]
        row_list = list()
        for row_data in sheet.values:
            row_list.append(row_data)
        row_list.pop(0)
        if row is None:
            return row_list
        else:
            return row_list[row]