"""Config.""" import os from abacus_common_logic.utils.ssh import get_private_key from dotenv import load_dotenv load_dotenv() ENVIRONMENT = os.environ.get('ENVIRONMENT', 'dev') SNOWFLAKE = { 'role': os.environ.get('SNOWFLAKE_ROLE'), 'warehouse': os.environ.get('SNOWFLAKE_WAREHOUSE'), 'db': os.environ.get('SNOWFLAKE_DATABASE', 'royalty_accounting'), 'schema': os.environ.get('SNOWFLAKE_SCHEMA', 'dev'), 'user': os.environ.get('SNOWFLAKE_USER'), 'password': os.environ.get('SNOWFLAKE_PASSWORD'), 'account': os.environ.get('SNOWFLAKE_ACCOUNT', 'orchard'), 'private_key': None } if ENVIRONMENT == 'dev' and os.environ.get('SNOWFLAKE_KEY_PASSPHRASE'): SNOWFLAKE['private_key'] = get_private_key( os.environ.get( 'SNOWFLAKE_PRIVATE_KEY_PATH', '{}/.ssh/snowflake/rsa_key.p8'.format(os.environ['HOME']) ), os.environ.get('SNOWFLAKE_KEY_PASSPHRASE') ) TRANSACTION_TYPE_SYNC = { 's3_path': os.environ.get('S3_TXN_TYPES_PATH'), 's3_filename': os.environ.get('S3_TXN_TYPES_FILENAME'), 'stage': os.environ.get('SNOWFLAKE_STAGE'), } STORE_SYNC = { 's3_path': os.environ.get('S3_STORES_PATH'), 's3_filename': os.environ.get('S3_STORES_FILENAME'), 'stage': os.environ.get('SNOWFLAKE_STAGE'), }