mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 21:12:16 +00:00
07c053fcff
* update * init * req * fix * req * meh * updates * syspath * update * local plotly, and scope logic * refactor * oops typo * plot dpi fixes * fix tests * fix * poetry update, reqs * some adjustments * added plot settings for scat/options Co-authored-by: Colin Delahunty <72827203+colin99d@users.noreply.github.com> Co-authored-by: didierlopes.eth <dro.lopes@campus.fct.unl.pt>
104 lines
2.9 KiB
Python
Executable File
104 lines
2.9 KiB
Python
Executable File
import io
|
|
import logging
|
|
from datetime import datetime, timedelta
|
|
|
|
from matplotlib import pyplot as plt
|
|
|
|
from bots import config_discordbot as cfg
|
|
from bots.helpers import image_border, load
|
|
from bots.imps import BOT_PLOT_DPI, bot_plot_scale
|
|
from openbb_terminal.common.technical_analysis import volatility_model
|
|
from openbb_terminal.decorators import log_start_end
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def donchian_command(
|
|
ticker="", upper_length="25", lower_length="100", start="", end=""
|
|
):
|
|
"""Displays chart with donchian channel [Yahoo Finance]"""
|
|
|
|
# Debug
|
|
if cfg.DEBUG:
|
|
logger.debug(
|
|
"ta donchian %s %s %s %s %s",
|
|
ticker,
|
|
upper_length,
|
|
lower_length,
|
|
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 upper_length.lstrip("-").isnumeric():
|
|
raise Exception("Number has to be an integer")
|
|
upper_length = float(upper_length)
|
|
if not lower_length.lstrip("-").isnumeric():
|
|
raise Exception("Number has to be an integer")
|
|
lower_length = float(lower_length)
|
|
|
|
ticker = ticker.upper()
|
|
df_stock = 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 = volatility_model.donchian(
|
|
df_stock["High"], df_stock["Low"], upper_length, lower_length
|
|
)
|
|
df_ta = df_ta.fillna(0.0)
|
|
|
|
# Output Data
|
|
fig, ax = plt.subplots(figsize=bot_plot_scale(), dpi=BOT_PLOT_DPI)
|
|
ax.plot(df_stock.index, df_stock["Adj Close"].values, lw=3)
|
|
ax.plot(df_ta.index, df_ta.iloc[:, 0].values, lw=1.5, label="upper")
|
|
ax.plot(df_ta.index, df_ta.iloc[:, 1].values, lw=1.5, ls="--")
|
|
ax.plot(df_ta.index, df_ta.iloc[:, 2].values, lw=1.5, label="lower")
|
|
ax.set_title(f"{ticker} donchian")
|
|
ax.set_xlim(df_stock.index[0], df_stock.index[-1])
|
|
ax.set_xlabel("Time")
|
|
ax.set_ylabel("Price ($)")
|
|
|
|
ax.legend([ticker, df_ta.columns[0], df_ta.columns[1], df_ta.columns[2]])
|
|
ax.fill_between(
|
|
df_ta.index,
|
|
df_ta.iloc[:, 0].values,
|
|
df_ta.iloc[:, 2].values,
|
|
alpha=0.1,
|
|
)
|
|
ax.grid(b=True, which="major", color="#666666", linestyle="-")
|
|
|
|
plt.gcf().autofmt_xdate()
|
|
fig.tight_layout(pad=1)
|
|
|
|
plt.legend()
|
|
imagefile = "ta_donchian.png"
|
|
|
|
dataBytesIO = io.BytesIO()
|
|
plt.savefig(dataBytesIO)
|
|
plt.close("all")
|
|
|
|
dataBytesIO.seek(0)
|
|
imagefile = image_border(imagefile, base64=dataBytesIO)
|
|
|
|
return {
|
|
"title": f"Stocks: Donchian-Channels {ticker}",
|
|
"imagefile": imagefile,
|
|
}
|