mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-08 02:11:44 +00:00
4458c98714
* Discord Bot: Added ta context * Fixed linting * TA Menu fix * Fix linting * More linting fixes * Debugging * Fixing review * Fixing review
127 lines
3.9 KiB
Python
127 lines
3.9 KiB
Python
import discord
|
|
import config_discordbot as cfg
|
|
from discordbot import gst_imgur
|
|
from datetime import datetime, timedelta
|
|
from matplotlib import pyplot as plt
|
|
import os
|
|
import helpers
|
|
|
|
from gamestonk_terminal.helper_funcs import plot_autoscale
|
|
from gamestonk_terminal.common.technical_analysis import volume_model
|
|
from gamestonk_terminal.config_plot import PLOT_DPI
|
|
|
|
|
|
async def ad_command(ctx, ticker="", open="False", start="", end=""):
|
|
"""Displays chart with accumulation/distribution line [Yahoo Finance]"""
|
|
|
|
try:
|
|
|
|
# Debug
|
|
if cfg.DEBUG:
|
|
print(f"!stocks.ta.ad {ticker} {open} {start} {end}")
|
|
|
|
# Check for argument
|
|
if ticker == "":
|
|
raise Exception("Stock ticker is required")
|
|
|
|
if start == "":
|
|
start = datetime.now() - timedelta(days=365)
|
|
else:
|
|
start = datetime.strptime(start, cfg.DATE_FORMAT)
|
|
|
|
if end == "":
|
|
end = datetime.now()
|
|
else:
|
|
end = datetime.strptime(end, cfg.DATE_FORMAT)
|
|
|
|
if open == "false" or open == "False" or open == "FALSE":
|
|
open = False
|
|
elif open == "true" or open == "True" or open == "TRUE":
|
|
open = True
|
|
else:
|
|
raise Exception("open argument has to be true or false")
|
|
|
|
ticker = ticker.upper()
|
|
df_stock = helpers.load(ticker, start)
|
|
if df_stock.empty:
|
|
raise Exception("Stock ticker is invalid")
|
|
|
|
# Retrieve Data
|
|
df_stock = df_stock.loc[(df_stock.index >= start) & (df_stock.index < end)]
|
|
|
|
bar_colors = [
|
|
"r" if x[1].Open < x[1].Close else "g" for x in df_stock.iterrows()
|
|
]
|
|
|
|
divisor = 1_000_000
|
|
df_vol = df_stock["Volume"].dropna()
|
|
df_vol = df_vol.values / divisor
|
|
df_ta = volume_model.ad(df_stock, open)
|
|
df_cal = df_ta.values
|
|
df_cal = df_cal / divisor
|
|
|
|
fig, axes = plt.subplots(
|
|
3,
|
|
1,
|
|
gridspec_kw={"height_ratios": [2, 1, 1]},
|
|
figsize=plot_autoscale(),
|
|
dpi=PLOT_DPI,
|
|
)
|
|
ax = axes[0]
|
|
ax.plot(df_stock.index, df_stock["Adj Close"].values, "k", lw=2)
|
|
ax.set_title(f"{ticker} AD")
|
|
ax.set_xlim(df_stock.index[0], df_stock.index[-1])
|
|
ax.set_ylabel("Price")
|
|
ax.grid(b=True, which="major", color="#666666", linestyle="-")
|
|
|
|
ax2 = axes[1]
|
|
ax2.set_ylabel("Volume [M]")
|
|
ax2.bar(
|
|
df_stock.index,
|
|
df_vol,
|
|
color=bar_colors,
|
|
alpha=0.8,
|
|
width=0.3,
|
|
)
|
|
ax2.set_xlim(df_stock.index[0], df_stock.index[-1])
|
|
|
|
ax3 = axes[2]
|
|
ax3.set_ylabel("A/D [M]")
|
|
ax3.set_xlabel("Time")
|
|
ax3.plot(df_ta.index, df_cal, "b", lw=1)
|
|
ax3.set_xlim(df_stock.index[0], df_stock.index[-1])
|
|
ax3.axhline(0, linewidth=2, color="k", ls="--")
|
|
ax3.grid(b=True, which="major", color="#666666", linestyle="-")
|
|
|
|
plt.gcf().autofmt_xdate()
|
|
fig.tight_layout(pad=1)
|
|
|
|
plt.savefig("ta_ad.png")
|
|
uploaded_image = gst_imgur.upload_image("ta_ad.png", title="something")
|
|
image_link = uploaded_image.link
|
|
if cfg.DEBUG:
|
|
print(f"Image URL: {image_link}")
|
|
title = "Stocks: Accumulation/Distribution Line " + ticker
|
|
embed = discord.Embed(title=title, colour=cfg.COLOR)
|
|
embed.set_author(
|
|
name=cfg.AUTHOR_NAME,
|
|
icon_url=cfg.AUTHOR_ICON_URL,
|
|
)
|
|
embed.set_image(url=image_link)
|
|
os.remove("ta_ad.png")
|
|
|
|
await ctx.send(embed=embed)
|
|
|
|
except Exception as e:
|
|
embed = discord.Embed(
|
|
title="ERROR Stocks: Accumulation/Distribution Line",
|
|
colour=cfg.COLOR,
|
|
description=e,
|
|
)
|
|
embed.set_author(
|
|
name=cfg.AUTHOR_NAME,
|
|
icon_url=cfg.AUTHOR_ICON_URL,
|
|
)
|
|
|
|
await ctx.send(embed=embed)
|