1 # Licensed to the Apache Software Foundation (ASF) under one or more
2 # contributor license agreements. See the NOTICE file distributed with
3 # this work for additional information regarding copyright ownership.
4 # The ASF licenses this file to You under the Apache License, Version 2.0
5 # (the "License"); you may not use this file except in compliance with
6 # the License. You may obtain a copy of the License at
8 # http://www.apache.org/licenses/LICENSE-2.0
10 # Unless required by applicable law or agreed to in writing, software
11 # distributed under the License is distributed on an "AS IS" BASIS,
12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
17 Base class for task executors.
20 from aria import logger
21 from aria.orchestrator import events
24 class BaseExecutor(logger.LoggerMixin):
26 Base class for task executors.
28 def _execute(self, ctx):
29 raise NotImplementedError
31 def execute(self, ctx):
35 :param task: task to execute
40 # In this case the task is missing a function. This task still gets to an
41 # executor, but since there is nothing to run, we by default simply skip the
43 self._task_started(ctx)
44 self._task_succeeded(ctx)
52 def terminate(self, task_id):
54 Terminate the executing task
60 def _task_started(ctx):
61 events.start_task_signal.send(ctx)
64 def _task_failed(ctx, exception, traceback=None):
65 events.on_failure_task_signal.send(ctx, exception=exception, traceback=traceback)
68 def _task_succeeded(ctx):
69 events.on_success_task_signal.send(ctx)
72 class StubTaskExecutor(BaseExecutor): # pylint: disable=abstract-method
73 def execute(self, ctx, *args, **kwargs):
74 with ctx.persist_changes:
75 ctx.task.status = ctx.task.SUCCESS