import json
import re

# 读取学生完成的项目数据
with open('网页未导入数据/学生完成的项目.json', 'r', encoding='utf-8') as f:
    student_projects = json.load(f)

# 筛选出食品相关的项目
food_related_keywords = ['食品', '营养', '餐厅', '配餐', '灌装', '进口', '电商', '生鲜']

food_projects_by_unit = {}

for item in student_projects:
    unit_name = item.get('单元名称查询', '')
    projects = item.get('学生完成的项目名称', [])

    # 检查是否包含食品相关关键词
    is_food_related = False
    for project in projects:
        for keyword in food_related_keywords:
            if keyword in project:
                is_food_related = True
                break
        if is_food_related:
            break

    if is_food_related and unit_name:
        if unit_name not in food_projects_by_unit:
            food_projects_by_unit[unit_name] = []

        # 清理项目名称（去掉序号）
        cleaned_projects = []
        for project in projects:
            # 去掉开头的序号（如 "1." "2."）
            cleaned_project = re.sub(r'^\d+\.', '', project).strip()
            if any(keyword in cleaned_project for keyword in food_related_keywords):
                cleaned_projects.append(cleaned_project)

        food_projects_by_unit[unit_name].extend(cleaned_projects)

print(f"找到 {len(food_projects_by_unit)} 个食品相关单元")
for unit, projects in food_projects_by_unit.items():
    print(f"\n{unit}:")
    for project in projects:
        print(f"  - {project}")

# 根据食品产业特点，创建更合适的单元分类
food_industry_units = {
    "食品研发与创新": [
        "食品灌装生产线PLC编程与传感器信号采集项目",
        "食品原料进口检疫检验与关务手续合规管理项目"
    ],
    "食品电商与营销": [
        "跨境电商食品类商品智能合规审核与清关支持系统项目",
        "食品电商订单售后服务流程优化与客户满意度提升项目",
        "食品电商平台订单履约时效监控与客户满意度提升项目",
        "食品电商多渠道客服（电话+在线+社群）整合管理项目"
    ],
    "营养配餐与健康管理": [
        "儿童一周营养配餐食谱编制方案"  # 这个是唯一可查看的详情项目
    ],
    "食品贸易与安全": [
        "南美大豆贸易与中国食品安全战略研究项目",
        "食品进口关税估价争议解决与合规应对项目"
    ],
    "食品质量控制": [
        "生鲜电商售后退赔流程数字化改造与服务效率提升项目"
    ]
}

# 生成JavaScript代码
js_content = '''// 我的项目库数据
const myProjectsData = [
'''

for unit_name, projects in food_industry_units.items():
    js_content += f'  {{\n'
    js_content += f'    "unitName": "{unit_name}",\n'
    js_content += f'    "projects": [\n'
    for project in projects:
        js_content += f'      "{project}",\n'
    # 移除最后的逗号
    js_content = js_content.rstrip(',\n') + '\n'
    js_content += f'    ]\n'
    js_content += f'  }},\n'

# 移除最后的逗号并闭合数组
js_content = js_content.rstrip(',\n') + '\n'
js_content += '];\n'

print(f"\n生成的JavaScript代码片段：")
print(js_content)

# 保存到文件
with open('my_projects_data.js', 'w', encoding='utf-8') as f:
    f.write(js_content)

print(f"\n已保存到 my_projects_data.js")

# 统计信息
total_projects = sum(len(projects) for projects in food_industry_units.values())
print(f"\n统计信息：")
print(f"  总单元数: {len(food_industry_units)}")
print(f"  总项目数: {total_projects}")
print(f"  唯一可查看项目: 儿童一周营养配餐食谱编制方案")