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

import json

# 读取头像列表获取更多选择
with open('网页未导入数据/头像列表.json', 'r', encoding='utf-8') as f:
    avatar_list = json.load(f)

# 选择不同的头像（选择列表中的其他头像）
# 选择索引 10, 15, 20 的头像
selected_avatars = [
    avatar_list[10]["file_url"].split("?")[0] if len(avatar_list) > 10 else "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/199b35fbd6ed5e3ea6d6d064cf895a13.jpg",  # 第1名
    avatar_list[15]["file_url"].split("?")[0] if len(avatar_list) > 15 else "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/13c5709a3993fdf353d147209f8145cb.jpg",  # 第2名  
    avatar_list[20]["file_url"].split("?")[0] if len(avatar_list) > 20 else "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/2e8af9fd9b8425946272fec27af82910.jpg",  # 第3名
]

print(f"选择的新头像：")
print(f"第1名: {selected_avatars[0]}")
print(f"第2名: {selected_avatars[1]}")
print(f"第3名: {selected_avatars[2]}")

# 读取mockData.js
with open('src/data/mockData.js', 'r', encoding='utf-8') as f:
    content = f.read()

# 更新前三名学生的头像
updates = [
    {
        "name": "于语涵",
        "old_avatar": "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/02393125baa474d558c484c0677664b1.jpg",
        "new_avatar": selected_avatars[0]
    },
    {
        "name": "陈沐谦", 
        "old_avatar": "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/07a0a14c8c8d5476b2c8d54de12e6a06.jpg",
        "new_avatar": selected_avatars[1]
    },
    {
        "name": "魏思涵",
        "old_avatar": "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/0852e24fc55e6e60c68a2adea1c8eb7b.jpg", 
        "new_avatar": selected_avatars[2]
    }
]

# 替换头像URL
for update in updates:
    content = content.replace(update["old_avatar"], update["new_avatar"])
    print(f"✓ 更新 {update['name']} 的头像")

# 写回文件
with open('src/data/mockData.js', 'w', encoding='utf-8') as f:
    f.write(content)

print("\n✅ 成功更换前三名学生的头像！")