mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-30 05:52:23 +00:00
2b52337da9
* loop and update sources * clean update sources dict * use only allowed sources and update hub sources on login * silently update hub * ruff * update only allowed sources now * clean up some funcs * fix dict on tests * mypy * remove unused method * change class arg name sources_dict * change attribute name in sources model * remove allowed from SourcesModel * change from user_choices to choices * change default msg
59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
from threading import Thread
|
|
from typing import Any, Type, TypeVar
|
|
|
|
from pydantic import ValidationError
|
|
|
|
from openbb_terminal.core.models import BaseModel
|
|
|
|
T = TypeVar("T", bound=BaseModel)
|
|
|
|
|
|
def load_dict_to_model(dictionary: dict, model: Type[T]) -> T:
|
|
"""Load variables to model.
|
|
|
|
Parameters
|
|
----------
|
|
dictionary : dict
|
|
Variables dictionary.
|
|
model : Type[T]
|
|
Model to load.
|
|
|
|
Returns
|
|
-------
|
|
T
|
|
Model with validated data.
|
|
"""
|
|
model_name = model.__name__.strip("Model").lower()
|
|
try:
|
|
return model(**dictionary) # 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 = dictionary.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(**dictionary) # type: ignore
|
|
except Exception:
|
|
print(f"Error loading {model_name}, using defaults.")
|
|
return model() # type: ignore
|
|
|
|
|
|
def run_thread(target, kwargs):
|
|
"""Run a thread.
|
|
|
|
Parameters
|
|
----------
|
|
target : function
|
|
The target function.
|
|
args : tuple
|
|
The arguments.
|
|
"""
|
|
thread = Thread(target=target, kwargs=kwargs)
|
|
thread.start()
|