mirror of
https://gitlab.com/simple-stock-bots/simple-stock-bot.git
synced 2025-06-16 15:17:28 +00:00
Switched to black formatter
This commit is contained in:
parent
b7663ae097
commit
027d45b6f2
75
stockBot.py
75
stockBot.py
@ -1,4 +1,4 @@
|
|||||||
# Work with Python 3.6
|
# Work with Python 3.7
|
||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
import re
|
import re
|
||||||
@ -7,14 +7,14 @@ import urllib.request
|
|||||||
import telegram
|
import telegram
|
||||||
from telegram.ext import CommandHandler, Filters, MessageHandler, Updater
|
from telegram.ext import CommandHandler, Filters, MessageHandler, Updater
|
||||||
|
|
||||||
import credentials
|
|
||||||
import tickerInfo
|
import tickerInfo
|
||||||
|
|
||||||
TOKEN = "TELEGRAM BOT TOKEN HERE"
|
TOKEN = "TELEGRAM_TOKEN
|
||||||
|
|
||||||
# Enable logging
|
# Enable logging
|
||||||
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
logging.basicConfig(
|
||||||
level=logging.INFO)
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", level=logging.INFO
|
||||||
|
)
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
@ -23,7 +23,7 @@ logger = logging.getLogger(__name__)
|
|||||||
# update. Error handlers also receive the raised TelegramError object in error.
|
# update. Error handlers also receive the raised TelegramError object in error.
|
||||||
def start(bot, update):
|
def start(bot, update):
|
||||||
"""Send a message when the command /start is issued."""
|
"""Send a message when the command /start is issued."""
|
||||||
update.message.reply_text('I am started and ready to go!')
|
update.message.reply_text("I am started and ready to go!")
|
||||||
|
|
||||||
|
|
||||||
def help(bot, update):
|
def help(bot, update):
|
||||||
@ -35,57 +35,70 @@ def stockInfo(bot, update):
|
|||||||
message = update.message.text
|
message = update.message.text
|
||||||
try:
|
try:
|
||||||
# regex to find tickers in messages, looks for up to 4 word characters following a dollar sign and captures the 4 word characters
|
# regex to find tickers in messages, looks for up to 4 word characters following a dollar sign and captures the 4 word characters
|
||||||
tickers = re.findall('[$](\w{1,4})', message)
|
tickers = re.findall("[$](\w{1,4})", message)
|
||||||
|
|
||||||
# Checks if !news is called, and prints news embed if it is
|
# Checks if !news is called, and prints news embed if it is
|
||||||
if message.startswith('!news'):
|
if message.startswith("!news"):
|
||||||
tickerData = tickerInfo.tickerQuote(tickers)
|
tickerData = tickerInfo.tickerQuote(tickers)
|
||||||
for ticker in tickers:
|
for ticker in tickers:
|
||||||
ticker = ticker.upper()
|
ticker = ticker.upper()
|
||||||
# bot.send_photo(bot.get_updates()
|
# bot.send_photo(bot.get_updates()
|
||||||
# [-1].message.chat_id, tickerData[ticker + 'Image'])
|
# [-1].message.chat_id, tickerData[ticker + 'Image'])
|
||||||
name = tickerData[ticker + 'Name']
|
name = tickerData[ticker + "Name"]
|
||||||
price = tickerData[ticker + 'Price']
|
price = tickerData[ticker + "Price"]
|
||||||
change = tickerData[ticker + 'Change']
|
change = tickerData[ticker + "Change"]
|
||||||
|
|
||||||
message = 'The current stock price of ' + \
|
message = (
|
||||||
name + ' is $**' + str(price) + '**'
|
"The current stock price of " + name + " is $**" + str(price) + "**"
|
||||||
|
)
|
||||||
if change > 0:
|
if change > 0:
|
||||||
message = message + \
|
message = (
|
||||||
', the stock is currently **up ' + str(change) + '%**'
|
message + ", the stock is currently **up " + str(change) + "%**"
|
||||||
|
)
|
||||||
elif change < 0:
|
elif change < 0:
|
||||||
message = message + \
|
message = (
|
||||||
', the stock is currently **down' + str(change) + '%**'
|
message
|
||||||
|
+ ", the stock is currently **down"
|
||||||
|
+ str(change)
|
||||||
|
+ "%**"
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
message = message + ", the stock hasn't shown any movement today."
|
message = message + ", the stock hasn't shown any movement today."
|
||||||
|
|
||||||
news = tickerInfo.stockNewsList(ticker)
|
news = tickerInfo.stockNewsList(ticker)
|
||||||
for source in news:
|
for source in news:
|
||||||
message = message + \
|
message = message + "\n[" + source + "](" + news[source] + ")"
|
||||||
'\n[' + source + '](' + news[source] + ')'
|
|
||||||
|
|
||||||
update.message.reply_text(
|
update.message.reply_text(
|
||||||
text=message, parse_mode=telegram.ParseMode.MARKDOWN)
|
text=message, parse_mode=telegram.ParseMode.MARKDOWN
|
||||||
|
)
|
||||||
|
|
||||||
else: # If news isnt called, print normal stock price
|
else: # If news isnt called, print normal stock price
|
||||||
tickerData = tickerInfo.tickerQuote(tickers)
|
tickerData = tickerInfo.tickerQuote(tickers)
|
||||||
for ticker in tickers:
|
for ticker in tickers:
|
||||||
ticker = ticker.upper()
|
ticker = ticker.upper()
|
||||||
name = tickerData[ticker + 'Name']
|
name = tickerData[ticker + "Name"]
|
||||||
price = tickerData[ticker + 'Price']
|
price = tickerData[ticker + "Price"]
|
||||||
change = tickerData[ticker + 'Change']
|
change = tickerData[ticker + "Change"]
|
||||||
message = 'The current stock price of ' + \
|
message = (
|
||||||
name + ' is $**' + str(price) + '**'
|
"The current stock price of " + name + " is $**" + str(price) + "**"
|
||||||
|
)
|
||||||
if change > 0:
|
if change > 0:
|
||||||
message = message + ', the stock is currently **up ' + \
|
message = (
|
||||||
str(change) + '%**'
|
message + ", the stock is currently **up " + str(change) + "%**"
|
||||||
|
)
|
||||||
elif change < 0:
|
elif change < 0:
|
||||||
message = message + ', the stock is currently **down ' + \
|
message = (
|
||||||
str(change) + '%**'
|
message
|
||||||
|
+ ", the stock is currently **down "
|
||||||
|
+ str(change)
|
||||||
|
+ "%**"
|
||||||
|
)
|
||||||
else:
|
else:
|
||||||
move = ", the stock hasn't shown any movement today."
|
move = ", the stock hasn't shown any movement today."
|
||||||
update.message.reply_text(
|
update.message.reply_text(
|
||||||
text=message, parse_mode=telegram.ParseMode.MARKDOWN)
|
text=message, parse_mode=telegram.ParseMode.MARKDOWN
|
||||||
|
)
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
@ -122,5 +135,5 @@ def main():
|
|||||||
updater.idle()
|
updater.idle()
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == "__main__":
|
||||||
main()
|
main()
|
||||||
|
|||||||
@ -4,39 +4,40 @@ import json
|
|||||||
|
|
||||||
def tickerQuote(tickers):
|
def tickerQuote(tickers):
|
||||||
stockData = {}
|
stockData = {}
|
||||||
IEXURL = 'https://api.iextrading.com/1.0/stock/market/batch?symbols=' + \
|
IEXURL = (
|
||||||
",".join(tickers) + '&types=quote'
|
"https://api.iextrading.com/1.0/stock/market/batch?symbols="
|
||||||
print('Gathering Quote from ' + IEXURL)
|
+ ",".join(tickers)
|
||||||
|
+ "&types=quote"
|
||||||
|
)
|
||||||
|
print("Gathering Quote from " + IEXURL)
|
||||||
with urllib.request.urlopen(IEXURL) as url:
|
with urllib.request.urlopen(IEXURL) as url:
|
||||||
IEXData = json.loads(url.read().decode())
|
IEXData = json.loads(url.read().decode())
|
||||||
|
|
||||||
for ticker in tickers:
|
for ticker in tickers:
|
||||||
ticker = ticker.upper()
|
ticker = ticker.upper()
|
||||||
stockData[ticker + 'Name'] = IEXData[ticker]['quote']['companyName']
|
stockData[ticker + "Name"] = IEXData[ticker]["quote"]["companyName"]
|
||||||
stockData[ticker + 'Price'] = \
|
stockData[ticker + "Price"] = IEXData[ticker]["quote"]["latestPrice"]
|
||||||
IEXData[ticker]['quote']['latestPrice']
|
stockData[ticker + "Change"] = IEXData[ticker]["quote"]["changePercent"] * 100
|
||||||
stockData[ticker +
|
stockData[ticker + "Image"] = stockLogo(ticker)
|
||||||
'Change'] = IEXData[ticker]['quote']['changePercent'] * 100
|
print("Quote Gathered")
|
||||||
stockData[ticker + 'Image'] = stockLogo(ticker)
|
|
||||||
print('Quote Gathered')
|
|
||||||
return stockData
|
return stockData
|
||||||
|
|
||||||
|
|
||||||
def stockNewsList(ticker):
|
def stockNewsList(ticker):
|
||||||
print('Gather News on ' + ticker)
|
print("Gather News on " + ticker)
|
||||||
news = {
|
news = {
|
||||||
'Bravos': 'https://bravos.co/' + ticker,
|
"Bravos": "https://bravos.co/" + ticker,
|
||||||
'Seeking Alpha': 'https://seekingalpha.com/symbol/' + ticker,
|
"Seeking Alpha": "https://seekingalpha.com/symbol/" + ticker,
|
||||||
'MSN Money': 'https://www.msn.com/en-us/money/stockdetails?symbol=' + ticker,
|
"MSN Money": "https://www.msn.com/en-us/money/stockdetails?symbol=" + ticker,
|
||||||
'Yahoo Finance': 'https://finance.yahoo.com/quote/' + ticker,
|
"Yahoo Finance": "https://finance.yahoo.com/quote/" + ticker,
|
||||||
'Wall Street Journal': 'https://quotes.wsj.com/' + ticker,
|
"Wall Street Journal": "https://quotes.wsj.com/" + ticker,
|
||||||
'The Street': 'https://www.thestreet.com/quote/' + ticker + '.html',
|
"The Street": "https://www.thestreet.com/quote/" + ticker + ".html",
|
||||||
'Zacks': 'https://www.zacks.com/stock/quote/' + ticker
|
"Zacks": "https://www.zacks.com/stock/quote/" + ticker,
|
||||||
}
|
}
|
||||||
print('News gathered.')
|
print("News gathered.")
|
||||||
return news
|
return news
|
||||||
|
|
||||||
|
|
||||||
def stockLogo(ticker):
|
def stockLogo(ticker):
|
||||||
logoURL = 'https://g.foolcdn.com/art/companylogos/mark/' + ticker + '.png'
|
logoURL = "https://g.foolcdn.com/art/companylogos/mark/" + ticker + ".png"
|
||||||
return logoURL
|
return logoURL
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user