From 5e08ce8c1f403f4aed9ecc12b0261bd00543d2e5 Mon Sep 17 00:00:00 2001 From: sichan Date: Mon, 17 Jun 2024 23:40:16 +0800 Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E4=BA=86=E9=80=9A=E7=94=A8ex?= =?UTF-8?q?cel=E4=B8=8A=E4=BC=A0=E8=A7=A3=E6=9E=90=E5=92=8C=E4=BF=9D?= =?UTF-8?q?=E5=AD=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../templates/excel_preview_table.html | 64 +++++++++++++++++-- excel_parser/urls.py | 3 +- excel_parser/utils.py | 60 +++++++++++++++++ excel_parser/views.py | 40 +++++++++++- static/js/custom.js | 8 +++ templates/items_list.html | 2 + 6 files changed, 169 insertions(+), 8 deletions(-) create mode 100644 excel_parser/utils.py create mode 100644 static/js/custom.js diff --git a/excel_parser/templates/excel_preview_table.html b/excel_parser/templates/excel_preview_table.html index 0b5139a..0f912ef 100644 --- a/excel_parser/templates/excel_preview_table.html +++ b/excel_parser/templates/excel_preview_table.html @@ -40,16 +40,21 @@ th { background-color: #f2f2f2; } + .error-cell { + background-color: #f8d7da; /* 红色背景 */ + border-color: #f5c2c7; /* 红色边框 */ + } +
- 返回列表 + 返回
@@ -80,9 +85,6 @@
@@ -93,6 +95,60 @@ + + diff --git a/excel_parser/urls.py b/excel_parser/urls.py index c28e888..de520e6 100644 --- a/excel_parser/urls.py +++ b/excel_parser/urls.py @@ -4,5 +4,6 @@ from excel_parser.views import * urlpatterns = [ path('excel_preview/', excel_preview, name='excel_preview'), path('dl_excel_tpl//', dl_excel_tpl, name='dl_excel_tpl'), - path('common_parse/', common_parse, name="ep_common_parse") + path('common_parse/', common_parse, name="ep_common_parse"), + path('common_save/', common_save, name="ep_common_save"), ] \ No newline at end of file diff --git a/excel_parser/utils.py b/excel_parser/utils.py new file mode 100644 index 0000000..d926ca7 --- /dev/null +++ b/excel_parser/utils.py @@ -0,0 +1,60 @@ +from django.core.exceptions import ValidationError +from django.apps import apps + + +class ExcelDataSaver: + def __init__(self, model_name, data): + app_name, model_name = model_name.split('.') + self.model = apps.get_model(app_name, model_name) + self.data = data + self.field_mapping, self.reverse_field_mapping = self._generate_field_mapping() + self.errors = [] + + def _generate_field_mapping(self): + field_mapping = {} + reverse_field_mapping = {} + for field in self.model._meta.get_fields(): + if hasattr(field, 'verbose_name'): + field_mapping[field.verbose_name] = field.name + reverse_field_mapping[field.name] = field.verbose_name + return field_mapping, reverse_field_mapping + + def validate_data(self): + for row_num, row in enumerate(self.data): + instance = self.model() + for field, value in row.items(): + actual_field = self.field_mapping.get(field) + if not actual_field: + self.errors.append(f"第【{row_num + 1}】行找不到对应的模型字段: 【{self.reverse_field_mapping[field] + ':' + field}】") + continue + setattr(instance, actual_field, value) + + # 在设置所有字段值之后再调用 full_clean + try: + instance.full_clean() + except ValidationError as e: + for field, errors in e.message_dict.items(): + for error in errors: + self.errors.append(f"第【{row_num + 1}】行【{self.reverse_field_mapping[field]}】字段输入错误: {error}") + return len(self.errors) == 0 + + def save_data(self): + if self.validate_data(): + for row in self.data: + row_data = {self.field_mapping[field]: value for field, value in row.items() if field in self.field_mapping} + instance = self.model(**row_data) + instance.save() + return True + return False + + def get_errors(self): + return self.errors + + def add_custom_validation(self, custom_validation_func): + self.custom_validation_func = custom_validation_func + self.validate_data = self._validate_data_with_custom + + def _validate_data_with_custom(self): + basic_valid = self.validate_data() + custom_valid = self.custom_validation_func(self.data) + return basic_valid and custom_valid diff --git a/excel_parser/views.py b/excel_parser/views.py index 36bd58b..b377893 100644 --- a/excel_parser/views.py +++ b/excel_parser/views.py @@ -1,3 +1,4 @@ +import json import urllib.parse from datetime import datetime @@ -8,6 +9,8 @@ from django.shortcuts import render from django.urls import reverse from openpyxl.reader.excel import load_workbook +from excel_parser.utils import ExcelDataSaver + @login_required def dl_excel_tpl(request, template_name): @@ -62,6 +65,7 @@ def common_parse(request): # 获取上传的 Excel 文件和模板名称 excel_file = request.FILES['file'] template_name = request.POST.get('template_name') + excel_valid_model = request.POST.get('excel_valid_model') # 加载上传的 Excel 文件 wb = load_workbook(excel_file) @@ -100,6 +104,7 @@ def common_parse(request): # 存储解析数据到 session 中 request.session['columns'] = uploaded_columns request.session['table_data'] = data + request.session['excel_valid_model'] = excel_valid_model # 返回新的页面 URL return JsonResponse({'redirect_url': reverse("excel_preview")}) @@ -122,10 +127,39 @@ def excel_preview(request): 返回: HttpResponse: 渲染 'excel_preview_table.html' 模板，并包含会话中的列名和表格数据。 """ - # 从会话中获取列名列表 + # 从会话中获取数据 columns = request.session.get('columns', []) - # 从会话中获取表格数据列表 table_data = request.session.get('table_data', []) + excel_valid_model = request.session.get('excel_valid_model', '') + + context = { + 'columns': columns, + 'table_data': table_data, + 'excel_valid_model': excel_valid_model + } # 渲染 HTML 模板 'excel_preview_table.html'，并传递列名和表格数据 - return render(request, 'excel_preview_table.html', {'columns': columns, 'table_data': table_data}) + return render(request, 'excel_preview_table.html', context) + + +def common_save(request): + if request.method == 'POST': + try: + body = json.loads(request.body) + data = body.get('data', None) + model_name = request.session.get('excel_valid_model', None) + + if not data or not model_name: + return JsonResponse({'status': 'error', 'message': '无效的数据或模型'}) + + data_saver = ExcelDataSaver(model_name, data) + + if data_saver.save_data(): + return JsonResponse({'status': 'success', 'message': '数据已成功保存'}) + else: + return JsonResponse({'status': 'error', 'message': '数据验证失败', 'errors': data_saver.get_errors()}) + + except json.JSONDecodeError: + return JsonResponse({'status': 'error', 'message': '无法解析请求的JSON数据'}) + + return JsonResponse({'status': 'error', 'message': '仅支持POST请求'}) diff --git a/static/js/custom.js b/static/js/custom.js new file mode 100644 index 0000000..da80b4f --- /dev/null +++ b/static/js/custom.js @@ -0,0 +1,8 @@ +function showAlert(type, message) { + const alertHtml = ` + `; + $('#alertContainer').html(alertHtml); +} \ No newline at end of file diff --git a/templates/items_list.html b/templates/items_list.html index f3e9514..9869dba 100644 --- a/templates/items_list.html +++ b/templates/items_list.html @@ -204,6 +204,7 @@ const deleteUrl = "{{ delete_url }}"; let targetIdToDelete = null; var templateName = "{{ excel_upload_config.template_name }}"; + var modelConfig = "{{ model_config }}"; $(document).ready(function () { function getCookie(name) { @@ -355,6 +356,7 @@ var formData = new FormData(); formData.append('file', inputFile); // 将文件添加到 FormData 中 formData.append('template_name', templateName); + formData.append('excel_valid_model', modelConfig); $.ajax({ url: '{{ excel_upload_config.parse_url }}',