fixed stock from skybot!

This commit is contained in:
Luke Rogers 2013-11-11 09:13:15 +13:00
parent 5e7d1e9c92
commit 12e2a445d7

View file

@ -1,48 +1,37 @@
import random
from util import hook, http from util import hook, http
import json
url = 'http://www.google.com/ig/api'
@hook.command @hook.command
def stock(inp): def stock(inp):
"""stock <symbol> -- Gets information about stock symbol <symbol>.""" """.stock <symbol> -- gets stock information"""
parsed = http.get_xml(url, stock=inp) url = ('http://query.yahooapis.com/v1/public/yql?format=json&'
'env=store%3A%2F%2Fdatatables.org%2Falltableswithkeys')
if len(parsed) != 1: parsed = http.get_json(url, q='select * from yahoo.finance.quote '
return "error getting stock info" 'where symbol in ("%s")' % inp) # heh, SQLI
# Stuff the results in a dict for easy string formatting quote = parsed['query']['results']['quote']
results = dict((el.tag, el.attrib['data'])
for el in parsed.xpath('//finance/*'))
# if we dont get a company name back, the symbol doesn't match a company # if we dont get a company name back, the symbol doesn't match a company
if not "company" in results: if quote['Change'] is None:
guess_data = json.loads(http.get("http://d.yimg.com/autoc.finance.yahoo.com/autoc", query=inp, return "unknown ticker symbol %s" % inp
callback="YAHOO.Finance.SymbolSuggest.ssCallback")[39:-1])
guess = guess_data['ResultSet']['Result'] change = float(quote['Change'])
if len(guess) > 0: price = float(quote['LastTradePriceOnly'])
guess = guess[0]["symbol"]
return stock(guess) if change < 0:
quote['color'] = "5"
else: else:
return "error: unable to get stock info for '{}'".format(inp) quote['color'] = "3"
if results['last'] == '0.00': quote['PercentChange'] = 100 * change / (price - change)
return "%(company)s - last known stock value was 0.00 %(currency)s" \
" as of %(trade_timestamp)s" % results
if results['change'][0] == '-': ret = "%(Name)s - %(LastTradePriceOnly)s " \
results['color'] = "5" "\x03%(color)s%(Change)s (%(PercentChange).2f%%)\x03 " \
else: "Day Range: %(DaysRange)s " \
results['color'] = "3" "MCAP: %(MarketCapitalization)s" % quote
ret = "%(company)s - %(last)s %(currency)s " \
"\x03%(color)s%(change)s (%(perc_change)s%%)\x03 " \
"as of %(trade_timestamp)s" % results
if results['delay'] != '0':
ret += " (delayed %s minutes)" % results['delay']
return ret return ret