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 from aria import workflow
18 from aria.orchestrator.workflows.api import task
19 from aria.orchestrator.workflows.exceptions import TaskException
22 INTERFACE_NAME = 'Maintenance'
23 ENABLE_OPERATION_NAME = 'enable'
24 DISABLE_OPERATION_NAME = 'disable'
28 def maintenance(ctx, graph, enabled):
30 Custom workflow to call the operations on the Maintenance interface.
33 for node in ctx.model.node.iter():
35 graph.add_tasks(task.OperationTask(node,
36 interface_name=INTERFACE_NAME,
37 operation_name=ENABLE_OPERATION_NAME if enabled
38 else DISABLE_OPERATION_NAME))