import os
import csv

# 创建目录结构
os.makedirs(r'C:\Users\jerry\PyCharmProjects\footviz\data\player_data', exist_ok=True)

# 创建德国球队和球员示例数据
german_players = {
    "Bayern Munich": [
        ["姓名", "国籍", "位置", "年龄", "出场次数", "进球", "助攻", "传球成功率%", "场均抢断", "场均拦截", "场均犯规", "射门/场", "过人/场", "评分"],
        ["Thomas Muller", "德国", "前锋", 34, 22, 8, 6, 85.2, 1.2, 0.8, 1.1, 2.1, 1.3, 7.2],
        ["Manuel Neuer", "德国", "门将", 38, 21, 0, 1, 92.1, 0.2, 0.3, 0.1, 0.0, 0.0, 7.1],
        ["Joshua Kimmich", "德国", "中场", 29, 23, 3, 8, 91.3, 2.1, 1.4, 1.8, 1.2, 1.1, 7.5]
    ],
    "Bayer Leverkusen": [
        ["姓名", "国籍", "位置", "年龄", "出场次数", "进球", "助攻", "传球成功率%", "场均抢断", "场均拦截", "场均犯规", "射门/场", "过人/场", "评分"],
        ["Bayer Leverkusen", "德国", "前锋", 21, 24, 12, 8, 78.9, 0.8, 0.5, 1.2, 3.2, 2.1, 7.3],
        ["Robert Andrich", "德国", "中场", 29, 22, 2, 3, 88.2, 2.3, 1.8, 2.1, 0.9, 0.8, 6.9]
    ]
}

# 创建中国足球队和球员示例数据
chinese_players = {
    "上海申花": [
        ["姓名", "国籍", "位置", "年龄", "出场次数", "进球", "助攻", "传球成功率%", "场均抢断", "场均拦截", "场均犯规", "射门/场", "过人/场", "评分"],
        ["吴曦", "中国", "中场", 34, 4, 1, 0, 84.2, 1.8, 1.5, 2.2, 1.0, 0.8, 6.8],
        ["于汉超", "中国", "前锋", 37, 3, 1, 1, 81.3, 0.7, 0.5, 1.1, 1.3, 1.2, 6.7]
    ],
    "上海海港": [
        ["姓名", "国籍", "位置", "年龄", "出场次数", "进球", "助攻", "传球成功率%", "场均抢断", "场均拦截", "场均犯规", "射门/场", "过人/场", "评分"],
        ["奥斯卡", "巴西", "中场", 32, 5, 2, 3, 90.1, 1.2, 0.8, 1.5, 2.1, 2.3, 8.1],
        ["武磊", "中国", "前锋", 33, 5, 3, 1, 82.7, 0.9, 0.6, 1.0, 2.8, 1.1, 7.3]
    ]
}

# 生成德国球员数据文件
for team, players in german_players.items():
    team_dir = f'C:\\Users\\jerry\\PyCharmProjects\\footviz\\data\\player_data\\{team}'
    os.makedirs(team_dir, exist_ok=True)
    
    header = players[0]
    for player_data in players[1:]:
        player_name = player_data[0]
        filename = f'{team_dir}\\{player_name}.csv'
        with open(filename, 'w', newline='', encoding='utf-8') as csvfile:
            writer = csv.writer(csvfile)
            writer.writerow(header)
            writer.writerow(player_data)

# 生成中国球员数据文件
for team, players in chinese_players.items():
    team_dir = f'C:\\Users\\jerry\\PyCharmProjects\\footviz\\data\\player_data\\{team}'
    os.makedirs(team_dir, exist_ok=True)
    
    header = players[0]
    for player_data in players[1:]:
        player_name = player_data[0]
        filename = f'{team_dir}\\{player_name}.csv'
        with open(filename, 'w', newline='', encoding='utf-8') as csvfile:
            writer = csv.writer(csvfile)
            writer.writerow(header)
            writer.writerow(player_data)

print("已创建德国和中国的球员数据文件")