jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-11-02 23:41:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

add params to transform query

Browse Source
This commit is contained in:
Diogo Sousa
2023-07-29 18:19:11 +01:00
parent 16cb5a22e9
commit ac9be5806b
2 changed files with 7 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
openbb_sdk/providers/fmp/openbb_fmp/models/balance_sheet.py
View File

@@ -5,7 +5,7 @@ from datetime import (
date as dateType,
datetime,
)
from typing import Dict, List, Literal, Optional
from typing import Any, Dict, List, Literal, Optional
from openbb_provider.abstract.data import Data
from openbb_provider.abstract.fetcher import Fetcher
@@ -118,9 +118,7 @@ class FMPBalanceSheetFetcher(
]
):
@staticmethod
def transform_query(
query: BalanceSheetQueryParams, extra_params: Optional[Dict] = None
) -> FMPBalanceSheetQueryParams:
def transform_query(params: Dict[str, Any]) -> FMPBalanceSheetQueryParams:
period = "annual" if query.period == "annually" else "quarter"
return FMPBalanceSheetQueryParams(
symbol=query.symbol, period=period, **extra_params if extra_params else {} # type: ignore
@@ -141,4 +139,4 @@ class FMPBalanceSheetFetcher(
def transform_data(
data: List[FMPBalanceSheetData],
) -> List[BalanceSheetData]:
return data_transformer(data, BalanceSheetData)
return data

10
openbb_sdk/providers/fmp/openbb_fmp/models/cash_flow.py
View File

@@ -5,7 +5,7 @@ from datetime import (
date as dateType,
datetime,
)
from typing import Dict, List, Literal, Optional
from typing import Any, Dict, List, Literal, Optional
from openbb_provider.abstract.data import Data
from openbb_provider.abstract.fetcher import Fetcher
@@ -107,9 +107,7 @@ class FMPCashFlowStatementFetcher(
]
):
@staticmethod
def transform_query(
query: CashFlowStatementQueryParams, extra_params: Optional[Dict] = None
) -> FMPCashFlowStatementQueryParams:
def transform_query(params: Dict[str, Any]) -> FMPCashFlowStatementQueryParams:
period = "annual" if query.period == "annually" else "quarter"
return FMPCashFlowStatementQueryParams(
symbol=query.symbol, period=period, **extra_params if extra_params else {} # type: ignore
@@ -129,5 +127,5 @@ class FMPCashFlowStatementFetcher(
@staticmethod
def transform_data(
data: List[FMPCashFlowStatementData],
) -> List[CashFlowStatementData]:
return data_transformer(data, CashFlowStatementData)
) -> List[FMPCashFlowStatementData]:
return data