mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-29 05:21:48 +00:00
78445ce007
* round 1 * black * more black * providers a-thru-fmp * charting settings * charting backend * black * charting_settings * missing nasdaq import * nasdaq mypy * fed money measure * some linting * black * more linting * more linting * charting imports * black * pylint * refactor fred * more linting * rest of the providers * didn't want that file * more cleanup * missed * ta_helpers * router views * chart ta_class * forgot a file * forgot another file * biztoc * package builder * technical router * fred params * bump finviz * fix tmx * more cleanup * try again * currency historical * equity search * missed files in commit * fix some tests * black * fix more test cassettes * more patchy * black * linters * more tests * black * mypy * delete unused model * more linting * forgot files * more linting * make up your mind, black, and stop moving pylint for me * more black * back to where we were before black destroyed it * .......... * .................. * unused fetcher test * remove standard model validator * commodity router * treasury prices * oecd tests * finra * container cleanup * black * forgotten file * linter * remove wrapped "" * fix merge * patches * ecb test * finra cassette * finviz cassettes * nasdaq cassettes * oecd cassettes * yfinance cassettes * more test cassettes * black * merge cleanup * cboe helpers * black * fix typo --------- Co-authored-by: Henrique Joaquim <henriquecjoaquim@gmail.com> Co-authored-by: Igor Radovanovic <74266147+IgorWounds@users.noreply.github.com>
100 lines
3.3 KiB
Python
100 lines
3.3 KiB
Python
"""Views for the ETF Extension."""
|
|
|
|
# pylint: disable=unused-argument
|
|
|
|
from typing import TYPE_CHECKING, Any, Dict, Tuple, Union
|
|
|
|
if TYPE_CHECKING:
|
|
from openbb_charting.core.openbb_figure import (
|
|
OpenBBFigure,
|
|
)
|
|
from plotly.graph_objs import Figure
|
|
|
|
|
|
class EtfViews:
|
|
"""Etf Views."""
|
|
|
|
@staticmethod
|
|
def etf_historical(
|
|
**kwargs,
|
|
) -> Tuple["OpenBBFigure", Dict[str, Any]]:
|
|
"""Etf Price Historical Chart."""
|
|
# pylint: disable=import-outside-toplevel
|
|
|
|
from openbb_charting.charts.price_historical import price_historical
|
|
|
|
return price_historical(**kwargs)
|
|
|
|
@staticmethod
|
|
def etf_price_performance(
|
|
**kwargs,
|
|
) -> Tuple["OpenBBFigure", Dict[str, Any]]:
|
|
"""Etf Price Performance Chart."""
|
|
# pylint: disable=import-outside-toplevel
|
|
from openbb_charting.charts.price_performance import price_performance
|
|
|
|
return price_performance(**kwargs)
|
|
|
|
@staticmethod
|
|
def etf_holdings(
|
|
**kwargs,
|
|
) -> Tuple[Union["OpenBBFigure", "Figure"], Dict[str, Any]]:
|
|
"""Equity Compare Groups Chart."""
|
|
# pylint: disable=import-outside-toplevel
|
|
from pandas import DataFrame # noqa
|
|
from openbb_core.app.utils import basemodel_to_df # noqa
|
|
from openbb_core.app.model.abstract.error import OpenBBError # noqa
|
|
from openbb_charting.charts.generic_charts import bar_chart # noqa
|
|
|
|
if "data" in kwargs and isinstance(kwargs["data"], DataFrame):
|
|
data = kwargs["data"]
|
|
elif "data" in kwargs and isinstance(kwargs["data"], list):
|
|
data = basemodel_to_df(kwargs["data"], index=None) # type: ignore
|
|
else:
|
|
data = basemodel_to_df(kwargs["obbject_item"], index=None) # type: ignore
|
|
|
|
if "weight" not in data.columns:
|
|
raise OpenBBError("No 'weight' column found in the data.")
|
|
|
|
orientation = kwargs.get("orientation", "h")
|
|
limit = kwargs.get("limit", 20)
|
|
symbol = kwargs["standard_params"].get("symbol") # type: ignore
|
|
title = kwargs.get("title", f"Top {limit} {symbol} Holdings")
|
|
layout_kwargs = kwargs.get("layout_kwargs", {})
|
|
|
|
data = data.sort_values("weight", ascending=False)
|
|
limit = min(limit, len(data)) # type: ignore
|
|
target = data.head(limit)[["symbol", "weight"]].set_index("symbol")
|
|
target = target.multiply(100)
|
|
axis_title = "Weight (%)"
|
|
|
|
fig = bar_chart(
|
|
target.reset_index(),
|
|
"symbol",
|
|
["weight"],
|
|
title=title, # type: ignore
|
|
xtitle=axis_title if orientation == "h" else None,
|
|
ytitle=axis_title if orientation == "v" else None,
|
|
orientation=orientation, # type: ignore
|
|
)
|
|
|
|
fig.update_layout(
|
|
hovermode="x" if orientation == "v" else "y",
|
|
margin=dict(r=0, l=50) if orientation == "h" else None,
|
|
)
|
|
|
|
fig.update_traces(
|
|
hovertemplate=(
|
|
"%{y:.3f}%<extra></extra>"
|
|
if orientation == "v"
|
|
else "%{x:.3f}%<extra></extra>"
|
|
)
|
|
)
|
|
|
|
if layout_kwargs:
|
|
fig.update_layout(**layout_kwargs) # type: ignore
|
|
|
|
content = fig.show(external=True).to_plotly_json() # type: ignore
|
|
|
|
return fig, content
|