jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-10-29 05:21:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

change start_date, end_date defaults

Browse Source
This commit is contained in:
montezdesousa
2023-07-11 12:52:54 +01:00
parent 5b1a13d6ef
commit 5fd8fdce27
9 changed files with 59 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
openbb_providers/builtin_providers/fmp/crypto_eod.py
View File

@@ -1,7 +1,7 @@
"""FMP Cryptos end of day fetcher."""
# IMPORT STANDARD
from datetime import date, datetime
from datetime import date, datetime, timedelta
from typing import Dict, List, Literal, Optional
# IMPORT INTERNAL
@@ -74,11 +74,13 @@ class FMPCryptoEODFetcher(
def transform_query(
query: CryptoEODQueryParams, extra_params: Optional[Dict] = None
) -> FMPCryptoEODQueryParams:
raw_end = query.end_date if query.end_date else datetime.now()
now = datetime.now()
start_date = query.start_date if query.start_date else now - timedelta(days=1)
end_date = query.end_date if query.end_date else now
return FMPCryptoEODQueryParams(
symbol=query.symbol,
start_date=query.start_date,
end_date=raw_end,
start_date=start_date,
end_date=end_date,
**extra_params if extra_params else {},
)

10
openbb_providers/builtin_providers/fmp/forex_eod.py
View File

@@ -1,7 +1,7 @@
"""FMP Forex end of day fetcher."""
# IMPORT STANDARD
from datetime import date, datetime
from datetime import date, datetime, timedelta
from typing import Dict, List, Literal, Optional
# IMPORT INTERNAL
@@ -74,11 +74,13 @@ class FMPForexEODFetcher(
def transform_query(
query: ForexEODQueryParams, extra_params: Optional[Dict] = None
) -> FMPForexEODQueryParams:
raw_end = query.end_date if query.end_date else datetime.now()
now = datetime.now()
start_date = query.start_date if query.start_date else now - timedelta(days=1)
end_date = query.end_date if query.end_date else now
return FMPForexEODQueryParams(
symbol=query.symbol,
start_date=query.start_date,
end_date=raw_end,
start_date=start_date,
end_date=end_date,
)
@staticmethod

10
openbb_providers/builtin_providers/fmp/stock_eod.py
View File

@@ -1,7 +1,7 @@
"""FMP Stocks end of day fetcher."""
# IMPORT STANDARD
from datetime import date, datetime
from datetime import date, datetime, timedelta
from typing import Dict, List, Literal, Optional
# IMPORT INTERNAL
@@ -74,11 +74,13 @@ class FMPStockEODFetcher(
def transform_query(
query: StockEODQueryParams, extra_params: Optional[Dict] = None
) -> FMPStockEODQueryParams:
raw_end = query.end_date if query.end_date else datetime.now()
now = datetime.now()
start_date = query.start_date if query.start_date else now - timedelta(days=1)
end_date = query.end_date if query.end_date else now
return FMPStockEODQueryParams(
symbol=query.symbol,
start_date=query.start_date,
end_date=raw_end,
start_date=start_date,
end_date=end_date,
**extra_params if extra_params else {},
)

9
openbb_providers/builtin_providers/polygon/crypto_eod.py
View File

@@ -1,7 +1,7 @@
"""Polygon crypto end of day fetcher."""
# IMPORT STANDARD
from datetime import datetime
from datetime import datetime, timedelta
from typing import Dict, List, Optional
# IMPORT INTERNAL
@@ -60,10 +60,13 @@ class PolygonCryptoEODFetcher(
def transform_query(
query: CryptoEODQueryParams, extra_params: Optional[Dict] = None
) -> PolygonCryptoEODQueryParams:
now = datetime.now()
start_date = query.start_date if query.start_date else now - timedelta(days=1)
end_date = query.end_date if query.end_date else now
return PolygonCryptoEODQueryParams(
symbol=query.symbol,
start_date=query.start_date,
end_date=query.end_date if query.end_date else datetime.now(),
start_date=start_date,
end_date=end_date,
**extra_params if extra_params else {},
)

9
openbb_providers/builtin_providers/polygon/forex_eod.py
View File

@@ -1,7 +1,7 @@
"""Polygon forex end of day fetcher."""
# IMPORT STANDARD
from datetime import datetime
from datetime import datetime, timedelta
from typing import Dict, List, Optional
# IMPORT INTERNAL
@@ -60,10 +60,13 @@ class PolygonForexEODFetcher(
def transform_query(
query: ForexEODQueryParams, extra_params: Optional[Dict] = None
) -> PolygonForexEODQueryParams:
now = datetime.now()
start_date = query.start_date if query.start_date else now - timedelta(days=1)
end_date = query.end_date if query.end_date else now
return PolygonForexEODQueryParams(
symbol=query.symbol,
start_date=query.start_date,
end_date=query.end_date if query.end_date else datetime.now(),
start_date=start_date,
end_date=end_date,
**extra_params if extra_params else {},
)

9
openbb_providers/builtin_providers/polygon/stock_eod.py
View File

@@ -1,7 +1,7 @@
"""Polygon stocks end of day fetcher."""
# IMPORT STANDARD
from datetime import datetime
from datetime import datetime, timedelta
from typing import Dict, List, Optional
# IMPORT INTERNAL
@@ -60,10 +60,13 @@ class PolygonStockEODFetcher(
def transform_query(
query: StockEODQueryParams, extra_params: Optional[Dict] = None
) -> PolygonStockEODQueryParams:
now = datetime.now()
start_date = query.start_date if query.start_date else now - timedelta(days=1)
end_date = query.end_date if query.end_date else now
return PolygonStockEODQueryParams(
symbol=query.symbol,
start_date=query.start_date,
end_date=query.end_date if query.end_date else datetime.now(),
start_date=start_date,
end_date=end_date,
**extra_params if extra_params else {},
)

11
openbb_providers/openbb_provider/model/data/crypto_eod.py
View File

@@ -18,15 +18,20 @@ class CryptoEODQueryParams(QueryParams, BaseSymbol):
---------
symbol : str
The symbol of the company.
start_date : date
start_date : Optional[date]
The start date of the stock data from which to retrieve the data.
If None, defaults to yesterday.
end_date : Optional[date]
The end date of the stock data up to which to retrieve the data.
If None, defaults to today.
"""
start_date: date = Field(description=DESCRIPTIONS.get("start_date", ""))
start_date: Optional[date] = Field(
description=DESCRIPTIONS.get("start_date", ""),
default=None,
)
end_date: Optional[date] = Field(
description=DESCRIPTIONS.get("end_date", ""), default=date.today()
description=DESCRIPTIONS.get("end_date", ""), default=None
)

11
openbb_providers/openbb_provider/model/data/forex_eod.py
View File

@@ -18,16 +18,19 @@ class ForexEODQueryParams(QueryParams, BaseSymbol):
---------
symbol : str
The symbol of the company.
start_date : date
start_date : Optional[date]
The start date of the stock data from which to retrieve the data.
If None, defaults to yesterday.
end_date : Optional[date]
The end date of the stock data up to which to retrieve the data.
If None, defaults to today.
"""
start_date: date = Field(description=DESCRIPTIONS.get("start_date", ""))
end_date: Optional[date] = Field(
description=DESCRIPTIONS.get("end_date", ""), default=date.today()
start_date: date = Field(
description=DESCRIPTIONS.get("start_date", ""),
default=None,
)
end_date: date = Field(description=DESCRIPTIONS.get("end_date", ""), default=None)
class ForexEODData(Data):

11
openbb_providers/openbb_provider/model/data/stock_eod.py
View File

@@ -17,15 +17,20 @@ class StockEODQueryParams(QueryParams, BaseSymbol):
---------
symbol : str
The symbol of the company.
start_date : date
start_date : Optional[date]
The start date of the stock data from which to retrieve the data.
If None, defaults to yesterday.
end_date : Optional[date]
The end date of the stock data up to which to retrieve the data.
If None, defaults to today.
"""
start_date: date = Field(description=DESCRIPTIONS.get("start_date", ""))
start_date: Optional[date] = Field(
description=DESCRIPTIONS.get("start_date", ""),
default=None,
)
end_date: Optional[date] = Field(
description=DESCRIPTIONS.get("end_date", ""), default=date.today()
description=DESCRIPTIONS.get("end_date", ""), default=None
)