This repository has been archived on 2023-04-13. You can view files and clone it, but cannot push or open issues or pull requests.
CloudBot/plugins/stock.py
2013-09-04 18:30:04 +08:00

49 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 '%s'" % 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