ALL-teach_sys/frontend_交通物流/fix_correct_splitting.py

#!/usr/bin/env python3
import json
import re

# 读取源数据
with open('网页未导入数据/交通物流产业/交通物流项目案例.json', 'r', encoding='utf-8') as f:
    source_data = json.load(f)

# 读取现有的mock文件来获取项目列表部分
with open('src/mocks/projectLibraryMock.js', 'r', encoding='utf-8') as f:
    content = f.read()

# 提取getMockProjectsList函数
projects_list_match = re.search(r'(export const getMockProjectsList.*?)export const getMockProjectDetail', content, re.DOTALL)
if not projects_list_match:
    print("无法找到getMockProjectsList函数")
    exit(1)

projects_list_func = projects_list_match.group(1).rstrip()

# 定义岗位等级映射
position_levels = {
    'WMS软件销售': '普通岗',
    '物流协调员': '普通岗',
    '仓储业务开发专员': '技术骨干岗',
    'AGV运维专员': '技术骨干岗',
    'AGV运维工程师': '高级管理岗',
    '物流审核员': '普通岗',
    '云物流调度负责人助理': '高级管理岗',
    '物流运营总监助理': '高级管理岗',
    '生产计划岗储备干部': '技术骨干岗',
    '国际物流储备经理': '高级管理岗',
    '供应链总监助理': '高级管理岗',
    'TMS销售专员': '普通岗',
    '海外运输经理助理': '高级管理岗',
    '海外物流协调员': '技术骨干岗',
    '海外仓设备运维工程师': '技术骨干岗',
    '售后支持专员': '普通岗',
    '采购专员': '普通岗',
    '港口安全员': '技术骨干岗'
}

# 构建新的getMockProjectDetail函数
detail_func_parts = ['export const getMockProjectDetail = (id) => {']
detail_func_parts.append('  const projects = [')

for idx, project in enumerate(source_data, 1):
    name = project['案例名称']
    content_text = project['项目案例内容']
    
    # 使用正确的标题进行分割
    # 1. 提取项目概述（从开始到"# 二、项目整体流程介绍"之前）
    overview_match = re.search(r'^(.*?)(?=# 二、项目整体流程介绍)', content_text, re.DOTALL | re.MULTILINE)
    overview = overview_match.group(1).strip() if overview_match else ''
    
    # 2. 提取项目流程（从"# 二、项目整体流程介绍"到"# 三、项目关键技术与创新点"之间）
    process_match = re.search(r'# 二、项目整体流程介绍(.*?)(?=# 三、项目关键技术与创新点|# 三、项目案例关键技术点|$)', content_text, re.DOTALL | re.MULTILINE)
    process = process_match.group(1).strip() if process_match else ''
    
    # 3. 提取关键技术点（从"# 三、项目关键技术与创新点"或"# 三、项目案例关键技术点"到结束）
    keypoints_match = re.search(r'# 三、项目关键技术与创新点(.*?)$', content_text, re.DOTALL | re.MULTILINE)
    if not keypoints_match:
        keypoints_match = re.search(r'# 三、项目案例关键技术点(.*?)$', content_text, re.DOTALL | re.MULTILINE)
    keypoints = keypoints_match.group(1).strip() if keypoints_match else ''
    
    # 如果仍然没有找到，回退到更通用的分割方式
    if not overview or not process or not keypoints:
        print(f"项目{idx}使用备用分割方式")
        # 分割成三个大致相等的部分
        text_parts = content_text.split('\n\n')
        total_parts = len(text_parts)
        
        if not overview:
            overview = '\n\n'.join(text_parts[:total_parts//3])
        if not process:
            process = '\n\n'.join(text_parts[total_parts//3:2*total_parts//3])
        if not keypoints:
            keypoints = '\n\n'.join(text_parts[2*total_parts//3:])
    
    # 重要：对模板字符串内容进行转义
    overview = overview.replace('\\', '\\\\').replace('`', '\\`').replace('${', '\\${')
    process = process.replace('\\', '\\\\').replace('`', '\\`').replace('${', '\\${')
    keypoints = keypoints.replace('\\', '\\\\').replace('`', '\\`').replace('${', '\\${')
    
    # 获取岗位列表并转换为对象数组
    positions_list = [
        'WMS软件销售', '物流协调员', '仓储业务开发专员', 'AGV运维专员',
        'AGV运维工程师', '物流审核员', '云物流调度负责人助理', 
        '物流运营总监助理', '生产计划岗储备干部', '国际物流储备经理'
    ]
    
    # 转换为对象数组格式
    applicable_positions = []
    for pos in positions_list[:5]:  # 每个项目取5个岗位
        applicable_positions.append({
            'position': pos,
            'level': position_levels.get(pos, '普通岗')
        })
    
    attachments = [
        {'name': f'{name}_项目方案.pdf', 'type': 'pdf'},
        {'name': f'{name}_技术文档.docx', 'type': 'docx'}
    ]
    
    # 如果流程或技术点为空，使用备用内容
    if not process.strip():
        process = '项目流程详情请参考项目概述'
    if not keypoints.strip():
        keypoints = '技术要点详情请参考项目概述'
    
    # 保持原有的Markdown标题
    if not process.startswith('#'):
        process = '## 项目整体流程介绍\\n\\n' + process
    if not keypoints.startswith('#'):
        keypoints = '## 项目案例关键技术点\\n\\n' + keypoints
    
    detail_func_parts.append(f'''    {{
      id: {idx},
      name: "{name}",
      title: "{name}",
      overview: `{overview}`,
      description: `{overview}`,
      process: `{process}`,
      keyPoints: `{keypoints}`,
      applicablePositions: {json.dumps(applicable_positions, ensure_ascii=False)},
      attachments: {json.dumps(attachments, ensure_ascii=False)}
    }}{',' if idx < len(source_data) else ''}''')

detail_func_parts.append('  ];')
detail_func_parts.append('')
detail_func_parts.append('  const project = projects.find(p => p.id === parseInt(id));')
detail_func_parts.append('  if (!project) {')
detail_func_parts.append('    return {')
detail_func_parts.append('      id: parseInt(id),')
detail_func_parts.append("      name: '项目未找到',")
detail_func_parts.append("      title: '项目未找到',")
detail_func_parts.append("      overview: '项目详情未找到',")
detail_func_parts.append("      description: '项目详情未找到',")
detail_func_parts.append("      process: '项目流程未找到',")
detail_func_parts.append("      keyPoints: '项目关键技术点未找到',")
detail_func_parts.append('      applicablePositions: [],')
detail_func_parts.append('      attachments: []')
detail_func_parts.append('    };')
detail_func_parts.append('  }')
detail_func_parts.append('')
detail_func_parts.append('  return project;')
detail_func_parts.append('};')

# 组合完整文件内容
new_content = projects_list_func + '\n\n' + '\n'.join(detail_func_parts)

# 写入文件
with open('src/mocks/projectLibraryMock.js', 'w', encoding='utf-8') as f:
    f.write(new_content)

print("✅ 成功修复内容分割逻辑，使用正确的标题进行分割")