"""Snowflake connector."""
import snowflake.connector

from config import SNOWFLAKE_CONFIG


def get_snowflake_connection():
    """Return a Snowflake connection."""
    return snowflake.connector.connect(
        user=SNOWFLAKE_CONFIG['user'],
        account=SNOWFLAKE_CONFIG['account'],
        database=SNOWFLAKE_CONFIG['database'],
        schema=SNOWFLAKE_CONFIG['schema'],
        warehouse=SNOWFLAKE_CONFIG['warehouse'],
        role=SNOWFLAKE_CONFIG['role'],
        private_key=SNOWFLAKE_CONFIG['private_key'],
    )