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>
134 lines
3.4 KiB
Python
Executable File
134 lines
3.4 KiB
Python
Executable File
import logging
|
|
|
|
import pandas_ta as ta
|
|
import plotly.graph_objects as go
|
|
|
|
from bots import imps, load_candle
|
|
from openbb_terminal.decorators import log_start_end
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# pylint: disable=R0912
|
|
# pylint: disable=R0913
|
|
@log_start_end(log=logger)
|
|
def candle_command(
|
|
ticker: str = "",
|
|
interval: int = 15,
|
|
past_days: int = 0,
|
|
extended_hours: bool = False,
|
|
start="",
|
|
end="",
|
|
news: bool = False,
|
|
heikin_candles: bool = False,
|
|
trendline: bool = False,
|
|
vwap: bool = False,
|
|
):
|
|
"""Shows candle plot of loaded ticker or crypto. [Source: Yahoo Finance or Binance API]
|
|
|
|
Parameters
|
|
----------
|
|
ticker : Stock Ticker
|
|
interval : Chart Minute Interval, 1440 for Daily
|
|
past_days: Past Days to Display. Default: 0(Not for Daily)
|
|
extended_hours: Display Pre/After Market Hours. Default: False
|
|
start: YYYY-MM-DD format
|
|
end: YYYY-MM-DD format
|
|
news: Display clickable news markers on interactive chart. Default: False
|
|
heikin_candles: Heikin Ashi candles. Default: False
|
|
trendline: Display trendline on Daily chart. Default: False
|
|
"""
|
|
|
|
logger.info(
|
|
"candle %s %s %s %s %s %s %s %s %s",
|
|
ticker,
|
|
interval,
|
|
past_days,
|
|
extended_hours,
|
|
start,
|
|
end,
|
|
news,
|
|
heikin_candles,
|
|
trendline,
|
|
)
|
|
|
|
# Retrieve Data
|
|
df_stock, start, end, bar_start = load_candle.stock_data(
|
|
ticker=ticker,
|
|
interval=interval,
|
|
past_days=past_days,
|
|
extended_hours=extended_hours,
|
|
start=start,
|
|
end=end,
|
|
news=news,
|
|
heikin_candles=heikin_candles,
|
|
)
|
|
|
|
df_stock = df_stock.loc[(df_stock.index >= start) & (df_stock.index < end)]
|
|
if vwap:
|
|
df_vwap = ta.vwap(
|
|
high=df_stock["High"],
|
|
low=df_stock["Low"],
|
|
close=df_stock["Close"],
|
|
volume=df_stock["Volume"],
|
|
offset=0,
|
|
)
|
|
df_stock = df_stock.join(df_vwap)
|
|
|
|
# Check that loading a stock was not successful
|
|
if df_stock.empty:
|
|
raise Exception(f"No data found for {ticker.upper()}.")
|
|
|
|
# Output Data
|
|
if interval != 1440:
|
|
df_stock = df_stock.loc[(df_stock.index >= bar_start) & (df_stock.index < end)]
|
|
|
|
plot = load_candle.candle_fig(
|
|
df_stock,
|
|
ticker,
|
|
interval,
|
|
extended_hours,
|
|
news,
|
|
bar=bar_start,
|
|
int_bar=interval,
|
|
trendline=trendline,
|
|
)
|
|
title = f"{plot['plt_title']} Chart"
|
|
fig = plot["fig"]
|
|
|
|
if interval != 1440 and vwap:
|
|
fig.add_trace(
|
|
go.Scatter(
|
|
name="VWAP",
|
|
x=df_stock.index,
|
|
y=df_stock["VWAP_D"],
|
|
opacity=0.65,
|
|
line=dict(color="#00e6c3", width=2),
|
|
showlegend=True,
|
|
),
|
|
secondary_y=True,
|
|
)
|
|
|
|
fig.update_layout(
|
|
margin=dict(l=0, r=0, t=40, b=20),
|
|
template=imps.PLT_CANDLE_STYLE_TEMPLATE,
|
|
title=f"<b>{title}</b>",
|
|
title_x=0.5,
|
|
title_font_size=14,
|
|
yaxis_title="",
|
|
)
|
|
imagefile = "candle.png"
|
|
|
|
# Check if interactive settings are enabled
|
|
plt_link = ""
|
|
if imps.INTERACTIVE:
|
|
plt_link = imps.inter_chart(fig, imagefile, callback=True)
|
|
|
|
imagefile = imps.image_border(imagefile, fig=fig)
|
|
|
|
return {
|
|
"title": title,
|
|
"description": plt_link,
|
|
"imagefile": imagefile,
|
|
}
|