#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
更新companyJobsData.json，添加企业图片数据
"""

import json
import os

# 文件路径
COMPANY_JOBS_DATA_PATH = "src/mocks/companyJobsData.json"
COMPANY_IMAGES_DATA_PATH = "网页未导入数据/土木水利产业/土木水利_内推岗位企业图片.json"

def main():
    # 读取企业图片数据
    with open(COMPANY_IMAGES_DATA_PATH, 'r', encoding='utf-8') as f:
        company_images_data = json.load(f)

    # 创建岗位名称到图片URL的映射
    image_map = {}
    for item in company_images_data:
        position_name = item.get("内推岗位名称")
        image_urls = item.get("BOSS照片链接", "")
        if position_name and image_urls:
            image_map[position_name] = image_urls

    print(f"读取到 {len(image_map)} 个岗位的企业图片数据")

    # 读取岗位数据
    with open(COMPANY_JOBS_DATA_PATH, 'r', encoding='utf-8') as f:
        company_jobs_data = json.load(f)

    print(f"总共 {len(company_jobs_data)} 个岗位")

    # 更新岗位数据，添加BOSS照片链接字段
    updated_count = 0
    for job in company_jobs_data:
        position_name = job.get("内推岗位名称")
        if position_name and position_name in image_map:
            job["BOSS照片链接"] = image_map[position_name]
            updated_count += 1
            print(f"✓ 更新岗位: {position_name}")

    print(f"\n成功更新 {updated_count} 个岗位的企业图片数据")
    print(f"未匹配到企业图片的岗位: {len(company_jobs_data) - updated_count} 个")

    # 写回文件
    with open(COMPANY_JOBS_DATA_PATH, 'w', encoding='utf-8') as f:
        json.dump(company_jobs_data, f, ensure_ascii=False, indent=2)

    print(f"\n数据已成功写入: {COMPANY_JOBS_DATA_PATH}")

if __name__ == "__main__":
    main()