#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
替换企业内推岗位数据
将能源产业的内推岗位数据替换到companyJobsData.json中
"""

import json
import os
from datetime import datetime

def main():
    # 读取能源产业内推岗位数据
    energy_data_path = '网页未导入数据/能源产业/能源内推岗位.json'
    target_path = 'src/mocks/companyJobsData.json'
    
    # 创建备份
    backup_path = f'{target_path}.backup_{datetime.now().strftime("%Y%m%d_%H%M%S")}'
    
    print(f"读取能源产业数据: {energy_data_path}")
    with open(energy_data_path, 'r', encoding='utf-8') as f:
        energy_jobs = json.load(f)
    
    print(f"共读取到 {len(energy_jobs)} 个岗位")
    
    # 如果目标文件存在，先备份
    if os.path.exists(target_path):
        print(f"备份原文件到: {backup_path}")
        with open(target_path, 'r', encoding='utf-8') as f:
            original_data = f.read()
        with open(backup_path, 'w', encoding='utf-8') as f:
            f.write(original_data)
    
    # 处理能源数据，确保字段完整
    processed_jobs = []
    for job in energy_jobs:
        # 添加必要的字段映射
        processed_job = {
            "内推岗位名称": job.get("内推岗位名称", ""),
            "就业管家课程": "能源",  # 标记为能源产业
            "薪资": job.get("薪资", ""),
            "工作地点": job.get("工作地点", ""),
            "学历要求": job.get("学历要求", ""),
            "招聘人数": job.get("招聘人数", ""),
            "职位标签": job.get("职位标签", []),
            "职位描述": job.get("职位描述", ""),
            "任职要求": job.get("任职要求", ""),
            "公司介绍": job.get("公司介绍", ""),
            "岗位标签": "就业",  # 统一标记
            "岗位相关标签": job.get("岗位相关标签", "专业相关岗位"),
            "福利标签": job.get("福利标签", []),
            "截止时间": job.get("岗位招聘截止时间", "2025/12/31")
        }
        processed_jobs.append(processed_job)
    
    # 写入新数据
    print(f"写入 {len(processed_jobs)} 个岗位到: {target_path}")
    with open(target_path, 'w', encoding='utf-8') as f:
        json.dump(processed_jobs, f, ensure_ascii=False, indent=2)
    
    print("✅ 数据替换完成!")
    print(f"- 原文件已备份: {backup_path}")
    print(f"- 新数据已写入: {target_path}")
    print(f"- 共替换 {len(processed_jobs)} 个能源产业岗位")
    
    # 验证数据
    with open(target_path, 'r', encoding='utf-8') as f:
        verify_data = json.load(f)
    print(f"✅ 验证: 文件包含 {len(verify_data)} 个岗位")
    
    # 显示前3个岗位名称作为验证
    print("\n前3个岗位:")
    for i, job in enumerate(verify_data[:3]):
        print(f"  {i+1}. {job['内推岗位名称']} - {job['工作地点']} - {job['薪资']}")

if __name__ == "__main__":
    main()