mirror of
https://github.com/williamkray/maubot-ticker.git
synced 2024-06-26 14:32:17 +00:00
72 lines
2.7 KiB
Python
72 lines
2.7 KiB
Python
|
from typing import Optional, Type
|
||
|
|
|
||
|
|
from mautrix.util.config import BaseProxyConfig, ConfigUpdateHelper
|
||
|
|
from maubot import Plugin, MessageEvent
|
||
|
|
from maubot.handlers import command
|
||
|
|
|
||
|
|
import requests
|
||
|
|
import json
|
||
|
|
|
||
|
|
class Config(BaseProxyConfig):
|
||
|
|
def do_update(self, helper: ConfigUpdateHelper) -> None:
|
||
|
|
helper.copy("apiKey")
|
||
|
|
helper.copy("stocktrigger")
|
||
|
|
helper.copy("cryptotrigger")
|
||
|
|
|
||
|
|
class TickerBot(Plugin):
|
||
|
|
async def start(self) -> None:
|
||
|
|
await super().start()
|
||
|
|
self.config.load_and_update()
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def get_config_class(cls) -> Type[BaseProxyConfig]:
|
||
|
|
return Config
|
||
|
|
|
||
|
|
@command.new(name=lambda self: self.config["stocktrigger"],
|
||
|
|
help="Look up information about a stock by its ticker symbol")
|
||
|
|
@command.argument("ticker", pass_raw=True, required=True)
|
||
|
|
async def handler(self, evt: MessageEvent, ticker: str) -> None:
|
||
|
|
await evt.mark_read()
|
||
|
|
|
||
|
|
tickerUpper = ticker.upper()
|
||
|
|
url = "https://apidojo-yahoo-finance-v1.p.rapidapi.com/market/v2/get-quotes"
|
||
|
|
querystring = {"symbols":tickerUpper}
|
||
|
|
headers = {
|
||
|
|
'x-rapidapi-key': self.config["apiKey"],
|
||
|
|
'x-rapidapi-host': "apidojo-yahoo-finance-v1.p.rapidapi.com"
|
||
|
|
}
|
||
|
|
|
||
|
|
try:
|
||
|
|
response = requests.request("GET", url, headers=headers, params=querystring)
|
||
|
|
except Exception as e:
|
||
|
|
evt.respond(f"request failed: {e.message}")
|
||
|
|
return None
|
||
|
|
try:
|
||
|
|
jsonresponse = response.json()['quoteResponse']['result'][0]
|
||
|
|
openDifference = f"{jsonresponse['regularMarketPrice'] - jsonresponse['regularMarketOpen']:.2f}"
|
||
|
|
openPercentDiff = f"{(float(openDifference) / jsonresponse['regularMarketOpen'] * 100):.2f}%"
|
||
|
|
except Exception as e:
|
||
|
|
evt.respond("No results, double check that you've chosen a real ticker symbol")
|
||
|
|
self.log.exception(e)
|
||
|
|
return None
|
||
|
|
|
||
|
|
if float(openDifference) < 0:
|
||
|
|
color = "red"
|
||
|
|
arrow = "\U000025BC"
|
||
|
|
else:
|
||
|
|
color = "green"
|
||
|
|
arrow = "\U000025B2"
|
||
|
|
|
||
|
|
prettyMessage = "<br />".join(
|
||
|
|
[
|
||
|
|
f"<b>Current data for {tickerUpper}:</b>" ,
|
||
|
|
f"",
|
||
|
|
f"<b>Price:</b> <font color=\"{color}\">${str(jsonresponse['regularMarketPrice'])}, \
|
||
|
|
{arrow}{openPercentDiff}</font> from market open @ ${str(jsonresponse['regularMarketOpen'])}",
|
||
|
|
f"<b>52 Week High:</b> ${str(jsonresponse['fiftyTwoWeekHigh'])}",
|
||
|
|
f"<b>52 Week Low:</b> ${str(jsonresponse['fiftyTwoWeekLow'])}"
|
||
|
|
]
|
||
|
|
)
|
||
|
|
|
||
|
|
await evt.respond(prettyMessage, allow_html=True)
|