mirror of
https://gitlab.com/simple-stock-bots/simple-stock-bot.git
synced 2025-06-16 15:17:28 +00:00
renamed ticker to symbol everywhere
This commit is contained in:
parent
38fcf6d054
commit
33c80bd75b
40
bot.py
40
bot.py
@ -7,7 +7,7 @@ from functions import *
|
|||||||
from telegram.ext import CommandHandler, Filters, MessageHandler, Updater
|
from telegram.ext import CommandHandler, Filters, MessageHandler, Updater
|
||||||
|
|
||||||
TELEGRAM_TOKEN = os.environ["TELEGRAM"]
|
TELEGRAM_TOKEN = os.environ["TELEGRAM"]
|
||||||
TICKER_REGEX = "[$]([a-zA-Z]{1,4})"
|
symbol_REGEX = "[$]([a-zA-Z]{1,4})"
|
||||||
|
|
||||||
# Enable logging
|
# Enable logging
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
@ -31,19 +31,19 @@ def help(bot, update):
|
|||||||
update.message.reply_text(text=message, parse_mode=telegram.ParseMode.MARKDOWN)
|
update.message.reply_text(text=message, parse_mode=telegram.ParseMode.MARKDOWN)
|
||||||
|
|
||||||
|
|
||||||
def tickerDetect(bot, update):
|
def symbolDetect(bot, update):
|
||||||
"""
|
"""
|
||||||
Runs on any message that doesn't have a command and searches for tickers, then returns the prices of any tickers found.
|
Runs on any message that doesn't have a command and searches for symbols, then returns the prices of any symbols found.
|
||||||
"""
|
"""
|
||||||
message = update.message.text
|
message = update.message.text
|
||||||
chat_id = update.message.chat_id
|
chat_id = update.message.chat_id
|
||||||
tickers = getTickers(message)
|
symbols = getSymbols(message)
|
||||||
|
|
||||||
if tickers:
|
if symbols:
|
||||||
# Let user know bot is working
|
# Let user know bot is working
|
||||||
bot.send_chat_action(chat_id=chat_id, action=telegram.ChatAction.TYPING)
|
bot.send_chat_action(chat_id=chat_id, action=telegram.ChatAction.TYPING)
|
||||||
|
|
||||||
for symbol, reply in tickerDataReply(tickers).items():
|
for symbol, reply in symbolDataReply(symbols).items():
|
||||||
|
|
||||||
update.message.reply_text(
|
update.message.reply_text(
|
||||||
text=reply, parse_mode=telegram.ParseMode.MARKDOWN
|
text=reply, parse_mode=telegram.ParseMode.MARKDOWN
|
||||||
@ -52,16 +52,16 @@ def tickerDetect(bot, update):
|
|||||||
|
|
||||||
def dividend(bot, update):
|
def dividend(bot, update):
|
||||||
"""
|
"""
|
||||||
waits for /dividend or /div command and then finds dividend info on that ticker.
|
waits for /dividend or /div command and then finds dividend info on that symbol.
|
||||||
"""
|
"""
|
||||||
message = update.message.text
|
message = update.message.text
|
||||||
chat_id = update.message.chat_id
|
chat_id = update.message.chat_id
|
||||||
tickers = getTickers(message)
|
symbols = getSymbols(message)
|
||||||
|
|
||||||
if tickers:
|
if symbols:
|
||||||
bot.send_chat_action(chat_id=chat_id, action=telegram.ChatAction.TYPING)
|
bot.send_chat_action(chat_id=chat_id, action=telegram.ChatAction.TYPING)
|
||||||
|
|
||||||
for symbol, reply in tickerDividend(tickers).items():
|
for symbol, reply in symbolDividend(symbols).items():
|
||||||
|
|
||||||
update.message.reply_text(
|
update.message.reply_text(
|
||||||
text=reply, parse_mode=telegram.ParseMode.MARKDOWN
|
text=reply, parse_mode=telegram.ParseMode.MARKDOWN
|
||||||
@ -70,16 +70,16 @@ def dividend(bot, update):
|
|||||||
|
|
||||||
def news(bot, update):
|
def news(bot, update):
|
||||||
"""
|
"""
|
||||||
waits for /news command and then finds news info on that ticker.
|
waits for /news command and then finds news info on that symbol.
|
||||||
"""
|
"""
|
||||||
message = update.message.text
|
message = update.message.text
|
||||||
chat_id = update.message.chat_id
|
chat_id = update.message.chat_id
|
||||||
tickers = getTickers(message)
|
symbols = getSymbols(message)
|
||||||
|
|
||||||
if tickers:
|
if symbols:
|
||||||
bot.send_chat_action(chat_id=chat_id, action=telegram.ChatAction.TYPING)
|
bot.send_chat_action(chat_id=chat_id, action=telegram.ChatAction.TYPING)
|
||||||
|
|
||||||
for symbol, reply in tickerNews(tickers).items():
|
for symbol, reply in symbolNews(symbols).items():
|
||||||
|
|
||||||
update.message.reply_text(
|
update.message.reply_text(
|
||||||
text=reply, parse_mode=telegram.ParseMode.MARKDOWN
|
text=reply, parse_mode=telegram.ParseMode.MARKDOWN
|
||||||
@ -88,16 +88,16 @@ def news(bot, update):
|
|||||||
|
|
||||||
def info(bot, update):
|
def info(bot, update):
|
||||||
"""
|
"""
|
||||||
waits for /info command and then finds info on that ticker.
|
waits for /info command and then finds info on that symbol.
|
||||||
"""
|
"""
|
||||||
message = update.message.text
|
message = update.message.text
|
||||||
chat_id = update.message.chat_id
|
chat_id = update.message.chat_id
|
||||||
tickers = getTickers(message)
|
symbols = getSymbols(message)
|
||||||
|
|
||||||
if tickers:
|
if symbols:
|
||||||
bot.send_chat_action(chat_id=chat_id, action=telegram.ChatAction.TYPING)
|
bot.send_chat_action(chat_id=chat_id, action=telegram.ChatAction.TYPING)
|
||||||
|
|
||||||
for symbol, reply in tickerInfo(tickers).items():
|
for symbol, reply in symbolInfo(symbols).items():
|
||||||
|
|
||||||
update.message.reply_text(
|
update.message.reply_text(
|
||||||
text=reply, parse_mode=telegram.ParseMode.MARKDOWN
|
text=reply, parse_mode=telegram.ParseMode.MARKDOWN
|
||||||
@ -126,7 +126,7 @@ def main():
|
|||||||
dp.add_handler(CommandHandler("info", info))
|
dp.add_handler(CommandHandler("info", info))
|
||||||
|
|
||||||
# on noncommand i.e message - echo the message on Telegram
|
# on noncommand i.e message - echo the message on Telegram
|
||||||
dp.add_handler(MessageHandler(Filters.text, tickerDetect))
|
dp.add_handler(MessageHandler(Filters.text, symbolDetect))
|
||||||
|
|
||||||
# log all errors
|
# log all errors
|
||||||
dp.add_error_handler(error)
|
dp.add_error_handler(error)
|
||||||
@ -141,4 +141,4 @@ def main():
|
|||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
85
functions.py
85
functions.py
@ -5,27 +5,27 @@ import urllib.request
|
|||||||
import requests
|
import requests
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
IEX_TOKEN = os.environ["IEX"]
|
IEX_TOKEN = "sk_130b8e8f75ba4e14a5683ff37a655584" # os.environ["IEX"]
|
||||||
|
|
||||||
|
|
||||||
def getTickers(text: str):
|
def getSymbols(text: str):
|
||||||
"""
|
"""
|
||||||
Takes a blob of text and returns any stock tickers found.
|
Takes a blob of text and returns any stock symbols found.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
TICKER_REGEX = "[$]([a-zA-Z]{1,4})"
|
SYMBOL_REGEX = "[$]([a-zA-Z]{1,4})"
|
||||||
|
|
||||||
return list(set(re.findall(TICKER_REGEX, text)))
|
return list(set(re.findall(SYMBOL_REGEX, text)))
|
||||||
|
|
||||||
|
|
||||||
def tickerDataReply(tickers: list):
|
def symbolDataReply(symbols: list):
|
||||||
"""
|
"""
|
||||||
Takes a list of tickers and returns a list of strings with information about the ticker.
|
Takes a list of symbols and returns a list of strings with information about the symbol.
|
||||||
"""
|
"""
|
||||||
tickerReplies = {}
|
symbolReplies = {}
|
||||||
for ticker in tickers:
|
for symbol in symbols:
|
||||||
IEXURL = (
|
IEXURL = (
|
||||||
f"https://cloud.iexapis.com/stable/stock/{ticker}/quote?token={IEX_TOKEN}"
|
f"https://cloud.iexapis.com/stable/stock/{symbol}/quote?token={IEX_TOKEN}"
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
with urllib.request.urlopen(IEXURL) as url:
|
with urllib.request.urlopen(IEXURL) as url:
|
||||||
@ -42,74 +42,87 @@ def tickerDataReply(tickers: list):
|
|||||||
else:
|
else:
|
||||||
reply += ", the stock hasn't shown any movement today."
|
reply += ", the stock hasn't shown any movement today."
|
||||||
except:
|
except:
|
||||||
reply = f"The ticker: {ticker} was not found."
|
reply = f"The symbol: {symbol} was not found."
|
||||||
|
|
||||||
tickerReplies[ticker] = reply
|
symbolReplies[symbol] = reply
|
||||||
|
|
||||||
return tickerReplies
|
return symbolReplies
|
||||||
|
|
||||||
|
|
||||||
def tickerDividend(tickers: list):
|
def symbolDividend(symbols: list):
|
||||||
messages = {}
|
messages = {}
|
||||||
|
|
||||||
for ticker in tickers:
|
for symbol in symbols:
|
||||||
IEXurl = f"https://cloud.iexapis.com/stable/data-points/{ticker}/NEXTDIVIDENDDATE?token={IEX_TOKEN}"
|
IEXurl = f"https://cloud.iexapis.com/stable/data-points/{symbol}/NEXTDIVIDENDDATE?token={IEX_TOKEN}"
|
||||||
date = requests.get(IEXurl).json()
|
response = requests.get(IEXurl)
|
||||||
if date:
|
if response.status_code is 200:
|
||||||
|
|
||||||
|
date = response.json()
|
||||||
# Pattern IEX uses for dividend date.
|
# Pattern IEX uses for dividend date.
|
||||||
pattern = "%Y-%m-%d"
|
pattern = "%Y-%m-%d"
|
||||||
divDate = datetime.strptime(date, pattern)
|
divDate = datetime.strptime(date, pattern)
|
||||||
#
|
#
|
||||||
daysDelta = divDate - datetime.now()
|
daysDelta = (divDate - datetime.now()).days
|
||||||
datePretty = divDate.strftime("%A, %B %w")
|
datePretty = divDate.strftime("%A, %B %w")
|
||||||
|
if daysDelta < 0:
|
||||||
|
messages[
|
||||||
|
symbol
|
||||||
|
] = f"{symbol.upper()} dividend was on {datePretty} and a new date hasn't been announced yet."
|
||||||
|
elif daysDelta > 0:
|
||||||
|
messages[
|
||||||
|
symbol
|
||||||
|
] = f"{symbol.upper()} dividend is on {datePretty} which is in {daysDelta} Days."
|
||||||
|
else:
|
||||||
|
messages[symbol] = f"{symbol.upper()} is today."
|
||||||
|
|
||||||
messages[
|
|
||||||
ticker
|
|
||||||
] = f"{ticker.upper()} dividend is on {datePretty} which is in {daysDelta} Days."
|
|
||||||
else:
|
else:
|
||||||
messages[ticker] = f"{ticker} either doesn't exist or pays no dividend."
|
messages[symbol] = f"{symbol} either doesn't exist or pays no dividend."
|
||||||
|
|
||||||
return messages
|
return messages
|
||||||
|
|
||||||
|
|
||||||
def tickerNews(tickers: list):
|
def symbolNews(symbols: list):
|
||||||
messages = {}
|
messages = {}
|
||||||
|
|
||||||
for ticker in tickers:
|
for symbol in symbols:
|
||||||
IEXurl = f"https://cloud.iexapis.com/stable/stock/{ticker}/news/last/3?token={IEX_TOKEN}"
|
IEXurl = f"https://cloud.iexapis.com/stable/stock/{symbol}/news/last/3?token={IEX_TOKEN}"
|
||||||
with urllib.request.urlopen(IEXurl) as url:
|
with urllib.request.urlopen(IEXurl) as url:
|
||||||
data = json.loads(url.read().decode())
|
data = json.loads(url.read().decode())
|
||||||
if data:
|
if data:
|
||||||
messages[ticker] = f"News for **{ticker.upper()}**:\n"
|
messages[symbol] = f"News for **{symbol.upper()}**:\n"
|
||||||
for news in data:
|
for news in data:
|
||||||
message = f"\t[{news['headline']}]({news['url']})\n\n"
|
message = f"\t[{news['headline']}]({news['url']})\n\n"
|
||||||
messages[ticker] = messages[ticker] + message
|
messages[symbol] = messages[symbol] + message
|
||||||
else:
|
else:
|
||||||
messages[
|
messages[
|
||||||
ticker
|
symbol
|
||||||
] = f"No news found for: {ticker}\nEither today is boring or the ticker does not exist."
|
] = f"No news found for: {symbol}\nEither today is boring or the symbol does not exist."
|
||||||
|
|
||||||
return messages
|
return messages
|
||||||
|
|
||||||
|
|
||||||
def tickerInfo(tickers: list):
|
def symbolInfo(symbols: list):
|
||||||
messages = {}
|
messages = {}
|
||||||
|
|
||||||
for ticker in tickers:
|
for symbol in symbols:
|
||||||
IEXurl = (
|
IEXurl = (
|
||||||
f"https://cloud.iexapis.com/stable/stock/{ticker}/company?token={IEX_TOKEN}"
|
f"https://cloud.iexapis.com/stable/stock/{symbol}/company?token={IEX_TOKEN}"
|
||||||
)
|
)
|
||||||
with urllib.request.urlopen(IEXurl) as url:
|
with urllib.request.urlopen(IEXurl) as url:
|
||||||
data = json.loads(url.read().decode())
|
data = json.loads(url.read().decode())
|
||||||
if data:
|
if data:
|
||||||
messages[
|
messages[
|
||||||
ticker
|
symbol
|
||||||
] = f"Company Name: [{data['companyName']}]({data['website']})\nIndustry: {data['industry']}\nSector: {data['sector']}\nCEO: {data['CEO']}\nDescription: {data['description']}\n"
|
] = f"Company Name: [{data['companyName']}]({data['website']})\nIndustry: {data['industry']}\nSector: {data['sector']}\nCEO: {data['CEO']}\nDescription: {data['description']}\n"
|
||||||
|
|
||||||
else:
|
else:
|
||||||
messages[
|
messages[
|
||||||
ticker
|
symbol
|
||||||
] = f"No information found for: {ticker}\nEither today is boring or the ticker does not exist."
|
] = f"No information found for: {symbol}\nEither today is boring or the symbol does not exist."
|
||||||
|
|
||||||
return messages
|
return messages
|
||||||
|
|
||||||
|
|
||||||
|
# symbolDividend(["psec", "f", "tsla"])
|
||||||
|
|
||||||
|
print(symbolDividend(["f"]))
|
||||||
|
Loading…
x
Reference in New Issue
Block a user