import os
import shutil
from pathlib import Path

def main():
    # 设置路径
    base_path = Path(__file__).parent
    vue_public_path = base_path / "footviz-vue" / "public"
    media_path = base_path / "media"
    
    print(f"基础路径: {base_path}")
    print(f"Vue公共路径: {vue_public_path}")
    print(f"媒体路径: {media_path}")
    
    # 检查源目录是否存在
    if not vue_public_path.exists():
        print("错误: Vue公共目录不存在")
        return
    
    # 创建目标目录
    media_path.mkdir(exist_ok=True)
    
    # 需要转移的目录列表
    directories_to_transfer = [
        "team-visuals",
        "gf_css",
        "images",
        "league_icons",
        "scripts",
        "static",
        "styles",
        "team_images",
        "team_player_images"
    ]
    
    # 转移每个目录
    for directory in directories_to_transfer:
        source_dir = vue_public_path / directory
        target_dir = media_path / directory
        
        if source_dir.exists():
            print(f"正在转移 {directory}...")
            
            # 如果目标目录已存在，先删除
            if target_dir.exists():
                shutil.rmtree(target_dir)
            
            # 复制整个目录
            shutil.copytree(source_dir, target_dir)
            print(f"{directory} 转移完成")
        else:
            print(f"警告: {directory} 不存在于源目录中")
    
    print("所有资源转移完成!")

if __name__ == "__main__":
    main()