mirror of
https://gitlab.com/simple-stock-bots/simple-discord-stock-bot.git
synced 2025-06-16 07:16:41 +00:00
parity with telegram merge request 26
This commit is contained in:
parent
c6db51a5cc
commit
a7fb09e341
151
IEX_Symbol.py
151
IEX_Symbol.py
@ -1,6 +1,7 @@
|
||||
"""Class with functions for running the bot with IEX Cloud.
|
||||
"""
|
||||
|
||||
import logging
|
||||
import os
|
||||
from datetime import datetime
|
||||
from logging import warning
|
||||
@ -47,6 +48,30 @@ class IEX_Symbol:
|
||||
schedule.every().day.do(self.get_symbol_list)
|
||||
schedule.every().day.do(self.clear_charts)
|
||||
|
||||
def get(self, endpoint, params: dict = {}, timeout=5) -> dict:
|
||||
|
||||
url = "https://cloud.iexapis.com/stable" + endpoint
|
||||
|
||||
# set token param if it wasn't passed.
|
||||
params["token"] = params.get("token", self.IEX_TOKEN)
|
||||
|
||||
resp = r.get(url, params=params, timeout=timeout)
|
||||
|
||||
# Make sure API returned a proper status code
|
||||
try:
|
||||
resp.raise_for_status()
|
||||
except r.exceptions.HTTPError as e:
|
||||
logging.error(e)
|
||||
return {}
|
||||
|
||||
# Make sure API returned valid JSON
|
||||
try:
|
||||
resp_json = resp.json()
|
||||
return resp_json
|
||||
except r.exceptions.JSONDecodeError as e:
|
||||
logging.error(e)
|
||||
return {}
|
||||
|
||||
def clear_charts(self) -> None:
|
||||
"""
|
||||
Clears cache of chart data.
|
||||
@ -71,14 +96,8 @@ class IEX_Symbol:
|
||||
If `return_df` is set to `True` returns a dataframe, otherwise returns `None`.
|
||||
"""
|
||||
|
||||
reg_symbols = r.get(
|
||||
f"https://cloud.iexapis.com/stable/ref-data/symbols?token={self.IEX_TOKEN}",
|
||||
timeout=5,
|
||||
).json()
|
||||
otc_symbols = r.get(
|
||||
f"https://cloud.iexapis.com/stable/ref-data/otc/symbols?token={self.IEX_TOKEN}",
|
||||
timeout=5,
|
||||
).json()
|
||||
reg_symbols = self.get("/ref-data/symbols")
|
||||
otc_symbols = self.get("/ref-data/otc/symbols")
|
||||
|
||||
reg = pd.DataFrame(data=reg_symbols)
|
||||
otc = pd.DataFrame(data=otc_symbols)
|
||||
@ -105,7 +124,7 @@ class IEX_Symbol:
|
||||
"""
|
||||
resp = r.get(
|
||||
"https://pjmps0c34hp7.statuspage.io/api/v2/status.json",
|
||||
timeout=5,
|
||||
timeout=15,
|
||||
)
|
||||
|
||||
if resp.status_code == 200:
|
||||
@ -174,14 +193,7 @@ class IEX_Symbol:
|
||||
Formatted markdown
|
||||
"""
|
||||
|
||||
IEXurl = f"https://cloud.iexapis.com/stable/stock/{symbol.id}/quote?token={self.IEX_TOKEN}"
|
||||
|
||||
response = r.get(
|
||||
IEXurl,
|
||||
timeout=5,
|
||||
)
|
||||
if response.status_code == 200:
|
||||
IEXData = response.json()
|
||||
if IEXData := self.get(f"/stock/{symbol.id}/quote"):
|
||||
|
||||
if symbol.symbol.upper() in self.otc_list:
|
||||
return f"OTC - {symbol.symbol.upper()}, {IEXData['companyName']} most recent price is: $**{IEXData['latestPrice']}**"
|
||||
@ -249,13 +261,11 @@ class IEX_Symbol:
|
||||
if symbol.symbol.upper() in self.otc_list:
|
||||
return "OTC stocks do not currently support any commands."
|
||||
|
||||
IEXurl = f"https://cloud.iexapis.com/stable/stock/{symbol}/dividends/next?token={self.IEX_TOKEN}"
|
||||
response = r.get(
|
||||
IEXurl,
|
||||
timeout=5,
|
||||
)
|
||||
if response.status_code == 200 and response.json():
|
||||
IEXData = response.json()[0]
|
||||
if resp := self.get(f"/stock/{symbol.id}/dividends/next"):
|
||||
try:
|
||||
IEXData = resp[0]
|
||||
except IndexError as e:
|
||||
return f"${symbol.id.upper()} either doesn't exist or pays no dividend."
|
||||
keys = (
|
||||
"amount",
|
||||
"currency",
|
||||
@ -313,28 +323,19 @@ class IEX_Symbol:
|
||||
if symbol.symbol.upper() in self.otc_list:
|
||||
return "OTC stocks do not currently support any commands."
|
||||
|
||||
IEXurl = f"https://cloud.iexapis.com/stable/stock/{symbol}/news/last/15?token={self.IEX_TOKEN}"
|
||||
response = r.get(
|
||||
IEXurl,
|
||||
timeout=5,
|
||||
)
|
||||
if response.status_code == 200:
|
||||
data = response.json()
|
||||
if data:
|
||||
line = []
|
||||
if data := self.get(f"/stock/{symbol.id}/news/last/15"):
|
||||
line = []
|
||||
|
||||
for news in data:
|
||||
if news["lang"] == "en" and not news["hasPaywall"]:
|
||||
line.append(
|
||||
f"*{news['source']}*: [{news['headline']}]({news['url']})"
|
||||
)
|
||||
for news in data:
|
||||
if news["lang"] == "en" and not news["hasPaywall"]:
|
||||
line.append(
|
||||
f"*{news['source']}*: [{news['headline']}]({news['url']})"
|
||||
)
|
||||
|
||||
return f"News for **{symbol.id.upper()}**:\n" + "\n".join(line[:5])
|
||||
|
||||
else:
|
||||
return f"No news found for: {symbol}\nEither today is boring or the symbol does not exist."
|
||||
else:
|
||||
return f"No news found for: {symbol}\nEither today is boring or the symbol does not exist."
|
||||
|
||||
return f"News for **{symbol.id.upper()}**:\n" + "\n".join(line[:5])
|
||||
return f"No news found for: {symbol.id}\nEither today is boring or the symbol does not exist."
|
||||
|
||||
def info_reply(self, symbol: Stock) -> str:
|
||||
"""Gets description for Stock
|
||||
@ -351,14 +352,7 @@ class IEX_Symbol:
|
||||
if symbol.symbol.upper() in self.otc_list:
|
||||
return "OTC stocks do not currently support any commands."
|
||||
|
||||
IEXurl = f"https://cloud.iexapis.com/stable/stock/{symbol}/company?token={self.IEX_TOKEN}"
|
||||
response = r.get(
|
||||
IEXurl,
|
||||
timeout=5,
|
||||
)
|
||||
|
||||
if response.status_code == 200:
|
||||
data = response.json()
|
||||
if data := self.get(f"/stock/{symbol.id}/company"):
|
||||
[data.pop(k) for k in list(data) if data[k] == ""]
|
||||
|
||||
if "description" in data:
|
||||
@ -381,14 +375,7 @@ class IEX_Symbol:
|
||||
if symbol.symbol.upper() in self.otc_list:
|
||||
return "OTC stocks do not currently support any commands."
|
||||
|
||||
IEXurl = f"https://cloud.iexapis.com/stable/stock/{symbol}/stats?token={self.IEX_TOKEN}"
|
||||
response = r.get(
|
||||
IEXurl,
|
||||
timeout=5,
|
||||
)
|
||||
|
||||
if response.status_code == 200:
|
||||
data = response.json()
|
||||
if data := self.get(f"/stock/{symbol.id}/stats"):
|
||||
[data.pop(k) for k in list(data) if data[k] == ""]
|
||||
|
||||
m = ""
|
||||
@ -412,25 +399,20 @@ class IEX_Symbol:
|
||||
else:
|
||||
return f"No information found for: {symbol}\nEither today is boring or the symbol does not exist."
|
||||
|
||||
def cap_reply(self, stock: Stock) -> str:
|
||||
def cap_reply(self, symbol: Stock) -> str:
|
||||
"""Get the Market Cap of a stock"""
|
||||
response = r.get(
|
||||
f"https://cloud.iexapis.com/stable/stock/{stock.id}/stats?token={self.IEX_TOKEN}",
|
||||
timeout=5,
|
||||
)
|
||||
if response.status_code == 200:
|
||||
|
||||
if data := self.get(f"/stable/stock/{symbol.id}/stats"):
|
||||
|
||||
try:
|
||||
data = response.json()
|
||||
|
||||
cap = data["marketcap"]
|
||||
except KeyError:
|
||||
return f"{stock.id} returned an error."
|
||||
return f"{symbol.id} returned an error."
|
||||
|
||||
message = f"The current market cap of {stock.name} is $**{cap:,.2f}**"
|
||||
message = f"The current market cap of {symbol.name} is $**{cap:,.2f}**"
|
||||
|
||||
else:
|
||||
message = f"The Coin: {stock.name} was not found or returned and error."
|
||||
message = f"The Stock: {symbol.name} was not found or returned and error."
|
||||
|
||||
return message
|
||||
|
||||
@ -453,13 +435,8 @@ class IEX_Symbol:
|
||||
if symbol.id.upper() not in list(self.symbol_list["symbol"]):
|
||||
return pd.DataFrame()
|
||||
|
||||
IEXurl = f"https://cloud.iexapis.com/stable/stock/{symbol}/intraday-prices?token={self.IEX_TOKEN}"
|
||||
response = r.get(
|
||||
IEXurl,
|
||||
timeout=5,
|
||||
)
|
||||
if response.status_code == 200:
|
||||
df = pd.DataFrame(response.json())
|
||||
if data := self.get(f"/stock/{symbol.id}/intraday-prices"):
|
||||
df = pd.DataFrame(data)
|
||||
df.dropna(inplace=True, subset=["date", "minute", "high", "low", "volume"])
|
||||
df["DT"] = pd.to_datetime(df["date"] + "T" + df["minute"])
|
||||
df = df.set_index("DT")
|
||||
@ -494,13 +471,11 @@ class IEX_Symbol:
|
||||
except KeyError:
|
||||
pass
|
||||
|
||||
response = r.get(
|
||||
f"https://cloud.iexapis.com/stable/stock/{symbol}/chart/1mm?token={self.IEX_TOKEN}&chartInterval=3&includeToday=false",
|
||||
timeout=5,
|
||||
)
|
||||
|
||||
if response.status_code == 200:
|
||||
df = pd.DataFrame(response.json())
|
||||
if data := self.get(
|
||||
f"/stock/{symbol.id}/chart/1mm",
|
||||
params={"chartInterval": 3, "includeToday": "false"},
|
||||
):
|
||||
df = pd.DataFrame(data)
|
||||
df.dropna(inplace=True, subset=["date", "minute", "high", "low", "volume"])
|
||||
df["DT"] = pd.to_datetime(df["date"] + "T" + df["minute"])
|
||||
df = df.set_index("DT")
|
||||
@ -518,14 +493,10 @@ class IEX_Symbol:
|
||||
list of $ID: NAME, CHANGE%
|
||||
"""
|
||||
|
||||
stocks = r.get(
|
||||
f"https://cloud.iexapis.com/stable/stock/market/list/mostactive?token={self.IEX_TOKEN}",
|
||||
timeout=5,
|
||||
)
|
||||
if stocks.status_code == 200:
|
||||
if data := self.get(f"/stock/market/list/mostactive"):
|
||||
return [
|
||||
f"`${s['symbol']}`: {s['companyName']}, {100*s['changePercent']:.2f}%"
|
||||
for s in stocks.json()
|
||||
for s in data
|
||||
]
|
||||
else:
|
||||
return ["Trending Stocks Currently Unavailable."]
|
||||
|
||||
160
cg_Crypto.py
160
cg_Crypto.py
@ -1,6 +1,7 @@
|
||||
"""Class with functions for running the bot with IEX Cloud.
|
||||
"""
|
||||
|
||||
import logging
|
||||
from datetime import datetime
|
||||
from typing import List, Optional, Tuple
|
||||
|
||||
@ -33,6 +34,25 @@ class cg_Crypto:
|
||||
self.get_symbol_list()
|
||||
schedule.every().day.do(self.get_symbol_list)
|
||||
|
||||
def get(self, endpoint, params: dict = {}, timeout=10) -> dict:
|
||||
|
||||
url = "https://api.coingecko.com/api/v3" + endpoint
|
||||
resp = r.get(url, params=params, timeout=timeout)
|
||||
# Make sure API returned a proper status code
|
||||
try:
|
||||
resp.raise_for_status()
|
||||
except r.exceptions.HTTPError as e:
|
||||
logging.error(e)
|
||||
return {}
|
||||
|
||||
# Make sure API returned valid JSON
|
||||
try:
|
||||
resp_json = resp.json()
|
||||
return resp_json
|
||||
except r.exceptions.JSONDecodeError as e:
|
||||
logging.error(e)
|
||||
return {}
|
||||
|
||||
def symbol_id(self, symbol) -> str:
|
||||
try:
|
||||
return self.symbol_list[self.symbol_list["symbol"] == symbol]["id"].values[
|
||||
@ -45,10 +65,7 @@ class cg_Crypto:
|
||||
self, return_df=False
|
||||
) -> Optional[Tuple[pd.DataFrame, datetime]]:
|
||||
|
||||
raw_symbols = r.get(
|
||||
"https://api.coingecko.com/api/v3/coins/list",
|
||||
timeout=5,
|
||||
).json()
|
||||
raw_symbols = self.get("/coins/list")
|
||||
symbols = pd.DataFrame(data=raw_symbols)
|
||||
|
||||
symbols["description"] = (
|
||||
@ -69,15 +86,16 @@ class cg_Crypto:
|
||||
str
|
||||
Human readable text on status of CoinGecko API
|
||||
"""
|
||||
status = r.get(
|
||||
"https://api.coingecko.com/api/v3/ping",
|
||||
status = self.get(
|
||||
"/ping",
|
||||
timeout=5,
|
||||
)
|
||||
|
||||
if status.status_code == 200:
|
||||
return f"CoinGecko API responded that it was OK in {status.elapsed.total_seconds()} Seconds."
|
||||
else:
|
||||
return f"CoinGecko API returned an error in {status.elapsed.total_seconds()} Seconds."
|
||||
try:
|
||||
status.raise_for_status()
|
||||
return f"CoinGecko API responded that it was OK with a {status.status_code} in {status.elapsed.total_seconds()} Seconds."
|
||||
except:
|
||||
return f"CoinGecko API returned an error code {status.status_code} in {status.elapsed.total_seconds()} Seconds."
|
||||
|
||||
def search_symbols(self, search: str) -> List[Tuple[str, str]]:
|
||||
"""Performs a fuzzy search to find coin symbols closest to a search term.
|
||||
@ -133,14 +151,16 @@ class cg_Crypto:
|
||||
markdown formatted string of the symbols price and movement.
|
||||
"""
|
||||
|
||||
response = r.get(
|
||||
f"https://api.coingecko.com/api/v3/simple/price?ids={coin.id}&vs_currencies={self.vs_currency}&include_24hr_change=true",
|
||||
timeout=5,
|
||||
)
|
||||
if response.status_code == 200:
|
||||
|
||||
if resp := self.get(
|
||||
"/simple/price",
|
||||
params={
|
||||
"ids": coin.id,
|
||||
"vs_currencies": self.vs_currency,
|
||||
"include_24hr_change": "true",
|
||||
},
|
||||
):
|
||||
try:
|
||||
data = response.json()[coin.id]
|
||||
data = resp[coin.id]
|
||||
|
||||
price = data[self.vs_currency]
|
||||
change = data[self.vs_currency + "_24h_change"]
|
||||
@ -160,7 +180,7 @@ class cg_Crypto:
|
||||
message += ", the coin hasn't shown any movement today."
|
||||
|
||||
else:
|
||||
message = f"The Coin: {coin.name} was not found."
|
||||
message = f"The price for {coin.name} is not available. If you suspect this is an error run `/status`"
|
||||
|
||||
return message
|
||||
|
||||
@ -177,13 +197,13 @@ class cg_Crypto:
|
||||
pd.DataFrame
|
||||
Returns a timeseries dataframe with high, low, and volume data if its available. Otherwise returns empty pd.DataFrame.
|
||||
"""
|
||||
response = r.get(
|
||||
f"https://api.coingecko.com/api/v3/coins/{symbol.id}/ohlc?vs_currency=usd&days=1",
|
||||
timeout=5,
|
||||
)
|
||||
if response.status_code == 200:
|
||||
|
||||
if resp := self.get(
|
||||
f"/coins/{symbol.id}/ohlc",
|
||||
params={"vs_currency": self.vs_currency, "days": 1},
|
||||
):
|
||||
df = pd.DataFrame(
|
||||
response.json(), columns=["Date", "Open", "High", "Low", "Close"]
|
||||
resp, columns=["Date", "Open", "High", "Low", "Close"]
|
||||
).dropna()
|
||||
df["Date"] = pd.to_datetime(df["Date"], unit="ms")
|
||||
df = df.set_index("Date")
|
||||
@ -205,14 +225,13 @@ class cg_Crypto:
|
||||
pd.DataFrame
|
||||
Returns a timeseries dataframe with high, low, and volume data if its available. Otherwise returns empty pd.DataFrame.
|
||||
"""
|
||||
response = r.get(
|
||||
f"https://api.coingecko.com/api/v3/coins/{symbol.id}/ohlc?vs_currency=usd&days=30",
|
||||
timeout=5,
|
||||
)
|
||||
|
||||
if response.status_code == 200:
|
||||
if resp := self.get(
|
||||
f"/coins/{symbol.id}/ohlc",
|
||||
params={"vs_currency": self.vs_currency, "days": 30},
|
||||
):
|
||||
df = pd.DataFrame(
|
||||
response.json(), columns=["Date", "Open", "High", "Low", "Close"]
|
||||
resp, columns=["Date", "Open", "High", "Low", "Close"]
|
||||
).dropna()
|
||||
df["Date"] = pd.to_datetime(df["Date"], unit="ms")
|
||||
df = df.set_index("Date")
|
||||
@ -233,12 +252,12 @@ class cg_Crypto:
|
||||
Preformatted markdown.
|
||||
"""
|
||||
|
||||
response = r.get(
|
||||
f"https://api.coingecko.com/api/v3/coins/{symbol.id}?localization=false",
|
||||
timeout=5,
|
||||
)
|
||||
if response.status_code == 200:
|
||||
data = response.json()
|
||||
if data := self.get(
|
||||
f"/coins/{symbol.id}",
|
||||
params={
|
||||
"localization": "false",
|
||||
},
|
||||
):
|
||||
|
||||
return f"""
|
||||
[{data['name']}]({data['links']['homepage'][0]}) Statistics:
|
||||
@ -265,14 +284,18 @@ class cg_Crypto:
|
||||
str
|
||||
Preformatted markdown.
|
||||
"""
|
||||
response = r.get(
|
||||
f"https://api.coingecko.com/api/v3/simple/price?ids={coin.id}&vs_currencies={self.vs_currency}&include_market_cap=true",
|
||||
timeout=5,
|
||||
)
|
||||
if response.status_code == 200:
|
||||
|
||||
if resp := self.get(
|
||||
f"/simple/price",
|
||||
params={
|
||||
"ids": coin.id,
|
||||
"vs_currencies": self.vs_currency,
|
||||
"include_market_cap": "true",
|
||||
},
|
||||
):
|
||||
print(resp)
|
||||
try:
|
||||
data = response.json()[coin.id]
|
||||
data = resp[coin.id]
|
||||
|
||||
price = data[self.vs_currency]
|
||||
cap = data[self.vs_currency + "_market_cap"]
|
||||
@ -302,12 +325,10 @@ class cg_Crypto:
|
||||
Preformatted markdown.
|
||||
"""
|
||||
|
||||
response = r.get(
|
||||
f"https://api.coingecko.com/api/v3/coins/{symbol.id}?localization=false",
|
||||
timeout=5,
|
||||
)
|
||||
if response.status_code == 200:
|
||||
data = response.json()
|
||||
if data := self.get(
|
||||
f"/coins/{symbol.id}",
|
||||
params={"localization": "false"},
|
||||
):
|
||||
try:
|
||||
return markdownify(data["description"]["en"])
|
||||
except KeyError:
|
||||
@ -324,28 +345,29 @@ class cg_Crypto:
|
||||
list of $$ID: NAME, CHANGE%
|
||||
"""
|
||||
|
||||
coins = r.get(
|
||||
"https://api.coingecko.com/api/v3/search/trending",
|
||||
timeout=5,
|
||||
)
|
||||
coins = self.get("/search/trending")
|
||||
try:
|
||||
trending = []
|
||||
if coins.status_code == 200:
|
||||
for coin in coins.json()["coins"]:
|
||||
c = coin["item"]
|
||||
for coin in coins["coins"]:
|
||||
c = coin["item"]
|
||||
|
||||
sym = c["symbol"].upper()
|
||||
name = c["name"]
|
||||
change = r.get(
|
||||
f"https://api.coingecko.com/api/v3/simple/price?ids={c['id']}&vs_currencies={self.vs_currency}&include_24hr_change=true"
|
||||
).json()[c["id"]]["usd_24h_change"]
|
||||
sym = c["symbol"].upper()
|
||||
name = c["name"]
|
||||
change = self.get(
|
||||
f"/simple/price",
|
||||
params={
|
||||
"ids": c["id"],
|
||||
"vs_currencies": self.vs_currency,
|
||||
"include_24hr_change": "true",
|
||||
},
|
||||
)[c["id"]]["usd_24h_change"]
|
||||
|
||||
msg = f"`$${sym}`: {name}, {change:.2f}%"
|
||||
msg = f"`$${sym}`: {name}, {change:.2f}%"
|
||||
|
||||
trending.append(msg)
|
||||
trending.append(msg)
|
||||
|
||||
except Exception as e:
|
||||
print(e)
|
||||
logging.warning(e)
|
||||
trending = ["Trending Coins Currently Unavailable."]
|
||||
|
||||
return trending
|
||||
@ -364,10 +386,14 @@ class cg_Crypto:
|
||||
"""
|
||||
query = ",".join([c.id for c in coins])
|
||||
|
||||
prices = r.get(
|
||||
f"https://api.coingecko.com/api/v3/simple/price?ids={query}&vs_currencies=usd&include_24hr_change=true",
|
||||
timeout=5,
|
||||
).json()
|
||||
prices = self.get(
|
||||
f"/simple/price",
|
||||
params={
|
||||
"ids": query,
|
||||
"vs_currencies": self.vs_currency,
|
||||
"include_24hr_change": "true",
|
||||
},
|
||||
)
|
||||
|
||||
replies = []
|
||||
for coin in coins:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user