NcmConfigDownload

Complexity

Total Complexity

11

Size/Duplication

Total Lines	70
Duplicated Lines	0 %

2 Methods

Rating	Name	Duplication	Size	Complexity
B	run()	0	40	6
B	transfer_complete()	0	27	5

#!/usr/bin/env python

# Licensed to the StackStorm, Inc ('StackStorm') 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.

import time

from st2actions.runners.pythonrunner import Action
from orionsdk import SwisClient


class NcmConfigDownload(Action):

    def transfer_complete(self, transfer_id, swis):
        ts = {}
        while True:
            transfer_data = swis.query(
                """SELECT TransferID,
                Action, Status, ErrorMessage, DeviceOutput
                FROM NCM.TransferResults WHERE TransferID=@transfer_id""",
                transfer_id=transfer_id)

            status = transfer_data['results'][0]['Status']

            if status == 1:
                time.sleep(10)
            elif status == 2:
                ts['status'] = "Complete"
                break
            elif status == 3:
                ts['status'] = "Error"
                ts['ErrorMessage'] = transfer_data['results'][0][
                    'ErrorMessage']
                break
            else:
                ts['status'] = "Unknown"
                ts['ErrorMessage'] = "Invalid stauts: {}".format(status)
                break

        return ts

    def run(self, node, platform, configs):
        """
        Mostly completed! Download configurations via Solarwinds Orion NCM ....

        see https://github.com/solarwinds/OrionSDK/wiki/NCM-Config-Transfer
        """

        # Set up the results
        results = {}

        try:
            swis = SwisClient(self.config['orion'][platform]['host'],
                              self.config['orion'][platform]['user'],
                              self.config['orion'][platform]['password'])
        except KeyError:
            raise ValueError("Orion host details not in the config.yaml")

        node_data = swis.query(
            "SELECT NodeID FROM Cirrus.Nodes WHERE NodeCaption=@node",
            node=node)
        node_ids = []

        try:
            node_ids.append(node_data['results'][0]['NodeID'])
        except IndexError:
            for config in configs:
                results[config] = {'status': "Error",
                                   'ErrorMessage': "Invalid NodeID"}
        else:
            for config in configs:
                orion_data = swis.invoke("Cirrus.ConfigArchive",
                                         "DownloadConfig",
                                         node_ids,
                                         config)
                self.logger.info("transfer_id: " + orion_data[0])
                transfer_id = orion_data[0]

                results[config] = self.transfer_complete(transfer_id, swis)

        return results