2024-06-02 01:48:07 +08:00
|
|
|
|
import json
|
2024-06-01 05:05:29 +08:00
|
|
|
|
import os
|
2024-06-01 00:57:14 +08:00
|
|
|
|
import urllib.parse
|
2024-05-30 17:06:23 +08:00
|
|
|
|
|
2024-06-04 16:50:30 +08:00
|
|
|
|
import openpyxl
|
|
|
|
|
from openpyxl.utils import get_column_letter
|
|
|
|
|
from openpyxl.worksheet.datavalidation import DataValidation
|
2024-06-02 01:48:07 +08:00
|
|
|
|
from django.apps import apps
|
2024-06-02 21:30:17 +08:00
|
|
|
|
from django.core.exceptions import ValidationError
|
2024-06-01 05:05:29 +08:00
|
|
|
|
from openpyxl import load_workbook
|
|
|
|
|
|
|
|
|
|
from django.conf import settings
|
2024-06-01 01:40:12 +08:00
|
|
|
|
from django.contrib.auth.decorators import login_required
|
2024-06-01 00:57:14 +08:00
|
|
|
|
from django.contrib.staticfiles import finders
|
2024-06-01 05:05:29 +08:00
|
|
|
|
from django.core.files.storage import default_storage
|
|
|
|
|
from django.http import FileResponse, HttpResponseNotFound, JsonResponse
|
2024-06-02 21:47:43 +08:00
|
|
|
|
from django.views.decorators.csrf import csrf_exempt, csrf_protect
|
2024-06-02 21:30:17 +08:00
|
|
|
|
from rest_framework.decorators import api_view
|
2024-06-04 16:50:30 +08:00
|
|
|
|
from rest_framework.serializers import ModelSerializer
|
2024-06-01 00:57:14 +08:00
|
|
|
|
|
|
|
|
|
|
2024-06-06 14:18:42 +08:00
|
|
|
|
@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
|
|
|
|
|
|
|
|
|
|
|
2024-06-01 01:40:12 +08:00
|
|
|
|
@login_required
|
2024-06-04 16:50:30 +08:00
|
|
|
|
def download_excel_template(request, template_name, fields):
|
2024-06-01 01:40:12 +08:00
|
|
|
|
"""
|
|
|
|
|
该视图提供下载指定的Excel模板文件。
|
2024-06-01 00:57:14 +08:00
|
|
|
|
|
2024-06-01 01:40:12 +08:00
|
|
|
|
参数:
|
|
|
|
|
request: HttpRequest对象。
|
|
|
|
|
template_name: 请求下载的Excel模板文件的名称,期望是一个字符串。
|
2024-06-04 16:50:30 +08:00
|
|
|
|
fields: 包含字段信息的字典,用于生成Excel文件的表头和格式。
|
2024-06-01 01:40:12 +08:00
|
|
|
|
|
|
|
|
|
返回:
|
2024-06-04 16:50:30 +08:00
|
|
|
|
FileResponse对象,允许用户下载生成的Excel文件。
|
2024-06-01 01:40:12 +08:00
|
|
|
|
如果文件未找到,则返回一个HttpResponseNotFound响应。
|
|
|
|
|
"""
|
2024-06-04 16:50:30 +08:00
|
|
|
|
fields = json.loads(urllib.parse.unquote(fields).replace("'", '"'))
|
|
|
|
|
|
|
|
|
|
# 创建一个新的工作簿
|
|
|
|
|
wb = openpyxl.Workbook()
|
|
|
|
|
ws = wb.active
|
|
|
|
|
|
|
|
|
|
# 设置表头行
|
2024-06-06 15:08:38 +08:00
|
|
|
|
headers = [field['label'] for field in fields if field['is_add']]
|
2024-06-04 16:50:30 +08:00
|
|
|
|
ws.append(headers)
|
2024-06-01 00:57:14 +08:00
|
|
|
|
|
2024-06-04 16:50:30 +08:00
|
|
|
|
# 保存生成的Excel文件到指定路径
|
2024-06-05 10:09:46 +08:00
|
|
|
|
static_dir = os.path.join(settings.BASE_DIR, 'static', 'excels')
|
2024-06-04 16:50:30 +08:00
|
|
|
|
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)
|
2024-06-01 00:57:14 +08:00
|
|
|
|
|
2024-06-01 01:40:12 +08:00
|
|
|
|
# 创建FileResponse对象,设置为以附件形式下载,设置MIME类型为Excel文件
|
2024-06-04 16:50:30 +08:00
|
|
|
|
response = FileResponse(open(temp_file_path, 'rb'),
|
|
|
|
|
content_type='application/vnd.openxmlformats-officedocument.spreadsheetml.sheet')
|
2024-06-05 10:09:46 +08:00
|
|
|
|
|
2024-06-01 00:57:14 +08:00
|
|
|
|
response['Content-Disposition'] = f'attachment; filename="{urllib.parse.quote(template_name)}"'
|
2024-06-04 16:50:30 +08:00
|
|
|
|
|
2024-06-01 00:57:14 +08:00
|
|
|
|
return response
|
2024-06-01 05:05:29 +08:00
|
|
|
|
|
|
|
|
|
|
2024-06-02 21:47:43 +08:00
|
|
|
|
@csrf_protect
|
2024-06-02 21:30:17 +08:00
|
|
|
|
@login_required
|
2024-06-01 05:05:29 +08:00
|
|
|
|
def common_excel_parse(request):
|
2024-06-01 10:13:10 +08:00
|
|
|
|
"""
|
|
|
|
|
该函数用于解析上传的Excel文件并返回数据。
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
request: HTTP请求对象,包含上传的Excel文件。
|
|
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
|
JsonResponse: 包含解析后的Excel数据的JSON响应,或包含错误消息的JSON响应。
|
|
|
|
|
|
|
|
|
|
Raises:
|
|
|
|
|
N/A
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
# 如果是POST请求并且有上传的文件
|
2024-06-01 05:05:29 +08:00
|
|
|
|
if request.method == 'POST' and request.FILES:
|
2024-06-01 10:13:10 +08:00
|
|
|
|
# 获取上传的Excel文件
|
2024-06-01 05:05:29 +08:00
|
|
|
|
excel_file = request.FILES.get('excel_file')
|
|
|
|
|
|
2024-06-01 10:13:10 +08:00
|
|
|
|
# 如果没有提供文件,返回错误响应
|
2024-06-01 05:05:29 +08:00
|
|
|
|
if not excel_file:
|
2024-06-04 16:50:30 +08:00
|
|
|
|
return JsonResponse({'error': '请先选择文件。'}, status=400)
|
2024-06-01 05:05:29 +08:00
|
|
|
|
|
2024-06-06 23:46:12 +08:00
|
|
|
|
# 获取模型名称
|
|
|
|
|
model_config = request.POST.get('model_config', '')
|
2024-06-01 05:05:29 +08:00
|
|
|
|
|
2024-06-02 21:30:17 +08:00
|
|
|
|
# 动态获取模型
|
|
|
|
|
try:
|
2024-06-06 23:46:12 +08:00
|
|
|
|
app_label, model_name = model_config.split('.')
|
|
|
|
|
model = apps.get_model(app_label, model_name)
|
|
|
|
|
except (LookupError, KeyError, ValueError):
|
2024-06-02 21:30:17 +08:00
|
|
|
|
return JsonResponse({'error': '模型配置错误。'}, status=400)
|
2024-06-01 05:05:29 +08:00
|
|
|
|
|
2024-06-02 21:30:17 +08:00
|
|
|
|
# 保存文件到服务器
|
|
|
|
|
file_name = default_storage.save(excel_file.name, excel_file)
|
|
|
|
|
file_path = os.path.join(settings.MEDIA_ROOT, file_name)
|
2024-06-01 05:05:29 +08:00
|
|
|
|
|
2024-06-04 16:50:30 +08:00
|
|
|
|
def create_dynamic_serializer(mod):
|
|
|
|
|
class DynamicSerializer(ModelSerializer):
|
|
|
|
|
class Meta:
|
|
|
|
|
model = mod
|
|
|
|
|
fields = '__all__'
|
|
|
|
|
|
|
|
|
|
return DynamicSerializer
|
|
|
|
|
|
2024-06-02 21:30:17 +08:00
|
|
|
|
try:
|
|
|
|
|
# 打开并解析Excel文件
|
|
|
|
|
workbook = load_workbook(file_path, data_only=True)
|
|
|
|
|
sheet = workbook.active
|
|
|
|
|
data = []
|
|
|
|
|
|
|
|
|
|
# 读取第一行作为表头
|
|
|
|
|
header_row = [cell.value for cell in sheet[1]]
|
|
|
|
|
|
2024-06-06 23:46:12 +08:00
|
|
|
|
# 获取字段名和 verbose_name,排除索引字段
|
|
|
|
|
model_fields = [field.name for field in model._meta.fields if not field.primary_key]
|
|
|
|
|
model_verbose_name = [field.verbose_name for field in model._meta.fields if not field.primary_key]
|
|
|
|
|
|
|
|
|
|
# 组成 fields_map
|
|
|
|
|
fields_map = dict(zip(model_fields, model_verbose_name))
|
|
|
|
|
|
|
|
|
|
# 检查表头是否与模型字段名对应
|
|
|
|
|
if header_row != model_verbose_name:
|
2024-06-02 21:30:17 +08:00
|
|
|
|
return JsonResponse({'error': '表头不匹配,请使用正确的Excel上传模板。'}, status=400)
|
|
|
|
|
|
|
|
|
|
for row in sheet.iter_rows(min_row=2, values_only=True):
|
|
|
|
|
if not all(value is None for value in row):
|
2024-06-06 23:46:12 +08:00
|
|
|
|
instance_data = dict(zip(model_fields, row))
|
2024-06-02 21:30:17 +08:00
|
|
|
|
instance = model(**instance_data)
|
|
|
|
|
try:
|
|
|
|
|
instance.full_clean()
|
|
|
|
|
data.append(instance)
|
|
|
|
|
except ValidationError as e:
|
|
|
|
|
return JsonResponse({'error': f'数据校验错误: {e.message_dict}'}, status=400)
|
|
|
|
|
|
2024-06-04 16:50:30 +08:00
|
|
|
|
# 动态获取序列化器
|
|
|
|
|
serializer_class = create_dynamic_serializer(model)
|
|
|
|
|
|
|
|
|
|
serializer = serializer_class(data, many=True)
|
|
|
|
|
|
2024-06-02 21:30:17 +08:00
|
|
|
|
# 清理,删除上传的文件
|
|
|
|
|
os.remove(file_path)
|
2024-06-06 23:46:12 +08:00
|
|
|
|
|
|
|
|
|
return JsonResponse({"table_data": serializer.data, "fields_map": fields_map}, safe=False)
|
2024-06-02 21:30:17 +08:00
|
|
|
|
except Exception as e:
|
|
|
|
|
# 清理,删除上传的文件
|
|
|
|
|
os.remove(file_path)
|
|
|
|
|
return JsonResponse({'error': f'解析文件时出错: {str(e)}'}, status=500)
|
2024-06-01 05:05:29 +08:00
|
|
|
|
|
|
|
|
|
return JsonResponse({'error': '请求错误'}, status=400)
|
2024-06-02 01:48:07 +08:00
|
|
|
|
|
|
|
|
|
|
2024-06-02 21:47:43 +08:00
|
|
|
|
@csrf_protect
|
2024-06-02 21:30:17 +08:00
|
|
|
|
@login_required
|
2024-06-02 01:48:07 +08:00
|
|
|
|
def save_excel_table_data(request):
|
|
|
|
|
if request.method == 'POST':
|
|
|
|
|
try:
|
|
|
|
|
data = json.loads(request.body)
|
2024-06-06 23:46:12 +08:00
|
|
|
|
model_config = data.get('model_config')
|
2024-06-02 21:30:17 +08:00
|
|
|
|
table_data = data.get('table_data')
|
|
|
|
|
|
2024-06-06 23:46:12 +08:00
|
|
|
|
if not model_config or not table_data:
|
2024-06-02 21:30:17 +08:00
|
|
|
|
return JsonResponse({'error': '缺少必要的参数'}, status=400)
|
2024-06-02 01:48:07 +08:00
|
|
|
|
|
2024-06-06 23:46:12 +08:00
|
|
|
|
# 分割 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)
|
2024-06-02 01:48:07 +08:00
|
|
|
|
|
2024-06-06 23:46:12 +08:00
|
|
|
|
# 创建模型实例列表
|
|
|
|
|
instances = []
|
2024-06-02 01:48:07 +08:00
|
|
|
|
for row_data in table_data:
|
2024-06-02 21:30:17 +08:00
|
|
|
|
try:
|
|
|
|
|
instance = Model(**row_data)
|
|
|
|
|
instance.full_clean() # 验证数据
|
2024-06-06 23:46:12 +08:00
|
|
|
|
instances.append(instance)
|
2024-06-02 21:30:17 +08:00
|
|
|
|
except ValidationError as e:
|
|
|
|
|
return JsonResponse({'error': f'数据校验错误: {e.message_dict}'}, status=400)
|
|
|
|
|
except Exception as e:
|
2024-06-06 23:46:12 +08:00
|
|
|
|
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)
|
2024-06-02 01:48:07 +08:00
|
|
|
|
|
|
|
|
|
return JsonResponse({'message': '表格数据保存成功'}, status=200)
|
2024-06-02 21:30:17 +08:00
|
|
|
|
except json.JSONDecodeError:
|
|
|
|
|
return JsonResponse({'error': '无效的JSON格式'}, status=400)
|
2024-06-02 01:48:07 +08:00
|
|
|
|
except Exception as e:
|
2024-06-02 21:30:17 +08:00
|
|
|
|
return JsonResponse({'error': f'服务器内部错误: {str(e)}'}, status=500)
|
|
|
|
|
return JsonResponse({'error': '无效的请求方法'}, status=400)
|