mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-29 05:21:48 +00:00
119 lines
3.9 KiB
Python
119 lines
3.9 KiB
Python
import os
|
|
from datetime import datetime, timedelta
|
|
import discord
|
|
from matplotlib import pyplot as plt
|
|
|
|
from gamestonk_terminal.helper_funcs import plot_autoscale
|
|
from gamestonk_terminal.common.technical_analysis import trend_indicators_model
|
|
from gamestonk_terminal.config_plot import PLOT_DPI
|
|
|
|
import discordbot.config_discordbot as cfg
|
|
from discordbot.run_discordbot import gst_imgur
|
|
import discordbot.helpers
|
|
|
|
|
|
async def adx_command(
|
|
ctx, ticker="", length="14", scalar="100", drift="1", start="", end=""
|
|
):
|
|
"""Displays chart with average directional movement index [Yahoo Finance]"""
|
|
|
|
try:
|
|
|
|
# Debug
|
|
if cfg.DEBUG:
|
|
print(f"!stocks.ta.adx {ticker} {length} {scalar} {drift} {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 not length.lstrip("-").isnumeric():
|
|
raise Exception("Number has to be an integer")
|
|
length = float(length)
|
|
if not scalar.lstrip("-").isnumeric():
|
|
raise Exception("Number has to be an integer")
|
|
scalar = float(scalar)
|
|
if not drift.lstrip("-").isnumeric():
|
|
raise Exception("Number has to be an integer")
|
|
drift = float(drift)
|
|
|
|
ticker = ticker.upper()
|
|
df_stock = discordbot.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)]
|
|
|
|
df_ta = trend_indicators_model.adx("1440min", df_stock, length, scalar, drift)
|
|
|
|
# Output Data
|
|
fig, ax = plt.subplots(2, 1, figsize=plot_autoscale(), dpi=PLOT_DPI)
|
|
ax0 = ax[0]
|
|
ax0.plot(df_stock.index, df_stock["Close"].values, "k", lw=2)
|
|
ax0.set_title(f"Average Directional Movement Index (ADX) on {ticker}")
|
|
ax0.set_xlim(df_stock.index[0], df_stock.index[-1])
|
|
ax0.set_ylabel("Share Price ($)")
|
|
ax0.grid(b=True, which="major", color="#666666", linestyle="-")
|
|
|
|
ax1 = ax[1]
|
|
ax1.plot(df_ta.index, df_ta.iloc[:, 0].values, "b", lw=2)
|
|
ax1.plot(df_ta.index, df_ta.iloc[:, 1].values, "g", lw=1)
|
|
ax1.plot(df_ta.index, df_ta.iloc[:, 2].values, "r", lw=1)
|
|
ax1.set_xlim(df_stock.index[0], df_stock.index[-1])
|
|
ax1.axhline(25, linewidth=3, color="k", ls="--")
|
|
ax1.legend(
|
|
[
|
|
f"ADX ({df_ta.columns[0]})",
|
|
f"+DI ({df_ta.columns[1]})",
|
|
f"- DI ({df_ta.columns[2]})",
|
|
],
|
|
loc="upper left",
|
|
)
|
|
ax1.set_xlabel("Time")
|
|
ax1.grid(b=True, which="major", color="#666666", linestyle="-")
|
|
|
|
ax1.set_ylim([0, 100])
|
|
|
|
fig.tight_layout()
|
|
plt.gcf().autofmt_xdate()
|
|
|
|
plt.savefig("ta_adx.png")
|
|
uploaded_image = gst_imgur.upload_image("ta_adx.png", title="something")
|
|
image_link = uploaded_image.link
|
|
if cfg.DEBUG:
|
|
print(f"Image URL: {image_link}")
|
|
title = "Stocks: Average-Directional-Movement-Index " + 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_adx.png")
|
|
|
|
await ctx.send(embed=embed)
|
|
|
|
except Exception as e:
|
|
embed = discord.Embed(
|
|
title="ERROR Stocks: Average-Directional-Movement-Index",
|
|
colour=cfg.COLOR,
|
|
description=e,
|
|
)
|
|
embed.set_author(
|
|
name=cfg.AUTHOR_NAME,
|
|
icon_url=cfg.AUTHOR_ICON_URL,
|
|
)
|
|
|
|
await ctx.send(embed=embed)
|