mirror of
https://github.com/ChuckBuilds/LEDMatrix.git
synced 2026-04-10 21:03:01 +00:00
679 lines
31 KiB
Python
679 lines
31 KiB
Python
import os
|
|
import time
|
|
import logging
|
|
import requests
|
|
import json
|
|
from typing import Dict, Any, Optional, List
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
from pathlib import Path
|
|
from datetime import datetime, timedelta, timezone
|
|
from src.display_manager import DisplayManager
|
|
|
|
# Constants
|
|
ESPN_NHL_SCOREBOARD_URL = "https://site.api.espn.com/apis/site/v2/sports/hockey/nhl/scoreboard"
|
|
|
|
class BaseNHLManager:
|
|
"""Base class for NHL managers with common functionality."""
|
|
def __init__(self, config: Dict[str, Any], display_manager: DisplayManager):
|
|
self.display_manager = display_manager
|
|
self.config = config
|
|
self.nhl_config = config.get("nhl_scoreboard", {})
|
|
self.is_enabled = self.nhl_config.get("enabled", False)
|
|
self.test_mode = self.nhl_config.get("test_mode", False)
|
|
self.logo_dir = self.nhl_config.get("logo_dir", "assets/sports/nhl_logos")
|
|
self.update_interval = self.nhl_config.get("update_interval_seconds", 60)
|
|
self.last_update = 0
|
|
self.current_game = None
|
|
self.fonts = self._load_fonts()
|
|
self.favorite_teams = self.nhl_config.get("favorite_teams", [])
|
|
self.logger = logging.getLogger('NHL')
|
|
|
|
# Get display dimensions from config
|
|
display_config = config.get("display", {})
|
|
hardware_config = display_config.get("hardware", {})
|
|
cols = hardware_config.get("cols", 64)
|
|
chain = hardware_config.get("chain_length", 1)
|
|
self.display_width = int(cols * chain)
|
|
self.display_height = hardware_config.get("rows", 32)
|
|
|
|
# Cache for loaded logos
|
|
self._logo_cache = {}
|
|
|
|
self.logger.info(f"Initialized NHL manager with display dimensions: {self.display_width}x{self.display_height}")
|
|
logging.info(f"[NHL] Test mode: {'enabled' if self.test_mode else 'disabled'}")
|
|
logging.info(f"[NHL] Favorite teams: {self.favorite_teams}")
|
|
|
|
def _load_fonts(self):
|
|
"""Load fonts used by the scoreboard."""
|
|
fonts = {}
|
|
try:
|
|
# Try to load the 4x6 font for scores
|
|
fonts['score'] = ImageFont.truetype("assets/fonts/4x6-font.ttf", 12)
|
|
fonts['time'] = ImageFont.truetype("assets/fonts/4x6-font.ttf", 10)
|
|
fonts['team'] = ImageFont.truetype("assets/fonts/4x6-font.ttf", 8)
|
|
fonts['status'] = ImageFont.truetype("assets/fonts/4x6-font.ttf", 9)
|
|
logging.info("[NHL] Successfully loaded 4x6 font for all text elements")
|
|
except IOError:
|
|
logging.warning("[NHL] 4x6 font not found, trying PressStart2P font.")
|
|
try:
|
|
# Try to load the PressStart2P font as a fallback
|
|
fonts['score'] = ImageFont.truetype("assets/fonts/PressStart2P-Regular.ttf", 12)
|
|
fonts['time'] = ImageFont.truetype("assets/fonts/PressStart2P-Regular.ttf", 10)
|
|
fonts['team'] = ImageFont.truetype("assets/fonts/PressStart2P-Regular.ttf", 8)
|
|
fonts['status'] = ImageFont.truetype("assets/fonts/PressStart2P-Regular.ttf", 9)
|
|
logging.info("[NHL] Successfully loaded PressStart2P font for all text elements")
|
|
except IOError:
|
|
logging.warning("[NHL] PressStart2P font not found, using default PIL font.")
|
|
# Use default PIL font as a last resort
|
|
fonts['score'] = ImageFont.load_default()
|
|
fonts['time'] = ImageFont.load_default()
|
|
fonts['team'] = ImageFont.load_default()
|
|
fonts['status'] = ImageFont.load_default()
|
|
return fonts
|
|
|
|
def _load_and_resize_logo(self, team_abbrev: str) -> Optional[Image.Image]:
|
|
"""Load and resize a team logo, with caching."""
|
|
if team_abbrev in self._logo_cache:
|
|
return self._logo_cache[team_abbrev]
|
|
|
|
logo_path = os.path.join(self.logo_dir, f"{team_abbrev}.png")
|
|
self.logger.debug(f"Loading logo from: {logo_path}")
|
|
|
|
try:
|
|
logo = Image.open(logo_path)
|
|
original_size = logo.size
|
|
self.logger.debug(f"Original logo size: {original_size}")
|
|
|
|
# Convert to RGBA if not already
|
|
if logo.mode != 'RGBA':
|
|
logo = logo.convert('RGBA')
|
|
|
|
# Calculate max size based on display dimensions
|
|
max_width = self.display_width // 4 # Quarter of display width
|
|
max_height = self.display_height // 2 # Half of display height
|
|
|
|
# Resize maintaining aspect ratio
|
|
logo.thumbnail((max_width, max_height), Image.Resampling.LANCZOS)
|
|
self.logger.debug(f"Resized logo size: {logo.size}")
|
|
|
|
# Cache the resized logo
|
|
self._logo_cache[team_abbrev] = logo
|
|
return logo
|
|
|
|
except Exception as e:
|
|
self.logger.error(f"Error loading logo for {team_abbrev}: {e}")
|
|
return None
|
|
|
|
def _fetch_data(self, date_str: str = None) -> Optional[Dict]:
|
|
"""Fetch data from ESPN API or load test data."""
|
|
url = ESPN_NHL_SCOREBOARD_URL
|
|
params = {}
|
|
if date_str:
|
|
params['dates'] = date_str
|
|
|
|
try:
|
|
response = requests.get(url, params=params, timeout=10)
|
|
response.raise_for_status()
|
|
data = response.json()
|
|
logging.info(f"[NHL] Successfully fetched data from ESPN API")
|
|
return data
|
|
except requests.exceptions.RequestException as e:
|
|
logging.error(f"[NHL] Error fetching data from ESPN: {e}")
|
|
return None
|
|
|
|
def _extract_game_details(self, game_event: Dict) -> Optional[Dict]:
|
|
"""Extract relevant game details from ESPN API response."""
|
|
if not game_event:
|
|
return None
|
|
|
|
try:
|
|
competition = game_event["competitions"][0]
|
|
status = competition["status"]
|
|
competitors = competition["competitors"]
|
|
game_date_str = game_event["date"]
|
|
|
|
# Parse game date/time
|
|
try:
|
|
start_time_utc = datetime.fromisoformat(game_date_str.replace("Z", "+00:00"))
|
|
except ValueError:
|
|
logging.warning(f"[NHL] Could not parse game date: {game_date_str}")
|
|
start_time_utc = None
|
|
|
|
home_team = next(c for c in competitors if c.get("homeAway") == "home")
|
|
away_team = next(c for c in competitors if c.get("homeAway") == "away")
|
|
|
|
# Format game time and date for display
|
|
game_time = ""
|
|
game_date = ""
|
|
if start_time_utc:
|
|
# Convert to local time
|
|
local_time = start_time_utc.astimezone()
|
|
game_time = local_time.strftime("%I:%M %p")
|
|
game_date = local_time.strftime("%b %d")
|
|
|
|
details = {
|
|
"start_time_utc": start_time_utc,
|
|
"status_text": status["type"]["shortDetail"],
|
|
"period": status.get("period", 0),
|
|
"clock": status.get("displayClock", "0:00"),
|
|
"is_live": status["type"]["state"] in ("in", "halftime"),
|
|
"is_final": status["type"]["state"] == "post",
|
|
"is_upcoming": status["type"]["state"] == "pre",
|
|
"home_abbr": home_team["team"]["abbreviation"],
|
|
"home_score": home_team.get("score", "0"),
|
|
"home_logo_path": os.path.join(self.logo_dir, f"{home_team['team']['abbreviation']}.png"),
|
|
"away_abbr": away_team["team"]["abbreviation"],
|
|
"away_score": away_team.get("score", "0"),
|
|
"away_logo_path": os.path.join(self.logo_dir, f"{away_team['team']['abbreviation']}.png"),
|
|
"game_time": game_time,
|
|
"game_date": game_date
|
|
}
|
|
|
|
# Validate logo files
|
|
for team in ["home", "away"]:
|
|
logo_path = details[f"{team}_logo_path"]
|
|
if not logo_path.is_file():
|
|
logging.warning(f"[NHL] {team.title()} logo not found: {logo_path}")
|
|
details[f"{team}_logo_path"] = None
|
|
else:
|
|
try:
|
|
with Image.open(logo_path) as img:
|
|
logging.debug(f"[NHL] {team.title()} logo is valid: {img.format}, size: {img.size}")
|
|
except Exception as e:
|
|
logging.error(f"[NHL] {team.title()} logo file exists but is not valid: {e}")
|
|
details[f"{team}_logo_path"] = None
|
|
|
|
return details
|
|
except Exception as e:
|
|
logging.error(f"[NHL] Error extracting game details: {e}")
|
|
return None
|
|
|
|
def display(self, force_clear: bool = False):
|
|
"""Display game information."""
|
|
if not self.current_game:
|
|
logging.warning("[NHL] No game data available to display")
|
|
return
|
|
|
|
try:
|
|
# Create a new black image
|
|
img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
draw = ImageDraw.Draw(img)
|
|
|
|
# Load and resize logos
|
|
home_logo = self._load_and_resize_logo(self.current_game["home_abbr"])
|
|
away_logo = self._load_and_resize_logo(self.current_game["away_abbr"])
|
|
|
|
# Draw home team logo (right side)
|
|
if home_logo:
|
|
home_x = 3 * self.display_width // 4 - home_logo.width // 2
|
|
home_y = self.display_height // 4 - home_logo.height // 2
|
|
temp_img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
temp_draw = ImageDraw.Draw(temp_img)
|
|
temp_img.paste(home_logo, (home_x, home_y), home_logo)
|
|
img.paste(temp_img, (0, 0))
|
|
|
|
# Draw away team logo (left side)
|
|
if away_logo:
|
|
away_x = self.display_width // 4 - away_logo.width // 2
|
|
away_y = self.display_height // 4 - away_logo.height // 2
|
|
temp_img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
temp_draw = ImageDraw.Draw(temp_img)
|
|
temp_img.paste(away_logo, (away_x, away_y), away_logo)
|
|
img.paste(temp_img, (0, 0))
|
|
|
|
# Draw scores in the format "AWAY - HOME"
|
|
home_score = str(self.current_game["home_score"])
|
|
away_score = str(self.current_game["away_score"])
|
|
|
|
# Format the score string with a hyphen separator
|
|
score_text = f"{away_score} - {home_score}"
|
|
|
|
# Calculate position for the score text (centered at the bottom)
|
|
score_width = draw.textlength(score_text, font=self.fonts['score'])
|
|
score_x = (self.display_width - score_width) // 2
|
|
score_y = 3 * self.display_height // 4 - 10
|
|
|
|
# Draw the score text
|
|
draw.text((score_x, score_y), score_text, font=self.fonts['score'], fill=(255, 255, 255))
|
|
|
|
# Draw game status (period and time or "FINAL")
|
|
status_text = self.current_game.get("status_text", "")
|
|
status_width = draw.textlength(status_text, font=self.fonts['status'])
|
|
status_x = (self.display_width - status_width) // 2
|
|
status_y = self.display_height // 2 - 5
|
|
|
|
draw.text((status_x, status_y), status_text, font=self.fonts['status'], fill=(255, 255, 255))
|
|
|
|
# Display the image
|
|
self.display_manager.image.paste(img, (0, 0))
|
|
self.display_manager.update_display()
|
|
logging.debug("[NHL] Successfully displayed game")
|
|
|
|
except Exception as e:
|
|
logging.error(f"[NHL] Error displaying game: {e}", exc_info=True)
|
|
|
|
class NHLLiveManager(BaseNHLManager):
|
|
"""Manager for live NHL games."""
|
|
def __init__(self, config: Dict[str, Any], display_manager: DisplayManager):
|
|
super().__init__(config, display_manager)
|
|
self.update_interval = self.nhl_config.get("live_update_interval", 30)
|
|
self.last_update = 0
|
|
self.logger.info("Initialized NHL Live Manager")
|
|
|
|
# Initialize with test game only if test mode is enabled
|
|
if self.test_mode:
|
|
self.current_game = {
|
|
"home_abbr": "TBL",
|
|
"away_abbr": "DAL",
|
|
"home_score": "3",
|
|
"away_score": "2",
|
|
"period": 2,
|
|
"clock": "12:34",
|
|
"home_logo_path": os.path.join(self.logo_dir, "TBL.png"),
|
|
"away_logo_path": os.path.join(self.logo_dir, "DAL.png"),
|
|
"game_time": "7:30 PM",
|
|
"game_date": "Apr 17"
|
|
}
|
|
logging.info("[NHL] Initialized NHLLiveManager with test game: TBL vs DAL")
|
|
else:
|
|
logging.info("[NHL] Initialized NHLLiveManager in live mode")
|
|
|
|
def update(self):
|
|
"""Update live game data."""
|
|
current_time = time.time()
|
|
if current_time - self.last_update >= self.update_interval:
|
|
self.logger.debug("Updating live game data")
|
|
self.last_update = current_time
|
|
if self.test_mode:
|
|
# For testing, we'll just update the clock to show it's working
|
|
if self.current_game:
|
|
minutes = int(self.current_game["clock"].split(":")[0])
|
|
seconds = int(self.current_game["clock"].split(":")[1])
|
|
seconds -= 1
|
|
if seconds < 0:
|
|
seconds = 59
|
|
minutes -= 1
|
|
if minutes < 0:
|
|
minutes = 19
|
|
if self.current_game["period"] < 3:
|
|
self.current_game["period"] += 1
|
|
else:
|
|
self.current_game["period"] = 1
|
|
self.current_game["clock"] = f"{minutes:02d}:{seconds:02d}"
|
|
logging.debug(f"[NHL] Updated test game clock: {self.current_game['clock']}")
|
|
else:
|
|
# Fetch live game data from ESPN API
|
|
data = self._fetch_data()
|
|
if data and "events" in data:
|
|
# Find the first live game involving favorite teams
|
|
for event in data["events"]:
|
|
details = self._extract_game_details(event)
|
|
if details and details["is_live"]:
|
|
if not self.favorite_teams or (
|
|
details["home_abbr"] in self.favorite_teams or
|
|
details["away_abbr"] in self.favorite_teams
|
|
):
|
|
self.current_game = details
|
|
logging.info(f"[NHL] Found live game: {details['away_abbr']} vs {details['home_abbr']}")
|
|
break
|
|
else:
|
|
# No live games found
|
|
self.current_game = None
|
|
logging.info("[NHL] No live games found")
|
|
|
|
def display(self, force_clear: bool = False):
|
|
"""Display live game information."""
|
|
if not self.current_game:
|
|
logging.warning("[NHL] No game data available to display")
|
|
return
|
|
|
|
try:
|
|
# Create a new black image
|
|
img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
draw = ImageDraw.Draw(img)
|
|
|
|
# Load and resize logos
|
|
home_logo = self._load_and_resize_logo(self.current_game["home_abbr"])
|
|
away_logo = self._load_and_resize_logo(self.current_game["away_abbr"])
|
|
|
|
# Draw home team logo (right side)
|
|
if home_logo:
|
|
home_x = 3 * self.display_width // 4 - home_logo.width // 2
|
|
home_y = self.display_height // 4 - home_logo.height // 2
|
|
temp_img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
temp_draw = ImageDraw.Draw(temp_img)
|
|
temp_img.paste(home_logo, (home_x, home_y), home_logo)
|
|
img.paste(temp_img, (0, 0))
|
|
|
|
# Draw away team logo (left side)
|
|
if away_logo:
|
|
away_x = self.display_width // 4 - away_logo.width // 2
|
|
away_y = self.display_height // 4 - away_logo.height // 2
|
|
temp_img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
temp_draw = ImageDraw.Draw(temp_img)
|
|
temp_img.paste(away_logo, (away_x, away_y), away_logo)
|
|
img.paste(temp_img, (0, 0))
|
|
|
|
# Draw scores in the format "AWAY - HOME"
|
|
home_score = str(self.current_game["home_score"])
|
|
away_score = str(self.current_game["away_score"])
|
|
|
|
# Format the score string with a hyphen separator
|
|
score_text = f"{away_score} - {home_score}"
|
|
|
|
# Calculate position for the score text (centered at the bottom)
|
|
score_width = draw.textlength(score_text, font=self.fonts['score'])
|
|
score_x = (self.display_width - score_width) // 2
|
|
score_y = 3 * self.display_height // 4 - 10
|
|
|
|
# Draw the score text
|
|
draw.text((score_x, score_y), score_text, font=self.fonts['score'], fill=(255, 255, 255))
|
|
|
|
# Format period text (1st, 2nd, 3rd, OT)
|
|
period = self.current_game.get("period", 1)
|
|
period_text = ""
|
|
if period == 1:
|
|
period_text = "1st"
|
|
elif period == 2:
|
|
period_text = "2nd"
|
|
elif period == 3:
|
|
period_text = "3rd"
|
|
elif period > 3:
|
|
period_text = "OT"
|
|
|
|
# Get clock time
|
|
clock = self.current_game.get("clock", "0:00")
|
|
|
|
# Combine period and clock
|
|
time_text = f"{period_text} {clock}"
|
|
|
|
# Calculate position for the time text (centered at the top)
|
|
time_width = draw.textlength(time_text, font=self.fonts['time'])
|
|
time_x = (self.display_width - time_width) // 2
|
|
time_y = 5
|
|
|
|
# Draw the time text
|
|
draw.text((time_x, time_y), time_text, font=self.fonts['time'], fill=(255, 255, 255))
|
|
|
|
# Display the image
|
|
self.display_manager.image.paste(img, (0, 0))
|
|
self.display_manager.update_display()
|
|
logging.debug("[NHL] Successfully displayed live game")
|
|
|
|
except Exception as e:
|
|
logging.error(f"[NHL] Error displaying live game: {e}", exc_info=True)
|
|
|
|
class NHLRecentManager(BaseNHLManager):
|
|
"""Manager for recently completed NHL games."""
|
|
def __init__(self, config: Dict[str, Any], display_manager: DisplayManager):
|
|
super().__init__(config, display_manager)
|
|
self.update_interval = self.nhl_config.get("recent_update_interval", 300) # 5 minutes
|
|
self.last_update = 0
|
|
self.logger.info("Initialized NHL Recent Manager")
|
|
self.recent_hours = self.nhl_config.get("recent_game_hours", 48) # Default 48 hours
|
|
self.current_game = None
|
|
|
|
if self.test_mode:
|
|
# Initialize with a test game
|
|
self.current_game = {
|
|
"home_abbr": "TBL",
|
|
"away_abbr": "DAL",
|
|
"home_score": "4",
|
|
"away_score": "2",
|
|
"status_text": "Final",
|
|
"home_logo_path": os.path.join(self.logo_dir, "TBL.png"),
|
|
"away_logo_path": os.path.join(self.logo_dir, "DAL.png"),
|
|
"game_time": "7:30 PM",
|
|
"game_date": "Apr 17"
|
|
}
|
|
logging.info("[NHL] Initialized NHLRecentManager with test game: TBL vs DAL")
|
|
else:
|
|
logging.info("[NHL] Initialized NHLRecentManager in live mode")
|
|
|
|
def update(self):
|
|
"""Update recent game data."""
|
|
current_time = time.time()
|
|
if current_time - self.last_update >= self.update_interval:
|
|
self.logger.debug("Updating recent game data")
|
|
self.last_update = current_time
|
|
if self.test_mode:
|
|
# In test mode, just keep the test game
|
|
pass
|
|
else:
|
|
# Fetch data for the last 48 hours
|
|
cutoff_time = datetime.now(timezone.utc) - timedelta(hours=self.recent_hours)
|
|
data = self._fetch_data()
|
|
|
|
if data and "events" in data:
|
|
# Find the most recent completed game involving favorite teams
|
|
most_recent_game = None
|
|
most_recent_time = None
|
|
|
|
for event in data["events"]:
|
|
details = self._extract_game_details(event)
|
|
if details and details["is_final"] and details["start_time_utc"]:
|
|
# Check if game is within our time window
|
|
if details["start_time_utc"] > cutoff_time:
|
|
# Check if it involves favorite teams (if any are configured)
|
|
if not self.favorite_teams or (
|
|
details["home_abbr"] in self.favorite_teams or
|
|
details["away_abbr"] in self.favorite_teams
|
|
):
|
|
# Keep the most recent game
|
|
if most_recent_time is None or details["start_time_utc"] > most_recent_time:
|
|
most_recent_game = details
|
|
most_recent_time = details["start_time_utc"]
|
|
|
|
self.current_game = most_recent_game
|
|
if most_recent_game:
|
|
logging.info(f"[NHL] Found recent game: {most_recent_game['away_abbr']} vs {most_recent_game['home_abbr']}")
|
|
else:
|
|
logging.info("[NHL] No recent games found")
|
|
|
|
def display(self, force_clear: bool = False):
|
|
"""Display recent game information."""
|
|
if not self.current_game:
|
|
logging.warning("[NHL] No recent game data available to display")
|
|
return
|
|
|
|
try:
|
|
# Create a new black image
|
|
img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
draw = ImageDraw.Draw(img)
|
|
|
|
# Load and resize logos
|
|
home_logo = self._load_and_resize_logo(self.current_game["home_abbr"])
|
|
away_logo = self._load_and_resize_logo(self.current_game["away_abbr"])
|
|
|
|
# Draw home team logo (right side)
|
|
if home_logo:
|
|
home_x = 3 * self.display_width // 4 - home_logo.width // 2
|
|
home_y = self.display_height // 4 - home_logo.height // 2
|
|
temp_img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
temp_draw = ImageDraw.Draw(temp_img)
|
|
temp_img.paste(home_logo, (home_x, home_y), home_logo)
|
|
img.paste(temp_img, (0, 0))
|
|
|
|
# Draw away team logo (left side)
|
|
if away_logo:
|
|
away_x = self.display_width // 4 - away_logo.width // 2
|
|
away_y = self.display_height // 4 - away_logo.height // 2
|
|
temp_img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
temp_draw = ImageDraw.Draw(temp_img)
|
|
temp_img.paste(away_logo, (away_x, away_y), away_logo)
|
|
img.paste(temp_img, (0, 0))
|
|
|
|
# Draw scores in the format "AWAY - HOME"
|
|
home_score = str(self.current_game["home_score"])
|
|
away_score = str(self.current_game["away_score"])
|
|
|
|
# Format the score string with a hyphen separator
|
|
score_text = f"{away_score} - {home_score}"
|
|
|
|
# Calculate position for the score text (centered at the bottom)
|
|
score_width = draw.textlength(score_text, font=self.fonts['score'])
|
|
score_x = (self.display_width - score_width) // 2
|
|
score_y = 3 * self.display_height // 4 - 10
|
|
|
|
# Draw the score text
|
|
draw.text((score_x, score_y), score_text, font=self.fonts['score'], fill=(255, 255, 255))
|
|
|
|
# Draw "FINAL" status
|
|
final_text = "FINAL"
|
|
|
|
# Calculate position for the final text (centered in the middle)
|
|
final_width = draw.textlength(final_text, font=self.fonts['status'])
|
|
final_x = (self.display_width - final_width) // 2
|
|
final_y = self.display_height // 2 - 5
|
|
|
|
# Draw the final text
|
|
draw.text((final_x, final_y), final_text, font=self.fonts['status'], fill=(255, 255, 255))
|
|
|
|
# Display the image
|
|
self.display_manager.image.paste(img, (0, 0))
|
|
self.display_manager.update_display()
|
|
logging.debug("[NHL] Successfully displayed recent game")
|
|
|
|
except Exception as e:
|
|
logging.error(f"[NHL] Error displaying recent game: {e}", exc_info=True)
|
|
|
|
class NHLUpcomingManager(BaseNHLManager):
|
|
"""Manager for upcoming NHL games."""
|
|
def __init__(self, config: Dict[str, Any], display_manager: DisplayManager):
|
|
super().__init__(config, display_manager)
|
|
self.update_interval = self.nhl_config.get("upcoming_update_interval", 300) # 5 minutes
|
|
self.last_update = 0
|
|
self.logger.info("Initialized NHL Upcoming Manager")
|
|
self.current_game = None
|
|
|
|
if self.test_mode:
|
|
# Initialize with a test game
|
|
self.current_game = {
|
|
"home_abbr": "TBL",
|
|
"away_abbr": "DAL",
|
|
"status_text": "7:30 PM ET",
|
|
"home_logo_path": os.path.join(self.logo_dir, "TBL.png"),
|
|
"away_logo_path": os.path.join(self.logo_dir, "DAL.png"),
|
|
"game_time": "7:30 PM",
|
|
"game_date": "Apr 17"
|
|
}
|
|
logging.info("[NHL] Initialized NHLUpcomingManager with test game: TBL vs DAL")
|
|
else:
|
|
logging.info("[NHL] Initialized NHLUpcomingManager in live mode")
|
|
|
|
def update(self):
|
|
"""Update upcoming game data."""
|
|
current_time = time.time()
|
|
if current_time - self.last_update >= self.update_interval:
|
|
self.logger.debug("Updating upcoming game data")
|
|
self.last_update = current_time
|
|
if self.test_mode:
|
|
# In test mode, just keep the test game
|
|
pass
|
|
else:
|
|
# Fetch today's and tomorrow's data
|
|
today = datetime.now(timezone.utc).date()
|
|
tomorrow = today + timedelta(days=1)
|
|
|
|
# Format dates for API (YYYYMMDD)
|
|
today_str = today.strftime('%Y%m%d')
|
|
tomorrow_str = tomorrow.strftime('%Y%m%d')
|
|
|
|
# Fetch data for both days
|
|
today_data = self._fetch_data(today_str)
|
|
tomorrow_data = self._fetch_data(tomorrow_str)
|
|
|
|
# Combine events from both days
|
|
all_events = []
|
|
if today_data and "events" in today_data:
|
|
all_events.extend(today_data["events"])
|
|
if tomorrow_data and "events" in tomorrow_data:
|
|
all_events.extend(tomorrow_data["events"])
|
|
|
|
# Find the next upcoming game involving favorite teams
|
|
next_game = None
|
|
next_game_time = None
|
|
|
|
for event in all_events:
|
|
details = self._extract_game_details(event)
|
|
if details and details["is_upcoming"] and details["start_time_utc"]:
|
|
# Check if it involves favorite teams (if any are configured)
|
|
if not self.favorite_teams or (
|
|
details["home_abbr"] in self.favorite_teams or
|
|
details["away_abbr"] in self.favorite_teams
|
|
):
|
|
# Keep the soonest upcoming game
|
|
if next_game_time is None or details["start_time_utc"] < next_game_time:
|
|
next_game = details
|
|
next_game_time = details["start_time_utc"]
|
|
|
|
self.current_game = next_game
|
|
if next_game:
|
|
logging.info(f"[NHL] Found upcoming game: {next_game['away_abbr']} vs {next_game['home_abbr']}")
|
|
else:
|
|
logging.info("[NHL] No upcoming games found")
|
|
|
|
def display(self, force_clear: bool = False):
|
|
"""Display upcoming game information."""
|
|
if not self.current_game:
|
|
logging.warning("[NHL] No upcoming game data available to display")
|
|
return
|
|
|
|
try:
|
|
# Create a new black image
|
|
img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
draw = ImageDraw.Draw(img)
|
|
|
|
# Load and resize logos
|
|
home_logo = self._load_and_resize_logo(self.current_game["home_abbr"])
|
|
away_logo = self._load_and_resize_logo(self.current_game["away_abbr"])
|
|
|
|
# Draw home team logo (right side)
|
|
if home_logo:
|
|
home_x = 3 * self.display_width // 4 - home_logo.width // 2
|
|
home_y = self.display_height // 4 - home_logo.height // 2
|
|
temp_img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
temp_draw = ImageDraw.Draw(temp_img)
|
|
temp_img.paste(home_logo, (home_x, home_y), home_logo)
|
|
img.paste(temp_img, (0, 0))
|
|
|
|
# Draw away team logo (left side)
|
|
if away_logo:
|
|
away_x = self.display_width // 4 - away_logo.width // 2
|
|
away_y = self.display_height // 4 - away_logo.height // 2
|
|
temp_img = Image.new('RGB', (self.display_width, self.display_height), 'black')
|
|
temp_draw = ImageDraw.Draw(temp_img)
|
|
temp_img.paste(away_logo, (away_x, away_y), away_logo)
|
|
img.paste(temp_img, (0, 0))
|
|
|
|
# Format the game time and date
|
|
game_time = self.current_game.get("game_time", "")
|
|
game_date = self.current_game.get("game_date", "")
|
|
|
|
# Format the time and date text
|
|
time_text = f"{game_time}"
|
|
date_text = f"{game_date}"
|
|
|
|
# Calculate position for the time text (centered at the bottom)
|
|
time_width = draw.textlength(time_text, font=self.fonts['score'])
|
|
time_x = (self.display_width - time_width) // 2
|
|
time_y = 3 * self.display_height // 4 - 10
|
|
|
|
# Draw the time text
|
|
draw.text((time_x, time_y), time_text, font=self.fonts['score'], fill=(255, 255, 255))
|
|
|
|
# Calculate position for the date text (centered in the middle)
|
|
date_width = draw.textlength(date_text, font=self.fonts['status'])
|
|
date_x = (self.display_width - date_width) // 2
|
|
date_y = self.display_height // 2 - 5
|
|
|
|
# Draw the date text
|
|
draw.text((date_x, date_y), date_text, font=self.fonts['status'], fill=(255, 255, 255))
|
|
|
|
# Display the image
|
|
self.display_manager.image.paste(img, (0, 0))
|
|
self.display_manager.update_display()
|
|
logging.debug("[NHL] Successfully displayed upcoming game")
|
|
|
|
except Exception as e:
|
|
logging.error(f"[NHL] Error displaying upcoming game: {e}", exc_info=True) |