john/SWIPAMforvRA8
1
0
Fork 0
mirror of https://github.com/jbowdre/SWIPAMforvRA8.git synced 2024-12-04 12:22:19 +00:00
Code Issues Projects Releases Packages Wiki Activity

Flesh out / clean up get_ip_ranges

Browse source
This commit is contained in:
jbowdre 2020-10-19 14:39:04 -05:00
parent 278ca69290
commit 1be1100a6d
3 changed files with 20 additions and 116 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/main/python/get_ip_ranges/requirements.txt
View file

@ -1,3 +1,4 @@
requests==2.21.0 requests==2.21.0
orionsdk orionsdk
json ipaddress

122
src/main/python/get_ip_ranges/source.py
View file

@ -13,7 +13,7 @@ import requests
from vra_ipam_utils.ipam import IPAM from vra_ipam_utils.ipam import IPAM
import logging import logging
from orionsdk import SwisClient from orionsdk import SwisClient
import json import ipaddress
''' '''
Example payload: Example payload:
@ -40,117 +40,31 @@ def handler(context, inputs):
return ipam.get_ip_ranges() return ipam.get_ip_ranges()
def do_get_ip_ranges(self, auth_credentials, cert): def do_get_ip_ranges(self, auth_credentials, cert):
# Your implemention goes here
username = auth_credentials["privateKeyId"] username = auth_credentials["privateKeyId"]
password = auth_credentials["privateKey"] password = auth_credentials["privateKey"]
hostname = self.inputs["endpoint"]["endpointProperties"]["hostName"] hostname = self.inputs["endpoint"]["endpointProperties"]["hostName"]
requests.packages.urllib3.disable_warnings()
swis = SwisClient(hostname, username, password) swis = SwisClient(hostname, username, password)
result_ranges = []
dResults = swis.query("SELECT DISTINCT GroupID, FriendlyName, Address, CIDR, Comments, Location, i.CustomProperties.Site_ID FROM IPAM.GroupNode i WHERE GroupTypeText LIKE 'Subnet' AND Address LIKE '10.28.248.%'") qResult = swis.query("SELECT DISTINCT GroupID AS id, FriendlyName AS name, Address AS addressSpaceId, CIDR AS subnetPrefixLength, Comments AS description, i.CustomProperties.Gateway as gatewayAddress, i.CustomProperties.DNS_Servers as dnsServers, i.CustomProperties.Site_ID AS siteId FROM IPAM.GroupNode i WHERE GroupTypeText LIKE 'Subnet' AND i.CustomProperties.VRA_Range = TRUE")
jResults = json.dumps(dResults) for range in qResult['results']:
network = ipaddress.ip_network(str(range['addressSpaceId']) + '/' + str(range['subnetPrefixLength']))
range['ipVersion'] = 'IPv' + str(network.version)
range['startIPAddress'] = str(network[10])
## If many IP ranges are expected on the IPAM server, it is considered a best practice range['endIPAddress'] = str(network[-6])
## to return them page by page instead of all at once. range['dnsServerAddresses'] = [server.strip() for server in str(range['dnsServers']).split(',')]
## The vRA IPAM Service will propagate a pageToken string with each consecutive request range['tags'] = [{
## until all pages are exhausted "key": "Site",
# pageToken = self.inputs['pagingAndSorting'].get('pageToken', None) ## The first request that vRA sends is with 'None' pageToken "value": range['siteId']
}]
result_ranges.append(range)
## Plug your implementation here to collect all the ranges from the external IPAM system
result_ranges, next_page_token = collect_ranges(pageToken)
result = { result = {
"ipRanges": result_ranges "ipRanges": result_ranges
} }
## Return the next page token so that vRA can process the first page and then fetch the second page or ranges with the next request
if next_page_token is not None:
result["nextPageToken"] = next_page_token
return result return result
def collect_ranges(pageToken):
logging.info("Collecting ranges")
range1 = {
"id": "us1lab-servers-10.28.248.128_26",
"name": "US1-Lab Servers 10.28.248.128/26",
"startIPAddress": "10.28.248.129",
"endIPAddress": "10.28.248.189",
"description": "labNet",
"ipVersion": "IPv4",
"addressSpaceId": "default",
"subnetPrefixLength": "26",
"gatewayAddress": "10.28.248.190",
"dnsServerAddresses": ["140.165.150.35", "140.165.235.36"],
"dnsSearchDomains": ["tdy.teledyne.com"],
"domain": "tdy.teledyne.com",
"tags": [{
"key": "Building",
"value": "7"
}],
"properties": {
}
}
# range2 = {
# "id": "range2",
# "name": "sample name 2",
# "startIPAddress": "10.23.117.1",
# "endIPAddress": "10.23.117.254",
# "description": "sampleDescription",
# "ipVersion": "IPv4",
# "addressSpaceId": "default",
# "subnetPrefixLength": "24",
# "dnsServerAddresses": ["10.10.17.3", "10.10.22.1"],
# "dnsSearchDomains": ["com", "net", "test.local"],
# "domain": "test.local",
# "tags": [{
# "key": "Building",
# "value": "VMware main facility"
# }],
# "properties": {
# }
# }
result = []
# next_page_token = None
# if pageToken is None:
# result = [range1]
# next_page_token = "87811419dec2112cda2aa29685685d650ac1f61f"
# else:
# result = [range2]
result = [range1]
return result, next_page_token

11
src/main/python/validate_endpoint/source.py
View file

@ -59,18 +59,7 @@ def do_validate_endpoint(self, auth_credentials, cert):
else: else:
raise Exception("Failed to connect") raise Exception("Failed to connect")
# response = requests.get("https://" + self.inputs["endpointProperties"]["hostName"], verify=cert, auth=(username, password))
# if response.status_code == 200:
# return {
# "message": "Validated successfully",
# "statusCode": "200"
# }
# elif response.status_code == 401:
# logging.error(f"Invalid credentials error: {str(response.content)}")
# raise Exception(f"Invalid credentials error: {str(response.content)}")
# else:
# raise Exception(f"Failed to connect: {str(response.content)}")
except Exception as e: except Exception as e:
""" In case of SSL validation error, a InvalidCertificateException is raised. """ In case of SSL validation error, a InvalidCertificateException is raised.
So that the IPAM SDK can go ahead and fetch the server certificate So that the IPAM SDK can go ahead and fetch the server certificate