#!/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
# 陈沐谦使用: 54461d443f4927823b56661d0aa6205b.jpg (id: 15)
chen_new_avatar = "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/54461d443f4927823b56661d0aa6205b.jpg"

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

# 当前陈沐谦的头像URL
chen_old_avatar = "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/377cc7767c3c9401c3987c844ce73e4b.jpg"

# 当前何泽的头像URL
he_old_avatar = "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/42c1b7e063d8835bceb6eb4ec45da315.jpg"

# 替换陈沐谦的所有头像
content = content.replace(chen_old_avatar, chen_new_avatar)

# 替换何泽的所有头像
content = content.replace(he_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"陈沐谦新头像: 54461d443f4927823b56661d0aa6205b.jpg")
print(f"何泽新头像: 54bf2cbc6433120fd99d7d85b7c70f3a.jpg")

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

# 替换ClassRankModal中陈沐谦的头像
modal_content = modal_content.replace(chen_old_avatar, chen_new_avatar)

# 替换ClassRankModal中何泽的头像
modal_content = modal_content.replace(he_old_avatar, he_new_avatar)

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

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

# 统计替换的位置
with open('src/data/mockData.js', 'r', encoding='utf-8') as f:
    check_content = f.read()
    chen_count = check_content.count(chen_new_avatar)
    he_count = check_content.count(he_new_avatar)
    print(f"✅ 陈沐谦头像在mockData.js中出现 {chen_count} 次")
    print(f"✅ 何泽头像在mockData.js中出现 {he_count} 次")