python之openpyxl模块

一 . Python操作EXCEL库的简介

1.1 Python官方库操作excel

Python官方库一般使用xlrd库来读取Excel文件，使用xlwt库来生成Excel文件，使用xlutils库复制和修改Excel文件，这三个库只支持到Excel2003。

1.2 第三方库openpyxl介绍

第三方库openpyxl（可读写excel表），专门处理Excel2007及以上版本产生的xlsx文件，xls和xlsx之间转换容易。 注意：如果文字编码是“gb2312” 读取后就会显示乱码，请先转成Unicode

本文将详细介绍第三方库openpyxl的基本用法

第三方库openpyxl的安装：

<1>下载路径：https://pypi.python.org/pypi/openpyxl

<2>解压到指定文件目录：tar -xzvf openpyxl.tar.gz

<3>进入目录，找到setup.py文件，执行命令：python3 setup.py install 如果报错No module named setuptools 就使用命令“easy_install openpyxl”，easy_install for win32，会自动安装setuptools； 或者直接用cmd命令：pip3 install openpyxl安装

二. openpyxl库基本操作总结

2.1 openpyxl 的基本操作

openpyxl中有三个不同层次的类，Workbook是对工作簿的抽象，Worksheet是对表格的抽象，Cell是对单元格的抽象，每一个类都包含了许多属性和方法。

(1)Excel基本操作

操作Excel的一般场景:

a)打开或者创建一个Excel需要创建一个Workbook对象

b)获取一个表则需要先创建一个Workbook对象，然后使用该对象的方法来得到一个Worksheet对象

c)如果要获取表中的数据，那么得到Worksheet对象以后再从中获取代表单元格的Cell对象

(2)Workbook对象知识点总结

一个Workbook对象代表一个Excel文档，因此在操作Excel之前，都应该先创建一个Workbook对象。对于创建一个新的Excel文档，直接进行Workbook类的调用即可，对于一个已经存在的Excel文档，可以使用openpyxl模块的load_workbook函数进行读取，该函数包含多个参数，但只有filename参数为必传参数。filename 是一个文件名，也可以是一个打开的文件对象。

创建workbook对象:

import openpyxl
>>> wb = openpyxl.Workbook(‘hello.xlxs‘)
>>> wb = openpyxl.load_workbook(‘file_name.xlxs‘)

注意：Workbook和load_workbook相同，返回的都是一个Workbook对象。

Workbook对象提供了很多属性和方法，其中，大部分方法都与sheet有关，其所有属性和方法如下：

['_sheets', '_pivots', '_active_sheet_index', 'defined_names', '_external_links', 'properties', 'security', '_Workbook__write_only', 'shared_strings', '_fonts', '_alignments', '_borders', '_fills', '_number_formats', '_protections', '_colors', '_cell_styles', '_named_styles', '_table_styles', 'loaded_theme', 'vba_archive', 'is_template', '_differential_styles', 'code_name', 'epoch', 'encoding', 'iso_dates', 'rels', 'calculation', 'views', '_data_only', '_read_only', '_keep_links', 'guess_types', 'template', '__module__', '__doc__', 'path', '__init__', '_setup_styles', 'read_only', 'data_only', 'write_only', 'keep_links', 'get_active_sheet', 'excel_base_date', 'active', 'create_sheet', '_add_sheet', 'remove', 'remove_sheet', 'create_chartsheet', 'get_sheet_by_name', '__contains__', 'index', 'get_index', '__getitem__', '__delitem__', '__iter__', 'get_sheet_names', 'worksheets', 'chartsheets', 'sheetnames', 'create_named_range', 'add_named_style', 'named_styles', 'get_named_ranges', 'add_named_range', 'get_named_range', 'remove_named_range', 'mime_type', 'save', 'style_names', 'copy_worksheet', 'close', '__dict__', '__weakref__', '__repr__', '__hash__', '__str__', '__getattribute__', '__setattr__', '__delattr__', '__lt__', '__le__', '__eq__', '__ne__', '__gt__', '__ge__', '__new__', '__reduce_ex__', '__reduce__', '__subclasshook__', '__init_subclass__', '__format__', '__sizeof__', '__dir__', '__class__']

 Workbook提供的部分常用属性如下：

