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

fix prov interface typing

Browse Source
This commit is contained in:
montezdesousa
2023-07-11 11:46:02 +01:00
parent a616111dd8
commit 5eb24734d6
1 changed files with 31 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
openbb_sdk/openbb_sdk_core/openbb_sdk_core/app/provider_interface.py
View File

@@ -290,7 +290,7 @@ class ProviderInterface:
@dataclass
class StockNews(ProviderChoices):
provider: Literal["benzinga", "polygon", "openbb"]
provider: Literal["benzinga", "polygon"]
"""
result: Dict = {}
@@ -308,35 +308,6 @@ class ProviderInterface:
return result
def __merge_data_dc(self, data: Dict) -> Dict[str, StandardData]:
result: Dict = {}
for query_name, dataclasses in data.items():
standard = dataclasses["standard"]
extra = dataclasses["extra"]
fields = standard.__fields__.copy()
fields.update(extra.__fields__)
fields_dict: Dict[str, Tuple[Any, Any]] = {}
for name, field in fields.items():
fields_dict[name] = (
field.type_,
Field(
default=field.default, description=field.field_info.description
),
)
class Config(BaseConfig):
extra = Extra.allow
result[query_name] = create_model( # type: ignore
query_name,
__config__=Config,
**fields_dict, # type: ignore
)
return result
def __generate_data_dc(
self,
) -> Dict[str, Dict[str, Union[StandardData, ExtraData]]]:
@@ -374,6 +345,36 @@ class ProviderInterface:
return result
def __merge_data_dc(self, data: Dict[str, Dict[str, Union[StandardData, ExtraData]]]) -> Dict[str, BaseModel]:
"""Merge standard data with extra data into a single BaseModel"""
result: Dict = {}
for query_name, dataclasses in data.items():
standard = dataclasses["standard"]
extra = dataclasses["extra"]
fields = standard.__fields__.copy()
fields.update(extra.__fields__)
fields_dict: Dict[str, Tuple[Any, Any]] = {}
for name, field in fields.items():
fields_dict[name] = (
field.type_,
Field(
default=field.default, description=field.field_info.description
),
)
class Config(BaseConfig):
extra = Extra.allow
result[query_name] = create_model( # type: ignore
query_name,
__config__=Config,
**fields_dict, # type: ignore
)
return result
__provider_interface = ProviderInterface()