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

import json
import shutil
from datetime import datetime

def convert_energy_interview_status():
    """将能源产业面试状态数据转换为系统格式"""

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

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

    # 转换数据格式
    converted_status = []

    for item in energy_status:
        # 构建阶段日期字段
        stage = item['岗位内推流程']
        time = item['流程时间'].split(' ')[0]  # 取日期部分

        # 映射流程阶段名称
        stage_map = {
            'HR评估': 'HR评估',
            '面试': '面试',
            '面试结果': '面试',
            'Offer': 'Offer'
        }

        stage_name = stage_map.get(stage, stage)

        # 构建面试状态文本
        status_text = item['内容']

        # 创建转换后的记录
        converted_item = {
            "查询岗位名称": item['查询岗位名称'],
            "阶段日期": f"{stage_name}：{time}",
            "面试状态": status_text
        }

        converted_status.append(converted_item)

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

    print(f"已成功替换面试状态数据:")
    print(f"- 能源产业面试状态记录数: {len(converted_status)} 条")
    print(f"- 备份文件: {backup_file}")

    # 显示前5条记录
    print("\n已替换的面试状态示例:")
    for i, item in enumerate(converted_status[:5], 1):
        print(f"{i}. {item['查询岗位名称']}: {item['阶段日期']} - {item['面试状态']}")

if __name__ == "__main__":
    convert_energy_interview_status()