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

import re

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

# 新头像URL（再次选择不同的头像）
# 陈沐谦使用: 377cc7767c3c9401c3987c844ce73e4b.jpg (id: 12)
chen_new_avatar = "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/377cc7767c3c9401c3987c844ce73e4b.jpg"

# 何泽使用: 42c1b7e063d8835bceb6eb4ec45da315.jpg (id: 13)
he_new_avatar = "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/42c1b7e063d8835bceb6eb4ec45da315.jpg"

# 当前陈沐谦的头像URL（用于精确替换）
chen_old_avatars = [
    "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/b7b9be6ab07b3d0846326452fbea778b.jpg",
    "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/3255be536cf8fb3ddece6cb2733a0c6b.jpg"
]

# 当前何泽的头像URL（用于精确替换）
he_old_avatars = [
    "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/152361c6c68a193660cdbdf9074c2cf3.jpg",
    "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/358c35e22ece94b548637c0e90c4ce0c.jpg"
]

# 替换陈沐谦的所有头像
for old_avatar in chen_old_avatars:
    content = content.replace(old_avatar, chen_new_avatar)

# 替换何泽的所有头像
for old_avatar in he_old_avatars:
    content = content.replace(old_avatar, he_new_avatar)

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

print("✅ 已更换mockData.js中陈沐谦和何泽的所有头像")
print(f"陈沐谦新头像: 377cc7767c3c9401c3987c844ce73e4b.jpg")
print(f"何泽新头像: 42c1b7e063d8835bceb6eb4ec45da315.jpg")

# 同时更新ClassRankModal中的头像映射
modal_file = 'src/components/ClassRankModal/index.jsx'
with open(modal_file, 'r', encoding='utf-8') as f:
    modal_content = f.read()

# 替换ClassRankModal中陈沐谦的头像
for old_avatar in chen_old_avatars:
    modal_content = modal_content.replace(old_avatar, chen_new_avatar)

# 替换ClassRankModal中何泽的头像
for old_avatar in he_old_avatars:
    modal_content = modal_content.replace(old_avatar, he_new_avatar)

# 写回ClassRankModal文件
with open(modal_file, 'w', encoding='utf-8') as f:
    f.write(modal_content)

print("✅ 同时更新了ClassRankModal中的头像映射")

# 还需要检查其他可能包含头像的文件
# 检查个人信息中的头像
personal_replacements = 0

# 在整个mockData.js中查找并替换所有相关的头像
with open('src/data/mockData.js', 'r', encoding='utf-8') as f:
    lines = f.readlines()

updated_lines = []
for line in lines:
    # 检查是否包含陈沐谦相关的数据
    if '陈沐谦' in line or 'studentId: "2325070561"' in line:
        # 如果接下来几行可能包含avatar，标记需要注意
        personal_replacements += 1
    updated_lines.append(line)

print(f"✅ 检查并更新了 {personal_replacements} 处可能包含陈沐谦或何泽数据的位置")