• active：获取当前活跃的Worksheet
• worksheets：以列表的形式返回所有的Worksheet(表格)
• read_only：判断是否以read_only模式打开Excel文档
• write_only:判断是否以write_only模式打开Excel文档
• encoding：获取文档的字符集编码
• properties：获取文档的元数据，如标题，创建者，创建日期等
• sheetnames：获取工作簿中的表（列表）

#常用属性练习
    workbook_path= os.path.join(CONFIG_PATH,'testdata.xlsx')
    wb = openpyxl.load_workbook(workbook_path)
    ws = wb.active
    print(ws)#<Worksheet "Sheet1">
    ws_list=wb.worksheets
    print(ws_list)#[<Worksheet "Sheet1">, <Worksheet "Sheet2">, <Worksheet "Sheet3">]
    print(wb.read_only) #False
    print(wb.encoding)#utf-8
    print(wb.write_only)#False
    print(wb.sheetnames)#['Sheet1', 'Sheet2', 'Sheet3']
    print(wb.properties)
    #<openpyxl.packaging.core.DocumentProperties object>
    #Parameters:
    #creator='openpyxl', title=None, description=None, subject=None, identifier=None, language=None, created=datetime.datetime(2018, 8, 18, 2, 14, 50),
    # modified=datetime.datetime(2018, 8, 18, 5, 12, 7), lastModifiedBy='wuxiaoli', category=None, contentStatus=None, version=None, revision=None,
    #  keywords=None, lastPrinted=None

 Workbook提供的部分常用方法如下：

• get_sheet_names：获取所有表格的名称(新版已经不建议使用，通过Workbook的sheetnames属性即可获取)
• get_sheet_by_name：通过表格名称获取Worksheet对象(新版也不建议使用，通过Worksheet[‘表名‘]获取)
• get_active_sheet：获取活跃的表格(新版建议通过active属性获取)
• remove_sheet：删除一个表格
• create_sheet：创建一个空的表格
• copy_worksheet：在Workbook内拷贝表格

 #workbook常用方法练习   
    workbook_path= os.path.join(CONFIG_PATH,'testdata.xlsx')
    wb = openpyxl.load_workbook(workbook_path)
    sheet_names_list=wb.get_sheet_names()#['Sheet1', 'Sheet2', 'Sheet3']
    ws1 = wb.get_sheet_by_name('Sheet2')#<Worksheet "Sheet2">
    ws2 = wb.get_active_sheet()#<Worksheet "Sheet1">
    ws3 = wb.create_sheet('text')#<Worksheet "text">
    print(wb.sheetnames)
    #wb.remove_sheet(ws3)
    wb.copy_worksheet(from_worksheet=ws1)

(3)worksheet对象知识点总结

有了Worksheet对象以后，我们可以通过这个Worksheet对象获取表格的属性，得到单元格中的数据，修改表格中的内容。openpyxl提供了非常灵活的方式来访问表格中的单元格和数据，Worksheet对象的所有属性和方法如下:

