From 2c6cfa1b6c45dd5763b0d9a319c3bb4c486f4979 Mon Sep 17 00:00:00 2001 From: Danglewood <85772166+deeleeramone@users.noreply.github.com> Date: Fri, 23 Feb 2024 00:29:41 -0800 Subject: add missing field descriptions to yfinance equity price historical (#6116) Co-authored-by: Igor Radovanovic <74266147+IgorWounds@users.noreply.github.com> --- .../openbb_yfinance/models/equity_historical.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/openbb_platform/providers/yfinance/openbb_yfinance/models/equity_historical.py b/openbb_platform/providers/yfinance/openbb_yfinance/models/equity_historical.py index 5fa2e971fe0..bd6253fa92e 100644 --- a/openbb_platform/providers/yfinance/openbb_yfinance/models/equity_historical.py +++ b/openbb_platform/providers/yfinance/openbb_yfinance/models/equity_historical.py @@ -73,6 +73,20 @@ class YFinanceEquityHistoricalQueryParams(EquityHistoricalQueryParams): class YFinanceEquityHistoricalData(EquityHistoricalData): """Yahoo Finance Equity Historical Price Data.""" + __alias_dict__ = { + "split_ratio": "stock_splits", + "dividend": "dividends", + } + + split_ratio: Optional[float] = Field( + default=None, + description="Ratio of the equity split, if a split occurred.", + ) + dividend: Optional[float] = Field( + default=None, + description="Dividend amount (split-adjusted), if a dividend was paid.", + ) + @field_validator("date", mode="before", check_fields=False) def date_validate(cls, v): # pylint: disable=E0213 """Return formatted datetime.""" @@ -164,6 +178,11 @@ class YFinanceEquityHistoricalFetcher( if query.interval in ["1d", "1W", "1M", "3M"]: data["date"] = data["date"].dt.strftime("%Y-%m-%d") + if "dividends" in data.columns and data.dividends.sum() == 0: + data.drop(columns=["dividends"], inplace=True) + if "stock_splits" in data.columns and all(data.stock_splits) == 0: + data.drop(columns=["stock_splits"], inplace=True) + return data.to_dict("records") @staticmethod -- cgit v1.2.3