From 5cb0ce030b66f5bcea74fd7c25265b4006a33554 Mon Sep 17 00:00:00 2001 From: teh_coderer Date: Fri, 17 Mar 2023 20:42:25 -0400 Subject: [PATCH] Revert "sdk generation tests, trailmap fixes" This reverts commit 2575784a3320ea051b0230ffcea6ab682f11f5ac. --- generate_sdk.py | 14 +++++--------- openbb_terminal/core/sdk/sdk_helpers.py | 2 +- openbb_terminal/core/sdk/trail_map.csv | 1 + openbb_terminal/sdk.py | 1 - tests/test_generate_sdk.py | 6 ------ 5 files changed, 7 insertions(+), 17 deletions(-) delete mode 100644 tests/test_generate_sdk.py diff --git a/generate_sdk.py b/generate_sdk.py index 788bf8998af..c194984db91 100644 --- a/generate_sdk.py +++ b/generate_sdk.py @@ -535,7 +535,7 @@ class BuildCategoryModelClasses: subprocess.check_call(["black", "openbb_terminal"]) # nosec: B603, B607 -def generate_sdk(sort: bool = False) -> bool: +def generate_sdk(sort: bool = False) -> None: """Generate the SDK. Parameters @@ -544,15 +544,11 @@ def generate_sdk(sort: bool = False) -> bool: 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 - return True + console.print("[yellow]Generating SDK...[/]") + BuildCategoryModelClasses(trailmaps).build() + console.print("[green]SDK Generated Successfully.[/]") + return if __name__ == "__main__": diff --git a/openbb_terminal/core/sdk/sdk_helpers.py b/openbb_terminal/core/sdk/sdk_helpers.py index 1eee2ad7e94..0bf69fa28f5 100644 --- a/openbb_terminal/core/sdk/sdk_helpers.py +++ b/openbb_terminal/core/sdk/sdk_helpers.py @@ -356,7 +356,7 @@ cfg.start_required_configurations() cfg.start_plot_backend() logger = logging.getLogger(__name__) -cfg.theme.applyMPLstyle() +theme.applyMPLstyle() \r\r\r """ return "\r".join(sdk_imports.splitlines()) diff --git a/openbb_terminal/core/sdk/trail_map.csv b/openbb_terminal/core/sdk/trail_map.csv index 2b63f1bc47c..450bb9e30dd 100644 --- a/openbb_terminal/core/sdk/trail_map.csv +++ b/openbb_terminal/core/sdk/trail_map.csv @@ -282,6 +282,7 @@ 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, diff --git a/openbb_terminal/sdk.py b/openbb_terminal/sdk.py index 2ea66e77c71..8779f27c106 100644 --- a/openbb_terminal/sdk.py +++ b/openbb_terminal/sdk.py @@ -27,7 +27,6 @@ cfg.start_required_configurations() cfg.start_plot_backend() logger = logging.getLogger(__name__) -cfg.theme.applyMPLstyle() class OpenBBSDK: diff --git a/tests/test_generate_sdk.py b/tests/test_generate_sdk.py deleted file mode 100644 index 4cf5b6caf28..00000000000 --- a/tests/test_generate_sdk.py +++ /dev/null @@ -1,6 +0,0 @@ -from generate_sdk import generate_sdk - - -def test_sdk_generation(): - """Test the sdk markdown generator""" - assert generate_sdk() is True