Function init_data

src/couchdb/init_data.py:129–154

Load a scenario's data (or the default) into CouchDB. Returns {collection: (db, n)}. Resolves the manifest first, so an unknown ``scenario_id`` raises FileNotFoundError before anything is dropped. ``reset_first=True`` then drops databases so collections absent from the manifest are left

(
    scenario_id=None,
    force: bool = True,
    reset_first: bool = False,
    managed_only: bool = False,
)

# Load
# --------------------------------------------------------------------------- #
def init_data(
    scenario_id=None,
    force: bool = True,
    reset_first: bool = False,
    managed_only: bool = False,
) -> dict:
    """Load a scenario's data (or the default) into CouchDB. Returns {collection: (db, n)}.

    Resolves the manifest first, so an unknown ``scenario_id`` raises FileNotFoundError
    before anything is dropped. ``reset_first=True`` then drops databases so collections
    absent from the manifest are left empty rather than carrying over.
    """
    manifest, base_dir = _resolve_manifest(
        scenario_id
    )  # validate first (raises on unknown id)
    if reset_first:
        reset(managed_only=managed_only)
    results = {}
    for key, spec in manifest.items():
        results[key] = loader.load_collection(
            key, spec, drop=force, base_dir=base_dir
        )  # database name = key
        logger.info(
            "Scenario %s: '%s' → %s (%d docs).", scenario_id, key, *results[key]
        )
    return results


def main() -> None: