#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import json
import re

def update_energy_study_data():
    """
    更新学习相关数据，使其符合能源产业特点
    """
    
    # 读取mockData.js文件
    with open("src/data/mockData.js", 'r', encoding='utf-8') as f:
        content = f.read()
    
    # 创建备份
    with open("src/data/mockData.js.backup_before_study_update", 'w', encoding='utf-8') as f:
        f.write(content)
    
    # 更新学习进度模块
    learning_updates = {
        # 已完成模块
        r'"基础理论",\s*"设计软件应用",\s*"创意思维训练",\s*"项目管理基础",': '"能源基础理论", "智能制造基础", "新能源材料概论", "自动化控制原理",',
        
        # 进行中模块  
        r'"展会策划实务",\s*"文旅产业研究",\s*"数字创意设计",\s*"品牌营销策略",': '"储能系统设计", "光伏发电技术", "智能电网运维", "新能源材料制备",',
        
        # 即将开始模块
        r'"实习实训",\s*"毕业设计",': '"能源企业实训", "新能源项目设计",',
    }
    
    # 更新技能标签
    skills_pattern = r'skills: \[\s*"[^"]*",[\s\S]*?\],'
    new_skills = '''skills: [
      "Python编程",
      "PLC控制系统",
      "AutoCAD制图",
      "储能技术",
      "光伏系统设计",
      "电气自动化",
      "新能源材料",
      "数据分析",
    ],'''
    
    # 更新证书
    certificates_pattern = r'certificates: \[\s*\{[\s\S]*?\},\s*\{[\s\S]*?\},\s*\],'
    new_certificates = '''certificates: [
      {
        name: "电气工程师助理",
        issueDate: "2024-12-15",
        issuer: "中国电力企业联合会",
      },
      {
        name: "新能源技术应用",
        issueDate: "2024-11-20",
        issuer: "国家能源局",
      },
    ],'''
    
    # 更新项目
    projects_pattern = r'projects: \[\s*\{[\s\S]*?\},\s*\{[\s\S]*?\},\s*\],'
    new_projects = '''projects: [
      {
        name: "校园光伏发电系统设计",
        role: "主设计师",
        period: "2024.10 - 2024.12",
        status: "已完成",
      },
      {
        name: "储能电站运维监控系统",
        role: "技术开发",
        period: "2024.11 - 至今",
        status: "进行中",
      },
    ],'''
    
    # 应用所有更新
    updated_content = content
    
    # 更新学习模块
    for pattern, replacement in learning_updates.items():
        updated_content = re.sub(pattern, replacement, updated_content)
        print(f"✓ 更新学习模块")
    
    # 更新技能标签
    updated_content = re.sub(skills_pattern, new_skills, updated_content)
    print(f"✓ 更新技能标签")
    
    # 更新证书
    updated_content = re.sub(certificates_pattern, new_certificates, updated_content)
    print(f"✓ 更新证书信息")
    
    # 更新项目
    updated_content = re.sub(projects_pattern, new_projects, updated_content)
    print(f"✓ 更新项目信息")
    
    # 写入更新后的内容
    with open("src/data/mockData.js", 'w', encoding='utf-8') as f:
        f.write(updated_content)
    
    print(f"\n✅ 成功更新能源产业相关的学习数据")
    print(f"📋 更新内容:")
    print(f"  - 已完成模块: 能源基础理论、智能制造基础等")
    print(f"  - 进行中模块: 储能系统设计、光伏发电技术等")
    print(f"  - 技能标签: Python编程、PLC控制系统、储能技术等")
    print(f"  - 证书: 电气工程师助理、新能源技术应用")
    print(f"  - 项目: 校园光伏发电系统、储能电站运维监控系统")

if __name__ == "__main__":
    update_energy_study_data()