from snowflake_connector.etl_connector import SQLLoader
from snowflake_connector.etl_connector import SnowflakeSQLExecutor

# Load SQL templates
sql_loader = SQLLoader(__file__)


class SFExecutor(SnowflakeSQLExecutor):
    def get_es_ids(self):
        """Get all IDs (via Snowflake)."""
        # Option 1: Try with params first
        params = dict(schema=self.sf_config['schema'])
        return self.fetchall_query(
            sql_loader, 'get_es_ids', params)