"""Snowflake connector."""

import config

import snowflake.connector


def get_snowflake_context():
    """Get Snowflake connection context.

    Returns:
        SnowflakeConnection

    """
    try:
        snowflake_context = snowflake.connector.connect(
            account=config.SNOWFLAKE_ACCOUNT,
            user=config.SNOWFLAKE_USER,
            password=config.SNOWFLAKE_PASSWORD,
            database=config.SNOWFLAKE_DATABASE,
            schema=config.SNOWFLAKE_SCHEMA,
            warehouse=config.SNOWFLAKE_WAREHOUSE,
            timezone=config.SNOWFLAKE_TIMEZONE
        )
        return snowflake_context
    except Exception as ex:
        print(ex)
        return None