mirror of
https://github.com/ChuckBuilds/LEDMatrix.git
synced 2026-04-13 22:13:00 +00:00
Stock News
Stock news Ticker
This commit is contained in:
@@ -30,7 +30,8 @@
|
|||||||
"weather": 15,
|
"weather": 15,
|
||||||
"stocks": 45,
|
"stocks": 45,
|
||||||
"hourly_forecast": 15,
|
"hourly_forecast": 15,
|
||||||
"daily_forecast": 15
|
"daily_forecast": 15,
|
||||||
|
"news": 30
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"clock": {
|
"clock": {
|
||||||
@@ -49,5 +50,12 @@
|
|||||||
"ASTS", "SCHD", "INTC", "NVDA", "T", "VOO", "SPYG", "SMCI"
|
"ASTS", "SCHD", "INTC", "NVDA", "T", "VOO", "SPYG", "SMCI"
|
||||||
],
|
],
|
||||||
"display_format": "{symbol}: ${price} ({change}%)"
|
"display_format": "{symbol}: ${price} ({change}%)"
|
||||||
|
},
|
||||||
|
"news": {
|
||||||
|
"enabled": true,
|
||||||
|
"update_interval": 300,
|
||||||
|
"scroll_speed": 1,
|
||||||
|
"scroll_delay": 0.05,
|
||||||
|
"max_headlines_per_symbol": 1
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
51
integrate_news_ticker.py
Normal file
51
integrate_news_ticker.py
Normal file
@@ -0,0 +1,51 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
import time
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
from src.config_manager import ConfigManager
|
||||||
|
from src.display_manager import DisplayManager
|
||||||
|
from src.news_manager import NewsManager
|
||||||
|
from src.stock_manager import StockManager
|
||||||
|
|
||||||
|
def main():
|
||||||
|
"""Integrate news ticker with the existing display system."""
|
||||||
|
try:
|
||||||
|
# Load configuration
|
||||||
|
config_manager = ConfigManager()
|
||||||
|
config = config_manager.config
|
||||||
|
|
||||||
|
# Initialize display manager
|
||||||
|
display_manager = DisplayManager(config.get('display', {}))
|
||||||
|
|
||||||
|
# Initialize stock manager
|
||||||
|
stock_manager = StockManager(config, display_manager)
|
||||||
|
|
||||||
|
# Initialize news manager
|
||||||
|
news_manager = NewsManager(config, display_manager)
|
||||||
|
|
||||||
|
print("News ticker integration test started. Press Ctrl+C to exit.")
|
||||||
|
print("Displaying stock data and news headlines...")
|
||||||
|
|
||||||
|
# Display stock data and news headlines in a loop
|
||||||
|
while True:
|
||||||
|
# Display stock data
|
||||||
|
stock_manager.display_stocks()
|
||||||
|
|
||||||
|
# Display news headlines for a limited time (30 seconds)
|
||||||
|
start_time = time.time()
|
||||||
|
while time.time() - start_time < 30:
|
||||||
|
news_manager.display_news()
|
||||||
|
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
print("\nTest interrupted by user.")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Error: {e}")
|
||||||
|
finally:
|
||||||
|
# Clean up
|
||||||
|
if 'display_manager' in locals():
|
||||||
|
display_manager.clear()
|
||||||
|
display_manager.update_display()
|
||||||
|
display_manager.cleanup()
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
174
src/news_manager.py
Normal file
174
src/news_manager.py
Normal file
@@ -0,0 +1,174 @@
|
|||||||
|
import time
|
||||||
|
import logging
|
||||||
|
import requests
|
||||||
|
import json
|
||||||
|
import random
|
||||||
|
from typing import Dict, Any, List, Tuple
|
||||||
|
from datetime import datetime
|
||||||
|
import os
|
||||||
|
import urllib.parse
|
||||||
|
import re
|
||||||
|
from src.config_manager import ConfigManager
|
||||||
|
|
||||||
|
# Configure logging
|
||||||
|
logging.basicConfig(level=logging.INFO)
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
class NewsManager:
|
||||||
|
def __init__(self, config: Dict[str, Any], display_manager):
|
||||||
|
self.config = config
|
||||||
|
self.config_manager = ConfigManager()
|
||||||
|
self.display_manager = display_manager
|
||||||
|
self.stocks_config = config.get('stocks', {})
|
||||||
|
self.news_config = config.get('news', {})
|
||||||
|
self.last_update = 0
|
||||||
|
self.news_data = {}
|
||||||
|
self.current_news_index = 0
|
||||||
|
self.scroll_position = 0
|
||||||
|
self.scroll_speed = self.news_config.get('scroll_speed', 1) # Pixels to move per frame
|
||||||
|
self.scroll_delay = self.news_config.get('scroll_delay', 0.05) # Delay between scroll updates
|
||||||
|
self.headers = {
|
||||||
|
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36'
|
||||||
|
}
|
||||||
|
# Initialize with first update
|
||||||
|
self.update_news_data()
|
||||||
|
|
||||||
|
def _fetch_news_for_symbol(self, symbol: str) -> List[Dict[str, Any]]:
|
||||||
|
"""Fetch news headlines for a stock symbol."""
|
||||||
|
try:
|
||||||
|
# Using Yahoo Finance API to get news
|
||||||
|
encoded_symbol = urllib.parse.quote(symbol)
|
||||||
|
url = f"https://query1.finance.yahoo.com/v1/finance/search?q={encoded_symbol}&lang=en-US®ion=US"esCount=0&newsCount={self.news_config.get('max_headlines_per_symbol', 5)}"
|
||||||
|
|
||||||
|
response = requests.get(url, headers=self.headers, timeout=5)
|
||||||
|
if response.status_code != 200:
|
||||||
|
logger.error(f"Failed to fetch news for {symbol}: HTTP {response.status_code}")
|
||||||
|
return []
|
||||||
|
|
||||||
|
data = response.json()
|
||||||
|
news_items = data.get('news', [])
|
||||||
|
|
||||||
|
# Process and format news items
|
||||||
|
formatted_news = []
|
||||||
|
for item in news_items:
|
||||||
|
formatted_news.append({
|
||||||
|
"title": item.get('title', ''),
|
||||||
|
"publisher": item.get('publisher', ''),
|
||||||
|
"link": item.get('link', ''),
|
||||||
|
"published": item.get('providerPublishTime', 0)
|
||||||
|
})
|
||||||
|
|
||||||
|
logger.info(f"Fetched {len(formatted_news)} news items for {symbol}")
|
||||||
|
return formatted_news
|
||||||
|
|
||||||
|
except requests.exceptions.RequestException as e:
|
||||||
|
logger.error(f"Network error fetching news for {symbol}: {e}")
|
||||||
|
return []
|
||||||
|
except (ValueError, IndexError, KeyError) as e:
|
||||||
|
logger.error(f"Error parsing news data for {symbol}: {e}")
|
||||||
|
return []
|
||||||
|
except Exception as e:
|
||||||
|
logger.error(f"Unexpected error fetching news for {symbol}: {e}")
|
||||||
|
return []
|
||||||
|
|
||||||
|
def update_news_data(self):
|
||||||
|
"""Update news data for all configured stock symbols."""
|
||||||
|
current_time = time.time()
|
||||||
|
update_interval = self.news_config.get('update_interval', 300) # Default to 5 minutes
|
||||||
|
|
||||||
|
# If not enough time has passed, keep using existing data
|
||||||
|
if current_time - self.last_update < update_interval:
|
||||||
|
return
|
||||||
|
|
||||||
|
# Get symbols from config
|
||||||
|
symbols = self.stocks_config.get('symbols', [])
|
||||||
|
if not symbols:
|
||||||
|
logger.warning("No stock symbols configured for news")
|
||||||
|
return
|
||||||
|
|
||||||
|
# Create temporary storage for new data
|
||||||
|
new_data = {}
|
||||||
|
success = False
|
||||||
|
|
||||||
|
for symbol in symbols:
|
||||||
|
# Add a small delay between requests to avoid rate limiting
|
||||||
|
time.sleep(random.uniform(0.1, 0.3))
|
||||||
|
news_items = self._fetch_news_for_symbol(symbol)
|
||||||
|
if news_items:
|
||||||
|
new_data[symbol] = news_items
|
||||||
|
success = True
|
||||||
|
|
||||||
|
if success:
|
||||||
|
# Only update the displayed data when we have new data
|
||||||
|
self.news_data = new_data
|
||||||
|
self.last_update = current_time
|
||||||
|
self.current_news_index = 0
|
||||||
|
self.scroll_position = 0
|
||||||
|
logger.info(f"Updated news data for {len(new_data)} symbols")
|
||||||
|
else:
|
||||||
|
logger.error("Failed to fetch news for any configured stocks")
|
||||||
|
|
||||||
|
def display_news(self):
|
||||||
|
"""Display news headlines by scrolling them across the screen."""
|
||||||
|
if not self.news_config.get('enabled', False):
|
||||||
|
return
|
||||||
|
|
||||||
|
# Start update in background if needed
|
||||||
|
if time.time() - self.last_update >= self.news_config.get('update_interval', 300):
|
||||||
|
self.update_news_data()
|
||||||
|
|
||||||
|
if not self.news_data:
|
||||||
|
logger.warning("No news data available to display")
|
||||||
|
return
|
||||||
|
|
||||||
|
# Get all news items from all symbols
|
||||||
|
all_news = []
|
||||||
|
for symbol, news_items in self.news_data.items():
|
||||||
|
for item in news_items:
|
||||||
|
all_news.append({
|
||||||
|
"symbol": symbol,
|
||||||
|
"title": item["title"],
|
||||||
|
"publisher": item["publisher"]
|
||||||
|
})
|
||||||
|
|
||||||
|
if not all_news:
|
||||||
|
return
|
||||||
|
|
||||||
|
# Get the current news item to display
|
||||||
|
current_news = all_news[self.current_news_index]
|
||||||
|
|
||||||
|
# Format the news text
|
||||||
|
news_text = f"{current_news['symbol']}: {current_news['title']}"
|
||||||
|
|
||||||
|
# Get text dimensions
|
||||||
|
bbox = self.display_manager.draw.textbbox((0, 0), news_text, font=self.display_manager.small_font)
|
||||||
|
text_width = bbox[2] - bbox[0]
|
||||||
|
|
||||||
|
# Clear the display
|
||||||
|
self.display_manager.clear()
|
||||||
|
|
||||||
|
# Draw the news text at the current scroll position
|
||||||
|
self.display_manager.draw_text(
|
||||||
|
news_text,
|
||||||
|
x=self.display_manager.matrix.width - self.scroll_position,
|
||||||
|
y=16, # Center vertically
|
||||||
|
color=(255, 255, 255), # White
|
||||||
|
small_font=True
|
||||||
|
)
|
||||||
|
|
||||||
|
# Update the display
|
||||||
|
self.display_manager.update_display()
|
||||||
|
|
||||||
|
# Update scroll position
|
||||||
|
self.scroll_position += self.scroll_speed
|
||||||
|
|
||||||
|
# If we've scrolled past the end of the text, move to the next news item
|
||||||
|
if self.scroll_position > text_width + self.display_manager.matrix.width:
|
||||||
|
self.scroll_position = 0
|
||||||
|
self.current_news_index = (self.current_news_index + 1) % len(all_news)
|
||||||
|
|
||||||
|
# Add a small delay to control scroll speed
|
||||||
|
time.sleep(self.scroll_delay)
|
||||||
|
|
||||||
|
# Return True if we've displayed all news items
|
||||||
|
return self.current_news_index == 0 and self.scroll_position == 0
|
||||||
44
test_news_ticker.py
Normal file
44
test_news_ticker.py
Normal file
@@ -0,0 +1,44 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
import time
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
from src.config_manager import ConfigManager
|
||||||
|
from src.display_manager import DisplayManager
|
||||||
|
from src.news_manager import NewsManager
|
||||||
|
|
||||||
|
def main():
|
||||||
|
"""Test the news ticker functionality."""
|
||||||
|
try:
|
||||||
|
# Load configuration
|
||||||
|
config_manager = ConfigManager()
|
||||||
|
config = config_manager.config
|
||||||
|
|
||||||
|
# Initialize display manager
|
||||||
|
display_manager = DisplayManager(config.get('display', {}))
|
||||||
|
|
||||||
|
# Initialize news manager
|
||||||
|
news_manager = NewsManager(config, display_manager)
|
||||||
|
|
||||||
|
print("News ticker test started. Press Ctrl+C to exit.")
|
||||||
|
print("Displaying news headlines for configured stock symbols...")
|
||||||
|
|
||||||
|
# Display news headlines for a limited time (30 seconds)
|
||||||
|
start_time = time.time()
|
||||||
|
while time.time() - start_time < 30:
|
||||||
|
news_manager.display_news()
|
||||||
|
|
||||||
|
print("Test completed successfully.")
|
||||||
|
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
print("\nTest interrupted by user.")
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Error: {e}")
|
||||||
|
finally:
|
||||||
|
# Clean up
|
||||||
|
if 'display_manager' in locals():
|
||||||
|
display_manager.clear()
|
||||||
|
display_manager.update_display()
|
||||||
|
display_manager.cleanup()
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
Reference in New Issue
Block a user