added plotting of historical pricing

bot.py

@@ -1,7 +1,9 @@
 # Works with Python 3.8
+import io
 import logging
 import os
 
+import mplfinance as mpf
 import telegram
 from telegram import InlineQueryResultArticle, InputTextMessageContent
 from telegram.ext import (
@@ -117,6 +119,10 @@ def search(update, context):
     message = update.message.text
     chat_id = update.message.chat_id
 
+    usage = """
+    To use this command 
+    """
+
     queries = s.search_symbols(message)[:6]
     if queries:
         reply = "*Search Results:*\n`$ticker: Company Name`\n"
@@ -125,6 +131,42 @@ def search(update, context):
         update.message.reply_text(text=reply, parse_mode=telegram.ParseMode.MARKDOWN)
 
 
+def historical(update, context):
+    # TODO: Document usage of this command. https://iexcloud.io/docs/api/#historical-prices
+
+    message = update.message.text
+    chat_id = update.message.chat_id
+
+    try:
+        cmd, symbol, period = message.split(" ")
+        symbol = symbol.replace("$", "")
+    except TypeError:
+        update.message.reply_text(
+            text="example: /historical tsla 5d", parse_mode=telegram.ParseMode.MARKDOWN
+        )
+
+    h = s.historical_reply(symbol, period)
+    if h.empty:
+        update.message.reply_text(
+            text="Invalid symbol or time period, please see `/help` or `/historical help` for usage details.",
+            parse_mode=telegram.ParseMode.MARKDOWN,
+        )
+
+    context.bot.send_chat_action(
+        chat_id=chat_id, action=telegram.ChatAction.UPLOAD_PHOTO
+    )
+
+    buf = io.BytesIO()
+    mpf.plot(h, type="candle", savefig=dict(fname=buf, dpi=400))
+    buf.seek(0)
+
+    update.message.reply_photo(
+        photo=buf,
+        caption=f"*{period} chart for {symbol}*",
+        parse_mode=telegram.ParseMode.MARKDOWN,
+    )
+
+
 def inline_query(update, context):
     """
     Handles inline query.
@@ -176,7 +218,8 @@ def main():
     dp.add_handler(CommandHandler("news", news))
     dp.add_handler(CommandHandler("info", info))
     dp.add_handler(CommandHandler("search", search))
-
+    dp.add_handler(CommandHandler("historical", historical))
+    dp.add_handler(CommandHandler("h", historical))
     # on noncommand i.e message - echo the message on Telegram
     dp.add_handler(MessageHandler(Filters.text, symbol_detect))
 

functions.py

@@ -5,8 +5,8 @@ from datetime import datetime, timedelta
 
 import pandas as pd
 import requests as r
-from fuzzywuzzy import fuzz
 import schedule
+from fuzzywuzzy import fuzz
 
 
 class Symbol:
@@ -211,3 +211,19 @@ Market data is provided by [IEX Cloud](https://iexcloud.io)
                 ] = f"No information found for: {symbol}\nEither today is boring or the symbol does not exist."
 
         return infoMessages
+
+    def historical_reply(self, symbol: str, period: str):
+        if symbol.upper() not in list(self.symbol_list["symbol"]):
+            return pd.DataFrame()
+
+        time_periods = ["max", "5y", "2y", "1y", "ytd", "6m", "3m", "1m", "5d"]
+        if period not in time_periods:
+            return pd.DataFrame()
+
+        IEXurl = f"https://cloud.iexapis.com/stable/stock/{symbol}/chart/{period}?token={self.IEX_TOKEN}"
+        response = r.get(IEXurl)
+        if response.status_code == 200:
+            df = pd.DataFrame(response.json())
+            df["date"] = pd.to_datetime(df["date"])
+            df = df.set_index("date")
+            return df

requirements.txt

@@ -4,3 +4,4 @@ pandas==1.0.3
 fuzzywuzzy==0.18.0
 python-Levenshtein==0.12.0
 schedule==0.6.0
+mplfinance==0.12.6a3