5 from typing import Any, Dict, List, Optional, Tuple
7 import yfinance as yf # type: ignore
8 import plotly.graph_objects as go
11 import kiosk_constants
15 logger = logging.getLogger(__file__)
18 class stock_quote_renderer(renderer.abstaining_renderer):
19 """Render the stock prices page."""
23 name_to_timeout_dict: Dict[str, int],
25 display_subs: Dict[str, str] = None,
27 super().__init__(name_to_timeout_dict)
28 self.symbols = symbols
29 self.display_subs = display_subs
30 self.info_cache: Dict[yf.ticker.Ticker, Dict] = {}
32 def cache_info(self, ticker: yf.ticker.Ticker) -> Dict:
33 if ticker in self.info_cache:
34 return self.info_cache[ticker]
36 self.info_cache[ticker] = i
39 def get_ticker_name(self, ticker: yf.Ticker) -> str:
40 """Get friendly name of a ticker."""
41 info = self.cache_info(ticker)
42 if "shortName" in info:
43 return info["shortName"]
47 def get_item_from_dict(keys: List[str], dictionary: Dict[str, Any]) -> Optional[Any]:
50 result = dictionary.get(key, None)
55 def get_price(self, ticker: yf.Ticker) -> Optional[float]:
56 """Get most recent price of a ticker."""
59 price = stock_quote_renderer.get_item_from_dict(
68 price = stock_quote_renderer.get_item_from_dict(
72 self.cache_info(ticker))
76 # Finally, fall back on history
77 hist = ticker.history(period="1d").to_dict()['Close']
80 for k, v in hist.items():
81 if latest is None or k > latest:
86 print(f"Price: fell back on latest close {latest_price} at {latest}")
90 def make_chart(symbol: str, ticker: yf.Ticker, period: str) -> str:
91 base_filename = f'stock_chart_{symbol}.png'
92 output_filename = os.path.join(kiosk_constants.pages_dir, base_filename)
93 transparent = go.Layout(
94 paper_bgcolor='rgba(0,0,0,0)',
95 plot_bgcolor='rgba(0,0,0,0)',
96 xaxis_rangeslider_visible=False,
98 hist = ticker.history(period=period, interval="1wk")
108 chart.update_xaxes(visible=False, showticklabels=False)
109 chart.update_yaxes(side="right")
110 chart.write_image(output_filename, format="png", width=600, height=350)
111 print(f"Write {output_filename}...")
118 last_close = stock_quote_renderer.get_item_from_dict(
125 last_close = stock_quote_renderer.get_item_from_dict(
127 self.cache_info(ticker))
130 return self.get_price(ticker)
132 def get_change_and_delta(
136 ) -> Tuple[float, float]:
137 """Given the current price, look up opening price and compute delta."""
138 last_price = self.get_last_close(ticker)
139 delta = price - last_price
140 return (delta / last_price * 100.0, delta)
142 def periodic_render(self, key: str) -> bool:
143 """Write an up-to-date stock page."""
144 with file_writer.file_writer("stock_3_86400.html") as f:
145 f.write("<H1>Stock Quotes</H1><HR>")
146 f.write("<TABLE WIDTH=99%>")
148 for symbol in self.symbols:
149 ticker = yf.Ticker(symbol)
151 logger.debug(f"Unknown symbol {symbol} -- ignored.")
153 name = self.get_ticker_name(ticker)
155 logger.debug(f'Bad name for {symbol} -- skipped.')
157 price = self.get_price(ticker)
159 logger.debug(f"No price information for {symbol} -- skipped.")
161 (percent_change, delta) = self.get_change_and_delta(
164 chart_filename = stock_quote_renderer.make_chart(symbol, ticker, "1y")
165 print(f"delta: {delta}, change: {percent_change}")
166 if percent_change < 0:
167 cell_color = "#b00000"
168 elif percent_change > 0:
169 cell_color = "#009000"
171 cell_color = "#a0a0a0"
172 if symbols_finished % 4 == 0:
173 if symbols_finished > 0:
176 symbols_finished += 1
177 if self.display_subs is not None and symbol in self.display_subs:
178 symbol = self.display_subs[symbol]
181 <TD WIDTH=20% HEIGHT=150 BGCOLOR="{cell_color}">
183 <DIV style="position:relative;
185 <!-- Symbol {symbol} -->
186 <DIV style="position:absolute;
189 -webkit-transform:rotate(-90deg);
191 font-family: helvetica, arial, sans-serif;
193 -webkit-text-stroke: 2px black;
197 <!-- Current price, Change today and percent change today, name -->
198 <DIV style="position:absolute;
202 font-family:helvetica, arial, sans-serif;
206 <I>({percent_change:.1f}%)</I><BR>
209 <IMG SRC="{chart_filename}" WIDTH=100%>
213 f.write("</TR></TABLE>")
217 x = stock_quote_renderer({}, ["MSFT", "GOOG", "BTC-USD", "ABHYX", "GC=F", "VNQ"], { "BTC-USD": "BTC", "GC=F": "GOLD" })
218 x.periodic_render(None)