Files
OpenBB/openbb_terminal/core/models/credentials_model.py
James Maslek 47500fa97a help me mamma
2023-04-13 10:34:15 -04:00

27 lines
818 B
Python

import json
from dataclasses import make_dataclass
from pydantic.dataclasses import dataclass as pydanticdataclass
from openbb_terminal.core.config.paths import MISCELLANEOUS_DIRECTORY
from openbb_terminal.core.models.base_model import BaseModel
with open(MISCELLANEOUS_DIRECTORY / "models" / "hub_credentials.json") as f:
HUB_CREDENTIALS = json.load(f)
with open(MISCELLANEOUS_DIRECTORY / "models" / "local_credentials.json") as f:
LOCAL_CREDENTIALS = json.load(f)
dc = make_dataclass(
cls_name="CredentialsModel",
fields=[
(k, str, "REPLACE_ME") for k in list(HUB_CREDENTIALS) + list(LOCAL_CREDENTIALS)
],
bases=(BaseModel,),
)
dc.__repr__ = dc.__base__.__repr__ # type: ignore
CredentialsModel = pydanticdataclass(
dc, config=dict(validate_assignment=True, frozen=True)
)