mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-06-09 08:14:49 +08:00
023531fac7
* set ChartStyle.plt_style from UserSettings * fix discrepencies between raw figure and pywry for light/dark
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, 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) # type: ignore
|
|
|
|
@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
|