mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-29 05:21:48 +00:00
4500132aa3
* add historical market cap * add to equity view * chart things * raise not return * pass all kwargs to line_chart * tests.. * docstring * static assets * black * chart integration test --------- Co-authored-by: Igor Radovanovic <74266147+IgorWounds@users.noreply.github.com>
86 lines
2.6 KiB
Python
86 lines
2.6 KiB
Python
"""Views for the Equity Extension."""
|
|
|
|
from typing import TYPE_CHECKING, Any, Dict, Tuple
|
|
|
|
if TYPE_CHECKING:
|
|
from openbb_charting.core.openbb_figure import (
|
|
OpenBBFigure,
|
|
)
|
|
|
|
|
|
class EquityViews:
|
|
"""Equity Views."""
|
|
|
|
@staticmethod
|
|
def equity_price_historical( # noqa: PLR0912
|
|
**kwargs,
|
|
) -> Tuple["OpenBBFigure", Dict[str, Any]]:
|
|
"""Equity Price Historical Chart."""
|
|
# pylint: disable=import-outside-toplevel
|
|
from openbb_charting.charts.price_historical import price_historical
|
|
|
|
return price_historical(**kwargs)
|
|
|
|
@staticmethod
|
|
def equity_price_performance( # noqa: PLR0912
|
|
**kwargs,
|
|
) -> Tuple["OpenBBFigure", Dict[str, Any]]:
|
|
"""Equity Price Performance Chart."""
|
|
# pylint: disable=import-outside-toplevel
|
|
from openbb_charting.charts.price_performance import price_performance
|
|
|
|
return price_performance(**kwargs)
|
|
|
|
@staticmethod
|
|
def equity_historical_market_cap( # noqa: PLR0912
|
|
**kwargs,
|
|
) -> Tuple["OpenBBFigure", Dict[str, Any]]:
|
|
"""Equity Historical Market Cap Chart."""
|
|
# pylint: disable=import-outside-toplevel
|
|
from openbb_charting.charts.generic_charts import line_chart
|
|
from openbb_core.app.utils import basemodel_to_df
|
|
from pandas import DataFrame
|
|
|
|
title = kwargs.pop("title", "Historical Market Cap")
|
|
|
|
data = DataFrame()
|
|
|
|
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=kwargs.get("index", "date")) # type: ignore
|
|
else:
|
|
data = basemodel_to_df(
|
|
kwargs["obbject_item"],
|
|
index=kwargs.get("index", "date"), # type: ignore
|
|
)
|
|
|
|
if "date" in data.columns:
|
|
data = data.set_index("date")
|
|
|
|
if data.empty:
|
|
raise ValueError("Data is empty")
|
|
|
|
df = data.pivot(columns="symbol", values="market_cap")
|
|
|
|
scatter_kwargs = kwargs.pop("scatter_kwargs", {})
|
|
|
|
if "hovertemplate" not in scatter_kwargs:
|
|
scatter_kwargs["hovertemplate"] = "%{y}"
|
|
|
|
ytital = kwargs.pop("ytitle", "Market Cap ($)")
|
|
y = kwargs.pop("y", df.columns.tolist())
|
|
|
|
fig = line_chart(
|
|
data=df,
|
|
title=title,
|
|
y=y,
|
|
ytitle=ytital,
|
|
same_axis=True,
|
|
scatter_kwargs=scatter_kwargs,
|
|
**kwargs,
|
|
)
|
|
content = fig.show(external=True).to_plotly_json()
|
|
|
|
return fig, content
|