mirror of
https://github.com/ChuckBuilds/LEDMatrix.git
synced 2026-04-10 21:03:01 +00:00
129 lines
3.8 KiB
Python
129 lines
3.8 KiB
Python
#!/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()
|