"""Snowflake executor for Amazon DataPulse Ingestion Workflow."""

from snowflake_connector.etl_connector import SQLLoader

from feed_ingestion.common.staging_raw_sf import base_executor
from feed_ingestion.flows.amazon_datapulse import config


class AmazonDataPulseExecutor(base_executor.SnowflakeSQLExecutorSR):
    """SnowflakeSQLExecutor for amazon_datapulse flow."""

    sql_loader = SQLLoader(__file__)

    @property
    def feed_name(self):
        """Name of the feed."""
        return config.feed_name

    @property
    def staging_raw_table(self):
        """Name of the staging_raw table."""
        # temporary support only fraud_report. Will be extended later.
        return config.reports['fraud_report']['staging_raw_table']