From 809c51ace02c29d84c9ebd449186fd9fbf574d7f Mon Sep 17 00:00:00 2001 From: hjoaquim Date: Tue, 23 Jan 2024 10:29:41 +0000 Subject: [PATCH] date filtering --- .../yfinance/openbb_yfinance/models/company_news.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/openbb_platform/providers/yfinance/openbb_yfinance/models/company_news.py b/openbb_platform/providers/yfinance/openbb_yfinance/models/company_news.py index d6d48e2839b..ed65c1b22a8 100644 --- a/openbb_platform/providers/yfinance/openbb_yfinance/models/company_news.py +++ b/openbb_platform/providers/yfinance/openbb_yfinance/models/company_news.py @@ -9,6 +9,7 @@ from openbb_core.provider.standard_models.company_news import ( CompanyNewsData, CompanyNewsQueryParams, ) +from openbb_core.provider.utils.helpers import filter_by_dates from pydantic import Field, field_validator from yfinance import Ticker # type: ignore @@ -74,6 +75,7 @@ class YFinanceCompanyNewsFetcher( """Transform query params.""" return YFinanceCompanyNewsQueryParams(**params) + # pylint: disable=unused-argument @staticmethod def extract_data( # pylint: disable=unused-argument query: YFinanceCompanyNewsQueryParams, @@ -86,6 +88,7 @@ class YFinanceCompanyNewsFetcher( return data + # pylint: disable=unused-argument @staticmethod def transform_data( query: YFinanceCompanyNewsQueryParams, @@ -93,4 +96,7 @@ class YFinanceCompanyNewsFetcher( **kwargs: Any, ) -> List[YFinanceCompanyNewsData]: """Transform data.""" - return [YFinanceCompanyNewsData.model_validate(d) for d in data] + modeled_data = [YFinanceCompanyNewsData.model_validate(d) for d in data] + return filter_by_dates( + modeled_data, query.start_date, query.end_date, query.date + )