['_WorkbookChild__parent', '_WorkbookChild__title', 'HeaderFooter', 'row_dimensions', 'column_dimensions', 'page_breaks', '_cells', '_charts', '_images', '_rels', '_drawing', '_comments', 'merged_cells', '_tables', '_pivots', 'data_validations', '_hyperlinks', 'sheet_state', 'page_setup', 'print_options', '_print_rows', '_print_cols', '_print_area', 'page_margins', 'views', 'protection', '_current_row', 'auto_filter', 'sort_state', 'paper_size', 'formula_attributes', 'orientation', 'conditional_formatting', 'legacy_drawing', 'sheet_properties', 'sheet_format', '__module__', '__doc__', '_rel_type', '_path', 'mime_type', 'BREAK_NONE', 'BREAK_ROW', 'BREAK_COLUMN', 'SHEETSTATE_VISIBLE', 'SHEETSTATE_HIDDEN', 'SHEETSTATE_VERYHIDDEN', 'PAPERSIZE_LETTER', 'PAPERSIZE_LETTER_SMALL', 'PAPERSIZE_TABLOID', 'PAPERSIZE_LEDGER', 'PAPERSIZE_LEGAL', 'PAPERSIZE_STATEMENT', 'PAPERSIZE_EXECUTIVE', 'PAPERSIZE_A3', 'PAPERSIZE_A4', 'PAPERSIZE_A4_SMALL', 'PAPERSIZE_A5', 'ORIENTATION_PORTRAIT', 'ORIENTATION_LANDSCAPE', '__init__', '_setup', 'sheet_view', 'selected_cell', 'active_cell', 'show_gridlines', 'show_summary_below', 'show_summary_right', 'vba_code', 'get_cell_collection', 'freeze_panes', 'add_print_title', 'cell', '_get_cell', '_add_cell', '__getitem__', '__setitem__', '__iter__', '__delitem__', 'min_row', 'max_row', 'min_column', 'max_column', 'calculate_dimension', 'dimensions', 'iter_rows', '_cells_by_row', 'rows', 'values', 'iter_cols', '_cells_by_col', 'columns', 'get_squared_range', 'get_named_range', 'set_printer_settings', 'add_data_validation', 'add_chart', 'add_image', 'add_table', 'add_pivot', 'merge_cells', '_clean_merge_range', 'merged_cell_ranges', 'unmerge_cells', 'append', '_move_cells', 'insert_rows', 'insert_cols', 'delete_rows', 'delete_cols', '_invalid_row', '_add_column', '_add_row', '_write', 'print_title_rows', 'print_title_cols', 'print_titles', 'print_area', '_id', '_default_title', '__repr__', 'parent', 'encoding', 'title', 'oddHeader', 'oddFooter', 'evenHeader', 'evenFooter', 'firstHeader', 'firstFooter', 'path', '__dict__', '__weakref__', '__hash__', '__str__', '__getattribute__', '__setattr__', '__delattr__', '__lt__', '__le__', '__eq__', '__ne__', '__gt__', '__ge__', '__new__', '__reduce_ex__', '__reduce__', '__subclasshook__', '__init_subclass__', '__format__', '__sizeof__', '__dir__', '__class__']

Worksheet提供的部分常用属性如下：

• title：表格的标题
• dimensions：表格的大小，这里的大小是指含有数据的表格的大小，即：左上角的坐标:右下角的坐标
• max_row：表格的最大行
• min_row：表格的最小行
• max_column：表格的最大列
• min_column：表格的最小列
• rows：按行获取单元格(Cell对象) - 生成器
• columns：按列获取单元格(Cell对象) - 生成器
• freeze_panes：冻结窗格
• values：按行获取表格的内容(数据)  - 生成器

注意：freeze_panes，参数比较特别，主要用于在表格较大时冻结顶部的行或左边的行。对于冻结的行，在用户滚动时，是始终可见的，可以设置为一个Cell对象或一个端元个坐标的字符串，单元格上面的行和左边的列将会冻结(单元格所在的行和列不会被冻结)。例如我们要冻结第一行那么设置A2为freeze_panes，如果要冻结第一列，freeze_panes取值为B1，如果要同时冻结第一行和第一列，那么需要设置B2为freeze_panes，freeze_panes值为none时 表示 不冻结任何列。

 Workbook提供的部分常用方法如下：

• iter_rows：按行获取所有单元格，内置属性有(min_row,max_row,min_col,max_col)
• iter_columns：按列获取所有的单元格
• append：在表格末尾添加数据
• merged_cells：合并多个单元格
• unmerged_cells：移除合并的单元格

(4)cell对象知识点总结

Cell对象比较简单，常用的属性如下:

• row：单元格所在的行
• column：单元格坐在的列
• value：单元格的值
• coordinate：单元格的坐标

(5)Excel基本操作总结

<1>openpyxl的三种最重要的数据结构

NULL空值：对应于python中的None，表示这个cell里面没有数据。
numberic： 数字型，统一按照浮点数来进行处理。对应于python中的float。
string： 字符串型，对应于python中的unicode。

<2>Excel 文件的三个对象

workbook： 工作簿，一个excel工作簿包含多个sheet。
sheet：工作表，一个workbook有多个，表名识别，如“sheet1”,“sheet2”等。
cell： 单元格，存储数据对象

<3>导入

from openpyxl import Workbook
from openpyxl import load_workbook
from openpyxl.styles import PatternFill, Border, Side, Alignment, Protection, Font, Color, Fill
from openpyxl.styles import colors
from openpyxl.styles import Fill,fills
from openpyxl.formatting.rule import ColorScaleRule

<4>加载workbook

加载workbook时注意openpyxl只支持xlsx格式，老版的xls格式需要其他方法去加载。 
wb = load_workbook('file_name.xlsx')或wb = load_workbook(filename = r'file_namexlsx')

