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

move classmethod

Browse Source
This commit is contained in:
montezdesousa
2023-07-11 11:33:31 +01:00
parent 8eb70b1f4f
commit e8772f1590
1 changed files with 38 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -197,6 +197,44 @@ class ProviderInterface:
return standard, extra
@classmethod
def __extract_data(
cls,
providers: Any,
) -> Tuple[Dict[str, Tuple[str, Any, Any]], Dict[str, Tuple[str, Any, Any]]]:
standard: Dict[str, Tuple[str, Any, Any]] = {}
extra: Dict[str, Tuple[str, Any, Any]] = {}
for provider_name, query_details in providers.items():
if provider_name == "openbb":
for name, field in query_details["Data"]["fields"].items():
incoming = cls.__create_field(name, field, provider_name)
standard[incoming.name] = (
incoming.name,
incoming.type_,
incoming.default,
)
else:
for name, field in query_details["Data"]["fields"].items():
if name not in providers["openbb"]["Data"]["fields"]:
incoming = cls.__create_field(name, field, provider_name)
if incoming.name in extra:
current = DataclassField(*extra[incoming.name])
updated = cls.__merge_fields(current, incoming)
else:
updated = incoming
extra[updated.name] = (
updated.name,
updated.type_,
updated.default,
)
return standard, extra
def __generate_params_dc(
self,
) -> Dict[str, Dict[str, Union[StandardParams, ExtraParams]]]:
@@ -238,44 +276,6 @@ class ProviderInterface:
}
return result
@classmethod
def __extract_data(
cls,
providers: Any,
) -> Tuple[Dict[str, Tuple[str, Any, Any]], Dict[str, Tuple[str, Any, Any]]]:
standard: Dict[str, Tuple[str, Any, Any]] = {}
extra: Dict[str, Tuple[str, Any, Any]] = {}
for provider_name, query_details in providers.items():
if provider_name == "openbb":
for name, field in query_details["Data"]["fields"].items():
incoming = cls.__create_field(name, field, provider_name)
standard[incoming.name] = (
incoming.name,
incoming.type_,
incoming.default,
)
else:
for name, field in query_details["Data"]["fields"].items():
if name not in providers["openbb"]["Data"]["fields"]:
incoming = cls.__create_field(name, field, provider_name)
if incoming.name in extra:
current = DataclassField(*extra[incoming.name])
updated = cls.__merge_fields(current, incoming)
else:
updated = incoming
extra[updated.name] = (
updated.name,
updated.type_,
updated.default,
)
return standard, extra
def __generate_query_providers_dc(self) -> Dict[str, ProviderChoices]:
"""Generate dataclasses for provider choices by query.