mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 13:01:38 +00:00
59d8b36bb0
* Fixes * Update stocks_helper.py * update git-actions set-output to new format * Update stocks_helper.py * Update terminal_helper.py * removed LineAnnotateDrawer from qa_view * lint * few changes * updates * sdk auto gen modules done * Update stocks_helper.py * updates to changed imports, and remove first sdk_modules * Update generate_sdk.py * Update generate_sdk.py * pylint * revert stocks_helper * Update generate_sdk.py * Update sdk.py * Update generate_sdk.py * full auto generation, added sdk.py/controllers creation * missed enable forecasting * added running black in subprocess after sdk files generation completes * removed deleted sdk_arg_logger * comment out tests * property doc fix * clean up * Update generate_sdk.py * make trailmap classes useable for doc generation * Update generate_sdk.py * added lineon to trailmap class for linking to func in markdown * changed lineon to dict * added full_path to trailmap for linking in docs * updated portfolio * feat: initial files * feat: added meta head * feat: added funcdef * added func_def to trailmap attributes for markdown in docs, added missing type hints to covid functions * feat: added view and merged with jaun * Update generate_sdk.py * Update generate_sdk.py * Update generate_sdk.py * Update generate_sdk.py * init * fix returns * fix: random stuff * fix: random * fixed encoding issue on windows * fix: generate tabs * update * Update generate_sdk_markdown.py * Create .pydocstyle.ini * added type hint classes for views * fixes * alt, ba * alt-economy * Update finviz_compare_model.py * fixs * Update substack_model.py * Update generate_sdk.py * last of my section * porfolio * po * Update optimizer_model.py * fixing more things * few more * keys done * update * fixes * Update generate_sdk_markdown.py * Update generate_sdk_markdown.py * mypy forecast fix * Update generate_sdk_markdown.py * Update generate_sdk_markdown.py * Update generate_sdk_markdown.py * fixes * forecast fixes * one more fix * Update coinbase_model.py * Update generate_sdk_markdown.py Co-authored-by: Colin Delahunty <72827203+colin99d@users.noreply.github.com> Co-authored-by: James Maslek <jmaslek11@gmail.com> Co-authored-by: jose-donato <zmcdonato@gmail.com> Co-authored-by: andrewkenreich <andrew.kenreich@gmail.com>
349 lines
9.5 KiB
Python
349 lines
9.5 KiB
Python
"""Volume View"""
|
|
__docformat__ = "numpy"
|
|
|
|
import logging
|
|
import os
|
|
from typing import Optional, List
|
|
|
|
import matplotlib.pyplot as plt
|
|
import pandas as pd
|
|
|
|
from openbb_terminal.config_terminal import theme
|
|
from openbb_terminal.common.technical_analysis import volume_model
|
|
from openbb_terminal.config_plot import PLOT_DPI
|
|
from openbb_terminal.decorators import log_start_end
|
|
from openbb_terminal.helper_funcs import (
|
|
export_data,
|
|
plot_autoscale,
|
|
reindex_dates,
|
|
is_valid_axes_count,
|
|
)
|
|
from openbb_terminal.rich_config import console
|
|
from openbb_terminal.common.technical_analysis import ta_helpers
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def display_ad(
|
|
data: pd.DataFrame,
|
|
use_open: bool = False,
|
|
symbol: str = "",
|
|
export: str = "",
|
|
external_axes: Optional[List[plt.Axes]] = None,
|
|
):
|
|
"""Plots AD technical indicator
|
|
|
|
Parameters
|
|
----------
|
|
data : pd.DataFrame
|
|
Dataframe of ohlc prices
|
|
use_open : bool
|
|
Whether to use open prices in calculation
|
|
symbol : str
|
|
Ticker symbol
|
|
export: str
|
|
Format to export data as
|
|
external_axes : Optional[List[plt.Axes]], optional
|
|
External axes (3 axes are expected in the list), by default None
|
|
"""
|
|
divisor = 1_000_000
|
|
df_vol = data["Volume"] / divisor
|
|
df_vol.name = "Adj Volume"
|
|
df_ta = volume_model.ad(data, use_open)
|
|
# check if AD exists in dataframe
|
|
if "AD" in df_ta.columns:
|
|
df_cal = df_ta["AD"] / divisor
|
|
elif "ADo" in df_ta.columns:
|
|
df_cal = df_ta["ADo"] / divisor
|
|
else:
|
|
console.print("AD not found in dataframe")
|
|
return
|
|
df_cal.name = "Adj AD"
|
|
|
|
plot_data = pd.merge(data, df_vol, how="outer", left_index=True, right_index=True)
|
|
plot_data = pd.merge(
|
|
plot_data, df_ta, how="outer", left_index=True, right_index=True
|
|
)
|
|
plot_data = pd.merge(
|
|
plot_data, df_cal, how="outer", left_index=True, right_index=True
|
|
)
|
|
plot_data = reindex_dates(plot_data)
|
|
|
|
# This plot has 3 axes
|
|
if external_axes is None:
|
|
_, axes = plt.subplots(
|
|
3,
|
|
1,
|
|
sharex=True,
|
|
figsize=plot_autoscale(),
|
|
dpi=PLOT_DPI,
|
|
)
|
|
ax1, ax2, ax3 = axes
|
|
elif is_valid_axes_count(external_axes, 3):
|
|
(ax1, ax2, ax3) = external_axes
|
|
else:
|
|
return
|
|
|
|
close_col = ta_helpers.check_columns(data)
|
|
if close_col is None:
|
|
return
|
|
ax1.plot(plot_data.index, plot_data[close_col].values)
|
|
ax1.set_title(f"{symbol} AD", x=0.04, y=1)
|
|
ax1.set_xlim(plot_data.index[0], plot_data.index[-1])
|
|
ax1.set_ylabel("Price")
|
|
theme.style_primary_axis(
|
|
ax1,
|
|
data_index=plot_data.index.to_list(),
|
|
tick_labels=plot_data["date"].to_list(),
|
|
)
|
|
|
|
ax2.set_ylabel("Volume [M]")
|
|
bar_colors = [
|
|
theme.down_color if x[1].Open < x[1].Close else theme.up_color
|
|
for x in plot_data.iterrows()
|
|
]
|
|
ax2.bar(
|
|
plot_data.index,
|
|
plot_data["Adj Volume"].values,
|
|
color=bar_colors,
|
|
width=theme.volume_bar_width,
|
|
)
|
|
ax2.set_xlim(plot_data.index[0], plot_data.index[-1])
|
|
theme.style_primary_axis(
|
|
ax2,
|
|
data_index=plot_data.index.to_list(),
|
|
tick_labels=plot_data["date"].to_list(),
|
|
)
|
|
|
|
ax3.set_ylabel("A/D [M]")
|
|
ax3.plot(plot_data.index, plot_data["Adj AD"])
|
|
ax3.set_xlim(plot_data.index[0], plot_data.index[-1])
|
|
ax3.axhline(0, linestyle="--")
|
|
theme.style_primary_axis(
|
|
ax3,
|
|
data_index=plot_data.index.to_list(),
|
|
tick_labels=plot_data["date"].to_list(),
|
|
)
|
|
|
|
if external_axes is None:
|
|
theme.visualize_output()
|
|
|
|
export_data(
|
|
export,
|
|
os.path.dirname(os.path.abspath(__file__)).replace("common", "stocks"),
|
|
"ad",
|
|
df_ta,
|
|
)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def display_adosc(
|
|
data: pd.DataFrame,
|
|
fast: int = 3,
|
|
slow: int = 10,
|
|
use_open: bool = False,
|
|
symbol: str = "",
|
|
export: str = "",
|
|
external_axes: Optional[List[plt.Axes]] = None,
|
|
):
|
|
"""Plots AD Osc Indicator
|
|
|
|
Parameters
|
|
----------
|
|
data : pd.DataFrame
|
|
Dataframe of ohlc prices
|
|
use_open : bool
|
|
Whether to use open prices in calculation
|
|
fast: int
|
|
Length of fast window
|
|
slow : int
|
|
Length of slow window
|
|
symbol : str
|
|
Stock ticker
|
|
export : str
|
|
Format to export data
|
|
external_axes : Optional[List[plt.Axes]], optional
|
|
External axes (3 axes are expected in the list), by default None
|
|
"""
|
|
divisor = 1_000_000
|
|
df_vol = data["Volume"] / divisor
|
|
df_vol.name = "Adj Volume"
|
|
df_ta = volume_model.adosc(data, use_open, fast, slow)
|
|
df_cal = df_ta[df_ta.columns[0]] / divisor
|
|
df_cal.name = "Adj ADOSC"
|
|
|
|
plot_data = pd.merge(data, df_vol, how="outer", left_index=True, right_index=True)
|
|
plot_data = pd.merge(
|
|
plot_data, df_ta, how="outer", left_index=True, right_index=True
|
|
)
|
|
plot_data = pd.merge(
|
|
plot_data, df_cal, how="outer", left_index=True, right_index=True
|
|
)
|
|
plot_data = reindex_dates(plot_data)
|
|
|
|
# This plot has 3 axes
|
|
if external_axes is None:
|
|
_, axes = plt.subplots(
|
|
3,
|
|
1,
|
|
sharex=True,
|
|
figsize=plot_autoscale(),
|
|
dpi=PLOT_DPI,
|
|
)
|
|
ax1, ax2, ax3 = axes
|
|
elif is_valid_axes_count(external_axes, 3):
|
|
(ax1, ax2, ax3) = external_axes
|
|
else:
|
|
return
|
|
|
|
ax1.set_title(f"{symbol} AD Oscillator")
|
|
ax1.plot(plot_data.index, plot_data["Adj Close"].values)
|
|
ax1.set_xlim(plot_data.index[0], plot_data.index[-1])
|
|
ax1.set_ylabel("Price")
|
|
theme.style_primary_axis(
|
|
ax1,
|
|
data_index=plot_data.index.to_list(),
|
|
tick_labels=plot_data["date"].to_list(),
|
|
)
|
|
|
|
ax2.set_ylabel("Volume [M]")
|
|
bar_colors = [
|
|
theme.down_color if x[1].Open < x[1].Close else theme.up_color
|
|
for x in plot_data.iterrows()
|
|
]
|
|
ax2.bar(
|
|
plot_data.index,
|
|
plot_data["Adj Volume"],
|
|
color=bar_colors,
|
|
width=theme.volume_bar_width,
|
|
)
|
|
ax2.set_xlim(plot_data.index[0], plot_data.index[-1])
|
|
theme.style_primary_axis(
|
|
ax2,
|
|
data_index=plot_data.index.to_list(),
|
|
tick_labels=plot_data["date"].to_list(),
|
|
)
|
|
|
|
ax3.set_ylabel("AD Osc [M]")
|
|
ax3.plot(plot_data.index, plot_data["Adj ADOSC"], label="AD Osc")
|
|
ax3.set_xlim(plot_data.index[0], plot_data.index[-1])
|
|
theme.style_primary_axis(
|
|
ax3,
|
|
data_index=plot_data.index.to_list(),
|
|
tick_labels=plot_data["date"].to_list(),
|
|
)
|
|
|
|
if external_axes is None:
|
|
theme.visualize_output()
|
|
|
|
export_data(
|
|
export,
|
|
os.path.dirname(os.path.abspath(__file__)).replace("common", "stocks"),
|
|
"adosc",
|
|
df_ta,
|
|
)
|
|
|
|
|
|
@log_start_end(log=logger)
|
|
def display_obv(
|
|
data: pd.DataFrame,
|
|
symbol: str = "",
|
|
export: str = "",
|
|
external_axes: Optional[List[plt.Axes]] = None,
|
|
):
|
|
"""Plots OBV technical indicator
|
|
|
|
Parameters
|
|
----------
|
|
data : pd.DataFrame
|
|
Dataframe of ohlc prices
|
|
symbol : str
|
|
Ticker
|
|
export: str
|
|
Format to export data as
|
|
external_axes : Optional[List[plt.Axes]], optional
|
|
External axes (1 axis is expected in the list), by default None
|
|
"""
|
|
divisor = 1_000_000
|
|
df_vol = data["Volume"] / divisor
|
|
df_vol.name = "Adj Volume"
|
|
df_ta = volume_model.obv(data)
|
|
df_cal = df_ta[df_ta.columns[0]] / divisor
|
|
df_cal.name = "Adj OBV"
|
|
|
|
plot_data = pd.merge(data, df_vol, how="outer", left_index=True, right_index=True)
|
|
plot_data = pd.merge(
|
|
plot_data, df_ta, how="outer", left_index=True, right_index=True
|
|
)
|
|
plot_data = pd.merge(
|
|
plot_data, df_cal, how="outer", left_index=True, right_index=True
|
|
)
|
|
plot_data = reindex_dates(plot_data)
|
|
|
|
# This plot has 3 axes
|
|
if external_axes is None:
|
|
_, axes = plt.subplots(
|
|
3,
|
|
1,
|
|
sharex=True,
|
|
figsize=plot_autoscale(),
|
|
dpi=PLOT_DPI,
|
|
)
|
|
ax1, ax2, ax3 = axes
|
|
elif is_valid_axes_count(external_axes, 3):
|
|
(ax1, ax2, ax3) = external_axes
|
|
else:
|
|
return
|
|
|
|
close_col = ta_helpers.check_columns(data)
|
|
if close_col is None:
|
|
return
|
|
ax1.plot(plot_data.index, plot_data[close_col].values)
|
|
ax1.set_title(f"{symbol} OBV")
|
|
ax1.set_xlim(plot_data.index[0], plot_data.index[-1])
|
|
ax1.set_ylabel("Price")
|
|
theme.style_primary_axis(
|
|
ax1,
|
|
data_index=plot_data.index.to_list(),
|
|
tick_labels=plot_data["date"].to_list(),
|
|
)
|
|
|
|
ax2.set_xlim(plot_data.index[0], plot_data.index[-1])
|
|
ax2.set_ylabel("Volume [M]")
|
|
bar_colors = [
|
|
theme.down_color if x[1].Open < x[1].Close else theme.up_color
|
|
for x in plot_data.iterrows()
|
|
]
|
|
ax2.bar(
|
|
plot_data.index,
|
|
plot_data["Adj Volume"],
|
|
color=bar_colors,
|
|
alpha=0.8,
|
|
width=theme.volume_bar_width,
|
|
)
|
|
theme.style_primary_axis(
|
|
ax2,
|
|
data_index=plot_data.index.to_list(),
|
|
tick_labels=plot_data["date"].to_list(),
|
|
)
|
|
|
|
ax3.set_ylabel("OBV [M]")
|
|
ax3.plot(plot_data.index, plot_data["Adj OBV"])
|
|
ax3.set_xlim(plot_data.index[0], plot_data.index[-1])
|
|
theme.style_primary_axis(
|
|
ax3,
|
|
data_index=plot_data.index.to_list(),
|
|
tick_labels=plot_data["date"].to_list(),
|
|
)
|
|
|
|
if external_axes is None:
|
|
theme.visualize_output()
|
|
|
|
export_data(
|
|
export,
|
|
os.path.dirname(os.path.abspath(__file__)).replace("common", "stocks"),
|
|
"obv",
|
|
df_ta,
|
|
)
|