import os


def environ_get(variable_name, default_value=None):
    airflow_name = f'AIRFLOW__CUSTOM__{variable_name}'
    airflow_value = os.environ.get(airflow_name)

    if airflow_value:
        return airflow_value

    return os.environ.get(variable_name, default_value)

SECRETS_PREFIX = environ_get('SECRETS_PREFIX')

EXTRACT_SALES_SERVICE_NAME = environ_get('EXTRACT_SALES_SERVICE_NAME')
SUBNET_IDS = environ_get('SUBNET_IDS')

SF_ACCOUNT = environ_get('SF_ACCOUNT')
SF_DATABASE = environ_get('SF_DATABASE')
SF_SCHEMA = environ_get('SF_SCHEMA')
SF_USER = environ_get('SF_USER')
SF_ROLE = environ_get('SF_ROLE')
SF_WAREHOUSE = environ_get('SF_WAREHOUSE')

OUTFILE_BUCKET = environ_get('OUTFILE_BUCKET')
OUTFILE_AWS_KEY_ID = environ_get('OUTFILE_AWS_KEY_ID')
OUTFILE_AWS_SECRET_KEY = environ_get('OUTFILE_AWS_SECRET_KEY')