mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-05-06 22:12:12 +08:00
sdk generation tests, trailmap fixes
This commit is contained in:
@@ -535,7 +535,7 @@ class BuildCategoryModelClasses:
|
||||
subprocess.check_call(["black", "openbb_terminal"]) # nosec: B603, B607
|
||||
|
||||
|
||||
def generate_sdk(sort: bool = False) -> None:
|
||||
def generate_sdk(sort: bool = False) -> bool:
|
||||
"""Generate the SDK.
|
||||
|
||||
Parameters
|
||||
@@ -544,11 +544,15 @@ def generate_sdk(sort: bool = False) -> None:
|
||||
Whether to sort the CSVs, by default False
|
||||
"""
|
||||
trailmaps = get_trailmaps(sort)
|
||||
try:
|
||||
console.print("[yellow]Generating SDK...[/]")
|
||||
BuildCategoryModelClasses(trailmaps).build()
|
||||
console.print("[green]SDK Generated Successfully.[/]")
|
||||
except Exception as e:
|
||||
console.print(f"[red]Error generating SDK: {e}[/]")
|
||||
return False
|
||||
|
||||
console.print("[yellow]Generating SDK...[/]")
|
||||
BuildCategoryModelClasses(trailmaps).build()
|
||||
console.print("[green]SDK Generated Successfully.[/]")
|
||||
return
|
||||
return True
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
@@ -356,7 +356,7 @@ cfg.start_required_configurations()
|
||||
cfg.start_plot_backend()
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
theme.applyMPLstyle()
|
||||
cfg.theme.applyMPLstyle()
|
||||
\r\r\r
|
||||
"""
|
||||
return "\r".join(sdk_imports.splitlines())
|
||||
|
||||
1
openbb_terminal/core/sdk/trail_map.csv
vendored
1
openbb_terminal/core/sdk/trail_map.csv
vendored
@@ -282,7 +282,6 @@ keys.messari,keys_model.set_messari_key,
|
||||
keys.mykeys,keys_model.get_keys,
|
||||
keys.news,keys_model.set_news_key,
|
||||
keys.oanda,keys_model.set_oanda_key,
|
||||
keys.openbb,keys_model.set_openbb_personal_access_token,
|
||||
keys.polygon,keys_model.set_polygon_key,
|
||||
keys.quandl,keys_model.set_quandl_key,
|
||||
keys.reddit,keys_model.set_reddit_key,
|
||||
|
||||
|
@@ -27,6 +27,7 @@ cfg.start_required_configurations()
|
||||
cfg.start_plot_backend()
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
cfg.theme.applyMPLstyle()
|
||||
|
||||
|
||||
class OpenBBSDK:
|
||||
|
||||
6
tests/test_generate_sdk.py
Normal file
6
tests/test_generate_sdk.py
Normal file
@@ -0,0 +1,6 @@
|
||||
from generate_sdk import generate_sdk
|
||||
|
||||
|
||||
def test_sdk_generation():
|
||||
"""Test the sdk markdown generator"""
|
||||
assert generate_sdk() is True
|
||||
Reference in New Issue
Block a user