-
Notifications
You must be signed in to change notification settings - Fork 25
/
swinds.py
executable file
·159 lines (123 loc) · 5.45 KB
/
swinds.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
#!/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 ([email protected])
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()