mirror of
https://github.com/ChuckBuilds/LEDMatrix.git
synced 2026-04-10 13:02:59 +00:00
new name extraction for broadcast names
This commit is contained in:
Chuck
@@ -38,6 +38,7 @@ class OddsTickerManager:
|
||||
"FS2": "fs2",
|
||||
"MLBN": "mlbn",
|
||||
"MLB Network": "mlbn",
|
||||
"MLB.TV": "mlbn",
|
||||
"NBC": "nbc",
|
||||
"NFLN": "nfln",
|
||||
"NFL Network": "nfln",
|
||||
@@ -51,7 +52,18 @@ class OddsTickerManager:
|
||||
"Paramount+": "cbs",
|
||||
"Hulu": "espn",
|
||||
"Disney+": "espn",
|
||||
"Apple TV+": "nbc"
|
||||
"Apple TV+": "nbc",
|
||||
# Regional sports networks
|
||||
"MASN": "cbs",
|
||||
"MASN2": "cbs",
|
||||
"MAS+": "cbs",
|
||||
"SportsNet": "nbc",
|
||||
"FanDuel SN": "fox",
|
||||
"FanDuel SN DET": "fox",
|
||||
"FanDuel SN FL": "fox",
|
||||
"SportsNet PIT": "nbc",
|
||||
"Padres.TV": "espn",
|
||||
"CLEGuardians.TV": "espn"
|
||||
}
|
||||
|
||||
def __init__(self, config: Dict[str, Any], display_manager: DisplayManager):
|
||||
@@ -374,7 +386,21 @@ class OddsTickerManager:
|
||||
broadcast_info = []
|
||||
broadcasts = event.get('competitions', [{}])[0].get('broadcasts', [])
|
||||
if broadcasts:
|
||||
broadcast_info = [b.get('media', {}).get('shortName', '') for b in broadcasts if b.get('media', {}).get('shortName')]
|
||||
# Handle new ESPN API format where broadcast names are in 'names' array
|
||||
for broadcast in broadcasts:
|
||||
if 'names' in broadcast:
|
||||
# New format: broadcast names are in 'names' array
|
||||
broadcast_names = broadcast.get('names', [])
|
||||
broadcast_info.extend(broadcast_names)
|
||||
elif 'media' in broadcast and 'shortName' in broadcast['media']:
|
||||
# Old format: broadcast name is in media.shortName
|
||||
short_name = broadcast['media']['shortName']
|
||||
if short_name:
|
||||
broadcast_info.append(short_name)
|
||||
|
||||
# Remove duplicates and filter out empty strings
|
||||
broadcast_info = list(set([name for name in broadcast_info if name]))
|
||||
|
||||
logger.info(f"Found broadcast channels for game {game_id}: {broadcast_info}")
|
||||
logger.debug(f"Raw broadcasts data for game {game_id}: {broadcasts}")
|
||||
# Log the first broadcast structure for debugging
|
||||
|
||||
110
test_new_broadcast_format.py
Normal file
110
test_new_broadcast_format.py
Normal file
@@ -0,0 +1,110 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
Test script for the new broadcast extraction logic
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
# Add the src directory to the path
|
||||
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'src'))
|
||||
|
||||
from odds_ticker_manager import OddsTickerManager
|
||||
from config_manager import ConfigManager
|
||||
|
||||
def test_broadcast_extraction():
|
||||
"""Test the new broadcast extraction logic"""
|
||||
|
||||
# Load config
|
||||
config_manager = ConfigManager()
|
||||
config = config_manager.load_config()
|
||||
|
||||
# Create a mock display manager
|
||||
class MockDisplayManager:
|
||||
def __init__(self):
|
||||
self.matrix = type('Matrix', (), {'width': 64, 'height': 32})()
|
||||
self.image = None
|
||||
self.draw = None
|
||||
|
||||
def update_display(self):
|
||||
pass
|
||||
|
||||
display_manager = MockDisplayManager()
|
||||
|
||||
# Create odds ticker manager
|
||||
odds_ticker = OddsTickerManager(config, display_manager)
|
||||
|
||||
# Test the broadcast extraction logic with sample data from the API
|
||||
test_broadcasts = [
|
||||
# Sample from the API response
|
||||
[
|
||||
{'market': 'away', 'names': ['MLB.TV', 'MAS+', 'MASN2']},
|
||||
{'market': 'home', 'names': ['CLEGuardians.TV']}
|
||||
],
|
||||
[
|
||||
{'market': 'away', 'names': ['MLB.TV', 'FanDuel SN DET']},
|
||||
{'market': 'home', 'names': ['SportsNet PIT']}
|
||||
],
|
||||
[
|
||||
{'market': 'away', 'names': ['MLB.TV', 'Padres.TV']},
|
||||
{'market': 'home', 'names': ['FanDuel SN FL']}
|
||||
],
|
||||
# Test with old format too
|
||||
[
|
||||
{'media': {'shortName': 'ESPN'}},
|
||||
{'media': {'shortName': 'FOX'}}
|
||||
]
|
||||
]
|
||||
|
||||
for i, broadcasts in enumerate(test_broadcasts):
|
||||
print(f"\n--- Test Case {i+1} ---")
|
||||
print(f"Input broadcasts: {broadcasts}")
|
||||
|
||||
# Simulate the extraction logic
|
||||
broadcast_info = []
|
||||
for broadcast in broadcasts:
|
||||
if 'names' in broadcast:
|
||||
# New format: broadcast names are in 'names' array
|
||||
broadcast_names = broadcast.get('names', [])
|
||||
broadcast_info.extend(broadcast_names)
|
||||
elif 'media' in broadcast and 'shortName' in broadcast['media']:
|
||||
# Old format: broadcast name is in media.shortName
|
||||
short_name = broadcast['media']['shortName']
|
||||
if short_name:
|
||||
broadcast_info.append(short_name)
|
||||
|
||||
# Remove duplicates and filter out empty strings
|
||||
broadcast_info = list(set([name for name in broadcast_info if name]))
|
||||
|
||||
print(f"Extracted broadcast info: {broadcast_info}")
|
||||
|
||||
# Test logo mapping
|
||||
if broadcast_info:
|
||||
logo_name = None
|
||||
sorted_keys = sorted(odds_ticker.BROADCAST_LOGO_MAP.keys(), key=len, reverse=True)
|
||||
|
||||
for b_name in broadcast_info:
|
||||
for key in sorted_keys:
|
||||
if key in b_name:
|
||||
logo_name = odds_ticker.BROADCAST_LOGO_MAP[key]
|
||||
print(f" Matched '{key}' to '{logo_name}' for '{b_name}'")
|
||||
break
|
||||
if logo_name:
|
||||
break
|
||||
|
||||
print(f" Final mapped logo: '{logo_name}'")
|
||||
|
||||
if logo_name:
|
||||
logo_path = os.path.join('assets', 'broadcast_logos', f"{logo_name}.png")
|
||||
print(f" Logo file exists: {os.path.exists(logo_path)}")
|
||||
else:
|
||||
print(" No broadcast info extracted")
|
||||
|
||||
if __name__ == "__main__":
|
||||
print("Testing New Broadcast Extraction Logic")
|
||||
print("=" * 50)
|
||||
|
||||
test_broadcast_extraction()
|
||||
|
||||
print("\n" + "=" * 50)
|
||||
print("Test complete. Check if the broadcast extraction and mapping works correctly.")
|
||||
Reference in New Issue
Block a user