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

import json
import os
from datetime import datetime

def main():
    # 读取能源产业内推岗位数据
    energy_data_path = '网页未导入数据/能源产业/能源内推岗位.json'
    target_path = 'src/data/companyJobsNew.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)
    
    # 写入新数据（直接使用原始格式，不需要转换）
    print(f"写入 {len(energy_jobs)} 个岗位到: {target_path}")
    with open(target_path, 'w', encoding='utf-8') as f:
        json.dump(energy_jobs, f, ensure_ascii=False, indent=2)
    
    print("✅ 数据替换完成!")
    print(f"- 原文件已备份: {backup_path}")
    print(f"- 新数据已写入: {target_path}")
    print(f"- 共替换 {len(energy_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()