ldooks/dragon-bot
ldooks/dragon-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Luke Robles 2023-05-30 11:45:26 -07:00
parent 68fa959da0
commit 579a48fa65
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/stock.py
View File

@ -8,7 +8,7 @@ def parse_message(symbols, verbose):
for s in symbols.split(): for s in symbols.split():
try: try:
embeds.append(get_stock(s, verbose=verbose)) embeds.append(get_stock(s, verbose=verbose))
except: except Exception as e:
bad_tickers.append(s) bad_tickers.append(s)
if bad_tickers: if bad_tickers:
embeds.append(_make_error_embed(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() share_name = share_name.upper()
# Fake headers to make yahoo happy # Fake headers to make yahoo happy
headers = { 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 = ( request_string = (
"https://query2.finance.yahoo.com/v11/finance/quoteSummary/%s?modules=price" "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_thumbnail(url=meme_url)
embed.set_author(name=request["shortName"]) embed.set_author(name=request["shortName"])
if "postMarketPrice" in request: if "raw" in request["postMarketPrice"]:
embed.add_field( embed.add_field(
name="After Hours price", name="After Hours price",
value="$%s" % request["postMarketPrice"]["raw"], value="$%s" % request["postMarketPrice"]["raw"],