sdk generation tests, trailmap fixes

This commit is contained in:
teh_coderer
2023-03-17 20:41:56 -04:00
parent 7638ac33bb
commit 2575784a33
5 changed files with 17 additions and 7 deletions

View File

@@ -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__":

View File

@@ -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())

View File

@@ -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,
1 trail model view
282 keys.mykeys keys_model.get_keys
283 keys.news keys_model.set_news_key
284 keys.oanda keys_model.set_oanda_key
keys.openbb keys_model.set_openbb_personal_access_token
285 keys.polygon keys_model.set_polygon_key
286 keys.quandl keys_model.set_quandl_key
287 keys.reddit keys_model.set_reddit_key

View File

@@ -27,6 +27,7 @@ cfg.start_required_configurations()
cfg.start_plot_backend()
logger = logging.getLogger(__name__)
cfg.theme.applyMPLstyle()
class OpenBBSDK:

View 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