Changes.

[kiosk.git] / stock_renderer.py

diff --git a/stock_renderer.py b/stock_renderer.py
index 2897edb7fb474b872ffa367096f50e19338bcb5f..b13af908235908428ea1e0dc747fca0424624058 100644 (file)
--- a/stock_renderer.py
+++ b/stock_renderer.py
@@ -13,7 +13,7 @@ import kiosk_constants
 import renderer
 
 
-logger = logging.getLogger(__file__)
+logger = logging.getLogger(__name__)
 
 
 class stock_quote_renderer(renderer.abstaining_renderer):
@@ -134,12 +134,24 @@ class stock_quote_renderer(renderer.abstaining_renderer):
         )
 
         hist = ticker.history(period=period, interval="1wk")
+        if isinstance(ticker, yq.Ticker):
+            _open = "open"
+            _high = "high"
+            _low = "low"
+            _close = "adjclose"
+        elif isinstance(ticker, yf.Ticker):
+            _open = "Open"
+            _high = "High"
+            _low = "Low"
+            _close = "Close"
+        else:
+            raise Exception("Bad Ticker type")
         chart = go.Figure(
             data=go.Candlestick(
-                open=hist["Open"],
-                high=hist["High"],
-                low=hist["Low"],
-                close=hist["Close"],
+                open=hist[_open],
+                high=hist[_high],
+                low=hist[_low],
+                close=hist[_close],
             ),
             layout=transparent,
         )