#!/usr/bin/env python
"""Flow's exec. Run it with --help for usage."""
import logging

from feed_ingestion.flows.tiktok import config
from feed_ingestion.util import exec_utils

logger = logging.getLogger(__name__)

# Licensors that are ready for production (Update as needed)
PROD_READY_LICENSORS = ['theorchard', 'sme', 'smejp', 'smejpintl', 'altafonte']


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

    flow_name = config.feed_name
    LICENSORS = PROD_READY_LICENSORS
    REPORTS = config.reports
    REPORT_CONTEXT_KEY = 'reports'

    def __arguments(self):
        """Add args."""
        exec_utils.add_build_dbt_arg(
            self.parser,
        )
        exec_utils.add_arg_bool(
            self.parser,
            name='use_sme_s3',
            env_var='USE_SME_S3',
            context_key='use_sme_s3',
            help='When True the '
                 'licensor theorchard loads from sme-ca-prod-partners',
        )

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


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