Hotfix/SDK Version Dunder (#4269)

* add version dunder

* Update test_sdk.py

* Update trailmap.py
This commit is contained in:
teh_coderer
2023-02-16 21:26:11 -06:00
committed by GitHub
parent b1ced18fb4
commit b5b8afa281
5 changed files with 9 additions and 3 deletions

View File

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

View File

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

View File

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

View File

@@ -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 })"

View File

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