LEDMatrix/test_stock_news_manager.py

#!/usr/bin/env python3
import time
import sys
import os
from src.config_manager import ConfigManager
from src.display_manager import DisplayManager
from src.stock_news_manager import StockNewsManager

print(f"Current working directory: {os.getcwd()}")

def main():
    """Test the StockNewsManager class directly."""
    display_manager = None
    try:
        # Load configuration
        config_manager = ConfigManager()
        config = config_manager.load_config()
        
        if not config:
            print("Error: Failed to load configuration")
            return
            
        display_config = config.get('display')
        if not display_config:
            print("Error: No display configuration found")
            return
        
        # Initialize display manager
        display_manager = DisplayManager(display_config)
        
        # Clear the display and show a test pattern
        display_manager.clear()
        display_manager.update_display()
        time.sleep(1)  # Give time to see the test pattern
        
        # Initialize news manager with the loaded config
        news_manager = StockNewsManager(config, display_manager)
        
        print("Testing news display. Press Ctrl+C to exit.")
        
        # Run the news display in a loop with proper timing
        last_update = time.time()
        while True:
            current_time = time.time()
            # Ensure we're not updating too frequently
            if current_time - last_update >= 0.001:  # 1ms minimum between updates
                news_manager.display_news()
                last_update = current_time
            
    except KeyboardInterrupt:
        print("\nTest interrupted by user")
    except Exception as e:
        print(f"Error during test: {e}")
        import traceback
        traceback.print_exc()
    finally:
        if display_manager:
            # Clear the display before exiting
            display_manager.clear()
            display_manager.update_display()
            display_manager.cleanup()
        print("Test completed")

if __name__ == "__main__":
    main()