Spaces:
Running on Zero
Running on Zero
| """Public taxonomy composition and deterministic ranking contracts.""" | |
| from __future__ import annotations | |
| from importlib.resources import files as resource_files | |
| from pathlib import Path | |
| from ._taxonomy_impl import ( | |
| DefaultRanker, | |
| DefaultTaxonomy, | |
| Ranker, | |
| Taxonomy, | |
| TaxonomyBuilder, | |
| ) | |
| # Keep repr(), documentation, and pickling pointed at the public module after | |
| # the pre-release namespace migration. | |
| for _public_type in ( | |
| DefaultRanker, | |
| DefaultTaxonomy, | |
| Ranker, | |
| Taxonomy, | |
| TaxonomyBuilder, | |
| ): | |
| _public_type.__module__ = __name__ | |
| class GeneralEconomyTaxonomy(DefaultTaxonomy): | |
| """The bundled, conservative taxonomy for economy and broad-market news.""" | |
| def copy_default_taxonomy( | |
| destination: str | Path, | |
| *, | |
| overwrite: bool = False, | |
| ) -> Path: | |
| """Copy the bundled general-economy taxonomy to an editable directory.""" | |
| root = Path(destination).expanduser().resolve() | |
| package = resource_files("range_curate._core") | |
| copies = ( | |
| (package.joinpath("configs", "default.yml"), root / "taxonomy.yml"), | |
| ( | |
| package.joinpath("catalogs", "entities.yml"), | |
| root / "catalogs" / "entities.yml", | |
| ), | |
| ( | |
| package.joinpath("catalogs", "lexical.yml"), | |
| root / "catalogs" / "lexical.yml", | |
| ), | |
| ) | |
| existing = [target for _source, target in copies if target.exists()] | |
| if existing and not overwrite: | |
| names = ", ".join(str(path) for path in existing) | |
| raise FileExistsError( | |
| f"taxonomy files already exist: {names}; pass overwrite=True to replace them" | |
| ) | |
| for source, target in copies: | |
| target.parent.mkdir(parents=True, exist_ok=True) | |
| target.write_bytes(source.read_bytes()) | |
| return root / "taxonomy.yml" | |
| __all__ = [ | |
| "DefaultRanker", | |
| "DefaultTaxonomy", | |
| "GeneralEconomyTaxonomy", | |
| "Ranker", | |
| "Taxonomy", | |
| "TaxonomyBuilder", | |
| "copy_default_taxonomy", | |
| ] | |