<5>新建worksheet

新建工作表用函数create_sheet()

create_sheet(title=None, index=None):
        """Create a worksheet (at an optional index).
        :param title: optional title of the sheet
        :type title: unicode
        :param index: optional position at which the sheet will be inserted
        :type index: int

        """
ws1 = wb.create_sheet() #默认插在最后
ws2 = wb.create_sheet(0) #插在开头建表后默认名按顺序，如sheet1，sheet2...
ws.title = "New Title" #修改表名称
简化 ws2 = wb.create_sheet(title="sheet_name")

<6>打开 worksheet

1)通过名字
    ws = wb["sheet_name"] 
    等同于 ws2 = wb.get_sheet_by_name('sheet_name')
    
2)不知道名字用index
    sheet_names = wb.get_sheet_names()
    ws = wb.get_sheet_by_name(sheet_names[index])# index为0为第一张表 

或者
    ws =wb.active
    等同于  ws = wb.get_active_sheet() #通过_active_sheet_index设定读取的表，默认0读第一个表
3) 获取活动表表名wb.get_active_sheet().title

<7>编辑单元格

1)读写单个单元格
c = ws['A4'] #read 等同于 c = ws.cell('A4') 
ws['A4'] = 4 #write 
#ws.cell有两种方式，行号列号从1开始
d = ws.cell(row = 4, column = 2) #行列读写
d = ws.cell('A4') 
写入cell值
    ws.cell(row = 4, column = 2).value = 'test')
    ws.cell(row = 4, column = 2, value = 'test')
2)读写多个单元格
cell_range = ws['A1':'C2']#读取A1到C1单元格数据
get_cell_collection()#读所有单元格数据


<8>获取最大行和最大列

print(sheet.max_row)
print(sheet.max_column)

<9>获取行和列的值

sheet.rows为生成器, 里面是每一行的数据，每一行又由一个tuple包裹。
sheet.columns类似，不过里面是每个tuple是每一列的单元格。
# 按行获取值，返回值为A1, B1, C1这样的顺序
for row in sheet.rows:
    for cell in row:
        print(cell.value)
# 按列获取值，返回值为A1, A2, A3这样的顺序
for column in sheet.columns:
    for cell in column:
        print(cell.value)
上面的代码就可以获得所有单元格的数据。如果要获得某行的数据呢？因sheet.rows是生成器类型，不能使用索引，但是转换成list之后便可使用索引，如list(sheet.rows)[2]这样就获取到第三行的tuple对象。
for cell in list(sheet.rows)[2]:
    print(cell.value)
也可以通过iter_rows的参数min_row, max_row指定并获取指定行内容，如下所示，获取第二行表格的内容:

for row in ws.iter_rows(min_row=2, max_row=2,):
    line = [cell.value for cell in row]
    print(line)

此外可使用方法ws.iter_rows()逐行读取数据
ws.iter_rows(range_string=None, row_offset=0, column_offset=0): range-string(string)-单元格的范围：例如('A1:C4') row_offset-添加行 column_offset-添加列 
   # 返回一个生成器, 注意取值时要用value,例如：
from openpyxl import load_workbook
from constant.constant import CONFIG_PATH
import os.path
test_data = os.path.join(CONFIG_PATH,'testdata.xlsx')
wb = load_workbook(test_data)
ws = wb.active
ws.title = 'testdata.xlsx'
for row in ws.iter_rows('A1:C2'):
    for cell in row:
        print(cell.value)

<10>获取任意区间的值

法1.使用range函数
可以使用range函数，下面的写法，获得了以A1为左上角，B3为右下角矩形区域的所有单元格。注意range从1开始的，因为在openpyxl中为了和Excel中的表达方式一致，并不和编程语言的习惯以0表示第一个值。
workbook_path= os.path.join(CONFIG_PATH,'testdata.xlsx')
wb = openpyxl.load_workbook(workbook_path)
ws = wb.get_sheet_by_name(name='Sheet1')
for i in range(1, 4):
    for j in range(1, 3):
        print(ws.cell(row=i, column=j))
#out
#<Cell 'Sheet1'.A1>
#<Cell 'Sheet1'.B1>
#<Cell 'Sheet1'.A2>
#<Cell 'Sheet1'.B2>
#<Cell 'Sheet1'.A3>
#<Cell 'Sheet1'.B3>
法2.使用切片，如sheet['A1':'B3']，返回一个tuple，该元组内部还是元组，由每行的单元格构成一个元组。

