48 lines
1.5 KiB
Python
Executable file
48 lines
1.5 KiB
Python
Executable file
from util import hook, http
|
|
|
|
import json
|
|
|
|
url = 'http://www.google.com/ig/api'
|
|
|
|
|
|
@hook.command
|
|
def stock(inp):
|
|
"""stock <symbol> -- Gets information about stock symbol <symbol>."""
|
|
|
|
parsed = http.get_xml(url, stock=inp)
|
|
|
|
if len(parsed) != 1:
|
|
return "error getting stock info"
|
|
|
|
# Stuff the results in a dict for easy string formatting
|
|
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 not "company" in results:
|
|
guess_data = json.loads(http.get("http://d.yimg.com/autoc.finance.yahoo.com/autoc", query=inp,
|
|
callback="YAHOO.Finance.SymbolSuggest.ssCallback")[39:-1])
|
|
guess = guess_data['ResultSet']['Result']
|
|
if len(guess) > 0:
|
|
guess = guess[0]["symbol"]
|
|
return stock(guess)
|
|
else:
|
|
return "error: unable to get stock info for '{}'".format(inp)
|
|
|
|
if results['last'] == '0.00':
|
|
return "%(company)s - last known stock value was 0.00 %(currency)s" \
|
|
" as of %(trade_timestamp)s" % results
|
|
|
|
if results['change'][0] == '-':
|
|
results['color'] = "5"
|
|
else:
|
|
results['color'] = "3"
|
|
|
|
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
|