#!/usr/bin/env python
"""Flow's exec. Run it with --help for usage."""
from feed_ingestion.flows.amazon_music import config
from feed_ingestion.util import exec_utils


class FlowExec(exec_utils.FlowExecDaily,
               exec_utils.LicensorMixin,
               exec_utils.ReportMixin):
    """Flow's exec."""

    flow_name = config.feed_name

    LICENSORS = config.licensors
    REPORTS = config.reports

    def __arguments(self):
        self.parser.add_arg(
            required=False,
            name='countries',
            context_key='countries',
            env_var='COUNTRIES',
            help='comma-separated list of countries example "AU,US"'
        )
        self.parser.add_arg(
            required=False,
            name='source',
            context_key='source',
            env_var='SOURCE',
            default=config.SOURCE_SFTP,
            choices=config.sources,
            help=(
                'Source of the staging data. "sftp" loads from partner '
                'drop files; "datapulse" loads from amazon_datapulse '
                'staging_raw tables in Snowflake.'
            ),
        )
        exec_utils.add_build_dbt_arg(
            self.parser,
        )

    def add_args(self):
        """Add args."""
        self.__arguments()
        super().add_args()


if __name__ == '__main__':
    FlowExec().main()