Source code for cowidev.testing.utils.orgs._config_loader

import yaml




class ConfigLoader:
    def __init__(self, config: dict) -> None:
        self.config = config



    @classmethod
    def from_yaml(cls, path: str):
        with open(path, "r") as f:
            config = yaml.safe_load(f)
        return cls(config)




    def countries_mapping(self):
        if "countries" not in self.config:
            return {}
        return _records_to_dict(self.config["countries"])




    def list_countries(self):
        return list(self.countries_mapping().values())






def _records_to_dict(records):
    return {record["org_name"]: record["owid_name"] for record in records}





def countries_mapping(config_path):
    """Get organization's country mapping.

    Note, only contains mappings for countries for which we source data from the organization.

    Returns:
        dict: ORG_COUNTRY_NAME -> OWID_COUNTRY_NAME
    """
    config_loader = ConfigLoader.from_yaml(config_path)
    return config_loader.countries_mapping()





def countries(config_path):
    """List countries for which we source data from the organization.

    Returns:
        list: List of countries sourced from the organization.
    """
    config_loader = ConfigLoader.from_yaml(config_path)
    return config_loader.config