OpenBB/discordbot/stocks/options/hist.py

import os
import asyncio
import discord

import mplfinance as mpf
from gamestonk_terminal.config_terminal import TRADIER_TOKEN
from gamestonk_terminal.stocks.options import tradier_model
import discordbot.config_discordbot as cfg


async def hist_command(
    ctx, ticker: str = None, expiry: str = "", strike: float = None, put: bool = False
):
    """Plot historical option prices

    Parameters
    ----------
    ticker: str
        Stock ticker
    expiry: str
        accepts 0-9
        0 being weeklies
        1+ for weeks out
        prompts reaction helper if empty
    strike: float
        Option strike price
    put: bool
        c for call
        p for put
    """

    async with ctx.typing():
        await asyncio.sleep(0.2)
        try:

            # Debug
            if cfg.DEBUG:
                print(f"!stocks.opt.hist {ticker} {strike} {put} {expiry}")

            error = ""
            if TRADIER_TOKEN == "REPLACE_ME":
                raise Exception("Tradier Token is required")

            # Check for argument
            if ticker is None:
                raise Exception("Stock ticker is required")

            if strike is None or put == "":
                raise Exception(
                    'A strike and c/p is required\n```bash\n"!stocks.opt.hist {ticker} {strike} {c/p}"```'
                )

            chain_id = None

            df_hist = tradier_model.get_historical_options(
                ticker, expiry, strike, put, chain_id
            )

            op_type = ["call", "put"][put]

            mc = mpf.make_marketcolors(
                up="green",
                down="red",
                edge="black",
                wick="black",
                volume="in",
                ohlc="i",
            )

            s = mpf.make_mpf_style(
                base_mpl_style="seaborn",
                marketcolors=mc,
                gridstyle=":",
                y_on_right=True,
            )

            mpf.plot(
                df_hist,
                type="candle",
                volume=True,
                title=f"\n{ticker.upper()} {strike} {op_type} expiring {expiry} Historical",
                tight_layout=True,
                style=s,
                figratio=(10, 7),
                figscale=1.10,
                figsize=(12, 5),
                update_width_config=dict(
                    candle_linewidth=1.0, candle_width=0.8, volume_linewidth=1.0
                ),
                savefig="opt_hist.png",
            )

            imagefile = "opt_hist.png"
            image = discord.File(imagefile)

            if cfg.DEBUG:
                print(f"Image: {imagefile}")
            title = f"{ticker.upper()} {strike} {op_type} expiring {expiry} Historical"
            embed = discord.Embed(title=title, colour=cfg.COLOR)
            embed.set_image(url="attachment://opt_hist.png")
            embed.set_author(
                name=cfg.AUTHOR_NAME,
                icon_url=cfg.AUTHOR_ICON_URL,
            )
            os.remove("opt_hist.png")

            await ctx.send(embed=embed, file=image)

        except TypeError:
            error = (
                'Invalid format.\n\nUse:\n```bash\n"!stocks.opt.hist {ticker} {strike} {c/p}"```'
                '```bash\n"!stocks.opt {ticker} {expiration} {strike} {c/p}"```'
            )
        except Exception as e:
            error = str(e)

        finally:
            if error != "":
                embed = discord.Embed(
                    title="ERROR Options: History",
                    colour=cfg.COLOR,
                    description=error,
                )
                embed.set_author(
                    name=cfg.AUTHOR_NAME,
                    icon_url=cfg.AUTHOR_ICON_URL,
                )

                await ctx.send(embed=embed, delete_after=30.0)