diff options
author | Danglewood <85772166+deeleeramone@users.noreply.github.com> | 2024-05-11 12:14:40 -0700 |
---|---|---|
committer | Danglewood <85772166+deeleeramone@users.noreply.github.com> | 2024-05-11 12:14:40 -0700 |
commit | e4705f293629f465f673c3e13bcf94e3be93fd45 (patch) | |
tree | bc93ed40fc995389efb4d12e66f1072853ec98d0 | |
parent | 0092af7a832003f9d44517615696c62449b5ba0a (diff) |
set default dates in transform query
-rw-r--r-- | openbb_platform/providers/fmp/openbb_fmp/models/economic_calendar.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/openbb_platform/providers/fmp/openbb_fmp/models/economic_calendar.py b/openbb_platform/providers/fmp/openbb_fmp/models/economic_calendar.py index ce99ae345ae..f118024876b 100644 --- a/openbb_platform/providers/fmp/openbb_fmp/models/economic_calendar.py +++ b/openbb_platform/providers/fmp/openbb_fmp/models/economic_calendar.py @@ -81,7 +81,12 @@ class FMPEconomicCalendarFetcher( @staticmethod def transform_query(params: Dict[str, Any]) -> FMPEconomicCalendarQueryParams: """Transform the query.""" - return FMPEconomicCalendarQueryParams(**params) + transformed_params = params + if not transformed_params.get("start_date"): + transformed_params["start_date"] = datetime.now().date() + if not transformed_params.get("end_date"): + transformed_params["end_date"] = (datetime.now() + timedelta(days=7)).date() + return FMPEconomicCalendarQueryParams(**transformed_params) @staticmethod async def aextract_data( @@ -95,11 +100,6 @@ class FMPEconomicCalendarFetcher( base_url = "https://financialmodelingprep.com/api/v3/economic_calendar?" # FMP allows only 3-month windows to be queried, we need to chunk to request. - if not query.start_date or query.start_date is None: - query.start_date = datetime.now().date() - if not query.end_date or query.end_date is None: - query.end_date = (datetime.now() + timedelta(days=7)).date() - def date_range(start_date, end_date): """Yield start and end dates for each 90-day period between start_date and end_date.""" delta = timedelta(days=90) |