#!/usr/bin/env python3
"""
Script to download all NCAA Football team logos from ESPN API
and save them with team abbreviations as filenames.
"""

import os
import requests
import json
from pathlib import Path
import time

def create_logo_directory():
    """Create the ncaaFBlogos directory if it doesn't exist."""
    logo_dir = Path("test/ncaaFBlogos")
    logo_dir.mkdir(parents=True, exist_ok=True)
    return logo_dir

def fetch_teams_data():
    """Fetch team data from ESPN API."""
    url = "https://site.api.espn.com/apis/site/v2/sports/football/college-football/teams"
    
    try:
        response = requests.get(url, timeout=30)
        response.raise_for_status()
        return response.json()
    except requests.exceptions.RequestException as e:
        print(f"Error fetching teams data: {e}")
        return None

def download_logo(url, filepath, team_name):
    """Download a logo from URL and save to filepath."""
    try:
        response = requests.get(url, timeout=30)
        response.raise_for_status()
        
        with open(filepath, 'wb') as f:
            f.write(response.content)
        
        print(f"✓ Downloaded: {team_name} -> {filepath.name}")
        return True
        
    except requests.exceptions.RequestException as e:
        print(f"✗ Failed to download {team_name}: {e}")
        return False

def normalize_abbreviation(abbreviation):
    """Normalize team abbreviation to lowercase for filename."""
    return abbreviation.lower()

def main():
    """Main function to download all NCAA FB team logos."""
    print("Starting NCAA Football logo download...")
    
    # Create directory
    logo_dir = create_logo_directory()
    print(f"Created/verified directory: {logo_dir}")
    
    # Fetch teams data
    print("Fetching teams data from ESPN API...")
    data = fetch_teams_data()
    
    if not data:
        print("Failed to fetch teams data. Exiting.")
        return
    
    # Extract teams
    teams = []
    try:
        sports = data.get('sports', [])
        for sport in sports:
            leagues = sport.get('leagues', [])
            for league in leagues:
                teams = league.get('teams', [])
                break
    except (KeyError, IndexError) as e:
        print(f"Error parsing teams data: {e}")
        return
    
    print(f"Found {len(teams)} teams")
    
    # Download logos
    downloaded_count = 0
    failed_count = 0
    
    for team_data in teams:
        team = team_data.get('team', {})
        
        # Extract team information
        abbreviation = team.get('abbreviation', '')
        display_name = team.get('displayName', 'Unknown')
        logos = team.get('logos', [])
        
        if not abbreviation or not logos:
            print(f"⚠ Skipping {display_name}: missing abbreviation or logos")
            continue
        
        # Get the default logo (first one is usually default)
        logo_url = logos[0].get('href', '')
        if not logo_url:
            print(f"⚠ Skipping {display_name}: no logo URL")
            continue
        
        # Create filename
        filename = f"{normalize_abbreviation(abbreviation)}.png"
        filepath = logo_dir / filename
        
        # Skip if already exists
        if filepath.exists():
            print(f"⏭ Skipping {display_name}: {filename} already exists")
            continue
        
        # Download logo
        if download_logo(logo_url, filepath, display_name):
            downloaded_count += 1
        else:
            failed_count += 1
        
        # Small delay to be respectful to the API
        time.sleep(0.1)
    
    print(f"\nDownload complete!")
    print(f"✓ Successfully downloaded: {downloaded_count} logos")
    print(f"✗ Failed downloads: {failed_count}")
    print(f"📁 Logos saved in: {logo_dir}")

if __name__ == "__main__":
    main()