Fix stock, was assuming post market price would always be populated

This commit is contained in:
Luke Robles 2023-05-30 11:45:26 -07:00
parent 68fa959da0
commit 579a48fa65

View File

@ -8,7 +8,7 @@ def parse_message(symbols, verbose):
for s in symbols.split():
try:
embeds.append(get_stock(s, verbose=verbose))
except:
except Exception as e:
bad_tickers.append(s)
if bad_tickers:
embeds.append(_make_error_embed(bad_tickers))
@ -87,7 +87,7 @@ def get_stock(share_name, verbose=False):
share_name = share_name.upper()
# Fake headers to make yahoo happy
headers = {
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_5) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/50.0.2661.102 Safari/537.36"
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/113.0.0.0 Safari/537.36"
}
request_string = (
"https://query2.finance.yahoo.com/v11/finance/quoteSummary/%s?modules=price"
@ -119,7 +119,7 @@ def get_stock(share_name, verbose=False):
embed.set_thumbnail(url=meme_url)
embed.set_author(name=request["shortName"])
if "postMarketPrice" in request:
if "raw" in request["postMarketPrice"]:
embed.add_field(
name="After Hours price",
value="$%s" % request["postMarketPrice"]["raw"],