mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 20:42:26 +00:00
d4207527eb
* remove ultima from deps * skip ultima test * black * change skip location * try except statement to avoid error on the ci
117 lines
4.7 KiB
Python
117 lines
4.7 KiB
Python
"""Install for development script."""
|
|
# noqa: S603,PLW1510,T201
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import toml
|
|
|
|
PLATFORM_PATH = Path(__file__).parent.resolve()
|
|
LOCK = PLATFORM_PATH / "poetry.lock"
|
|
PYPROJECT = PLATFORM_PATH / "pyproject.toml"
|
|
|
|
|
|
LOCAL_DEPS = """
|
|
[tool.poetry.dependencies]
|
|
python = ">=3.8,<3.12"
|
|
openbb-devtools = { path = "./extensions/devtools", develop = true }
|
|
openbb-provider = { path = "./platform/provider", develop = true }
|
|
openbb-core = { path = "./platform/core", develop = true }
|
|
|
|
openbb-benzinga = { path = "./providers/benzinga", develop = true }
|
|
openbb-fmp = { path = "./providers/fmp", develop = true }
|
|
openbb-fred = { path = "./providers/fred", develop = true }
|
|
openbb-intrinio = { path = "./providers/intrinio", develop = true }
|
|
openbb-oecd = { path = "./providers/oecd", develop = true }
|
|
openbb-polygon = { path = "./providers/polygon", develop = true }
|
|
openbb-sec = { path = "./providers/sec", develop = true }
|
|
openbb-tradingeconomics = { path = "./providers/tradingeconomics", develop = true }
|
|
|
|
openbb-crypto = { path = "./extensions/crypto", develop = true }
|
|
openbb-currency = { path = "./extensions/currency", develop = true }
|
|
openbb-economy = { path = "./extensions/economy", develop = true }
|
|
openbb-equity = { path = "./extensions/equity", develop = true }
|
|
openbb-etf = { path = "./extensions/etf", develop = true }
|
|
openbb-fixedincome = { path = "./extensions/fixedincome", develop = true }
|
|
openbb-index = { path = "./extensions/index", develop = true }
|
|
openbb-news = { path = "./extensions/news", develop = true }
|
|
openbb-regulators = { path = "./extensions/regulators", develop = true }
|
|
|
|
# Community dependencies
|
|
openbb-alpha-vantage = { path = "./providers/alpha_vantage", optional = true, develop = true }
|
|
openbb-biztoc = { path = "./providers/biztoc", optional = true, develop = true }
|
|
openbb-cboe = { path = "./providers/cboe", optional = true, develop = true }
|
|
openbb-ecb = { path = "./providers/ecb", optional = true, develop = true }
|
|
openbb-finra = { path = "./providers/finra", develop = true }
|
|
openbb-nasdaq = { path = "./providers/nasdaq", optional = true, develop = true }
|
|
openbb-seeking-alpha = { path = "./providers/seeking_alpha", optional = true, develop = true }
|
|
openbb-stockgrid = { path = "./providers/stockgrid" ,optional = true, develop = true }
|
|
openbb-wsj = { path = "./providers/wsj", develop = true }
|
|
openbb-yfinance = { path = "./providers/yfinance", optional = true, develop = true }
|
|
|
|
openbb-charting = { path = "./extensions/charting", optional = true, develop = true }
|
|
openbb-econometrics = { path = "./extensions/econometrics", optional = true, develop = true }
|
|
openbb-futures = { path = "./extensions/futures", optional = true, develop = true }
|
|
openbb-quantitative = { path = "./extensions/quantitative", optional = true, develop = true }
|
|
openbb-technical = { path = "./extensions/technical", optional = true, develop = true }
|
|
"""
|
|
|
|
pyproject_toml = toml.load(PYPROJECT)
|
|
pyproject_toml["tool"]["poetry"]["dependencies"] = toml.loads(LOCAL_DEPS)["tool"][
|
|
"poetry"
|
|
]["dependencies"]
|
|
|
|
TEMP_PYPROJECT = toml.dumps(pyproject_toml)
|
|
|
|
|
|
def install_local(_extras: bool = False):
|
|
"""Install the Platform locally for development purposes.
|
|
|
|
Installs the Platform in editable mode, instead of copying the source code to
|
|
the site-packages directory. This makes any changes immediately available
|
|
to the interpreter.
|
|
|
|
Parameters
|
|
----------
|
|
_extras : bool, optional
|
|
Whether to install the Platform with the extra dependencies, by default False
|
|
"""
|
|
original_lock = LOCK.read_text()
|
|
original_pyproject = PYPROJECT.read_text()
|
|
extras_args = ["-E", "all"] if _extras else []
|
|
|
|
try:
|
|
# we create a temporary pyproject.toml
|
|
with open(PYPROJECT, "w", encoding="utf-8", newline="\n") as f:
|
|
f.write(TEMP_PYPROJECT)
|
|
|
|
CMD = [sys.executable, "-m", "poetry"]
|
|
|
|
subprocess.run( # noqa: PLW1510
|
|
CMD + ["lock", "--no-update"], cwd=PLATFORM_PATH, check=True # noqa: S603
|
|
)
|
|
subprocess.run( # noqa: PLW1510
|
|
CMD + ["install"] + extras_args, cwd=PLATFORM_PATH, check=True # noqa: S603
|
|
)
|
|
|
|
except (Exception, KeyboardInterrupt) as e:
|
|
print(e) # noqa: T201
|
|
print("Restoring pyproject.toml and poetry.lock") # noqa: T201
|
|
|
|
# we restore the original pyproject.toml
|
|
with open(PYPROJECT, "w", encoding="utf-8", newline="\n") as f:
|
|
f.write(original_pyproject)
|
|
|
|
# we restore the original poetry.lock
|
|
with open(LOCK, "w", encoding="utf-8", newline="\n") as f:
|
|
f.write(original_lock)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
args = sys.argv[1:]
|
|
|
|
# pylint: disable=use-a-generator
|
|
extras = any([arg.lower() in ["-e", "--extras"] for arg in args])
|
|
|
|
install_local(extras)
|