jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-11-07 09:52:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

date filtering

Browse Source
This commit is contained in:
hjoaquim
2024-01-23 10:29:41 +00:00
parent c09782be6d
commit 809c51ace0
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
openbb_platform/providers/yfinance/openbb_yfinance/models/company_news.py
View File

@@ -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
)