workbook_path= os.path.join(CONFIG_PATH,'testdata.xlsx')
wb = openpyxl.load_workbook(workbook_path)
ws = wb.get_sheet_by_name(name='Sheet1')

for row_cell in ws['A1':'B3']:
     for cell in row_cell:
           print(cell)
#out
#<Cell 'Sheet1'.A1>
#<Cell 'Sheet1'.B1>
#<Cell 'Sheet1'.A2>
#<Cell 'Sheet1'.B2>
#<Cell 'Sheet1'.A3>
#<Cell 'Sheet1'.B3>

for cell in sheet['A1':'B3']:
    print(cell)
#out
#(<Cell 'Sheet1'.A1>, <Cell 'Sheet1'.B1>)
#(<Cell 'Sheet1'.A2>, <Cell 'Sheet1'.B2>)
#(<Cell 'Sheet1'.A3>, <Cell 'Sheet1'.B3>)

<11>根据字母获得列号，根据列号返回字母
from openpyxl.utils import get_column_letter, column_index_from_string

# 根据列的数字返回字母
print(get_column_letter(2))  # B
# 根据字母返回列的数字
print(column_index_from_string('D'))  # 4

<12>写单元格

# 直接给单元格赋值就行
sheet['A1'] = 'good'
# 利用公式在B9处写入平均值
sheet['B9'] = '=AVERAGE(B2:B8)'
#注意：如果是读取，需要加上data_only=True，这样读到B9返回的才是数字，如果不加这个参数，返回的将是公式本身'=AVERAGE(B2:B8)'

<13>使用append函数写数据

append函数可以一次添加多行数据，从第一行空白行开始（下面都是空白行）写入。
ws.append(iterable)
 #添加一行到当前sheet的最底部（即逐行追加从第一行开始） iterable必须是list,tuple,dict,range,generator类型的。 1,如果是list,将list从头到尾顺序添加。 2，如果是dict,按照相应的键添加相应的键值。
 ws.append([‘This is A1’, ‘This is B1’, ‘This is C1’])
 ws.append({‘A’ : ‘This is A1’, ‘C’ : ‘This is C1’})
 ws.append({1 : ‘This is A1’, 3 : ‘This is C1’})
# 添加一行如下：
row = [1 ,2, 3, 4, 5]
sheet.append(row)

 2.2 openpyxl练习

(1)Excel的读取操作练习

练习表格内容如下

import openpyxl


if __name__=='__main__':
    
    wb = openpyxl.load_workbook(r'D:\pythonstudy\2018-08-18.xlsx')
    # 取第一张表
    sheetnames = wb.sheetnames
    ws = wb[sheetnames[0]]
    # 显示表名，表行数，表列数
    print("Work Sheet Title:", ws.title)
    print("Work Sheet Rows:", ws.max_row)
    print("Work Sheet Cols:",ws.max_column)
   #获取指定行的值，如第三行
    row3_values = []
    row3_cell_list = list(ws.rows)[2]
    for cell in row3_cell_list:
        row3_values.append(cell.value)
    print(row3_values)#output:['A3', 'B3', 'C3', 'D3']
    
    #获取所有行的数据
    # 建立存储数据的字典
    data = {}
    # 获取表格所有值
    #法1:
    for i in range(0,ws.max_row):
        every_row_values = []
        every_row_cell_list = list(ws.rows)[i]
        for cell in every_row_cell_list:
            every_row_values.append(cell.value)
        data[i+1]= every_row_values
    print(data)
    #{1: ['A1', 'B1', 'C1', 'D1'], 2: ['A2', 'B2', 'C2', 'D2'], 3: ['A3', 'B3', 'C3', 'D3'], 4: ['A4', 'B4', 'C4', 'D4'], 5: ['A5', 'B5', 'C5', 'D5'],
    #  6: ['A6', 'B6', 'C6', 'D6'], 7: ['A7', 'B7', 'C7', 'D7'], 8: ['A8', 'B8', 'C8', 'D8'], 9: ['A9', 'B9', 'C9', 'D9'], 10: ['A10', 'B10', 'C10', 'D10'],
    # 11: ['A11', 'B11', 'C11', 'D11']}
    #法2
    for row in ws.rows:
    line = [cell.value for cell in row]
    print (line)#output:['A1', 'B1', 'C1', 'D1'],['A2', 'B2', 'C2', 'D2'], ['A3', 'B3', 'C3', 'D3'],['A4', 'B4', 'C4', 'D4'], ['A5', 'B5', 'C5', 'D5'],['A6', 'B6', 'C6', 'D6'],['A7', 'B7', 'C7', 'D7'],['A8', 'B8', 'C8', 'D8'],['A9', 'B9', 'C9', 'D9'],['A10', 'B10', 'C10', 'D10'],['A11', 'B11', 'C11', 'D11']
    #获取某个区间的值，例：获得了以A3为左上角，C6为右下角矩形区域的所有单元格
    #法1：使用range
    data1={}
    for i in range(2,6):
        every_row_values = []
        every_row_cell_list = list(ws.rows)[i]
        for cell in every_row_cell_list:
            every_row_values.append(cell.value)
        data1[i+1]=every_row_values
    print(data1)
