Changes ;)

[kiosk.git] / stock_renderer.py

diff --git a/stock_renderer.py b/stock_renderer.py
index 16273858ad3353e426f42ddfbe6b4670633adcb4..cfd2277a62d54c18a3eb27bef8212657e17c6a1e 100644 (file)
--- a/stock_renderer.py
+++ b/stock_renderer.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 
-from typing import Dict, List, Tuple
-import yfinance as yf
+from typing import Dict, List, Optional, Tuple
+import yfinance as yf  # type: ignore
 
 import file_writer
 import renderer
@@ -27,10 +27,12 @@ class stock_quote_renderer(renderer.debuggable_abstaining_renderer):
     def get_ticker_name(ticker: yf.ticker.Ticker) -> str:
         """Get friendly name of a ticker."""
         info = ticker.get_info()
-        return info["shortName"]
+        if "shortName" in info:
+            return info["shortName"]
+        return ticker
 
     @staticmethod
-    def get_price(ticker: yf.ticker.Ticker) -> float:
+    def get_price(ticker: yf.ticker.Ticker) -> Optional[float]:
         """Get most recent price of a ticker."""
         keys = [
             "bid",
@@ -53,8 +55,8 @@ class stock_quote_renderer(renderer.debuggable_abstaining_renderer):
     ) -> Tuple[float, float]:
         """Given the current price, look up opening price and compute delta."""
         keys = [
-            "open",
             "previousClose",
+            "open",
         ]
         info = ticker.get_info()
         for key in keys:
@@ -73,7 +75,6 @@ class stock_quote_renderer(renderer.debuggable_abstaining_renderer):
             symbols_finished = 0
             for symbol in self.symbols:
                 ticker = yf.Ticker(symbol)
-                print(type(ticker))
                 # print(ticker.get_info())
                 if ticker is None:
                     self.debug_print(f"Unknown symbol {symbol} -- ignored.")