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

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


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

    flow_name = config.feed_name
    REPORTS = list(config.reports.keys())

    def add_args(self):
        """Add args to parser."""
        self.parser.add_arg(
            'partition',
            env_var='PARTITION',
            context_key='partition',
            type=exec_utils.json_arg_type,
            metavar='JSON',
            help='Partition filter as JSON dict.',
            required=False,
        )
        super().add_args()


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