jerry/OpenBB
1
0
Fork 0
mirror of https://github.com/OpenBB-finance/OpenBB.git synced 2026-11-04 00:12:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

making keys working with the personal access token

Browse Source
This commit is contained in:
hjoaquim
2023-02-09 11:50:29 +00:00
parent b4c7038046
commit a03360b8f0
3 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
openbb_terminal/config_terminal.py
View File

@@ -196,4 +196,6 @@ API_TOKEN_TERMINAL_KEY = os.getenv("OPENBB_API_TOKEN_TERMINAL_KEY") or "REPLACE_
API_STOCKSERA_KEY = os.getenv("OPENBB_API_STOCKSERA_KEY") or "REPLACE_ME"
# https://my.openbb.co
OPENBB_PERSONAL_ACCESS_TOKEN = os.getenv("OPENBB_PERSONAL_ACCESS_TOKEN") or "REPLACE_ME"
OPENBB_PERSONAL_ACCESS_TOKEN = (
os.getenv("OPENBB_OPENBB_PERSONAL_ACCESS_TOKEN") or "REPLACE_ME"
)

13
openbb_terminal/keys_model.py
View File

@@ -248,7 +248,7 @@ def set_key(env_var_name: str, env_var_value: str, persist: bool = False) -> Non
not User.is_guest()
and User.is_sync_enabled()
and env_var_name not in cfg.SENSITIVE_KEYS
and env_var_name.startswith("API_")
and (env_var_name.startswith("API_") or env_var_name.startswith("OPENBB_"))
):
patch_user_configs(
key=env_var_name,
@@ -285,14 +285,17 @@ def get_keys(show: bool = False) -> pd.DataFrame:
# TODO: Refactor api variables without prefix API_ and extend API_SOURCE_KEY format
var_list = [v for v in dir(cfg) if v.startswith("API_")]
var_list = [v for v in dir(cfg) if v.startswith("API_") or v.startswith("OPENBB_")]
current_keys = {}
for cfg_var_name in var_list:
cfg_var_value = getattr(cfg, cfg_var_name)
if cfg_var_value != "REPLACE_ME":
current_keys[cfg_var_name[4:]] = cfg_var_value
if cfg_var_name.startswith("OPENBB_"):
current_keys[cfg_var_name] = cfg_var_value
else:
current_keys[cfg_var_name[4:]] = cfg_var_value
if current_keys:
df = pd.DataFrame.from_dict(current_keys, orient="index")
@@ -2653,7 +2656,7 @@ def set_openbb_personal_access_token(
>>> from openbb_terminal.sdk import openbb
>>> openbb.keys.openbb(key="example_key")
"""
set_key("OPENBB_PERSONAL_ACCESS_TOKEN", key, persist)
set_key("OPENBB_OPENBB_PERSONAL_ACCESS_TOKEN", key, persist)
return check_openbb_personal_access_token(show_output)
@@ -2702,5 +2705,5 @@ def check_openbb_personal_access_token(show_output: bool = False):
status = KeyStatus.DEFINED_TEST_FAILED
if show_output:
console.print(status.colorize() + "\n")
console.print(status.colorize())
return str(status)

2
openbb_terminal/session/sdk_session.py
View File

@@ -54,6 +54,8 @@ def login(
if not session:
session = get_session(email, password, token, keep_session)
else:
console.print("Using saved session to login.")
if session_model.login(session) in [
session_model.LoginStatus.FAILED,