Anson/Simple-Discord-Stock-Bot
1
0
Fork 0
mirror of https://gitlab.com/simple-stock-bots/simple-discord-stock-bot.git synced 2025-06-16 15:17:29 +00:00
Code Releases Activity

Added more comments and did some formatting

Browse Source
This commit is contained in:
Anson 2019-01-09 18:19:58 -07:00
parent 602a1aff19
commit f0d41a6310
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
stockBot.py
View File

@ -32,33 +32,41 @@ async def on_message(message):
channel = message.channel channel = message.channel
try: try:
# regex to find tickers in messages # 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})', content) tickers = re.findall('[$](\w{1,4})', content)
# get ticker information from bravos api # 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) + \ url = 'https://data.bravos.co/v1/quote?symbols=' + ",".join(tickers) + \
'&apikey=' + BRAVOS_API + '&format=json' '&apikey=' + BRAVOS_API + '&format=json'
# load json data from url as an object # load json data from url as an object
with urllib.request.urlopen(url) as url: with urllib.request.urlopen(url) as url:
data = json.loads(url.read().decode()) data = json.loads(url.read().decode())
for ticker in tickers: # iterate through the tickers and print relevant info one message at a time 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 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'] nameTicker = data[ticker.upper()]['name']
# Get Stock Ticker price from Object
priceTicker = data[ticker.upper()]['price'] priceTicker = data[ticker.upper()]['price']
# Checks if !news is called, and prints news embed if it is
if content.startswith('!news'): if content.startswith('!news'):
embed = displayembed(ticker, nameTicker, priceTicker) embed = displayembed(ticker, nameTicker, priceTicker)
await client.send_message(channel, embed=embed) await client.send_message(channel, embed=embed)
else: else: # If news embed isnt called, print normal stock price
await client.send_message(channel, 'The current stock price of ' + nameTicker + ' is $**' + str(priceTicker) + '**') await client.send_message(channel, 'The current stock price of ' + nameTicker + ' is $**' + str(priceTicker) + '**')
except KeyError:
except KeyError: # If searching for the ticker in loaded data fails, then Bravos didnt provide it, so tell the user.
await client.send_message(channel, ticker.upper() + ' does not exist.') await client.send_message(channel, ticker.upper() + ' does not exist.')
pass pass
except: except:
pass pass
# Prints an embed full of news about listed stock
def displayembed(ticker, nameTicker, priceTicker): def displayembed(ticker, nameTicker, priceTicker):
embed = discord.Embed( embed = discord.Embed(