59 lines
1.7 KiB
Python
59 lines
1.7 KiB
Python
|
"""CMS 导入 / 导出通用工具（基于 openpyxl）。"""
|
|||
|
|
import io
|
|||
|
|
|
|||
|
|
from django.http import HttpResponse
|
|||
|
|
from openpyxl import Workbook, load_workbook
|
|||
|
|
|
|||
|
|
XLSX_CONTENT_TYPE = (
|
|||
|
|
'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def xlsx_response(filename, headers, rows):
|
|||
|
|
"""生成 .xlsx 下载响应。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
filename: 下载文件名（含 .xlsx）
|
|||
|
|
headers: 表头列表
|
|||
|
|
rows: 二维数据（list[list]）；空则只导出表头（可作模板）
|
|||
|
|
"""
|
|||
|
|
wb = Workbook()
|
|||
|
|
ws = wb.active
|
|||
|
|
ws.append(list(headers))
|
|||
|
|
for row in rows:
|
|||
|
|
ws.append(list(row))
|
|||
|
|
buf = io.BytesIO()
|
|||
|
|
wb.save(buf)
|
|||
|
|
buf.seek(0)
|
|||
|
|
resp = HttpResponse(buf.getvalue(), content_type=XLSX_CONTENT_TYPE)
|
|||
|
|
resp['Content-Disposition'] = f'attachment; filename="{filename}"'
|
|||
|
|
return resp
|
|||
|
|
|
|||
|
|
|
|||
|
|
def rows_from_xlsx(file):
|
|||
|
|
"""解析上传的 .xlsx，首行为表头，返回 list[dict]（表头→单元格字符串）。
|
|||
|
|
|
|||
|
|
空行跳过；单元格统一转为去空格字符串（None → ''）。
|
|||
|
|
"""
|
|||
|
|
wb = load_workbook(file, read_only=True, data_only=True)
|
|||
|
|
ws = wb.active
|
|||
|
|
it = ws.iter_rows(values_only=True)
|
|||
|
|
try:
|
|||
|
|
header_cells = next(it)
|
|||
|
|
except StopIteration:
|
|||
|
|
return []
|
|||
|
|
headers = [str(h).strip() if h is not None else '' for h in header_cells]
|
|||
|
|
|
|||
|
|
result = []
|
|||
|
|
for raw in it:
|
|||
|
|
if raw is None or all(c is None or str(c).strip() == '' for c in raw):
|
|||
|
|
continue
|
|||
|
|
row = {}
|
|||
|
|
for i, h in enumerate(headers):
|
|||
|
|
if not h:
|
|||
|
|
continue
|
|||
|
|
val = raw[i] if i < len(raw) else None
|
|||
|
|
row[h] = '' if val is None else str(val).strip()
|
|||
|
|
result.append(row)
|
|||
|
|
return result
|