"""Configuration parameters.""" import os ENVIRONMENT = os.environ.get('Environment', 'dev') AWS_REGION = 'us-east-1' BACKFILL_KINESIS_STREAM = os.environ.get( 'BACKFILL_KINESIS_STREAM') MR_ACTIVE_TABLE_NAME = os.environ.get( 'MR_ACTIVE_TABLE_NAME') DDB_ADDITIONAL_READ_CAP = int(os.environ.get('DDB_ADDITIONAL_READ_CAP', 200)) REDIS_HOST = os.environ.get('REDIS_HOST') REDIS_PORT = os.environ.get('REDIS_PORT') REDIS_DB = os.environ.get('REDIS_DB') REDIS_PASSWORD = os.environ.get('REDIS_PASSWORD', 'devredis') MAX_RECORDS = int(os.environ.get('MAX_RECORDS', 10000000)) TIMEOUT = int(os.environ.get('TIMEOUT', 60 * 60 * 24)) TOTAL_SEGMENTS = int(os.environ.get('TOTAL_SEGMENTS', 3)) # Snowflake settings SNOWFLAKE_ACCOUNT = os.environ.get('SNOWFLAKE_ACCOUNT', 'orchard') SNOWFLAKE_USER = os.environ.get('SNOWFLAKE_USER') SNOWFLAKE_PASSWORD = os.environ.get('SNOWFLAKE_PASSWORD') SNOWFLAKE_DATABASE = os.environ.get('SNOWFLAKE_DATABASE') SNOWFLAKE_SCHEMA = os.environ.get('SNOWFLAKE_SCHEMA') SNOWFLAKE_WAREHOUSE = os.environ.get('SNOWFLAKE_WAREHOUSE', 'LOAD_WH') SNOWFLAKE_TIMEZONE = os.environ.get('SNOWFLAKE_TIMEZONE', 'UTC') if ENVIRONMENT != 'test': assert BACKFILL_KINESIS_STREAM, 'BACKFILL_KINESIS_STREAM is not set' assert all( (SNOWFLAKE_ACCOUNT, SNOWFLAKE_USER, SNOWFLAKE_PASSWORD, SNOWFLAKE_DATABASE, SNOWFLAKE_SCHEMA, SNOWFLAKE_WAREHOUSE, SNOWFLAKE_TIMEZONE)), \ 'Snowflake connection parameters are not set' assert MR_ACTIVE_TABLE_NAME, 'MR_ACTIVE_TABLE_NAME is not set' assert all((REDIS_HOST, REDIS_PORT, REDIS_DB)), \ 'some of REDIS connection parameters are not set'