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

diff --git a/stock_renderer.py b/stock_renderer.py
index 2897edb..ae65648 100644
--- 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,
         )
@@ -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)