tools/config.py

from __future__ import annotations

from pathlib import Path


APP_DIR = Path(__file__).resolve().parents[1]
PARAMETERS_DIR = APP_DIR / "parameters"

PARAMETER_FILE_MAP = {
    "cluster": "cluster_mapping.csv",
    "country": "country_region_mapping.csv",
    "region": "country_region_mapping.csv",
    "market summary": "market_summary_mapping.csv",
    "market_summary": "market_summary_mapping.csv",
    "product": "company_products.csv",
    "products": "company_products.csv",
    "company products": "company_products.csv",
    "threshold": "threshold_matrix.csv",
}

DEFAULT_MAX_ROWS = 500
DEFAULT_SAMPLE_ROWS = 15


def parameter_path(filename: str) -> Path:
    return Path(PARAMETERS_DIR) / filename