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

using the date filtering for all providers

Browse Source
This commit is contained in:
hjoaquim
2024-01-23 18:27:40 +00:00
parent 809c51ace0
commit 584d833da3
5 changed files with 22 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
openbb_platform/providers/fmp/openbb_fmp/models/company_news.py
View File

@@ -7,6 +7,7 @@ from openbb_core.provider.standard_models.company_news import (
CompanyNewsData,
CompanyNewsQueryParams,
)
from openbb_core.provider.utils.helpers import filter_by_dates
from openbb_fmp.utils.helpers import get_data_many
from pydantic import Field
@@ -75,4 +76,7 @@ class FMPCompanyNewsFetcher(
query: FMPCompanyNewsQueryParams, data: List[Dict], **kwargs: Any
) -> List[FMPCompanyNewsData]:
"""Return the transformed data."""
return [FMPCompanyNewsData.model_validate(d) for d in data]
modeled_data = [FMPCompanyNewsData.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/company_news.py
View File

@@ -12,6 +12,7 @@ from openbb_core.provider.standard_models.company_news import (
from openbb_core.provider.utils.helpers import (
ClientResponse,
amake_requests,
filter_by_dates,
get_querystring,
)
from pydantic import Field, field_validator
@@ -94,4 +95,7 @@ class IntrinioCompanyNewsFetcher(
query: IntrinioCompanyNewsQueryParams, data: List[Dict], **kwargs: Any
) -> List[IntrinioCompanyNewsData]:
"""Return the transformed data."""
return [IntrinioCompanyNewsData.model_validate(d) for d in data]
modeled_data = [IntrinioCompanyNewsData.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/polygon/openbb_polygon/models/company_news.py
View File

@@ -8,7 +8,7 @@ from openbb_core.provider.standard_models.company_news import (
CompanyNewsData,
CompanyNewsQueryParams,
)
from openbb_core.provider.utils.helpers import get_querystring
from openbb_core.provider.utils.helpers import filter_by_dates, get_querystring
from openbb_polygon.utils.helpers import get_data_many, get_date_condition
from pydantic import BaseModel, Field, field_validator
@@ -122,4 +122,7 @@ class PolygonCompanyNewsFetcher(
**kwargs: Any,
) -> List[PolygonCompanyNewsData]:
"""Transform data."""
return [PolygonCompanyNewsData.model_validate(d) for d in data]
modeled_data = [PolygonCompanyNewsData.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/company_news.py
View File

@@ -9,7 +9,7 @@ from openbb_core.provider.standard_models.company_news import (
CompanyNewsData,
CompanyNewsQueryParams,
)
from openbb_core.provider.utils.helpers import get_querystring
from openbb_core.provider.utils.helpers import filter_by_dates, get_querystring
from openbb_tiingo.utils.helpers import get_data_many
from pydantic import Field, field_validator
@@ -93,4 +93,7 @@ class TiingoCompanyNewsFetcher(
query: TiingoCompanyNewsQueryParams, data: List[Dict], **kwargs: Any
) -> List[TiingoCompanyNewsData]:
"""Return the transformed data."""
return [TiingoCompanyNewsData.model_validate(d) for d in data]
modeled_data = [TiingoCompanyNewsData.model_validate(d) for d in data]
return filter_by_dates(
modeled_data, query.start_date, query.end_date, query.date
)

3
openbb_platform/providers/ultima/openbb_ultima/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 openbb_ultima.utils.helpers import get_data
from pydantic import Field
@@ -93,4 +94,4 @@ class UltimaCompanyNewsFetcher(
item["publisher"] = item["publisher"]
item["risk_category"] = item["riskCategory"]
results.append(UltimaCompanyNewsData.model_validate(item))
return results
return filter_by_dates(results, query.start_date, query.end_date, query.date)