"""SQLAlchemy Snowflake database connector."""

from functools import partial

from sqlalchemy import create_engine

from availability_etl import config
from .utils import get_connection

engine = create_engine(config.SNOWFLAKE_DB_CONNECTION_STRING)

use_statements = (
    'USE ROLE "{}";'.format(config.SF_ROLE),
    'USE DATABASE "{}";'.format(config.SF_DB_NAME),
    'USE WAREHOUSE "{}";'.format(config.SF_WAREHOUSE),
)
get_snowflake_connection = partial(get_connection, engine, use_statements)