aria/orchestrator/workflows/executor/dry.py

# Licensed to the Apache Software Foundation (ASF) under one or more
# contributor license agreements.  See the NOTICE file distributed with
# this work for additional information regarding copyright ownership.
# The ASF licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with
# the License.  You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
Dry task executor.
"""

from datetime import datetime

from . import base


class DryExecutor(base.BaseExecutor):                                                               # pylint: disable=abstract-method
    """
    Dry task executor: prints task information without causing any side effects.
    """
    def execute(self, ctx):
        task = ctx.task
        # updating the task manually instead of calling self._task_started(task),
        # to avoid any side effects raising that event might cause
        task.started_at = datetime.utcnow()
        task.status = task.STARTED

        dry_msg = '<dry> {name} {task.interface_name}.{task.operation_name} {suffix}'
        logger = ctx.logger.info if task.function else ctx.logger.debug

        if hasattr(task.actor, 'source_node'):
            name = '{source_node.name}->{target_node.name}'.format(
                source_node=task.actor.source_node, target_node=task.actor.target_node)
        else:
            name = task.actor.name

        if task.function:
            logger(dry_msg.format(name=name, task=ctx.task, suffix='started...'))
            logger(dry_msg.format(name=name, task=ctx.task, suffix='successful'))
        else:
            logger(dry_msg.format(name=name, task=ctx.task, suffix='has no implementation'))

        # updating the task manually instead of calling self._task_succeeded(task),
        # to avoid any side effects raising that event might cause
        task.ended_at = datetime.utcnow()
        task.status = task.SUCCESS
        ctx.model.task.update(task)