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

import json
import shutil
from datetime import datetime

def replace_with_energy_jobs():
    """将岗位数据替换为能源产业岗位"""

    # 备份原有数据
    backup_file = f"src/data/companyJobsNew.json.backup_{datetime.now().strftime('%Y%m%d_%H%M%S')}"
    shutil.copy('src/data/companyJobsNew.json', backup_file)
    print(f"已创建备份文件: {backup_file}")

    # 读取能源岗位数据
    with open('网页未导入数据/能源产业/能源内推岗位.json', 'r', encoding='utf-8') as f:
        energy_jobs = json.load(f)

    # 转换数据格式，映射字段名称
    converted_jobs = []
    for job in energy_jobs:
        # 判断岗位标签类型
        position_tag = "就业"  # 默认为就业
        if "实习" in job.get('内推岗位名称', '') or "助理" in job.get('内推岗位名称', ''):
            position_tag = "实习"

        converted_job = {
            "内推岗位名称": job['内推岗位名称'],
            "岗位相关标签": job['岗位相关标签'],
            "岗位标签": position_tag,
            "薪资": job['薪资'],
            "工作地点": job['工作地点'],
            "学历要求": job['学历要求'],
            "招聘人数": job['招聘人数'],
            "截止时间": job['岗位招聘截止时间'],  # 字段名映射
            "职位标签": job['职位标签'],
            "福利标签": job['福利标签'],
            "职位描述": job['职位描述'],
            "任职要求": job['任职要求'],
            "公司介绍": job['公司介绍']
        }
        converted_jobs.append(converted_job)

    # 写入转换后的数据
    with open('src/data/companyJobsNew.json', 'w', encoding='utf-8') as f:
        json.dump(converted_jobs, f, ensure_ascii=False, indent=2)

    print(f"已成功替换岗位数据:")
    print(f"- 能源产业岗位数: {len(converted_jobs)} 个")
    print(f"- 备份文件: {backup_file}")

    # 显示前5个岗位
    print("\n已替换的岗位示例:")
    for i, job in enumerate(converted_jobs[:5], 1):
        print(f"{i}. {job['内推岗位名称']}: {job['薪资']} - {job['工作地点']} ({job['岗位标签']})")

if __name__ == "__main__":
    replace_with_energy_jobs()