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

filtering dates on world news

Browse Source
This commit is contained in:
hjoaquim
2024-01-23 18:44:41 +00:00
parent 490d7317f2
commit d8a619e58c
4 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
openbb_platform/providers/biztoc/openbb_biztoc/models/world_news.py
View File

@@ -8,6 +8,7 @@ from openbb_core.provider.standard_models.world_news import (
WorldNewsData,
WorldNewsQueryParams,
)
from openbb_core.provider.utils.helpers import filter_by_dates
from pandas import to_datetime
from pydantic import Field, field_validator
@@ -101,4 +102,7 @@ class BiztocWorldNewsFetcher(
query: BiztocWorldNewsQueryParams, data: List[Dict], **kwargs: Any
) -> List[BiztocWorldNewsData]:
"""Transform the data to the standard format."""
return [BiztocWorldNewsData.model_validate(d) for d in data]
modeled_data = [BiztocWorldNewsData.model_validate(d) for d in data]
return filter_by_dates(
modeled_data, query.start_date, query.end_date, query.date
)

7
openbb_platform/providers/fmp/openbb_fmp/models/world_news.py
View File

@@ -9,7 +9,7 @@ from openbb_core.provider.standard_models.world_news import (
WorldNewsData,
WorldNewsQueryParams,
)
from openbb_core.provider.utils.helpers import amake_requests
from openbb_core.provider.utils.helpers import amake_requests, filter_by_dates
from pydantic import Field, field_validator
@@ -77,4 +77,7 @@ class FMPWorldNewsFetcher(
if isinstance(d["image"], str):
d["image"] = [{"url": d["image"]}]
return [FMPWorldNewsData.model_validate(d) for d in data]
modeled_data = [FMPWorldNewsData.model_validate(d) for d in data]
return filter_by_dates(
modeled_data, query.start_date, query.end_date, query.date
)

6
openbb_platform/providers/intrinio/openbb_intrinio/models/world_news.py
View File

@@ -9,6 +9,7 @@ from openbb_core.provider.standard_models.world_news import (
WorldNewsData,
WorldNewsQueryParams,
)
from openbb_core.provider.utils.helpers import filter_by_dates
from openbb_intrinio.utils.helpers import get_data_many
from pydantic import Field, field_validator
@@ -68,4 +69,7 @@ class IntrinioWorldNewsFetcher(
query: IntrinioWorldNewsQueryParams, data: List[Dict], **kwargs: Any
) -> List[IntrinioWorldNewsData]:
"""Return the transformed data."""
return [IntrinioWorldNewsData.model_validate(d) for d in data]
modeled_data = [IntrinioWorldNewsData.model_validate(d) for d in data]
return filter_by_dates(
modeled_data, query.start_date, query.end_date, query.date
)

7
openbb_platform/providers/tiingo/openbb_tiingo/models/world_news.py
View File

@@ -9,7 +9,7 @@ from openbb_core.provider.standard_models.world_news import (
WorldNewsData,
WorldNewsQueryParams,
)
from openbb_core.provider.utils.helpers import get_querystring
from openbb_core.provider.utils.helpers import filter_by_dates, get_querystring
from pydantic import Field, field_validator
from ..utils.helpers import get_data_many
@@ -94,4 +94,7 @@ class TiingoWorldNewsFetcher(
query: TiingoWorldNewsQueryParams, data: List[Dict], **kwargs: Any
) -> List[TiingoWorldNewsData]:
"""Return the transformed data."""
return [TiingoWorldNewsData.model_validate(d) for d in data]
modeled_data = [TiingoWorldNewsData.model_validate(d) for d in data]
return filter_by_dates(
modeled_data, query.start_date, query.end_date, query.date
)