#output{3: ['A3', 'B3', 'C3', 'D3'], 4: ['A4', 'B4', 'C4', 'D4'], 5: ['A5', 'B5', 'C5', 'D5'], 6: ['A6', 'B6', 'C6', 'D6']}

    #法2:使用切片
    data_list = []
    for row_cell in ws['A3':'C6']:
        every_row_value = []
        for cell in row_cell:
            every_row_value.append(cell.value)
        data_list.append(every_row_value)
    print(data_list)#output:[['A3', 'B3', 'C3'], ['A4', 'B4', 'C4'], ['A5', 'B5', 'C5'], ['A6', 'B6', 'C6']]

 (2)封装一个读取表格的类，实用于读测试用例

#-*- coing=utf-8 -*-
from openpyxl import load_workbook
from constant.constant import CONFIG_PATH
import os.path

class ExcelReader(object):
    """
        Read the contents of the excel file, Return list.
        example:
        The contents of the excel file：
        | A  | B  | C  |
        | A1 | B1 | C1 |
        | A2 | B2 | C2 |
        print(ExcelReader(excel, title_line=True).data)，output：
        [{A: A1, B: B1, C:C1}, {A:A2, B:B2, C:C2}]
        print(ExcelReader(excel, title_line=False).data)，output：
        [[A,B,C], [A1,B1,C1], [A2,B2,C2]]
        You can specify sheet through index or name: 
        example:
        ExcelReader(excel, sheet=2)
        ExcelReader(excel, sheet='testdata')
    """

    def __init__(self, excel, sheet=0, title_line=True):
        if os.path.exists(excel):
           self.excel = excel
        else:
            raise FileNotFoundError('Excel file does not exist ！')
        self.sheet = sheet
        self.title_line = title_line
        self._data = list()

    @property
    def data(self):
        if not self._data:
            workbook = load_workbook(self.excel)
            if type(self.sheet) not in [int, str]:
                raise TypeError('Please pass in <type int> or <type str>, not {0}'.format(type(self.sheet)))
            elif type(self.sheet) == int:
                sheetnames = workbook.sheetnames
                ws = workbook[sheetnames[self.sheet]]
            else:
                ws = workbook[self.sheet]

            if self.title_line:
                for row in ws.iter_rows(min_row=1, max_row=1, ):
                    title = [cell.value for cell in row]# 首行为title

                for i in range(1, ws.max_row):
                    every_row_cell_list = list(ws.rows)[i]
                    every_row_value = []
                    for cell in every_row_cell_list:
                        every_row_value.append(cell.value)
                    self._data.append(dict(zip(title, every_row_value)))
            else:
                for row in ws.rows:
                    self._data.append([cell.value for cell in row])
        return self._data

if __name__ == '__main__':
    test_data = os.path.join(CONFIG_PATH,'testdata.xlsx')
    reader = ExcelReader(test_data,sheet= 0,title_line=False)
    print(reader.data)

 (3)Openpyxl常用操作的封装

#-*- coing=utf-8 -*-
from openpyxl import load_workbook
from openpyxl.utils import get_column_letter, column_index_from_string
from config.globalconfig import TEST_DATA_PATH
import os.path

