mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 21:12:16 +00:00
937d033160
* adding BaseModel * initial version of the system model * creating setup functions and removing constants from config_terminal * removing log_supress const * fix test * merge * system handler * fix import * centralized configs setup * repr to use the BaseModel * updating i18n dict location * ignoring warning on wrong type-var * ruff * Update openbb_terminal/core/models/system_model.py Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com> * Update openbb_terminal/core/models/system_model.py Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com> * import literal * loading env files before the plot backend is called * fix tests/openbb_terminal/test_terminal_helper.py * tests need to have i18n initialized * fix seeking_alpha tests --------- Co-authored-by: montezdesousa <79287829+montezdesousa@users.noreply.github.com>
87 lines
2.4 KiB
Python
87 lines
2.4 KiB
Python
# IMPORTS STANDARD
|
|
from typing import Any, Dict, Optional, Type, TypeVar
|
|
|
|
# IMPORTS THIRDPARTY
|
|
from dotenv import dotenv_values, set_key
|
|
from pydantic import ValidationError
|
|
|
|
# IMPORTS INTERNAL
|
|
from openbb_terminal.core.config.paths import (
|
|
PACKAGE_ENV_FILE,
|
|
REPOSITORY_ENV_FILE,
|
|
SETTINGS_ENV_FILE,
|
|
)
|
|
from openbb_terminal.core.models import BaseModel
|
|
|
|
T = TypeVar("T", bound=BaseModel)
|
|
|
|
|
|
def reading_env() -> Dict[str, Any]:
|
|
__env_dict: Dict[str, Optional[str]] = {}
|
|
|
|
if REPOSITORY_ENV_FILE.exists():
|
|
__env_dict.update(**dotenv_values(REPOSITORY_ENV_FILE))
|
|
|
|
if PACKAGE_ENV_FILE.exists():
|
|
__env_dict.update(**dotenv_values(PACKAGE_ENV_FILE))
|
|
|
|
if SETTINGS_ENV_FILE.exists():
|
|
__env_dict.update(**dotenv_values(SETTINGS_ENV_FILE))
|
|
|
|
__env_dict_filtered = {
|
|
k[len("OPENBBB_") - 1 :]: v
|
|
for k, v in __env_dict.items()
|
|
if k.startswith("OPENBB_")
|
|
}
|
|
|
|
return __env_dict_filtered
|
|
|
|
|
|
def load_env_to_model(env_dict: dict, model: Type[T]) -> T:
|
|
"""Load environment variables to model.
|
|
|
|
Parameters
|
|
----------
|
|
env_dict : dict
|
|
Environment variables dictionary.
|
|
model : Type[T]
|
|
Type of the model to validate (can be any type that is a subclass of `BaseModel`).
|
|
|
|
Returns
|
|
-------
|
|
T
|
|
Model with environment variables.
|
|
"""
|
|
model_name = model.__name__.strip("Model").lower()
|
|
try:
|
|
return model(**env_dict) # type: ignore
|
|
except ValidationError as error:
|
|
print(f"Error loading {model_name}:")
|
|
for err in error.errors():
|
|
loc = err.get("loc", None)
|
|
var_name = str(loc[0]) if loc else ""
|
|
msg = err.get("msg", "")
|
|
var = env_dict.pop(var_name, None)
|
|
fields: dict[str, Any] = model.get_fields()
|
|
if var and var_name in fields:
|
|
default = fields[var_name].default
|
|
print(f" {var_name}: {msg}, using default -> {default}")
|
|
|
|
return model(**env_dict) # type: ignore
|
|
except Exception:
|
|
print(f"Error loading {model_name}, using defaults.")
|
|
return model() # type: ignore
|
|
|
|
|
|
def write_to_dotenv(name: str, value: str) -> None:
|
|
"""Write to .env file.
|
|
|
|
Parameters
|
|
----------
|
|
name : str
|
|
Name of the variable.
|
|
value : str
|
|
Value of the variable.
|
|
"""
|
|
set_key(str(SETTINGS_ENV_FILE), name, str(value))
|