#!/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（选择不同的头像）
# 陈沐谦使用: 3255be536cf8fb3ddece6cb2733a0c6b.jpg (id: 10)
chen_new_avatar = "https://ddcz-1315997005.cos.ap-nanjing.myqcloud.com/static/avatar/douyin/3255be536cf8fb3ddece6cb2733a0c6b.jpg"

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

# 替换陈沐谦的头像 (所有出现的地方)
content = re.sub(
    r'(陈沐谦.*?avatar:\s*")[^"]+(")',
    rf'\1{chen_new_avatar}\2',
    content
)

# 替换何泽的头像 (所有出现的地方)
content = re.sub(
    r'(何泽.*?avatar:\s*")[^"]+(")',
    rf'\1{he_new_avatar}\2',
    content
)

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

print("✅ 已更换陈沐谦和何泽的头像")
print(f"陈沐谦新头像: 3255be536cf8fb3ddece6cb2733a0c6b.jpg")
print(f"何泽新头像: 358c35e22ece94b548637c0e90c4ce0c.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 = re.sub(
    r'case "陈沐谦":\s+avatarUrl = "[^"]+";',
    f'case "陈沐谦":\n                    avatarUrl = "{chen_new_avatar}";',
    modal_content
)

# 更新ClassRankModal中何泽的头像
modal_content = re.sub(
    r'case "何泽":\s+avatarUrl = "[^"]+";',
    f'case "何泽":\n                    avatarUrl = "{he_new_avatar}";',
    modal_content
)

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

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