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

import json
import shutil
from datetime import datetime

def main():
    print("🚀 更新视觉设计课程日历数据...")
    
    # 创建备份
    backup_name = f"src/data/mockData.js.backup_calendar_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
    shutil.copy('src/data/mockData.js', backup_name)
    print(f"📦 已创建备份: {backup_name}")
    
    # 复制视觉设计课程日历文件到data目录
    shutil.copy('网页未导入数据/视觉设计产业/视觉设计课程日历.json', 'src/data/visualDesignCalendar.json')
    print("📋 已复制视觉设计课程日历到data目录")
    
    # 读取mockData.js文件
    with open('src/data/mockData.js', 'r', encoding='utf-8') as f:
        content = f.read()
    
    # 替换import语句
    content = content.replace(
        'import calendarCoursesData from \'./intelligentManufacturingCalendar.json\';',
        'import calendarCoursesData from \'./visualDesignCalendar.json\';'
    )
    
    # 更新用户信息为视觉设计相关
    content = content.replace(
        'name: "王强"',
        'name: "张小同"'
    )
    
    content = content.replace(
        'major: "模具设计与制造"',
        'major: "数字媒体艺术设计"'
    )
    
    content = content.replace(
        'verticalDirection: "自动化设备智能调试"',
        'verticalDirection: "商业平面设计与三维建模"'
    )
    
    content = content.replace(
        'industry: "智能制造"',
        'industry: "视觉设计"'
    )
    
    # 写入更新后的文件
    with open('src/data/mockData.js', 'w', encoding='utf-8') as f:
        f.write(content)
    
    print("✅ 已更新mockData.js")
    
    # 验证视觉设计课程数据
    with open('src/data/visualDesignCalendar.json', 'r', encoding='utf-8') as f:
        visual_calendar = json.load(f)
    
    # 统计课程数据
    course_count = 0
    public_course_count = 0
    personal_course_count = 0
    executive_course_count = 0
    
    for record in visual_calendar:
        if record.get('公共课'):
            public_course_count += 1
            course_count += 1
        if record.get('个人课程表'):
            personal_course_count += 1
            course_count += 1
        if record.get('企业高管公开课'):
            executive_course_count += 1
    
    print(f"\\n📊 视觉设计课程统计:")
    print(f"   - 总记录数: {len(visual_calendar)}")
    print(f"   - 有课程内容的记录: {course_count}")
    print(f"   - 公共课记录: {public_course_count}")
    print(f"   - 个人课程记录: {personal_course_count}")
    print(f"   - 企业高管公开课记录: {executive_course_count}")
    
    print("\\n🎉 视觉设计课程日历数据更新完成！")
    print("\\n📝 主要变更:")
    print("   - 导入了视觉设计课程日历数据")
    print("   - 更新了用户信息为视觉设计专业")
    print("   - 保持了原有的数据转换逻辑")

if __name__ == "__main__":
    main()