Skip to content

Commit

Permalink
Merge branch 'develop' into bugfix/remove-unnecessary-env-var
Browse files Browse the repository at this point in the history
  • Loading branch information
hjoaquim authored Feb 23, 2024
2 parents 647bde2 + 2c6cfa1 commit 8de121a
Showing 1 changed file with 19 additions and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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."""
Expand Down Expand Up @@ -164,6 +178,11 @@ def extract_data(
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
Expand Down

0 comments on commit 8de121a

Please sign in to comment.