导入导出

说明

项目基于 openpyxl 实现 Excel 导入导出功能，支持通过 file_template 模块管理导入模板配置。路由定义在各业务模块的 router 文件中（如 src/modules/system/user/router.py）。

功能概述

Excel 导入导出是后台管理系统的常见需求，通过 openpyxl 库实现：

Excel 导出：
1. 将数据库中的数据导出为 Excel 文件，支持自定义表头和格式
2. 适用于生成报表、数据备份等场景

Excel 导入：
1. 将 Excel 文件中的数据导入到数据库，支持批量处理和校验
2. 适用于批量数据录入、数据迁移等场景

依赖说明

在 requirements.txt 中已包含：

openpyxl>=3.1.0

导出实现

导出接口通常在各业务模块的 router 中定义，返回文件流：

from openpyxl import Workbook
from flask import send_file
import io

@user_blue.route('/export', methods=['GET'])
@check_permission('sys:user:export')
def export():
    # 查询数据
    from src.modules.system.user.service import get_export_data
    data_list = get_export_data()

    # 创建工作簿
    wb = Workbook()
    ws = wb.active
    ws.title = "用户信息"

    # 写入表头
    ws.append(["用户名", "姓名", "手机号", "邮箱", "状态"])

    # 写入数据
    for item in data_list:
        ws.append([item.username, item.realname, item.phone, item.email, item.status_name])

    # 返回文件流
    output = io.BytesIO()
    wb.save(output)
    output.seek(0)

    return send_file(
        output,
        mimetype="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
        as_attachment=True,
        download_name="export.xlsx"
    )

导入实现

导入接口接收上传的 Excel 文件，解析后逐行校验并入库：

from openpyxl import load_workbook
from flask import request

@user_blue.route('/import', methods=['POST'])
@check_permission('sys:user:import')
def import_excel():
    file = request.files.get('file')
    if not file:
        return R.failed(msg="请上传文件")

    content = file.read()
    wb = load_workbook(io.BytesIO(content))
    ws = wb.active

    # 跳过表头，逐行解析
    errors = []
    for row_idx, row in enumerate(ws.iter_rows(min_row=2, values_only=True), start=2):
        username, realname, phone, email, status = row
        # 校验并入库 ...
        if error:
            errors.append(f"第{row_idx}行：{error}")

    if errors:
        return R.failed(msg="导入失败：" + "，".join(errors))
    return R.ok(msg="导入成功")

file_template 模块

file_template 模块用于管理导入模板的配置信息，定义模板的字段映射、校验规则等。

模块路径：src/modules/file_template/

模板配置包含：
- 模板名称
- 关联模块
- 字段映射关系
- 校验规则

温馨提示

导入时建议先下载模板文件，按模板格式填写数据后再导入，可减少格式错误。

错误处理

导入过程中遇到校验错误时，返回详细的错误信息：

{
    "code": 1,
    "msg": "导入失败：第3行用户名已存在，第5行手机号格式错误",
    "data": null,
    "ok": false
}

总结

Excel 导入导出功能基于 openpyxl 实现，导出使用 send_file 流式返回避免内存溢出，导入逐行校验并提供详细错误信息。file_template 模块可管理导入模板配置，便于维护字段映射和校验规则。