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

updated bot to work with new functions

This commit is contained in:
Anson 2019-05-28 05:06:59 -07:00
parent c49a2c321e
commit a195c87b6e

29
bot.py
View File

@ -40,35 +40,33 @@ def tickerDetect(bot, update):
chat_id = update.message.chat_id chat_id = update.message.chat_id
tickers = getTickers(message) tickers = getTickers(message)
# Let user know bot is working
if tickers: if tickers:
# 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)
replies = tickerDataReply(tickers)
for symbol, reply in replies.items(): for symbol, reply in tickerDataReply(tickers).items():
update.message.reply_text(text=reply, parse_mode=telegram.ParseMode.MARKDOWN) update.message.reply_text(
text=reply, parse_mode=telegram.ParseMode.MARKDOWN
)
def dividend(bot, update): def dividend(bot, update):
""" """
This Functions is incomplete. waits for /dividend or /div command and then finds dividend info on that ticker.
""" """
message = update.message.text message = update.message.text
chat_id = update.message.chat_id chat_id = update.message.chat_id
tickers = getTickers(message)
# regex to find tickers in messages, looks for up to 4 word characters following a dollar sign and captures the 4 word characters if tickers:
tickers = re.findall(TICKER_REGEX, message) 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 ticker in tickers: for symbol, reply in tickerDividend(tickers).items():
message = tickerDividend(ticker)
update.message.reply_text(text=message, parse_mode=telegram.ParseMode.MARKDOWN)
update.message.reply_text(
def error(bot, update, error): text=reply, parse_mode=telegram.ParseMode.MARKDOWN
"""Log Errors caused by Updates.""" )
logger.warning('Update "%s" caused error "%s"', update, error)
def main(): def main():
@ -83,6 +81,7 @@ def main():
dp.add_handler(CommandHandler("start", start)) dp.add_handler(CommandHandler("start", start))
dp.add_handler(CommandHandler("help", help)) dp.add_handler(CommandHandler("help", help))
dp.add_handler(CommandHandler("dividend", dividend)) dp.add_handler(CommandHandler("dividend", dividend))
dp.add_handler(CommandHandler("div", dividend))
# 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, tickerDetect))