Completed
Push — master ( 772307...2b2f12 )
by Tomaz
31s
created

NodeUnmanage.run()   B

Complexity

Conditions 2

Size

Total Lines 25

Duplication

Lines 0
Ratio 0 %
Metric Value
cc 2
dl 0
loc 25
rs 8.8571
1
# Licensed to the StackStorm, Inc ('StackStorm') 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
7
#
8
#     http://www.apache.org/licenses/LICENSE-2.0
9
#
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.
15
16
from datetime import datetime, timedelta
17
18
from lib.actions import OrionBaseAction
19
20
21
class NodeUnmanage(OrionBaseAction):
22
    def run(self, node, platform, minutes):
23
        """
24
        Unmanage an Orion node
25
        """
26
27
        if minutes > self.config['unmanage_max']:
28
            raise ValueError(
29
                "minutes ({}) greater than unmanage_max ({})".format(
30
                    minutes, self.config['unmanage_max']))
31
32
        self.connect(platform)
33
34
        NodeId = "N:{}".format(self.get_node_id(node))
35
        now = datetime.utcnow()
36
        later = now + timedelta(minutes=minutes)
37
38
        self.invoke("Orion.Nodes",
39
                    "Unmanage",
40
                    NodeId,
41
                    now,
42
                    later,
43
                    False)
44
45
        # The Invoke returns None, so return something.
46
        return True
47