Merge remote-tracking branch 'origin/pengsen' into wsc1
This commit is contained in:
commit
b662812af6
|
|
@ -83,18 +83,6 @@ urlpatterns = [
|
|||
path('edit_employee_commission_detail/<int:commission_id>/', edit_employee_commission_detail,
|
||||
name='edit_employee_commission_detail'),
|
||||
|
||||
path('common_excel_parse_exps/', common_excel_parse_exp, name='common_excel_parse_exp'),
|
||||
path('save_excel_table_data_exp/', save_excel_table_data_exp, name='save_excel_table_data_exp'),
|
||||
|
||||
path('common_excel_parse_gab/', common_excel_parse_gab, name='common_excel_parse_gab'),
|
||||
path('save_excel_table_data_gab/', save_excel_table_data_gab, name='save_excel_table_data_gab'),
|
||||
|
||||
path('common_excel_parse_pjt/', common_excel_parse_pjt, name='common_excel_parse_pjt'),
|
||||
path('save_excel_table_data_pjt/', save_excel_table_data_pjt, name='save_excel_table_data_pjt'),
|
||||
|
||||
path('common_excel_parse_rr/', common_excel_parse_rr, name='common_excel_parse_rr'),
|
||||
path('save_excel_table_data_rr/', save_excel_table_data_rr, name='save_excel_table_data_rr'),
|
||||
|
||||
path('common_excel_parse_pc/', common_excel_parse_pc, name='common_excel_parse_pc'),
|
||||
path('save_excel_table_data_pc/', save_excel_table_data_pc, name='save_excel_table_data_pc'),
|
||||
path('get_invoice_records/', get_invoice_records, name='get_invoice_records'),
|
||||
path('get_repayment_records/', get_repayment_records, name='get_repayment_records'),
|
||||
]
|
||||
|
|
|
|||
File diff suppressed because it is too large
Load Diff
|
|
@ -39,10 +39,4 @@ urlpatterns = [
|
|||
path('performance_add/', performance_add, name='performance_add'),
|
||||
path('performance_modify/', performance_modify, name='performance_modify'),
|
||||
path('performance_delete/', performance_delete, name='performance_delete'),
|
||||
|
||||
path('common_excel_parse/', common_excel_parse, name='common_excel_parse_attd'),
|
||||
path('common_excel_parse/', common_excel_parse_pe, name='common_excel_parse_pe'),
|
||||
|
||||
path('save_excel_table_data/', save_excel_table_data, name='save_excel_table_data_attd'),
|
||||
path('save_excel_table_data/', save_excel_table_data_pe, name='save_excel_table_data_pe'),
|
||||
]
|
||||
|
|
|
|||
|
|
@ -286,8 +286,8 @@ def attd_rec_list_view(request):
|
|||
],
|
||||
"excel_upload_config": {
|
||||
"template_url": reverse("download_template", kwargs={'template_name': template_name}),
|
||||
"parse_url": reverse("common_excel_parse_attd"),
|
||||
"save_url": reverse("save_excel_table_data_attd"),
|
||||
"parse_url": reverse("common_excel_parse"),
|
||||
"save_url": reverse("save_excel_table_data"),
|
||||
"fields_preview_config": {
|
||||
"employee": {"type": "text", "width": "180px"},
|
||||
"year_month": {"type": "date", "width": "110px"},
|
||||
|
|
@ -858,8 +858,8 @@ def performance_list_view(request):
|
|||
"table_exclude_field_name": ['performance_id'],
|
||||
"excel_upload_config": {
|
||||
"template_url": reverse("download_template", kwargs={'template_name': template_name}),
|
||||
"parse_url": reverse("common_excel_parse_pe"),
|
||||
"save_url": reverse("save_excel_table_data_pe"),
|
||||
"parse_url": reverse("common_excel_parse"),
|
||||
"save_url": reverse("save_excel_table_data"),
|
||||
"fields_preview_config": {
|
||||
"employee": {"type": "text", "width": "180px"},
|
||||
"year": {"type": "text", "width": "180px"},
|
||||
|
|
@ -937,123 +937,6 @@ def performance_delete(request):
|
|||
return JsonResponse({"message": "无效的请求方法"}, status=405)
|
||||
|
||||
|
||||
@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
|
||||
|
||||
def to_representation(self, instance):
|
||||
representation = super().to_representation(instance)
|
||||
if 'employee' in representation and instance.employee:
|
||||
representation['employee'] = instance.employee.name
|
||||
return representation
|
||||
|
||||
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]
|
||||
|
||||
if not all(item in model_verbose_name for item in header_row):
|
||||
return JsonResponse({'error': '表头不匹配，请使用正确的Excel上传模板。'}, status=400)
|
||||
|
||||
# 组成 fields_map
|
||||
fields_map = dict(zip(model_fields, header_row))
|
||||
fields_map_nf = dict(zip(header_row, model_fields))
|
||||
|
||||
# 创建一个映射，将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):
|
||||
instance_data = dict(zip(model_fields, row))
|
||||
|
||||
# 处理employee字段，假设模型中有employee字段并且Excel文件中的表头包含员工姓名
|
||||
employee_name = instance_data.get('employee')
|
||||
if employee_name:
|
||||
try:
|
||||
employee_instance = EmployeeInformation.objects.get(name=employee_name)
|
||||
instance_data['employee'] = employee_instance
|
||||
except EmployeeInformation.DoesNotExist:
|
||||
return JsonResponse({'error': f'找不到姓名为 {employee_name} 的员工信息。'}, status=400)
|
||||
|
||||
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
|
||||
|
|
@ -1115,180 +998,7 @@ def save_excel_table_data(request):
|
|||
return JsonResponse({'error': '无效的请求方法'}, status=400)
|
||||
|
||||
|
||||
@csrf_protect
|
||||
@login_required
|
||||
def common_excel_parse_pe(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
|
||||
|
||||
def to_representation(self, instance):
|
||||
representation = super().to_representation(instance)
|
||||
if 'employee' in representation and instance.employee:
|
||||
representation['employee'] = instance.employee.name
|
||||
return representation
|
||||
|
||||
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]
|
||||
|
||||
if not all(item in model_verbose_name for item in header_row):
|
||||
return JsonResponse({'error': '表头不匹配，请使用正确的Excel上传模板。'}, status=400)
|
||||
|
||||
# 组成 fields_map
|
||||
fields_map = dict(zip(model_fields, header_row))
|
||||
fields_map_nf = dict(zip(header_row, model_fields))
|
||||
|
||||
# 创建一个映射，将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):
|
||||
instance_data = dict(zip(model_fields, row))
|
||||
|
||||
# 处理employee字段，假设模型中有employee字段并且Excel文件中的表头包含员工姓名
|
||||
employee_name = instance_data.get('employee')
|
||||
if employee_name:
|
||||
try:
|
||||
employee_instance = EmployeeInformation.objects.get(name=employee_name)
|
||||
instance_data['employee'] = employee_instance
|
||||
except EmployeeInformation.DoesNotExist:
|
||||
return JsonResponse({'error': f'找不到姓名为 {employee_name} 的员工信息。'}, status=400)
|
||||
|
||||
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_pe(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)
|
||||
|
||||
# 获取模型中配置的不需要的字段
|
||||
exclude_fields = getattr(Model, 'excluded_fields', [])
|
||||
|
||||
# 创建模型实例列表
|
||||
instances = []
|
||||
for row_data in table_data:
|
||||
# 处理employee字段，假设模型中有employee字段并且table_data中包含员工姓名
|
||||
employee_name = row_data.get('employee')
|
||||
if employee_name:
|
||||
try:
|
||||
employee_instance = EmployeeInformation.objects.get(name=employee_name)
|
||||
row_data['employee'] = employee_instance
|
||||
except EmployeeInformation.DoesNotExist:
|
||||
return JsonResponse({'error': f'找不到姓名为 {employee_name} 的员工信息。'}, status=400)
|
||||
|
||||
try:
|
||||
# 排除不需要的字段
|
||||
instance_data = {key: value for key, value in row_data.items() if key not in exclude_fields}
|
||||
instance = Model(**instance_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)
|
||||
|
||||
|
|
|
|||
|
|
@ -24,7 +24,4 @@ urlpatterns = [
|
|||
path('sd_list/add/', sd_list_add, name='sd_list_add'),
|
||||
path('sd_list/modify/', sd_list_modify, name='sd_list_modify'),
|
||||
path('sd_list/delete/', sd_list_delete, name='sd_list_delete'),
|
||||
|
||||
path('common_excel_parse/', common_excel_parse, name='common_excel_parse_sd'),
|
||||
path('save_excel_table_data/', save_excel_table_data, name='save_excel_table_data_sd'),
|
||||
]
|
||||
|
|
|
|||
|
|
@ -419,8 +419,8 @@ def sd_list_view(request):
|
|||
"table_exclude_field_name": ['secondary_department_id'],
|
||||
"excel_upload_config": {
|
||||
"template_url": reverse("download_template", kwargs={'template_name': template_name}),
|
||||
"parse_url": reverse("common_excel_parse_sd"),
|
||||
"save_url": reverse("save_excel_table_data_sd"),
|
||||
"parse_url": reverse("common_excel_parse"),
|
||||
"save_url": reverse("save_excel_table_data"),
|
||||
"fields_preview_config": {
|
||||
"primary_department": {"type": "text", "width": "180px"},
|
||||
"secondary_department_name": {"type": "text", "width": "180px"},
|
||||
|
|
@ -496,127 +496,6 @@ def sd_list_delete(request):
|
|||
return JsonResponse({"message": "无效的请求方法"}, status=405)
|
||||
|
||||
|
||||
@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
|
||||
|
||||
def to_representation(self, instance):
|
||||
representation = super().to_representation(instance)
|
||||
if 'primary_department' in representation and instance.primary_department:
|
||||
representation['primary_department'] = instance.primary_department.department_name
|
||||
return representation
|
||||
|
||||
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]
|
||||
|
||||
if not all(item in model_verbose_name for item in header_row):
|
||||
return JsonResponse({'error': '表头不匹配，请使用正确的Excel上传模板。'}, status=400)
|
||||
|
||||
# 组成 fields_map
|
||||
fields_map = dict(zip(model_fields, header_row))
|
||||
fields_map_nf = dict(zip(header_row, model_fields))
|
||||
|
||||
# 创建一个映射，将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):
|
||||
instance_data = dict(zip(model_fields, row))
|
||||
|
||||
# 处理primary_department字段，假设模型中有primary_department字段并且Excel文件中的表头包含一级部门名称
|
||||
primary_department_name = instance_data.get('primary_department')
|
||||
if primary_department_name:
|
||||
try:
|
||||
primary_department_instance = PrimaryDepartment.objects.get(
|
||||
department_name=primary_department_name)
|
||||
instance_data['primary_department'] = primary_department_instance
|
||||
except PrimaryDepartment.DoesNotExist:
|
||||
return JsonResponse({'error': f'找不到名称为 {primary_department_name} 的一级部门信息。'},
|
||||
status=400)
|
||||
|
||||
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):
|
||||
|
|
|
|||
|
|
@ -99,7 +99,7 @@ def proj_ledger_list_view(request):
|
|||
"import_excel_button": True
|
||||
}
|
||||
|
||||
return render(request, 'items_list.html', context)
|
||||
return render(request, 'project_list.html', context)
|
||||
|
||||
|
||||
@login_required
|
||||
|
|
|
|||
|
|
@ -1,14 +1,16 @@
|
|||
from django.urls import path
|
||||
from common.views import *
|
||||
|
||||
|
||||
urlpatterns = [
|
||||
path('error_page/', error_page, name='error_page'),
|
||||
path('download_excel_template/<str:template_name>/', download_template, name='download_template'),
|
||||
path('download_excel_template/<str:template_name>/<str:fields>/', download_excel_template, name='download_excel_template'),
|
||||
path('download_excel_template/<str:template_name>/<str:fields>/', download_excel_template,
|
||||
name='download_excel_template'),
|
||||
path('common_excel_parse/', common_excel_parse, name='common_excel_parse'),
|
||||
path('save_excel_data/', save_excel_table_data, name='save_excel_table_data'),
|
||||
path('load_secondary_departments/', load_secondary_departments, name='load_secondary_departments'),
|
||||
|
||||
path('save_excel_table_data_fac_inv/', save_excel_table_data_fac_inv, name='save_excel_table_data_fac_inv'),
|
||||
path('save_excel_table_data_fac_rep/', save_excel_table_data_fac_rep, name='save_excel_table_data_fac_rep'),
|
||||
|
||||
]
|
||||
|
||||
|
||||
|
|
|
|||
137
common/views.py
137
common/views.py
|
|
@ -16,7 +16,10 @@ from django.http import FileResponse, HttpResponseNotFound, JsonResponse
|
|||
from django.views.decorators.csrf import csrf_protect
|
||||
from rest_framework.serializers import ModelSerializer
|
||||
|
||||
from application import fac_mgnt
|
||||
from application.fac_mgnt.models import InvoiceRecord
|
||||
from application.org_mgnt.models import SecondaryDepartment
|
||||
from application.pjt_mgnt.models import ProjectLedger
|
||||
|
||||
|
||||
def error_page(request):
|
||||
|
|
@ -128,35 +131,23 @@ def common_excel_parse(request):
|
|||
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]
|
||||
not field.primary_key and field.verbose_name in header_row]
|
||||
model_verbose_name = [field.verbose_name for field in model._meta.fields if
|
||||
not field.primary_key and field.name not in exclude_fields]
|
||||
not field.primary_key and field.verbose_name in header_row]
|
||||
|
||||
if not all(item in model_verbose_name for item in header_row):
|
||||
return JsonResponse({'error': '表头不匹配，请使用正确的Excel上传模板。'}, status=400)
|
||||
|
|
@ -167,43 +158,15 @@ def common_excel_parse(request):
|
|||
|
||||
# 创建一个映射，将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]
|
||||
|
||||
# 动态处理外键关系
|
||||
def get_related_instance(model, field_name, value):
|
||||
field = model._meta.get_field(field_name)
|
||||
if field.is_relation:
|
||||
related_model = field.related_model
|
||||
related_field_name_list = related_model._meta.fields # 获取关联模型的第一个字段名
|
||||
for related_field_name in related_field_name_list:
|
||||
try:
|
||||
related_instance = related_model.objects.get(**{related_field_name.name: value})
|
||||
value = related_instance
|
||||
return value
|
||||
except Exception:
|
||||
continue
|
||||
return None
|
||||
else:
|
||||
return value
|
||||
|
||||
instance_list = []
|
||||
for row in sheet.iter_rows(min_row=2, values_only=True):
|
||||
if not all(value is None for value in row):
|
||||
instance_data = {header_to_field_map[header]: value for header, value in zip(header_row, row)}
|
||||
preview_data = instance_data.copy()
|
||||
for field_name, value in instance_data.items():
|
||||
instance_data[field_name] = get_related_instance(model, field_name, value)
|
||||
|
||||
instance = model(**instance_data)
|
||||
try:
|
||||
instance.full_clean()
|
||||
data.append(preview_data)
|
||||
except ValidationError as e:
|
||||
return JsonResponse({'error': f'数据校验错误: {e.message_dict}'}, status=400)
|
||||
instance_list.append({header_to_field_map[header]: value for header, value in zip(header_row, row)})
|
||||
|
||||
# 清理，删除上传的文件
|
||||
os.remove(file_path)
|
||||
|
||||
return JsonResponse({"table_data": data, "fields_map": fields_map}, safe=False)
|
||||
return JsonResponse({"table_data": instance_list, "fields_map": fields_map}, safe=False)
|
||||
except Exception as e:
|
||||
# 清理，删除上传的文件
|
||||
os.remove(file_path)
|
||||
|
|
@ -275,3 +238,87 @@ def load_secondary_departments(request):
|
|||
'secondary_department_name')
|
||||
return JsonResponse(list(secondary_departments.values('secondary_department_id', 'secondary_department_name')),
|
||||
safe=False)
|
||||
|
||||
|
||||
@csrf_protect
|
||||
@login_required
|
||||
def save_excel_table_data_fac_inv(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:
|
||||
row_data['project_id'] = ProjectLedger.objects.get(project_name=row_data['project_name'])
|
||||
instance = Model(**row_data)
|
||||
try:
|
||||
instance.full_clean()
|
||||
instances.append(instance)
|
||||
except ValidationError as e:
|
||||
return JsonResponse({'error': f'数据校验错误: {e.message_dict}'}, status=400)
|
||||
|
||||
# 批量保存数据
|
||||
Model.objects.bulk_create(instances)
|
||||
|
||||
return JsonResponse({'success': '数据保存成功'}, status=201)
|
||||
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)
|
||||
|
||||
|
||||
@csrf_protect
|
||||
@login_required
|
||||
def save_excel_table_data_fac_rep(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:
|
||||
row_data['project_id'] = ProjectLedger.objects.get(project_name=row_data['project_name'])
|
||||
row_data['invoice'] = InvoiceRecord.objects.get(project_name=row_data['project_name'])
|
||||
instance = Model(**row_data)
|
||||
try:
|
||||
instance.full_clean()
|
||||
instances.append(instance)
|
||||
except ValidationError as e:
|
||||
return JsonResponse({'error': f'数据校验错误: {e.message_dict}'}, status=400)
|
||||
|
||||
# 批量保存数据
|
||||
Model.objects.bulk_create(instances)
|
||||
|
||||
return JsonResponse({'success': '数据保存成功'}, status=201)
|
||||
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)
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue