Zack Meier
176 lines
7.6 KiB
Python
176 lines
7.6 KiB
Python
import os
|
|
import argparse
|
|
import json
|
|
import urllib3
|
|
import requests
|
|
from requests_ntlm import HttpNtlmAuth
|
|
# Use pip install requests_ntlm to install the necessary modules
|
|
|
|
class API(object):
|
|
def __init__(self):
|
|
if not ((os.environ.get('ITD_SHAREPOINT_PASS')) and (os.environ.get('ITD_SHAREPOINT_USER'))):
|
|
print("\n*** ERROR: Environment Variables are not set for SharePoint ***")
|
|
print("Set OS environment variables ITD_SHAREPOINT_PASS and ITD_SHAREPOINT_USER")
|
|
exit(1)
|
|
else:
|
|
# Username is in the format of 'NDGOV\\<userName>'
|
|
sp_user = os.environ['ITD_SHAREPOINT_USER']
|
|
sp_pass = os.environ['ITD_SHAREPOINT_PASS']
|
|
|
|
self.server_url = "https://share.nd.gov/"
|
|
self.site_url = self.server_url + "itd/Computer-Systems/Distributed-Systems/VMWare/_api/web/"
|
|
self.context_url = self.server_url + "itd/Computer-Systems/Distributed-Systems/VMWare/_api/contextinfo/"
|
|
self.env = "All"
|
|
|
|
self.headers = {
|
|
"Accept":"application/json; odata=verbose",
|
|
"Content-Type":"application/json; odata=verbose",
|
|
"odata":"verbose",
|
|
"X-RequestForceAuthentication":"true"
|
|
}
|
|
|
|
self.auth = HttpNtlmAuth(sp_user, sp_pass)
|
|
|
|
def get_AllNodes(self):
|
|
|
|
list_name = 'VM Guests'
|
|
|
|
# Get list information
|
|
requests.packages.urllib3.disable_warnings()
|
|
r = requests.get(self.site_url + "lists/GetByTitle('%s')" % list_name, auth=self.auth, headers=self.headers, verify=False)
|
|
list_guid = r.json()['d']['Id']
|
|
list_itemcount = r.json()['d']['ItemCount']
|
|
|
|
# Query list items
|
|
api_items_url = self.site_url + "Lists(guid'%s')/Items" % list_guid
|
|
concat_items = []
|
|
# Parse the list
|
|
requests.packages.urllib3.disable_warnings()
|
|
cur_page = requests.get(api_items_url, auth=self.auth, headers=self.headers, verify=False)
|
|
concat_items += cur_page.json()['d']['results']
|
|
|
|
# Sharepoint uses pagination with '__next' in the JSON results to indicate additional pages of information
|
|
# are present. Loop through the pages and concatenate the information into one dataset
|
|
while '__next' in cur_page.json()['d']:
|
|
requests.packages.urllib3.disable_warnings()
|
|
cur_page = requests.get(cur_page.json()['d']['__next'], auth=self.auth, headers=self.headers, verify=False)
|
|
concat_items += cur_page.json()['d']['results']
|
|
|
|
return concat_items
|
|
|
|
def get_ActiveNodes(self, e):
|
|
|
|
self.env = e
|
|
list_name = 'VM Guests'
|
|
|
|
# Get list information
|
|
requests.packages.urllib3.disable_warnings()
|
|
r = requests.get(self.site_url + "lists/GetByTitle('%s')" % list_name, auth=self.auth, headers=self.headers, verify=False)
|
|
list_guid = r.json()['d']['Id']
|
|
list_itemcount = r.json()['d']['ItemCount']
|
|
|
|
# Query list items
|
|
api_items_url = self.site_url + "Lists(guid'%s')/Items" % list_guid
|
|
if self.env == 'All':
|
|
active_items_url = api_items_url + "?$filter=(Status eq 'Current') or (Status eq 'Change')"
|
|
else:
|
|
active_items_url = api_items_url + "?$filter=(Environment eq '%s') and ((Status eq 'Current') or (Status eq 'Change'))" % self.env
|
|
concat_items = []
|
|
|
|
# Parse the list
|
|
requests.packages.urllib3.disable_warnings()
|
|
cur_page = requests.get(active_items_url, auth=self.auth, headers=self.headers, verify=False)
|
|
concat_items += cur_page.json()['d']['results']
|
|
|
|
# Sharepoint uses pagination with '__next' in the JSON results to indicate additional pages of information
|
|
# are present. Loop through the pages and concatenate the information into one dataset
|
|
while '__next' in cur_page.json()['d']:
|
|
requests.packages.urllib3.disable_warnings()
|
|
cur_page = requests.get(cur_page.json()['d']['__next'], auth=self.auth, headers=self.headers, verify=False)
|
|
concat_items += cur_page.json()['d']['results']
|
|
|
|
return concat_items
|
|
|
|
def find_VM(self, hostname):
|
|
|
|
domains = [".nd.gov", ".ndtestdev.nd.dev", ".ndcloud.gov", ".testnd.gov", ".itd.nd.gov", ".ns.nd.gov", ".cloudapp.azure.com", ".uat.mmis.nd.gov", ".test.ndcloud.gov", ".k12tst.nd.us", ".legend.nd.gov", ".tst.mmis.dev", ".trn.mmis.nd.gov", ".sit.mmis.nd.gov", ".nd.gov-mag", ".uat.mmis.dev", ".mo.legend.nd.gov", ".video.nd.gov", ".mmis.nd.gov", ".sit.legend.nd.gov", ".trn.mmis.dev", ".prm.mmis.nd.gov", ".state.nd.us", ".k12.nd.us", ".stg.k12.nd.us", ".sit.mmis.dev", "ndnic.com"]
|
|
|
|
for domain in domains:
|
|
try:
|
|
data = self.get_Node(hostname + domain)
|
|
if 'Title' in data[0]:
|
|
hostname = data[0]['Title']
|
|
break
|
|
except:
|
|
continue
|
|
|
|
return(hostname)
|
|
|
|
def get_Node(self, server):
|
|
|
|
myNode = server
|
|
list_name = 'VM Guests'
|
|
|
|
# Get list information
|
|
requests.packages.urllib3.disable_warnings()
|
|
r = requests.get(self.site_url + "lists/GetByTitle('%s')" % list_name, auth=self.auth, headers=self.headers, verify=False)
|
|
list_guid = r.json()['d']['Id']
|
|
|
|
# Query list items
|
|
api_items_url = self.site_url + "Lists(guid'%s')/Items" % list_guid
|
|
singleServer_url = api_items_url + "?$filter=Title eq '%s'" % myNode
|
|
cur_page = requests.get(singleServer_url, auth=self.auth, headers=self.headers, verify=False)
|
|
|
|
return cur_page.json()['d']['results']
|
|
|
|
def get_Agency(self, aID):
|
|
|
|
agencyID = aID
|
|
list_name = 'Agency'
|
|
|
|
# Get list information
|
|
requests.packages.urllib3.disable_warnings()
|
|
r = requests.get(self.site_url + "lists/GetByTitle('%s')" % list_name, auth=self.auth, headers=self.headers, verify=False)
|
|
list_guid = r.json()['d']['Id']
|
|
|
|
# Query list items
|
|
api_items_url = self.site_url + "Lists(guid'%s')/Items" % list_guid
|
|
singleServer_url = api_items_url + "?$filter=ID eq '%s'" % agencyID
|
|
cur_page = requests.get(singleServer_url, auth=self.auth, headers=self.headers, verify=False)
|
|
|
|
return cur_page.json()['d']['results'][0]['Title']
|
|
|
|
def get_AppName(self, appID):
|
|
|
|
applicationID = appID
|
|
list_name = 'VM App Name'
|
|
|
|
# Get list information
|
|
requests.packages.urllib3.disable_warnings()
|
|
r = requests.get(self.site_url + "lists/GetByTitle('%s')" % list_name, auth=self.auth, headers=self.headers, verify=False)
|
|
list_guid = r.json()['d']['Id']
|
|
|
|
# Query list items
|
|
api_items_url = self.site_url + "Lists(guid'%s')/Items" % list_guid
|
|
singleServer_url = api_items_url + "?$filter=ID eq '%s'" % applicationID
|
|
cur_page = requests.get(singleServer_url, auth=self.auth, headers=self.headers, verify=False)
|
|
|
|
return cur_page.json()['d']['results'][0]['Title']
|
|
|
|
def get_AppNodes(self, appID):
|
|
|
|
applicationID = appID
|
|
list_name = 'VM Guests'
|
|
|
|
# Get list information
|
|
requests.packages.urllib3.disable_warnings()
|
|
r = requests.get(self.site_url + "lists/GetByTitle('%s')" % list_name, auth=self.auth, headers=self.headers, verify=False)
|
|
list_guid = r.json()['d']['Id']
|
|
|
|
# Query list items
|
|
api_items_url = self.site_url + "Lists(guid'%s')/Items" % list_guid
|
|
singleServer_url = api_items_url + "?$filter=AppName eq '%s'" % appID
|
|
cur_page = requests.get(singleServer_url, auth=self.auth, headers=self.headers, verify=False)
|
|
|
|
return cur_page.json()['d']['results']
|