From e7161608ce8e7dd29b7f011ad9695e3e88e8aa3f Mon Sep 17 00:00:00 2001 From: ChuckBuilds <33324927+ChuckBuilds@users.noreply.github.com> Date: Fri, 11 Apr 2025 12:39:42 -0500 Subject: [PATCH] Add stock logo download and display functionality --- src/stock_manager.py | 118 ++++++++++++++++++++++++++++++++++++++++--- 1 file changed, 110 insertions(+), 8 deletions(-) diff --git a/src/stock_manager.py b/src/stock_manager.py index 4f0613c4..fc1d272b 100644 --- a/src/stock_manager.py +++ b/src/stock_manager.py @@ -10,6 +10,7 @@ import urllib.parse import re from PIL import Image, ImageDraw import numpy as np +import hashlib # Configure logging logging.basicConfig(level=logging.INFO) @@ -37,6 +38,10 @@ class StockManager: self.last_fps_log_time = time.time() self.frame_times = [] + # Create assets/stocks directory if it doesn't exist + self.logo_dir = os.path.join('assets', 'stocks') + os.makedirs(self.logo_dir, exist_ok=True) + 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' } @@ -290,6 +295,87 @@ class StockManager: else: logger.error("Failed to fetch data for any configured stocks") + def _download_stock_logo(self, symbol: str) -> str: + """Download and save stock logo for a given symbol. + + Args: + symbol: Stock symbol (e.g., 'AAPL', 'MSFT') + + Returns: + Path to the saved logo image, or None if download failed + """ + try: + # Create a filename based on the symbol + filename = f"{symbol.lower()}.png" + filepath = os.path.join(self.logo_dir, filename) + + # Check if we already have the logo + if os.path.exists(filepath): + return filepath + + # Try to find logo from various sources + # 1. Yahoo Finance + yahoo_url = f"https://logo.clearbit.com/{symbol.lower()}.com" + + # 2. Alternative source if Yahoo fails + alt_url = f"https://storage.googleapis.com/iex/api/logos/{symbol}.png" + + # Try Yahoo first + response = requests.get(yahoo_url, headers=self.headers, timeout=5) + if response.status_code == 200 and 'image' in response.headers.get('Content-Type', ''): + with open(filepath, 'wb') as f: + f.write(response.content) + logger.info(f"Downloaded logo for {symbol} from Yahoo") + return filepath + + # Try alternative source + response = requests.get(alt_url, headers=self.headers, timeout=5) + if response.status_code == 200 and 'image' in response.headers.get('Content-Type', ''): + with open(filepath, 'wb') as f: + f.write(response.content) + logger.info(f"Downloaded logo for {symbol} from alternative source") + return filepath + + logger.warning(f"Could not download logo for {symbol}") + return None + + except Exception as e: + logger.error(f"Error downloading logo for {symbol}: {str(e)}") + return None + + def _get_stock_logo(self, symbol: str) -> Image.Image: + """Get stock logo image, or create a text-based fallback. + + Args: + symbol: Stock symbol (e.g., 'AAPL', 'MSFT') + + Returns: + PIL Image of the logo or text-based fallback + """ + # Try to download the logo if we don't have it + logo_path = self._download_stock_logo(symbol) + + if logo_path and os.path.exists(logo_path): + try: + # Open and resize the logo + logo = Image.open(logo_path) + + # Convert to RGBA if not already + if logo.mode != 'RGBA': + logo = logo.convert('RGBA') + + # Resize to fit in the display (assuming square logo) + max_size = min(self.display_manager.matrix.width // 2, + self.display_manager.matrix.height // 2) + logo = logo.resize((max_size, max_size), Image.LANCZOS) + + return logo + except Exception as e: + logger.error(f"Error processing logo for {symbol}: {str(e)}") + + # Fallback to text-based logo + return None + def _create_stock_display(self, symbol: str, data: Dict[str, Any], width: int, height: int, scroll_position: int = 0) -> Image.Image: """Create a single stock display with scrolling animation.""" # Create a wider image for scrolling @@ -303,14 +389,30 @@ class StockManager: # Calculate center position for the main content center_x = width // 2 - # Draw large stock logo on the left - logo_text = symbol[:1].upper() # First letter of symbol - bbox = draw.textbbox((0, 0), logo_text, font=self.display_manager.regular_font) - logo_width = bbox[2] - bbox[0] - logo_height = bbox[3] - bbox[1] - logo_x = center_x - width // 3 - logo_width // 2 - logo_y = (height - logo_height) // 2 - draw.text((logo_x, logo_y), logo_text, font=self.display_manager.regular_font, fill=color) + # Try to get stock logo + logo = self._get_stock_logo(symbol) + + if logo: + # Draw the logo on the left + logo_x = center_x - width // 3 - logo.width // 2 + logo_y = (height - logo.height) // 2 + + # Create a new image with alpha channel for the logo + logo_bg = Image.new('RGBA', (scroll_width, height), (0, 0, 0, 0)) + logo_bg.paste(logo, (logo_x, logo_y)) + + # Convert to RGB for the main image + logo_rgb = logo_bg.convert('RGB') + image.paste(logo_rgb, (0, 0)) + else: + # Fallback to text-based logo + logo_text = symbol[:1].upper() # First letter of symbol + bbox = draw.textbbox((0, 0), logo_text, font=self.display_manager.regular_font) + logo_width = bbox[2] - bbox[0] + logo_height = bbox[3] - bbox[1] + logo_x = center_x - width // 3 - logo_width // 2 + logo_y = (height - logo_height) // 2 + draw.text((logo_x, logo_y), logo_text, font=self.display_manager.regular_font, fill=color) # Draw stacked symbol, price, and change in the center # Symbol (always white)