"""Snowflake executor."""
from contextlib import contextmanager

import snowflake.connector
from snowflake.connector import DictCursor

from dbdeploy.util.snowflake import helpers


@contextmanager
def sfcursor(snowflake_account: str):  # type: ignore
    with snowflake.connector.connect(
        **helpers.get_sf_config(snowflake_account)
    ) as con:
        yield con.cursor(DictCursor)