import os

AWS_REGION = "eu-west-1"


class EnrichmentType:
    ENRICH_AGE = "EnrichAge"
    ENRICH_NAME = "EnrichName"
    ENRICH_UNIQUE_GENDER = "EnrichUniqueGender"
    ENRICH_FEATURE_ENGINEERING = "EnrichFeatureEngineering"
    ENRICH_GEOGRAPHICS_PELIAS = "EnrichGeographicsPelias"
    ENRICH_RFM = "EnrichRFM"
    ENRICH_SUPERFAN = "EnrichSuperfan"
    ENRICH_MACHINE_LEARNING_CLUSTERS = "EnrichMachineLearningClusters"


ENRICHMENTS = [
    EnrichmentType.ENRICH_AGE,
    EnrichmentType.ENRICH_NAME,
    EnrichmentType.ENRICH_UNIQUE_GENDER,
    EnrichmentType.ENRICH_FEATURE_ENGINEERING,
    EnrichmentType.ENRICH_GEOGRAPHICS_PELIAS,
    EnrichmentType.ENRICH_RFM,
    EnrichmentType.ENRICH_SUPERFAN,
    EnrichmentType.ENRICH_MACHINE_LEARNING_CLUSTERS,
]

ALGORITHMS = [
    EnrichmentType.ENRICH_RFM,
    EnrichmentType.ENRICH_SUPERFAN,
    EnrichmentType.ENRICH_MACHINE_LEARNING_CLUSTERS,
]

DEPENDENCIES = [
    [EnrichmentType.ENRICH_AGE, EnrichmentType.ENRICH_MACHINE_LEARNING_CLUSTERS],
    [EnrichmentType.ENRICH_NAME, EnrichmentType.ENRICH_UNIQUE_GENDER],
    [EnrichmentType.ENRICH_UNIQUE_GENDER, EnrichmentType.ENRICH_MACHINE_LEARNING_CLUSTERS],
    [EnrichmentType.ENRICH_FEATURE_ENGINEERING, EnrichmentType.ENRICH_RFM],
    [EnrichmentType.ENRICH_RFM, EnrichmentType.ENRICH_MACHINE_LEARNING_CLUSTERS],
    [EnrichmentType.ENRICH_FEATURE_ENGINEERING, EnrichmentType.ENRICH_SUPERFAN],
    [EnrichmentType.ENRICH_FEATURE_ENGINEERING, EnrichmentType.ENRICH_MACHINE_LEARNING_CLUSTERS],
    [EnrichmentType.ENRICH_GEOGRAPHICS_PELIAS, EnrichmentType.ENRICH_MACHINE_LEARNING_CLUSTERS],
]


def _get_appsync_profile():
    stage = os.getenv("STAGE", "dev")
    return "devel" if stage == "dev" else stage


APPSYNC_PROFILE = _get_appsync_profile()