mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 20:42:26 +00:00
131 lines
3.1 KiB
Python
131 lines
3.1 KiB
Python
"""Generate ``pyproject.toml`` and the package skeleton files."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class GeneratedProject:
|
|
"""Output of ``generate_pyproject``.
|
|
|
|
Parameters
|
|
----------
|
|
project_name : str
|
|
PyPI distribution name.
|
|
package_name : str
|
|
Snake-case Python package name.
|
|
source : str
|
|
Full ``pyproject.toml`` contents.
|
|
"""
|
|
|
|
project_name: str
|
|
package_name: str
|
|
source: str
|
|
|
|
|
|
def generate_pyproject(
|
|
*,
|
|
project_name: str,
|
|
package_name: str,
|
|
providers: list[str],
|
|
routers: list,
|
|
description: str,
|
|
version: str = "0.1.0",
|
|
spec_provenance: dict[str, str] | None = None,
|
|
) -> GeneratedProject:
|
|
"""Render a PEP 621 + Hatchling ``pyproject.toml`` for the bundled extension.
|
|
|
|
Parameters
|
|
----------
|
|
project_name : str
|
|
PyPI / distribution name.
|
|
package_name : str
|
|
Python package directory name (snake_case).
|
|
providers : list of str
|
|
Provider identifiers.
|
|
routers : list of GeneratedRouter
|
|
Top-level routers.
|
|
description : str
|
|
Free-form description for ``[project] description``.
|
|
version : str
|
|
Initial version string.
|
|
|
|
Returns
|
|
-------
|
|
GeneratedProject
|
|
Project metadata plus the ``pyproject.toml`` source.
|
|
"""
|
|
provider_entries = "\n".join(
|
|
f'{p} = "{package_name}.providers.{p}:{p}_provider"' for p in sorted(providers)
|
|
)
|
|
router_entries = "\n".join(
|
|
f'{r.entry_point_name} = "{package_name}.routers.{r.module_name}:router"'
|
|
for r in routers
|
|
if r.entry_point_name
|
|
)
|
|
provenance_block = _render_provenance_block(spec_provenance or {})
|
|
body = f"""[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|
|
|
|
|
|
[project]
|
|
name = "{project_name}"
|
|
version = "{version}"
|
|
description = {description!r}
|
|
requires-python = ">=3.10,<4"
|
|
license = {{ text = "Apache-2.0" }}
|
|
dependencies = [
|
|
"openbb-core>=2.0.0",
|
|
]
|
|
|
|
|
|
[tool.hatch.build.targets.wheel]
|
|
packages = ["{package_name}"]
|
|
|
|
|
|
[project.entry-points."openbb_provider_extension"]
|
|
{provider_entries}
|
|
|
|
[project.entry-points."openbb_core_extension"]
|
|
{router_entries}
|
|
{provenance_block}
|
|
[tool.ruff]
|
|
line-length = 100
|
|
target-version = "py310"
|
|
|
|
[tool.ruff.lint]
|
|
select = ["E", "F", "W", "I", "B", "UP", "D"]
|
|
ignore = [
|
|
"D100", "D101", "D102", "D103", "D104", "D105", "D107",
|
|
"D200", "D205", "D212", "D401", "D415",
|
|
"E501", "B008",
|
|
]
|
|
|
|
[tool.ruff.lint.pydocstyle]
|
|
convention = "numpy"
|
|
"""
|
|
return GeneratedProject(
|
|
project_name=project_name,
|
|
package_name=package_name,
|
|
source=body,
|
|
)
|
|
|
|
|
|
def _render_provenance_block(provenance: dict[str, str]) -> str:
|
|
"""Render ``[tool.openbb-codegen]`` (empty string when no provenance)."""
|
|
keys = (
|
|
"source_url",
|
|
"api_version",
|
|
"generator",
|
|
"generated_at",
|
|
"spec_version",
|
|
"spec_sha256",
|
|
)
|
|
entries = [(k, provenance[k]) for k in keys if provenance.get(k)]
|
|
if not entries:
|
|
return ""
|
|
body = "\n".join(f"{k} = {v!r}" for k, v in entries)
|
|
return f"\n[tool.openbb-codegen]\n{body}\n"
|