XH_Digital_Management/common/views.py

import json
import os
import urllib.parse

import openpyxl
from django.apps import apps
from django.core.exceptions import ValidationError
from openpyxl import load_workbook

from django.conf import settings
from django.contrib.auth.decorators import login_required
from django.contrib.staticfiles import finders
from django.core.files.storage import default_storage
from django.http import FileResponse, HttpResponseNotFound, JsonResponse
from django.views.decorators.csrf import csrf_protect
from rest_framework.serializers import ModelSerializer


@login_required
def download_template(request, template_name):
    """
    该视图提供下载指定的Excel模板文件。

    参数:
        request: HttpRequest对象。
        template_name: 请求下载的Excel模板文件的名称，期望是一个字符串。
        fields: 包含字段信息的字典，用于生成Excel文件的表头和格式。

    返回:
        FileResponse对象，允许用户下载生成的Excel文件。
        如果文件未找到，则返回一个HttpResponseNotFound响应。
    """
    # 使用finders.find()根据文件名在Django的静态文件目录中查找文件的绝对路径
    file_path = finders.find(f'excels/{template_name}')

    # 如果文件路径不存在，返回404响应
    if not file_path:
        return HttpResponseNotFound('<h1>文件未找到</h1>')

    # 创建FileResponse对象，设置为以附件形式下载，设置MIME类型为Excel文件
    response = FileResponse(open(file_path, 'rb'),
                            content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
    # 设置Content-Disposition头，告诉浏览器这是一个需要下载的文件
    # 使用urllib.parse.quote对文件名进行URL编码，以确保文件名中包含的特殊字符不会引起问题
    response['Content-Disposition'] = f'attachment; filename="{urllib.parse.quote(template_name)}"'
    return response


@login_required
def download_excel_template(request, template_name, fields):
    """
    该视图提供下载指定的Excel模板文件。

    参数:
        request: HttpRequest对象。
        template_name: 请求下载的Excel模板文件的名称，期望是一个字符串。
        fields: 包含字段信息的字典，用于生成Excel文件的表头和格式。

    返回:
        FileResponse对象，允许用户下载生成的Excel文件。
        如果文件未找到，则返回一个HttpResponseNotFound响应。
    """
    fields = json.loads(urllib.parse.unquote(fields).replace("'", '"'))

    # 创建一个新的工作簿
    wb = openpyxl.Workbook()
    ws = wb.active

    # 设置表头行
    headers = [field['label'] for field in fields if field['is_add']]
    ws.append(headers)

    # 保存生成的Excel文件到指定路径
    static_dir = os.path.join(settings.BASE_DIR, 'static', 'excels')
    if not os.path.exists(static_dir):
        os.makedirs(static_dir)
    temp_file_path = os.path.join(static_dir, template_name)
    wb.save(temp_file_path)

    # 创建FileResponse对象，设置为以附件形式下载，设置MIME类型为Excel文件
    response = FileResponse(open(temp_file_path, 'rb'),
                            content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')

    response['Content-Disposition'] = f'attachment; filename="{urllib.parse.quote(template_name)}"'

    return response


@csrf_protect
@login_required
def common_excel_parse(request):
    """
    该函数用于解析上传的Excel文件并返回数据。

    Args:
        request: HTTP请求对象，包含上传的Excel文件。

    Returns:
        JsonResponse: 包含解析后的Excel数据的JSON响应，或包含错误消息的JSON响应。

    Raises:
        N/A
    """

    # 如果是POST请求并且有上传的文件
    if request.method == 'POST' and request.FILES:
        # 获取上传的Excel文件
        excel_file = request.FILES.get('excel_file')

        # 如果没有提供文件，返回错误响应
        if not excel_file:
            return JsonResponse({'error': '请先选择文件。'}, status=400)

        # 获取模型名称
        model_config = request.POST.get('model_config', '')

        # 动态获取模型
        try:
            app_label, model_name = model_config.split('.')
            model = apps.get_model(app_label, model_name)
        except (LookupError, KeyError, ValueError):
            return JsonResponse({'error': '模型配置错误。'}, status=400)

        # 获取模型中配置的不需要的字段
        exclude_fields = getattr(model, 'excluded_fields', [])

        # 保存文件到服务器
        file_name = default_storage.save(excel_file.name, excel_file)
        file_path = os.path.join(settings.MEDIA_ROOT, file_name)

        def create_dynamic_serializer(mod, include):
            class DynamicSerializer(ModelSerializer):
                class Meta:
                    model = mod
                    fields = include

            return DynamicSerializer

        try:
            # 打开并解析Excel文件
            workbook = load_workbook(file_path, data_only=True)
            sheet = workbook.active
            data = []

            # 读取第一行作为表头
            header_row = [cell.value for cell in sheet[1]]

            # 获取字段名和 verbose_name，排除索引字段
            model_fields = [field.name for field in model._meta.fields if
                            not field.primary_key and field.name not in exclude_fields]
            model_verbose_name = [field.verbose_name for field in model._meta.fields if
                                  not field.primary_key and field.name not in exclude_fields]

            # 组成 fields_map
            fields_map = dict(zip(model_fields, model_verbose_name))
            fields_map_nf = dict(zip(model_verbose_name, model_fields))

            # 检查表头是否与模型字段名对应
            if not set(header_row).issubset(model_verbose_name):
                return JsonResponse({'error': '表头不匹配，请使用正确的Excel上传模板。'}, status=400)

            # 创建一个映射，将Excel表头映射到模型字段名
            header_to_field_map = {header: fields_map_nf[header] for header in header_row}
            header_fields = [header_to_field_map[header] for header in header_row]

            for row in sheet.iter_rows(min_row=2, values_only=True):
                if not all(value is None for value in row):
                    # 使用映射来确保每个Excel单元格的数据对应到正确的模型字段
                    instance_data = {header_to_field_map[header]: value for header, value in zip(header_row, row)}
                    instance = model(**instance_data)
                    try:
                        instance.full_clean()
                        data.append(instance)
                    except ValidationError as e:
                        return JsonResponse({'error': f'数据校验错误: {e.message_dict}'}, status=400)

            # 动态获取序列化器
            serializer_class = create_dynamic_serializer(model, include=header_fields)

            serializer = serializer_class(data, many=True)

            # 清理，删除上传的文件
            os.remove(file_path)

            return JsonResponse({"table_data": serializer.data, "fields_map": fields_map}, safe=False)
        except Exception as e:
            # 清理，删除上传的文件
            os.remove(file_path)
            return JsonResponse({'error': f'解析文件时出错: {str(e)}'}, status=500)

    return JsonResponse({'error': '请求错误'}, status=400)


@csrf_protect
@login_required
def save_excel_table_data(request):
    if request.method == 'POST':
        try:
            data = json.loads(request.body)
            model_config = data.get('model_config')
            table_data = data.get('table_data')

            if not model_config or not table_data:
                return JsonResponse({'error': '缺少必要的参数'}, status=400)

            # 分割 model_config 以获取 app_label 和 model_name
            try:
                app_label, model_name = model_config.split('.')
                Model = apps.get_model(app_label, model_name)
            except (ValueError, LookupError):
                return JsonResponse({'error': '无效的 model_config'}, status=400)

            # 创建模型实例列表
            instances = []
            for row_data in table_data:
                try:
                    instance = Model(**row_data)
                    instance.full_clean()  # 验证数据
                    instances.append(instance)
                except ValidationError as e:
                    return JsonResponse({'error': f'数据校验错误: {e.message_dict}'}, status=400)
                except Exception as e:
                    return JsonResponse({'error': f'创建实例时出错: {str(e)}'}, status=500)

            # 批量创建模型实例
            try:
                Model.objects.bulk_create(instances)
            except Exception as e:
                return JsonResponse({'error': f'批量保存数据时出错: {str(e)}'}, status=500)

            return JsonResponse({'message': '表格数据保存成功'}, status=200)
        except json.JSONDecodeError:
            return JsonResponse({'error': '无效的JSON格式'}, status=400)
        except Exception as e:
            return JsonResponse({'error': f'服务器内部错误: {str(e)}'}, status=500)
    return JsonResponse({'error': '无效的请求方法'}, status=400)