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

add more return type info

Browse Source
This commit is contained in:
Diogo Sousa
2023-07-30 14:24:06 +01:00
parent d5eff741d9
commit c0962b09fe
1 changed files with 12 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
openbb_sdk/sdk/provider/openbb_provider/registry_map.py
View File

@@ -57,7 +57,7 @@ class RegistryMap:
f = fetcher()
standard_query, extra_query = self.extract_info(f, "query_params")
standard_data, extra_data = self.extract_info(f, "data")
return_type = f.return_type
return_info = self.extract_return_info(f)
if model_name not in map_:
map_[model_name] = {}
@@ -69,7 +69,7 @@ class RegistryMap:
map_[model_name][p] = {
"QueryParams": extra_query,
"Data": extra_data,
"ReturnType": return_type,
"Return": return_info,
}
return map_
@@ -97,3 +97,13 @@ class RegistryMap:
extra_info["fields"][name] = field
return standard_info, extra_info
@staticmethod
def extract_return_info(fetcher: Fetcher):
"""Extract return info from fetcher."""
data_type = getattr(fetcher, "provider_data_type")
return_type = fetcher.return_type
return (data_type, return_type)