mirror of
https://gitlab.com/simple-stock-bots/simple-telegram-stock-bot.git
synced 2025-06-16 15:06:53 +00:00
Transfered over Discord code
This commit is contained in:
parent
6dcf4c1a59
commit
1f62e472df
119
stockBot.py
Normal file
119
stockBot.py
Normal file
@ -0,0 +1,119 @@
|
|||||||
|
# Work with Python 3.6
|
||||||
|
import json
|
||||||
|
import logging
|
||||||
|
import re
|
||||||
|
import urllib.request
|
||||||
|
|
||||||
|
import telegram
|
||||||
|
from telegram.ext import CommandHandler, Filters, MessageHandler, Updater
|
||||||
|
|
||||||
|
import credentials
|
||||||
|
|
||||||
|
# Make sure to update credentials.py with your secrets
|
||||||
|
TOKEN = credentials.secrets['TOKEN']
|
||||||
|
BRAVOS_API = credentials.secrets['BRAVOS_API']
|
||||||
|
|
||||||
|
"""Simple Bot to reply to Telegram messages.
|
||||||
|
This program is dedicated to the public domain under the CC0 license.
|
||||||
|
This Bot uses the Updater class to handle the bot.
|
||||||
|
First, a few handler functions are defined. Then, those functions are passed to
|
||||||
|
the Dispatcher and registered at their respective places.
|
||||||
|
Then, the bot is started and runs until we press Ctrl-C on the command line.
|
||||||
|
Usage:
|
||||||
|
Basic Echobot example, repeats messages.
|
||||||
|
Press Ctrl-C on the command line or send a signal to the process to stop the
|
||||||
|
bot.
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
# Enable logging
|
||||||
|
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||||
|
level=logging.INFO)
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
# Define a few command handlers. These usually take the two arguments bot and
|
||||||
|
# update. Error handlers also receive the raised TelegramError object in error.
|
||||||
|
def start(bot, update):
|
||||||
|
"""Send a message when the command /start is issued."""
|
||||||
|
update.message.reply_text('I am started and ready to go!')
|
||||||
|
|
||||||
|
|
||||||
|
def help(bot, update):
|
||||||
|
"""Send a message when the command /help is issued."""
|
||||||
|
update.message.reply_text("I don't know how to help yet!")
|
||||||
|
|
||||||
|
|
||||||
|
def stockInfo(bot, update):
|
||||||
|
message = update.message.text
|
||||||
|
try:
|
||||||
|
# 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)
|
||||||
|
|
||||||
|
# get ticker information from bravos api, turns tickers into comma separated list so that only one api call is needed per message
|
||||||
|
url = 'https://data.bravos.co/v1/quote?symbols=' + ",".join(tickers) + \
|
||||||
|
'&apikey=' + BRAVOS_API + '&format=json'
|
||||||
|
|
||||||
|
# load json data from url as an object
|
||||||
|
with urllib.request.urlopen(url) as url:
|
||||||
|
data = json.loads(url.read().decode())
|
||||||
|
|
||||||
|
for ticker in tickers: # iterate through the tickers and print relevant info one message at a time
|
||||||
|
try: # checks if data is a valid ticker, if it is not tells the user
|
||||||
|
|
||||||
|
# Get Stock ticker name from Data Object
|
||||||
|
nameTicker = data[ticker.upper()]['name']
|
||||||
|
# Get Stock Ticker price from Object
|
||||||
|
priceTicker = data[ticker.upper()]['price']
|
||||||
|
|
||||||
|
# Checks if !news is called, and prints news embed if it is
|
||||||
|
if message.startswith('!news'):
|
||||||
|
update.message.reply_text(
|
||||||
|
text='The current stock price of ' + nameTicker + ' is $<b>' + str(priceTicker) + '</b>', parse_mode=telegram.ParseMode.HTML)
|
||||||
|
else: # If news embed isnt called, print normal stock price
|
||||||
|
update.message.reply_text(
|
||||||
|
text='The current stock price of ' + nameTicker + ' is $<b>' + str(priceTicker) + '</b>', parse_mode=telegram.ParseMode.HTML)
|
||||||
|
|
||||||
|
except KeyError: # If searching for the ticker in loaded data fails, then Bravos didnt provide it, so tell the user.
|
||||||
|
update.message.reply_text(
|
||||||
|
ticker.upper() + ' does not exist.')
|
||||||
|
pass
|
||||||
|
except:
|
||||||
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def error(bot, update, error):
|
||||||
|
"""Log Errors caused by Updates."""
|
||||||
|
logger.warning('Update "%s" caused error "%s"', update, error)
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
"""Start the bot."""
|
||||||
|
# Create the EventHandler and pass it your bot's token.
|
||||||
|
updater = Updater(TOKEN)
|
||||||
|
|
||||||
|
# Get the dispatcher to register handlers
|
||||||
|
dp = updater.dispatcher
|
||||||
|
|
||||||
|
# on different commands - answer in Telegram
|
||||||
|
dp.add_handler(CommandHandler("start", start))
|
||||||
|
dp.add_handler(CommandHandler("help", help))
|
||||||
|
|
||||||
|
# on noncommand i.e message - echo the message on Telegram
|
||||||
|
dp.add_handler(MessageHandler(Filters.text, stockInfo))
|
||||||
|
|
||||||
|
# log all errors
|
||||||
|
dp.add_error_handler(error)
|
||||||
|
|
||||||
|
# Start the Bot
|
||||||
|
updater.start_polling()
|
||||||
|
|
||||||
|
# Run the bot until you press Ctrl-C or the process receives SIGINT,
|
||||||
|
# SIGTERM or SIGABRT. This should be used most of the time, since
|
||||||
|
# start_polling() is non-blocking and will stop the bot gracefully.
|
||||||
|
updater.idle()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
main()
|
Loading…
x
Reference in New Issue
Block a user