可以使用xlrd，也可以使用openpyxl，但是openpyxl读取不了.xls格式的Excel，需要转成xlsx，有点麻烦，所以使用了xlrd。

读取

  • 打开文件

xlrd也遇到了一点问题，文件如果是xls文件直接打开，在提取合并单元格的时候会拿不到数据，需要加上 formatting_info=True，formatting_info=True的时候打开.xlsx文件会报错NotImplementedError: formatting_info=True not yet implemented，加个if就好了

1
2
3
4
5
workbook = xlrd.open_workbook(path)
if path.split('.xl')[1] == 's':
    workbook = xlrd.open_workbook(path, formatting_info=True)
# 获取sheet
sheet = workbook.sheet_by_index(0)
  • 获取所有的合并单元格坐标

突突突

1
2
3
4
5
6
# 获取列数
r_num = sheet.nrows
# 获取行数
c_num = sheet.ncols
merge = sheet.merged_cells
print(merge)   # [(1, 5, 0, 1), (1, 5, 1, 2)], 对应上面两个合并的单元格
  • 组装数据
1
2
3
4
5
6
7
8
9
10
11
12
13
14
read_data =[]
for r in range(r_num):
    li = []
    for c in range(c_num):
        # 读取每个单元格里的数据，合并单元格只有单元格内的第一行第一列有数据，其余空间都为空
        cell_value = sheet.row_values(r)[c]
        # 判断空数据是否在合并单元格的坐标中，如果在就把数据填充进去
        if cell_value is None or cell_value == '':
            for (rlow, rhigh, clow, chigh) in merge:
                if rlow <= r < rhigh:
                    if clow <= c < chigh:
                        cell_value = sheet.cell_value(rlow, clow)
        li.append(cell_value)
    read_data.append(li)

写入

使用的是xlwt

  • 直接上代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
f = xlwt.Workbook()
sheet1 = f.add_sheet('sheet1', cell_overwrite_ok=True)
headers = ['序号', "姓名", '列表', '1月', '2月', '3月', '4月', '5月']
name = ['1-AAA', '2-BBB', '3-CCC', '4-DDD']
status = ['累计应发数', '累计专项扣除-五险一金', '累计免税收入', '累计扣个税']
# 生成第一行header
for i in range(0,len(headers)):
    sheet1.write(0,i,headers[i])

# 生成序号、姓名单元格
i, j = 1, 0
while i < 10*len(name) and j < len(name):
    sheet1.write_merge(i,i+9,0,0,name[j].split('-')[0])
    sheet1.write_merge(i,i+9,1,1,name[j].split('-')[1])
    i += 10
    j += 1

#生成列表
i = 0
while i < 10*len(column0):
    for j in range(0,len(status)):
        sheet1.write(j+i+1,3,status[j])
    i += 10
# 保存
f.save(path)

Django导出

  • 服务端代码
1
2
3
4
5
6
7
8
9
10
11
excel_stream = io.BytesIO()
# BytesIO流（在内存中读写）
f.save(excel_stream)
res = excel_stream.getvalue()
excel_stream.close()
response = HttpResponse(content_type='application/vnd.ms-excel')
from urllib import parse
file_name = str(path.name).split('.xls')[0] + '(计算后).xls'
response['Content-Disposition'] = 'attachment;filename=' + parse.quote(file_name)
response.write(res)
return response
  • 前端

写一个简单的form表单就行了