mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 04:52:07 +00:00
141dd68e68
* Removed prints * FIxed typing * FIxed typing * Fixed typing * Fixed typing * Fixed typing * Fixed typing * Fixed typing * Fix * Reverted stuff
56 lines
1.2 KiB
Python
56 lines
1.2 KiB
Python
import json
|
|
from typing import Dict, List
|
|
|
|
from pydantic.dataclasses import Field, dataclass
|
|
|
|
from openbb_terminal.core.config.paths import DATA_SOURCES_DEFAULT_FILE
|
|
from openbb_terminal.core.models.base_model import BaseModel
|
|
from openbb_terminal.core.sources.utils import flatten
|
|
|
|
# pylint: disable=useless-parent-delegation
|
|
|
|
|
|
def read_default_sources() -> Dict:
|
|
"""Read default sources from file.
|
|
|
|
Returns
|
|
-------
|
|
Dict
|
|
Dictionary with sources
|
|
"""
|
|
try:
|
|
with open(DATA_SOURCES_DEFAULT_FILE) as file:
|
|
return flatten(json.load(file))
|
|
except Exception as e:
|
|
print( # noqa: T201
|
|
f"\nFailed to read data sources file: "
|
|
f"{DATA_SOURCES_DEFAULT_FILE}\n{e}\n"
|
|
)
|
|
return {}
|
|
|
|
|
|
__allowed = read_default_sources()
|
|
|
|
|
|
def get_allowed_sources() -> Dict:
|
|
"""Get allowed sources.
|
|
|
|
Returns
|
|
-------
|
|
Dict
|
|
Dictionary with sources
|
|
"""
|
|
return __allowed
|
|
|
|
|
|
@dataclass(config=dict(validate_assignment=True))
|
|
class SourcesModel(BaseModel):
|
|
"""Model for sources."""
|
|
|
|
choices: Dict[str, List[str]] = Field(
|
|
default_factory=lambda: get_allowed_sources(),
|
|
)
|
|
|
|
def __repr__(self):
|
|
return super().__repr__()
|