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

from garcon_contrib.dynamo_feed_status import garcon_feed_status

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


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

    flow_name = config.feed_name
    LICENSORS = config.licensors

    def is_context_completed(self, context: dict, date: datetime.date):
        """Check if context is completed."""
        if context['licensor'] == 'smej':
            self.args.expected_status = (
                garcon_feed_status.STATUS_POPULATED_RAW_TABLE)
        else:
            self.args.expected_status = (
                garcon_feed_status.STATUS_INGESTED)

        return super().is_context_completed(context, date)

    def add_args(self):
        """Add args."""
        exec_utils.add_licensor_arg(
            parser=self.parser,
            choices=config.licensors
        )
        super().add_args()


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