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

import os
import datetime
import shutil
import json
from collections import defaultdict

def generate_industries_with_avatars():
    """
    生成带有岗位头像的industries数据
    """
    # 读取能源岗位简历数据
    with open("网页未导入数据/能源产业/能源岗位简历.json", 'r', encoding='utf-8') as f:
        energy_jobs = json.load(f)

    # 读取岗位等级数据（包含头像）
    with open("src/data/joblevel.json", 'r', encoding='utf-8') as f:
        joblevel_data = json.load(f)

    # 创建岗位名称到头像的映射
    position_avatars = {}
    for level_key, level_data in joblevel_data["data"].items():
        for position in level_data["list"]:
            position_avatars[position["position_name"]] = position.get("img", "")

    # 按岗位群分组
    job_groups = defaultdict(list)
    for job in energy_jobs:
        group_name = job.get("简历岗位群", "")
        if group_name:
            job_groups[group_name].append(job)

    # 生成industries数据结构
    industries = []
    group_id = 1

    for group_name, jobs in job_groups.items():
        industry = {
            "id": f"energy_{group_id}",
            "name": group_name,
            "positions": [],
            "questions": [
                {
                    "id": f"group_q{group_id}",
                    "question": f"# {group_name}面试题",
                    "subQuestions": []
                }
            ]
        }

        # 添加该岗位群下的所有岗位
        pos_id = 1
        for job in jobs:
            position_name = job["岗位名称"]
            avatar_url = position_avatars.get(position_name, "")

            position = {
                "id": f"energy_{group_id}_{pos_id}",
                "title": position_name,
                "level": job.get("岗位等级标签", "基础岗"),
                "avatar": avatar_url,
                "department": group_name,
                "type": "全职",
                "experience": "1-3年",
                "education": "大专",
                "salary": "8-15K",
                "location": "北京",
                "updateTime": "2024-01-20",
                "description": f"负责{position_name}相关工作",
                "requirements": []
            }
            industry["positions"].append(position)
            pos_id += 1

        industries.append(industry)
        group_id += 1

    return industries

def replace_industries_in_mock():
    """
    替换mock文件中的industries数据
    """
    mock_file = "src/mocks/resumeInterviewMock.js"

    # 备份文件
    backup_path = f"{mock_file}.backup_{datetime.datetime.now().strftime('%Y%m%d_%H%M%S')}"
    shutil.copy(mock_file, backup_path)
    print(f"✅ 已备份文件到：{backup_path}")

    # 生成新的industries数据
    industries = generate_industries_with_avatars()

    # 读取原文件内容
    with open(mock_file, 'r', encoding='utf-8') as f:
        content = f.read()

    # 查找industries数组的起始和结束位置
    import re

    # 生成新的industries字符串
    industries_str = json.dumps(industries, ensure_ascii=False, indent=2)

    # 替换industries数组
    pattern = r'const industries = \[[\s\S]*?\n\];'
    replacement = f'const industries = {industries_str};'

    new_content = re.sub(pattern, replacement, content, count=1)

    # 写入文件
    with open(mock_file, 'w', encoding='utf-8') as f:
        f.write(new_content)

    # 输出统计信息
    print(f"✅ 成功替换industries数据")
    print(f"📊 共 {len(industries)} 个岗位群")
    for i, industry in enumerate(industries, 1):
        print(f"   {i}. {industry['name']} ({len(industry['positions'])}个岗位)")

if __name__ == "__main__":
    replace_industries_in_mock()