import requests
import json

def test_api_endpoints():
    """
    测试API端点是否正常工作
    """
    base_url = "http://localhost:8000/api"
    
    # 测试国家接口
    try:
        response = requests.get(f"{base_url}/countries/")
        if response.status_code == 200:
            countries = response.json()
            print(f"国家接口测试成功，返回 {len(countries.get('results', []))} 条记录")
        else:
            print(f"国家接口测试失败，状态码: {response.status_code}")
    except Exception as e:
        print(f"国家接口测试出错: {str(e)}")
    
    # 测试球队接口
    try:
        response = requests.get(f"{base_url}/teams/")
        if response.status_code == 200:
            teams = response.json()
            print(f"球队接口测试成功，返回 {len(teams.get('results', []))} 条记录")
        else:
            print(f"球队接口测试失败，状态码: {response.status_code}")
    except Exception as e:
        print(f"球队接口测试出错: {str(e)}")
    
    # 测试球员接口
    try:
        response = requests.get(f"{base_url}/players/")
        if response.status_code == 200:
            players = response.json()
            print(f"球员接口测试成功，返回 {len(players.get('results', []))} 条记录")
        else:
            print(f"球员接口测试失败，状态码: {response.status_code}")
    except Exception as e:
        print(f"球员接口测试出错: {str(e)}")
    
    # 测试球员统计数据接口
    try:
        response = requests.get(f"{base_url}/player-stats/")
        if response.status_code == 200:
            player_stats = response.json()
            print(f"球员统计数据接口测试成功，返回 {len(player_stats.get('results', []))} 条记录")
        else:
            print(f"球员统计数据接口测试失败，状态码: {response.status_code}")
    except Exception as e:
        print(f"球员统计数据接口测试出错: {str(e)}")

if __name__ == "__main__":
    test_api_endpoints()