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):
        with ctx.persist_changes:
            # updating the task manually instead of calling self._task_started(task),
            # to avoid any side effects raising that event might cause
            ctx.task.started_at = datetime.utcnow()
            ctx.task.status = ctx.task.STARTED

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

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

            if ctx.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
            ctx.task.ended_at = datetime.utcnow()
            ctx.task.status = ctx.task.SUCCESS