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

import json
import re

def extract_energy_teachers():
    """从导师信息（通用）.json文件中提取能源产业导师"""
    with open('网页未导入数据/导师信息（通用）.json', 'r', encoding='utf-8') as f:
        data = json.load(f)
    
    energy_teachers = {}
    
    for teacher in data:
        if teacher.get('就业管家') == '能源':
            name = teacher['查询导师名称']
            
            # 确定导师类型
            teacher_type = "复合课导师"  # 默认
            if '公共课' in teacher.get('导师类型', ''):
                teacher_type = "公共课导师"
            elif 'AI' in teacher.get('导师类型', ''):
                teacher_type = "AI课导师"
            elif 'HR' in teacher.get('导师类型', '') or '企业' in teacher.get('导师类型', ''):
                teacher_type = "企业资深HR"
            
            # 获取导师特长或标签
            specialties = teacher.get('导师特长', teacher.get('导师标签', []))
            
            energy_teachers[name] = {
                'name': name,
                'introduction': teacher['导师介绍'],
                'specialties': specialties,
                'avatar': teacher.get('❌导师头像url链接', ''),
                'type': teacher_type,
                'verticalDirection': teacher.get('所属垂直方向', ''),
                'courses': []  # 初始化为空，后续会通过日历数据更新
            }
    
    return energy_teachers

def update_mockdata_teachers():
    """更新mockData.js中的teacherData"""
    
    # 提取能源产业导师
    energy_teachers = extract_energy_teachers()
    
    # 读取mockData.js
    with open('src/data/mockData.js', 'r', encoding='utf-8') as f:
        content = f.read()
    
    # 构建新的teacherData对象字符串
    teacher_data_str = "{\n"
    for name, teacher in energy_teachers.items():
        specialties_str = json.dumps(teacher['specialties'], ensure_ascii=False)
        
        # 转义特殊字符
        introduction = teacher['introduction'].replace('"', '\\"').replace('\n', '\\n')
        
        teacher_data_str += f'''    "{name}": {{
      name: "{teacher['name']}",
      introduction: "{introduction}",
      specialties: {specialties_str},
      avatar: "{teacher['avatar']}",
      type: "{teacher['type']}",
      verticalDirection: "{teacher['verticalDirection']}",
      courses: []
    }},\n'''
    
    # 移除最后一个逗号
    teacher_data_str = teacher_data_str.rstrip(',\n') + '\n  }'
    
    # 替换teacherData内容
    # 找到teacherData的起始位置
    pattern = r'teacherData:\s*updateTeacherCourses\(\{[\s\S]*?\},\s*allCalendarEvents\)'
    
    replacement = f'teacherData: updateTeacherCourses({teacher_data_str}, allCalendarEvents)'
    
    new_content = re.sub(pattern, replacement, content)
    
    # 写回文件
    with open('src/data/mockData.js', 'w', encoding='utf-8') as f:
        f.write(new_content)
    
    print(f"成功更新 {len(energy_teachers)} 位能源产业导师数据")
    print("导师名单：", list(energy_teachers.keys()))

if __name__ == "__main__":
    update_mockdata_teachers()