1
0
mirror of https://gitlab.com/simple-stock-bots/simple-stock-bot.git synced 2025-06-16 15:17:28 +00:00

Merge branch 'Features' into 'master'

replaced formatting with f strings and fixed news bug

See merge request MisterBiggs/simple-telegram-bot!8
This commit is contained in:
Anson Biggs 2019-02-22 19:43:07 +00:00
commit 0e14b183fa
2 changed files with 14 additions and 40 deletions

View File

@ -65,36 +65,22 @@ def news(bot, update):
price = tickerData[ticker + "Price"] price = tickerData[ticker + "Price"]
change = tickerData[ticker + "Change"] change = tickerData[ticker + "Change"]
message = ( message = f"The current stock price of {name} is $**{price}**"
"The current stock price of "
+ name
+ " is $**"
+ str(price)
+ "**"
)
if change > 0: if change > 0:
message = ( message = f"{message}, the stock is currently **up {change}%**"
message
+ ", the stock is currently **up "
+ str(change)
+ "%**"
)
elif change < 0: elif change < 0:
message = ( message = (
message f"{message}, the stock is currently **down {change}%**"
+ ", the stock is currently **down "
+ str(change)
+ "%**"
) )
else: else:
message = ( message = (
message + ", the stock hasn't shown any movement today." f"{message}, the stock hasn't shown any movement today."
) )
news = tickerInfo.stockNews(ticker) news = tickerInfo.stockNews(ticker)
for i in range(3): for i in range(3):
message = "{}\n\n[{}]({})".format( message = (
message, news["title"][i], news["link"][i] f"{message}\n\n[{news['title'][i]}]({news['link'][i]})"
) )
update.message.reply_text( update.message.reply_text(
@ -124,22 +110,13 @@ def stockInfo(bot, update):
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 = ( message = f"The current stock price of {name} is $**{price}**"
"The current stock price of " + name + " is $**" + str(price) + "**"
)
if change > 0: if change > 0:
message = ( message = f"{message}, the stock is currently **up {change}%**"
message + ", the stock is currently **up " + str(change) + "%**"
)
elif change < 0: elif change < 0:
message = ( message = f"{message}, the stock is currently **down {change}%**"
message
+ ", the stock is currently **down "
+ str(change)
+ "%**"
)
else: else:
message = message + ", the stock hasn't shown any movement today." message = f"{message}, 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
) )
@ -152,7 +129,6 @@ def stockInfo(bot, update):
def dividend(bot, update): def dividend(bot, update):
message = update.message.text message = update.message.text
chat_id = update.message.chat_id chat_id = update.message.chat_id
print("div")
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)

View File

@ -38,7 +38,7 @@ def stockNews(ticker):
"""Makes a bunch of strings that are links to news websites for an input ticker""" """Makes a bunch of strings that are links to news websites for an input ticker"""
print("Gather News on " + ticker) print("Gather News on " + ticker)
newsLink = "https://api.iextrading.com/1.0/stock/{}/news/last/5".format(ticker) newsLink = f"https://api.iextrading.com/1.0/stock/{ticker}/news/last/5"
with urllib.request.urlopen(newsLink) as url: with urllib.request.urlopen(newsLink) as url:
data = json.loads(url.read().decode()) data = json.loads(url.read().decode())
@ -52,12 +52,12 @@ def stockNews(ticker):
def stockLogo(ticker): def stockLogo(ticker):
"""returns a png of an input ticker""" """returns a png of an input ticker"""
logoURL = "https://g.foolcdn.com/art/companylogos/mark/" + ticker + ".png" logoURL = f"https://g.foolcdn.com/art/companylogos/mark/{ticker}.png"
return logoURL return logoURL
def stockInfo(ticker): def stockInfo(ticker):
infoURL = "https://api.iextrading.com/1.0/stock/{}/stats".format(ticker) infoURL = f"https://api.iextrading.com/1.0/stock/{ticker}/stats"
with urllib.request.urlopen(infoURL) as url: with urllib.request.urlopen(infoURL) as url:
data = json.loads(url.read().decode()) data = json.loads(url.read().decode())
@ -99,9 +99,7 @@ def stockDividend(ticker):
h, m = divmod(h, 3600) h, m = divmod(h, 3600)
m, s = divmod(m, 60) m, s = divmod(m, 60)
countdownMessage = "\n\nThe dividend is in: {:.0f} Days {:.0f} Hours {:.0f} Minutes {:.0f} Seconds.".format( countdownMessage = f"\n\nThe dividend is in: {d:.0f} Days {h:.0f} Hours {m:.0f} Minutes {s:.0f} Seconds."
d, h, m, s
)
message = line1 + countdownMessage message = line1 + countdownMessage
return message return message