Python Excel工具类封装, 给excel表头搞点颜色
封装Excel工具类
我们常用的excel工具类，读有xlrd，写有xlwt。有读有写，新一代库有pandas，openpyxl等等。
大家用法都差不多，今天博主就介绍新手最爱，我也爱的xlrd和xlwt。（不过xlwt似乎最多只支持65535条数据，此乃一坑）
缘起
老板给博主安排了一个导出excel的任务，而这个新项目里面还尚未编写此类公共方法，于是就想来一波尝试。
主要是想把xlwt和xlrd做一个整合，但并不是说要对一个文件读和写，因为应用场景一般是导入或者导出。
封装ExcelHelper类
读数据
我们需要先接受一个filename（文件的全路径），初始化helper对象，所以我们可以顺手写出这样的代码:
class ExcelHelper(object):
def __init__(self, filename):
self.filename = filename
接着我们编写xlrd的部分，虽然网上的例子一大把，但是有特色的还是少。我决定支持读数组，也支持读json数组。
啥子是json数组，因为我们excel一般都是表头+表数据组成，如果光给一条数据，你不知道他属于哪一列，那还得往上找它的表头，和之对应起来。
比如我们经常读出的数据是:
data = [
['姓名', '电话'],
['三毛', '17800000000']
]
当列的数据比较多的时候，我们给你一个三毛，你知道它的表头是姓名吗？还是小名？
如果是这样的数据呢？
data = [
{"姓名": "三毛", "电话": "17800000000"}
]
两种数据，不同的展示方式，其实内容相差无几。
- 编写read_data方法
def read_data(self):
"""
获取数据，不区分表头
:return:
"""
book = xlrd.open_workbook(self.filename)
sheets = book.sheets()
for s in sheets:
yield (s.row_values(i) for i in range(s.nrows))
这边用了yield关键字，之所以不用list，还是考虑到数据表数据如果很多，那么全部放到list，会占用很大内存空间，所以用了yield节省内存空间。
这里如果有疑问，大家可以查下生成器相关资料。
- 编写read_json相关方法
@staticmethod
def read_json_item(sheet):
"""
获取json数据
:param sheet:
:return:
"""
if sheet.nrows <= 1:
return (sheet.row_values(i) for i in range(sheet.nrows))
# 否则说明数据大于1行
header = sheet.row_values(0)
for i in range(1, sheet.nrows):
row_data = dict()
for j in range(sheet.ncols):
row_data[header[j]] = sheet.cell_value(i, j)
yield row_data
def read_json_data(self):
"""
获取List[dict]数据，也就是JSON数组
:return:
"""
book = xlrd.open_workbook(self.filename)
sheets = book.sheets()
for st in sheets:
yield ExcelHelper.read_json_item(st)
比较常规，excel可能会有多个sheet，所以我们遍历sheets。接着去每个sheet中拿到每行数据，由于要求json数组模式，所以我们需要判断下行数。
如果就1行，那就最多一个表头，没啥意义，所以我们直接切换到原生模式，一行一行读数据。
写数据
写数据的demo比较简单，考虑到传入json数组的时候，万一有小可爱传这样的数据，其实我们是不太好支持的:
a = [
{"名字": "张三", "称号": "法外狂徒"},
{"性格": "闷骚", "称号": "秒杀光环"}
]
可以看到2条数据对不上~所以不打算支持这样的数据。
-
编写write_data方法
我们知道，表头是个很重要的数据，我们要让她
与众不同一点，所以我们可以设置下它的样式。
@staticmethod
def get_style():
style = xlwt.XFStyle()
pattern = xlwt.Pattern()
pattern.pattern = xlwt.Pattern.SOLID_PATTERN
pattern.pattern_fore_colour = xlwt.Style.colour_map['sky_blue']
style.pattern = pattern
return style
创建style，设置背景色为纯洁的天蓝色，最后返回style。
这个style有什么用呢，我们在写入数据的时候可以指定单元格的样式。
接着编写write方法:
def write_excel_data(self, header, row_data, sheet_name="sheet1"):
wb = xlwt.Workbook()
ws = wb.add_sheet(sheet_name)
# 写入表头
for i, h in enumerate(header):
ws.write(0, i, h, self.style)
# 写入数据
for line, row in enumerate(row_data):
for c, item in enumerate(row):
ws.write(line + 1, c, str(item))
wb.save(self.filename)
接受参数是表头（数组），row_data(二维数组)，写入完毕后调用save方法。
如果有需要对多个sheet写入，请自行改造。
本节总体来说，只是写了一个excel的读写方法，亮点在于读json数组和表头搞颜色。
下次聊聊FastApi下载文件以及删除下载后的文件。
完整代码如下（其实是给我自己备份）:
import xlrd
import xlwt
class ExcelHelper(object):
def __init__(self, filename):
self.filename = filename
self.style = ExcelHelper.get_style()
@staticmethod
def get_style():
style = xlwt.XFStyle()
pattern = xlwt.Pattern()
pattern.pattern = xlwt.Pattern.SOLID_PATTERN
pattern.pattern_fore_colour = xlwt.Style.colour_map['sky_blue']
style.pattern = pattern
return style
def read_data(self):
"""
获取数据，不区分表头
:return:
"""
book = xlrd.open_workbook(self.filename)
sheets = book.sheets()
for s in sheets:
yield (s.row_values(i) for i in range(s.nrows))
@staticmethod
def read_json_item(sheet):
"""
获取json数据
:param sheet:
:return:
"""
if sheet.nrows <= 1:
return (sheet.row_values(i) for i in range(sheet.nrows))
# 否则说明数据大于1行
header = sheet.row_values(0)
for i in range(1, sheet.nrows):
row_data = dict()
for j in range(sheet.ncols):
row_data[header[j]] = sheet.cell_value(i, j)
yield row_data
def read_json_data(self):
"""
获取List[dict]数据，也就是JSON数组
:return:
"""
book = xlrd.open_workbook(self.filename)
sheets = book.sheets()
for st in sheets:
yield ExcelHelper.read_json_item(st)
def write_excel_data(self, header, row_data, sheet_name="sheet1"):
wb = xlwt.Workbook()
ws = wb.add_sheet(sheet_name)
# 写入表头
for i, h in enumerate(header):
ws.write(0, i, h, self.style)
# 写入数据
for line, row in enumerate(row_data):
for c, item in enumerate(row):
ws.write(line + 1, c, str(item))
wb.save(self.filename)
