mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 12:40:35 +00:00
Hotfix/SDK Version Dunder (#4269)
* add version dunder * Update test_sdk.py * Update trailmap.py
This commit is contained in:
@@ -228,6 +228,7 @@ class BuildCategoryModelClasses:
|
||||
if module:
|
||||
f.write(" def __init__(self):\r super().__init__()\r")
|
||||
elif sdk_root:
|
||||
f.write(" __version__ = obbff.VERSION\r\r")
|
||||
f.write(" def __init__(self):\r SDKLogger()\r")
|
||||
|
||||
def write_class_attributes(
|
||||
|
||||
@@ -20,6 +20,7 @@ from openbb_terminal.sdk_core import (
|
||||
controllers as ctrl,
|
||||
models as model,
|
||||
)
|
||||
from openbb_terminal import feature_flags as obbff
|
||||
from openbb_terminal.session.user import User
|
||||
|
||||
if User.is_guest():
|
||||
@@ -39,6 +40,8 @@ class OpenBBSDK:
|
||||
`whoami`: Display user info.\n
|
||||
"""
|
||||
|
||||
__version__ = obbff.VERSION
|
||||
|
||||
def __init__(self):
|
||||
SDKLogger()
|
||||
self.login = lib.sdk_session.login
|
||||
|
||||
@@ -324,6 +324,7 @@ from openbb_terminal.sdk_core import (
|
||||
controllers as ctrl,
|
||||
models as model,
|
||||
)
|
||||
from openbb_terminal import feature_flags as obbff
|
||||
from openbb_terminal.session.user import User
|
||||
|
||||
if User.is_guest():
|
||||
|
||||
@@ -112,11 +112,10 @@ class FuncAttr:
|
||||
|
||||
definition = definition.rstrip(", ")
|
||||
|
||||
if location_path[0] == "root":
|
||||
location_path[0] = ""
|
||||
|
||||
trail = ".".join([t for t in location_path if t != ""])
|
||||
sdk_name = class_attr if not view else f"{class_attr}_chart"
|
||||
|
||||
trail = trail.lstrip("root.") # pylint: disable=E1310
|
||||
sdk_path = f"{f'openbb.{trail}' if trail else 'openbb'}.{sdk_name}"
|
||||
|
||||
return f"{sdk_path}({definition })"
|
||||
|
||||
@@ -4,10 +4,12 @@
|
||||
def test_openbb(mocker):
|
||||
"""Test the openbb function"""
|
||||
mock = mocker.patch("openbb_terminal.base_helpers.load_dotenv_and_reload_configs")
|
||||
from openbb_terminal import feature_flags as obbff
|
||||
from openbb_terminal.sdk import openbb
|
||||
|
||||
mock.assert_called_once()
|
||||
assert "stocks" in dir(openbb)
|
||||
assert openbb.__version__ == obbff.VERSION
|
||||
|
||||
|
||||
# from importlib import import_module
|
||||
|
||||
Reference in New Issue
Block a user