"""Postgress datawarehouse related functions.""" from garcon_contrib.snowflake import garcon_snowflake as snowflake from flows import config def execute(sql): """Execute a Snowflake query. Args: sql (str): Query to run. Returns: list: query result. """ return snowflake.execute_with_py_conn( sql, snowflake.FetchEnum.ALL, sf_config=config.SNOWFLAKE_CREDENTIALS, result_key='results')['results']