87 lines
3.3 KiB
Python
87 lines
3.3 KiB
Python
from pyircbot.modulebase import ModuleBase, command
|
|
from pyircbot.modules.ModInfo import info
|
|
from threading import Thread
|
|
from time import sleep, time
|
|
import requests
|
|
import traceback
|
|
|
|
|
|
API_URL = "https://financialmodelingprep.com/api/v3/quotes/index?apikey={apikey}"
|
|
|
|
|
|
def bits(is_gain):
|
|
if is_gain:
|
|
return ("\x0303", "⬆", "+", )
|
|
return ("\x0304", "⬇", "-", )
|
|
|
|
|
|
class StockIndex(ModuleBase):
|
|
def __init__(self, bot, moduleName):
|
|
super().__init__(bot, moduleName)
|
|
self.session = requests.session()
|
|
self.updater = None
|
|
self.running = True
|
|
self.last_update = 0
|
|
self.start_cache_updater()
|
|
|
|
def start_cache_updater(self):
|
|
self.updater = Thread(target=self.cache_updater, daemon=True)
|
|
self.updater.start()
|
|
|
|
def ondisable(self):
|
|
self.running = False
|
|
self.updater.join()
|
|
|
|
def cache_updater(self):
|
|
while self.running:
|
|
try:
|
|
self.update_cache()
|
|
except:
|
|
traceback.print_exc()
|
|
delay = self.config.get("cache_update_interval", 600)
|
|
while self.running and delay > 0:
|
|
delay -= 1
|
|
sleep(1)
|
|
|
|
def update_cache(self):
|
|
data = self.session.get(API_URL.format(**self.config),
|
|
timeout=self.config.get("cache_update_timeout", 10)).json()
|
|
self.cache = {item["symbol"]: item for item in data}
|
|
self.last_update = time()
|
|
|
|
@info("djia", "get the current value of the DJIA", cmds=["djia"])
|
|
@command("djia", allow_private=True)
|
|
def cmd_djia(self, message, command):
|
|
self.send_quote("^DJI", "DJIA", message.replyto)
|
|
|
|
@info("nasdaq", "get the current value of the NASDAQ composite index", cmds=["nasdaq"])
|
|
@command("nasdaq", allow_private=True)
|
|
def cmd_nasdaq(self, message, command):
|
|
self.send_quote("^IXIC", "NASDAQ", message.replyto)
|
|
|
|
@info("vix", "get the current value of the vix/fear index", cmds=["vix"])
|
|
@command("vix", allow_private=True)
|
|
def cmd_vix(self, message, command):
|
|
self.send_quote("^VIX", "VIX", message.replyto)
|
|
|
|
def send_quote(self, key, symbol, to):
|
|
index = self.cache[key]
|
|
is_gain = index["price"] >= index["previousClose"]
|
|
color, arrow, plusmin = bits(is_gain)
|
|
|
|
change = float(index["price"]) - float(index["previousClose"])
|
|
percentchange = float(change) / float(index["previousClose"]) * 100
|
|
|
|
warn_thresh = self.config.get("warning_thresh", 1800)
|
|
|
|
warning = "" if time() - self.last_update < warn_thresh else " \x030(quote is out-of-date)"
|
|
|
|
self.bot.act_PRIVMSG(to, "{} ${:,.2f} {}{}{:,.2f} ({:.2f}%){}{}".format(symbol,
|
|
index["price"],
|
|
color,
|
|
plusmin,
|
|
change,
|
|
percentchange,
|
|
arrow,
|
|
warning))
|