"""Base Tasks for workflows."""

from garcon import task

from feed_sender.flows import helpers


@task.decorate(timeout=7200)
def set_status(activity, feed_name, context_date, status):
    """Mark feed as provided status.

    Args:
        activity (ActivityWorker): The activity worker.
        feed_name (str): Name of the feed.
        context_date (str): Reporting date.
        status (str): Feed status.
    """
    helpers.set_status(feed_name, context_date, status)
    if status == helpers.STATUS_SENT:
        activity.logger.info(
            'Workflow {} on {} has completed.'.format(
                feed_name, context_date))