X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=stock_renderer.py;h=75014d2cc23f2c48ed56a1e85366f5b13c67a4ce;hb=addd4980077f6e3857c5c035b49784dc3ceca49a;hp=2897edb7fb474b872ffa367096f50e19338bcb5f;hpb=6e31b94aea2ed2ba12de6b03f529ea2a018eb0a0;p=kiosk.git

diff --git a/stock_renderer.py b/stock_renderer.py
index 2897edb..75014d2 100644
--- a/stock_renderer.py
+++ b/stock_renderer.py
@@ -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,
         )
@@ -224,9 +236,9 @@ class stock_quote_renderer(renderer.abstaining_renderer):
 
 
 # Test
-# x = stock_quote_renderer(
-#    {},
-#    ["MSFT", "GOOG", "BTC-USD", "ABHYX", "GC=F", "VNQ"],
-#    {"BTC-USD": "BTC", "GC=F": "GOLD"},
-# )
-# x.periodic_render(None)
+x = stock_quote_renderer(
+    {},
+    ["MSFT", "GOOG", "BTC-USD", "ABHYX", "GC=F", "VNQ"],
+    {"BTC-USD": "BTC", "GC=F": "GOLD"},
+)
+x.periodic_render(None)