From 70d01ad73b4b6a548d94f9f1a4d4a5da80b18316 Mon Sep 17 00:00:00 2001
From: Scott Gasch <scott@gasch.org>
Date: Tue, 21 Feb 2023 11:06:51 -0800
Subject: [PATCH] Changes to make history work.

---
 stock_renderer.py | 32 ++++++++++++++++++++++----------
 1 file changed, 22 insertions(+), 10 deletions(-)

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)
-- 
2.45.2