From 59015ebf82e9b9ac1977d1ac72e8ef49bb2a4b51 Mon Sep 17 00:00:00 2001
From: Giovani <giovani@boldquest.net>
Date: Wed, 28 Jul 2021 21:30:14 -0400
Subject: [PATCH] feat: add ability to query stock prices

---
 pyvis.py | 39 +++++++++++++++++++++++++++++++++++----
 1 file changed, 35 insertions(+), 4 deletions(-)

diff --git a/pyvis.py b/pyvis.py
index 13c5268..7866393 100644
--- a/pyvis.py
+++ b/pyvis.py
@@ -5,11 +5,12 @@ import os
 from discord.ext import commands
 from dotenv import load_dotenv
 
-# load environment variables
+# load environment variables from .env file
 load_dotenv()
 DISCORD_BOT_TOKEN = os.getenv('DISCORD_BOT_TOKEN')
 DISCORD_GUILD_NAME = os.getenv('DISCORD_GUILD_NAME')
 LUNAR_CRUSH_API_KEY = os.getenv('LUNAR_CRUSH_API_KEY')
+ALPHAVANTAGE_API_KEY = os.getenv('ALPHAVANTAGE_API_KEY')
 
 bot = commands.Bot(command_prefix='!')
 
@@ -26,19 +27,49 @@ async def on_ready():
 async def resources(ctx):
     await ctx.send("Some resources for you here!")
 
+@bot.command(name="stock")
+async def stock(ctx, *tickers):
+    if len(tickers) == 0:
+        tickers = ["AAPL", "GOOG", "MSFT", "AMZN"]
+
+    # msg response setup
+    msg_response = "Stock prices for today!\n"
+    stock_found = False
+
+    # make the call for stock prices
+    for ticker in tickers:
+        http_response = requests.get(f'https://www.alphavantage.co/query?function=GLOBAL_QUOTE&symbol={ticker}&apikey={ALPHAVANTAGE_API_KEY}')
+        stock_data = json.loads(http_response.text)
+
+        # check if stock exists
+        if 'Global Quote' not in stock_data:
+            continue
+
+        # get the stock price
+        stock_price = stock_data['Global Quote']['05. price']
+        msg_response += f'**{ticker}**: ${round(float(stock_price), 2)}\n'  
+        stock_found = True
+
+
+    if not stock_found:
+        await ctx.send("No stock data found for given tickers!")
+        return
+
+    await ctx.send(msg_response) 
+
 @bot.command(name="crypto")
 async def crypto(ctx, *tickers):
     # default tickers
     if len(tickers) == 0:
-        tickers = ['BTC', 'ETH', 'LTC']
+        tickers = ["BTC", "ETH", "LTC"]
 
     # make the call for prices
-    http_response = requests.get("https://api.lunarcrush.com/v2?data=assets&key=" + LUNAR_CRUSH_API_KEY + "&symbol=" + ",".join(tickers))
+    http_response = requests.get(f"https://api.lunarcrush.com/v2?data=assets&key={LUNAR_CRUSH_API_KEY}&symbol={','.join(tickers)}")
     data = json.loads(http_response.text)
 
     # extract prices and derive message response
     msg_response = "Cryptocurrency prices for today!\n"
-    msg_response += "\n".join([f"**{item['symbol']}**: ${int(item['price'])}" for item in data["data"]])
+    msg_response += "\n".join([f"**{item['symbol']}**: ${round(float(item['price']), 2)}" for item in data["data"]])
 
     await ctx.send(msg_response)