import os from secrets_manager.python_ext import PythonSecretsManager from src.utils.private_key import decode_sm_pkey, get_pkey DEV_ENVIRONMENT = "dev" PROD_ENVIRONMENT = "prod" ENVIRONMENT = os.environ.get("Environment", DEV_ENVIRONMENT) SCRIPT_NAME = "distro-bug-monitoring" secrets_manager_client = PythonSecretsManager(environment=ENVIRONMENT, service_name=SCRIPT_NAME) SNOWFLAKE_USER_NAME = os.environ.get("SNOWFLAKE_USER_NAME") SNOWFLAKE_ACCOUNT = os.environ.get("SNOWFLAKE_ACCOUNT", "delphi.us-east-1") SNOWFLAKE_DATABASE = os.environ.get("SNOWFLAKE_DATABASE") SNOWFLAKE_SCHEMA = os.environ.get("SNOWFLAKE_SCHEMA") SNOWFLAKE_ROLE = os.environ.get("SNOWFLAKE_ROLE") SNOWFLAKE_PRIVATE_KEY_PATH = os.environ.get("SNOWFLAKE_PRIVATE_KEY_PATH") SNOWFLAKE_KEY_PASSPHRASE = os.environ.get("SNOWFLAKE_KEY_PASSPHRASE") SNOWFLAKE_PRIVATE_KEY = None if SNOWFLAKE_PRIVATE_KEY_PATH and SNOWFLAKE_KEY_PASSPHRASE: SNOWFLAKE_PRIVATE_KEY = get_pkey(SNOWFLAKE_PRIVATE_KEY_PATH, SNOWFLAKE_KEY_PASSPHRASE) else: sm_private_key = secrets_manager_client.get_cred("SNOWFLAKE_PRIVATE_KEY") or None if sm_private_key: SNOWFLAKE_PRIVATE_KEY = decode_sm_pkey(sm_private_key) SLACK_CHANNEL_ID = os.environ.get("SLACK_CHANNEL_ID") SLACK_BOT_USER_OAUTH_TOKEN = secrets_manager_client.get_cred("SLACK_BOT_USER_OAUTH_TOKEN") SLACK_APP_USERNAME = os.environ.get("SLACK_APP_USERNAME", "distro-bug-monitor")