class ExcelReader(object):

    def __init__(self, excel, sheet=0, title_line=True,column=None,row=None):
        """
        :param excel: excel path,type=str
        :param sheet: sheet name or sheet sequence number,type = str or int
        :param title_line: title_line = True,return dic,title_line = False,return list
        :param column: Non essential parameter,type = int or str,int express column number，str express column field name of first line
        :param row: Non essential parameter,type = int,express row number
        """
        if os.path.exists(excel):
           self.excel = excel
        else:
            raise FileNotFoundError('Excel file does not exist ！')
        self.sheet = sheet
        self.title_line = title_line
        self.column = column
        self.row = row
        self._data = []
        self._cell_data = str()
        self.workbook = load_workbook(self.excel)
        if type(self.sheet) not in [int, str]:
            raise TypeError('Please pass in <type int> or <type str>, not {0}'.format(type(self.sheet)))
        elif type(self.sheet) == int:
            sheetnames = self.workbook.sheetnames
            self.ws = self.workbook[sheetnames[self.sheet]]
        else:
            self.ws = self.workbook[self.sheet]

    @property
    def sheet_data(self):
        """
            Read the contents of the excel file, Return list.
            example:
            The contents of the excel file：
            | A  | B  | C  |
            | A1 | B1 | C1 |
            | A2 | B2 | C2 |
            print(ExcelReader(excel, title_line=True).data)，output：
            [{A: A1, B: B1, C:C1}, {A:A2, B:B2, C:C2}]
            print(ExcelReader(excel, title_line=False).data)，output：
            [[A,B,C], [A1,B1,C1], [A2,B2,C2]]
            You can specify sheet through index or name:
            example:
            ExcelReader(excel, sheet=2)
            ExcelReader(excel, sheet='testdata')
        """
        if not self._data:
            if self.title_line:
                for row in self.ws.iter_rows(min_row=1, max_row=1, ):
                    title = [cell.value for cell in row]# 首行为title

                for i in range(1, self.ws.max_row):
                    every_row_cell_list = list(self.ws.rows)[i]
                    every_row_value = []
                    for cell in every_row_cell_list:
                        every_row_value.append(cell.value)
                    self._data.append(dict(zip(title, every_row_value)))
            else:
                for row in self.ws.rows:
                    self._data.append([cell.value for cell in row])
        return self._data

    def column_letter(self):
        """Column number converted to letter form."""
        if isinstance(self.column, (int)):
            return get_column_letter(self.column)  # 数字转换为字母
        elif isinstance(self.column, (str)):
            for row in self.ws.iter_rows(min_row=1, max_row=1, ):
                columnFieldName = [cell.value for cell in row]
            if columnFieldName.count(self.column) == 1:  # 相同字段名存在的个数
                for index, cell in enumerate(columnFieldName):
                    if cell == self.column:
                        return get_column_letter(index + 1)
            elif columnFieldName.count(self.column) == 0:
                raise ValueError("Current sheet page '%s' no field name '%s' exists!" % (self.ws.title, self.column))
            else:
                raise ValueError("Current sheet page'%s'include duplicate field name '%s'" % (self.ws.title, self.column))
        else:
            raise TypeError("The column parameter type should be int or str, not '%s'" % (self.column.__class__))
    @property
    def cell_data(self):
        if not self._cell_data:
            if not isinstance(self.row, (int)):
                raise TypeError("The row parameter type should be int instead of '%s'!" % (self.row.__class__))
            column = self.column_letter()
            return self.ws[column + str(self.row)].value


class ExcelWritter(ExcelReader):
    def __init__(self, excel, sheet=0, column=None,row=None,value=''):
        ExcelReader.__init__(self,excel,sheet=0)
        self.value = value
        self.column =column
        self.row = row

    def write_cell(self):
        if not isinstance(self.row, (int)):
            raise TypeError("The row parameter type should be int instead of '%s'!" % (self.row.__class__))
        if not isinstance(self.value, (str)):
            raise TypeError("The value parameter type should be str instead of '%s'!" % (self.value.__class__))

        column = self.column_letter()
        self.ws[column + str(self.row)].value = self.value
        self.workbook.save(self.excel)

(4)写存在的xlsx文件，报错:PermissionError: [Errno 13] Permission denied的解决办法

此问题为文件被拒绝访问，主要是该文件已经被打开了。关闭此excel文件后在执行excel的相关操作就ok了。

 

 

>>>>>>>>>>待续