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.
20 from . import operation
23 class NodeToolBelt(object):
25 Node operation tool belt.
27 def __init__(self, operation_context):
28 self._op_context = operation_context
33 The host ip of the current node
36 assert isinstance(self._op_context, operation.NodeOperationContext)
37 return self._op_context.node.host.attributes.get('ip')
40 class RelationshipToolBelt(object):
42 Relationship operation tool belt.
44 def __init__(self, operation_context):
45 self._op_context = operation_context
48 def toolbelt(operation_context):
50 Get a toolbelt from to the current operation executor.
52 :param operation_context:
54 if isinstance(operation_context, operation.NodeOperationContext):
55 return NodeToolBelt(operation_context)
56 elif isinstance(operation_context, operation.RelationshipOperationContext):
57 return RelationshipToolBelt(operation_context)
59 raise RuntimeError("Operation context not supported")