swinds.py

#!/usr/bin/env python


'''
Custom dynamic inventory script for Ansible and Solar Winds, in Python.
This was tested on Python 2.7.6, Orion 2016.2.100, and Ansible  2.3.0.0.

(c) 2017, Chris Babcock (chris@bluegreenit.com)

https://github.com/cbabs/solarwinds-ansible-inv

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

NOTE:  This software is free to use for any reason or purpose. That said, the
author request that improvements be submitted back to the repo or forked
to something public.

'''
import argparse
import ConfigParser
import requests
import re

try:
    import json
except ImportError:
    import simplejson as json


config_file = 'swinds.ini'

# Get configuration variables
config = ConfigParser.ConfigParser()
config.readfp(open(config_file))



# Orion Server IP or DNS/hostname
server = config.get('solarwinds', 'npm_server')
# Orion Username
user = config.get('solarwinds', 'npm_user')
# Orion Password
password = config.get('solarwinds', 'npm_password')
# Field for groups
groupField = 'GroupName'
# Field for host
hostField = 'SysName'

payload = "query=SELECT C.Name as GroupName, N.SysName FROM Orion.Nodes as N JOIN Orion.ContainerMemberSnapshots as CM on N.NodeID = CM.EntityID JOIN Orion.Container as C on CM.ContainerID=C.ContainerID WHERE CM.EntityDisplayName = 'Node' AND N.Vendor = 'Cisco'"

use_groups = True
parentField = 'ParentGroupName'
childField = 'ChildGroupName'

group_payload = "query=SELECT C.Name as ParentGroupName, CM.Name as ChildGroupName FROM Orion.ContainerMemberSnapshots as CM JOIN Orion.Container as C on CM.ContainerID=C.ContainerID WHERE CM.EntityDisplayName = 'Group'"

#payload = "query=SELECT+" + hostField + "+," + groupField + "+FROM+Orion.Nodes"
url = "https://"+server+":17778/SolarWinds/InformationService/v3/Json/Query"
req = requests.get(url, params=payload, verify=False, auth=(user, password))

jsonget = req.json()


class SwInventory(object):

    # CLI arguments
    def read_cli(self):
        parser = argparse.ArgumentParser()
        parser.add_argument('--host')
        parser.add_argument('--list', action='store_true')
        self.options = parser.parse_args()

    def __init__(self):
        self.inventory = {}
        self.read_cli_args()

        # Called with `--list`.
        if self.args.list:
            self.inventory = self.get_list()
            if use_groups:
                self.groups = self.get_groups()
                self.add_groups_to_hosts(self.groups)
        # Called with `--host [hostname]`.
        elif self.args.host:
            # Not implemented, since we return _meta info `--list`.
            self.inventory = self.empty_inventory()
        # If no groups or vars are present, return empty inventory.
        else:
            self.inventory = self.empty_inventory()

        print(json.dumps(self.inventory, indent=2))
    def get_list(self):
        hostsData = jsonget
        dumped = eval(json.dumps(jsonget))

        # Inject data below to speed up script
        final_dict = {'_meta': {'hostvars': {}}}

        # Loop hosts in groups and remove special chars from group names
        for m in dumped['results']:
            # Allow Upper/lower letters and numbers. Replace everything else with underscore
            m[groupField] = self.clean_inventory_item(m[groupField])
            if m[groupField] in final_dict:
                final_dict[m[groupField]]['hosts'].append(m[hostField])
            else:
                final_dict[m[groupField]] = {'hosts': [m[hostField]]}
        return final_dict

        #if self.args.groups:
    def get_groups(self):
        req = requests.get(url, params=group_payload, verify=False, auth=(user, password))
        hostsData = req.json()
        dumped = eval(json.dumps(hostsData))
        
        parentField = 'ParentGroupName'
        childField = 'ChildGroupName'
        final_dict = {} 
        for m in dumped['results']:
            # Allow Upper/lower letters and numbers. Replace everything else with underscore
            m[parentField] = self.clean_inventory_item(m[parentField])
            m[childField] = self.clean_inventory_item(m[childField])
            if m[parentField] in final_dict:
                final_dict[m[parentField]]['children'].append(m[childField])
            else:
                final_dict[m[parentField]] = {'children': [m[childField]]}
        return final_dict

    def add_groups_to_hosts (self, groups):
        self.inventory.update(groups)

    @staticmethod
    def clean_inventory_item(item):
        item = re.sub('[^A-Za-z0-9]+', '_', item)
        return item

    # Empty inventory for testing.
    def empty_inventory(self):
        return {'_meta': {'hostvars': {}}}

    # Read the command line args passed to the script.
    def read_cli_args(self):
        parser = argparse.ArgumentParser()
        parser.add_argument('--list', action='store_true')
        parser.add_argument('--host', action='store')
        self.args = parser.parse_args()

# Get the inventory.
SwInventory()