h/plugins/stock.py

47 lines
1.5 KiB
Python

from util import hook, http
@hook.command
def stock(inp):
'''.stock <symbol> -- gets stock information'''
url = ('http://query.yahooapis.com/v1/public/yql?format=json&'
'env=store%3A%2F%2Fdatatables.org%2Falltableswithkeys')
parsed = http.get_json(url, q='select * from yahoo.finance.quotes '
'where symbol in ("%s")' % inp) # heh, SQLI
quote = parsed['query']['results']['quote']
# if we dont get a company name back, the symbol doesn't match a company
if quote['Change'] is None:
return "unknown ticker symbol %s" % inp
price = float(quote['LastTradePriceOnly'])
change = float(quote['Change'])
if quote['Open'] and quote['Bid'] and quote['Ask']:
open_price = float(quote['Open'])
bid = float(quote['Bid'])
ask = float(quote['Ask'])
if price < bid:
price = bid
elif price > ask:
price = ask
change = price - open_price
quote['LastTradePriceOnly'] = "%.2f" % price
quote['Change'] = ("+%.2f" % change) if change >= 0 else change
if change < 0:
quote['color'] = "5"
else:
quote['color'] = "3"
quote['PercentChange'] = 100 * change / (price - change)
ret = "%(Name)s - %(LastTradePriceOnly)s " \
"\x03%(color)s%(Change)s (%(PercentChange).2f%%)\x03 " \
"Day Range: %(DaysRange)s " \
"MCAP: %(MarketCapitalization)s" % quote
return ret