contivModelClient.py

# contivModel REST client

import urllib
import urllib2
import json
import argparse
import os

# Exit on error
def errorExit(str):
    print "############### Operation failed: " + str + " ###############"
    os._exit(1)

# HTTP Delete wrapper
def httpDelete(url):
    opener = urllib2.build_opener(urllib2.HTTPHandler)
    request = urllib2.Request(url)
    request.get_method = lambda: 'DELETE'
    try:
        ret = opener.open(request)
        return ret

    except urllib2.HTTPError, err:
        if err.code == 404:
            print "Page not found!"
        elif err.code == 403:
            print "Access denied!"
        else:
            print "HTTP Error response! Error code", err.code
        return "Error"
    except urllib2.URLError, err:
        print "URL error:", err.reason
        return "Error"

# HTTP POST wrapper
def httpPost(url, data):
    try:
        retData = urllib2.urlopen(url, data)
        return retData.read()
    except urllib2.HTTPError, err:
        if err.code == 404:
            print "Page not found!"
        elif err.code == 403:
            print "Access denied!"
        else:
            print "HTTP Error! Error code", err.code
        return "Error"
    except urllib2.URLError, err:
        print "URL error:", err.reason
        return "Error"

# Wrapper for HTTP get
def httpGet(url):
    try:
        retData = urllib2.urlopen(url)
        return retData.read()

    except urllib2.HTTPError, err:
        if err.code == 404:
            print "Page not found!"
        elif err.code == 403:
            print "Access denied!"
        else:
            print "HTTP Error! Error code", err.code
        return "Error"
    except urllib2.URLError, err:
        print "URL error:", err.reason
        return "Error"

# object model client
class objmodelClient:
	def __init__(self, baseUrl):
		self.baseUrl = baseUrl
	# Create appProfile
	def createAppProfile(self, obj):
	    postUrl = self.baseUrl + '/api/appProfiles/' + obj.tenantName + ":" + obj.networkName + ":" + obj.appProfileName  + '/'

	    jdata = json.dumps({ 
			"appProfileName": obj.appProfileName, 
			"endpointGroups": obj.endpointGroups, 
			"networkName": obj.networkName, 
			"tenantName": obj.tenantName, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("AppProfile create failure")

	# Delete appProfile
	def deleteAppProfile(self, tenantName, networkName, appProfileName):
	    # Delete AppProfile
	    deleteUrl = self.baseUrl + '/api/appProfiles/' + tenantName + ":" + networkName + ":" + appProfileName  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("AppProfile create failure")

	# List all appProfile objects
	def listAppProfile(self):
	    # Get a list of appProfile objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/appProfiles/')
	    if retData == "Error":
	        errorExit("list AppProfile failed")

	    return json.loads(retData)
	# Create endpointGroup
	def createEndpointGroup(self, obj):
	    postUrl = self.baseUrl + '/api/endpointGroups/' + obj.tenantName + ":" + obj.networkName + ":" + obj.groupName  + '/'

	    jdata = json.dumps({ 
			"endpointGroupId": obj.endpointGroupId, 
			"groupName": obj.groupName, 
			"networkName": obj.networkName, 
			"policies": obj.policies, 
			"tenantName": obj.tenantName, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("EndpointGroup create failure")

	# Delete endpointGroup
	def deleteEndpointGroup(self, tenantName, networkName, groupName):
	    # Delete EndpointGroup
	    deleteUrl = self.baseUrl + '/api/endpointGroups/' + tenantName + ":" + networkName + ":" + groupName  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("EndpointGroup create failure")

	# List all endpointGroup objects
	def listEndpointGroup(self):
	    # Get a list of endpointGroup objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/endpointGroups/')
	    if retData == "Error":
	        errorExit("list EndpointGroup failed")

	    return json.loads(retData)
	# Create global
	def createGlobal(self, obj):
	    postUrl = self.baseUrl + '/api/globals/' + obj.name  + '/'

	    jdata = json.dumps({ 
			"name": obj.name, 
			"network-infra-type": obj.network-infra-type, 
			"vlans": obj.vlans, 
			"vxlans": obj.vxlans, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("Global create failure")

	# Delete global
	def deleteGlobal(self, name):
	    # Delete Global
	    deleteUrl = self.baseUrl + '/api/globals/' + name  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("Global create failure")

	# List all global objects
	def listGlobal(self):
	    # Get a list of global objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/globals/')
	    if retData == "Error":
	        errorExit("list Global failed")

	    return json.loads(retData)
	# Create Bgp
	def createBgp(self, obj):
	    postUrl = self.baseUrl + '/api/Bgps/' + obj.hostname  + '/'

	    jdata = json.dumps({ 
			"as": obj.as, 
			"hostname": obj.hostname, 
			"neighbor": obj.neighbor, 
			"neighbor-as": obj.neighbor-as, 
			"routerip": obj.routerip, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("Bgp create failure")

	# Delete Bgp
	def deleteBgp(self, hostname):
	    # Delete Bgp
	    deleteUrl = self.baseUrl + '/api/Bgps/' + hostname  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("Bgp create failure")

	# List all Bgp objects
	def listBgp(self):
	    # Get a list of Bgp objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/Bgps/')
	    if retData == "Error":
	        errorExit("list Bgp failed")

	    return json.loads(retData)
	# Create network
	def createNetwork(self, obj):
	    postUrl = self.baseUrl + '/api/networks/' + obj.tenantName + ":" + obj.networkName  + '/'

	    jdata = json.dumps({ 
			"encap": obj.encap, 
			"gateway": obj.gateway, 
			"networkName": obj.networkName, 
			"nwType": obj.nwType, 
			"pktTag": obj.pktTag, 
			"subnet": obj.subnet, 
			"tenantName": obj.tenantName, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("Network create failure")

	# Delete network
	def deleteNetwork(self, tenantName, networkName):
	    # Delete Network
	    deleteUrl = self.baseUrl + '/api/networks/' + tenantName + ":" + networkName  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("Network create failure")

	# List all network objects
	def listNetwork(self):
	    # Get a list of network objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/networks/')
	    if retData == "Error":
	        errorExit("list Network failed")

	    return json.loads(retData)
	# Create policy
	def createPolicy(self, obj):
	    postUrl = self.baseUrl + '/api/policys/' + obj.tenantName + ":" + obj.policyName  + '/'

	    jdata = json.dumps({ 
			"policyName": obj.policyName, 
			"tenantName": obj.tenantName, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("Policy create failure")

	# Delete policy
	def deletePolicy(self, tenantName, policyName):
	    # Delete Policy
	    deleteUrl = self.baseUrl + '/api/policys/' + tenantName + ":" + policyName  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("Policy create failure")

	# List all policy objects
	def listPolicy(self):
	    # Get a list of policy objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/policys/')
	    if retData == "Error":
	        errorExit("list Policy failed")

	    return json.loads(retData)
	# Create rule
	def createRule(self, obj):
	    postUrl = self.baseUrl + '/api/rules/' + obj.tenantName + ":" + obj.policyName + ":" + obj.ruleId  + '/'

	    jdata = json.dumps({ 
			"action": obj.action, 
			"direction": obj.direction, 
			"fromEndpointGroup": obj.fromEndpointGroup, 
			"fromIpAddress": obj.fromIpAddress, 
			"fromNetwork": obj.fromNetwork, 
			"policyName": obj.policyName, 
			"port": obj.port, 
			"priority": obj.priority, 
			"protocol": obj.protocol, 
			"ruleId": obj.ruleId, 
			"tenantName": obj.tenantName, 
			"toEndpointGroup": obj.toEndpointGroup, 
			"toIpAddress": obj.toIpAddress, 
			"toNetwork": obj.toNetwork, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("Rule create failure")

	# Delete rule
	def deleteRule(self, tenantName, policyName, ruleId):
	    # Delete Rule
	    deleteUrl = self.baseUrl + '/api/rules/' + tenantName + ":" + policyName + ":" + ruleId  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("Rule create failure")

	# List all rule objects
	def listRule(self):
	    # Get a list of rule objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/rules/')
	    if retData == "Error":
	        errorExit("list Rule failed")

	    return json.loads(retData)
	# Create service
	def createService(self, obj):
	    postUrl = self.baseUrl + '/api/services/' + obj.tenantName + ":" + obj.appName + ":" + obj.serviceName  + '/'

	    jdata = json.dumps({ 
			"appName": obj.appName, 
			"command": obj.command, 
			"cpu": obj.cpu, 
			"endpointGroups": obj.endpointGroups, 
			"environment": obj.environment, 
			"imageName": obj.imageName, 
			"memory": obj.memory, 
			"networks": obj.networks, 
			"scale": obj.scale, 
			"serviceName": obj.serviceName, 
			"tenantName": obj.tenantName, 
			"volumeProfile": obj.volumeProfile, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("Service create failure")

	# Delete service
	def deleteService(self, tenantName, appName, serviceName):
	    # Delete Service
	    deleteUrl = self.baseUrl + '/api/services/' + tenantName + ":" + appName + ":" + serviceName  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("Service create failure")

	# List all service objects
	def listService(self):
	    # Get a list of service objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/services/')
	    if retData == "Error":
	        errorExit("list Service failed")

	    return json.loads(retData)
	# Create serviceInstance
	def createServiceInstance(self, obj):
	    postUrl = self.baseUrl + '/api/serviceInstances/' + obj.tenantName + ":" + obj.appName + ":" + obj.serviceName + ":" + obj.instanceId  + '/'

	    jdata = json.dumps({ 
			"appName": obj.appName, 
			"instanceId": obj.instanceId, 
			"serviceName": obj.serviceName, 
			"tenantName": obj.tenantName, 
			"volumes": obj.volumes, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("ServiceInstance create failure")

	# Delete serviceInstance
	def deleteServiceInstance(self, tenantName, appName, serviceName, instanceId):
	    # Delete ServiceInstance
	    deleteUrl = self.baseUrl + '/api/serviceInstances/' + tenantName + ":" + appName + ":" + serviceName + ":" + instanceId  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("ServiceInstance create failure")

	# List all serviceInstance objects
	def listServiceInstance(self):
	    # Get a list of serviceInstance objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/serviceInstances/')
	    if retData == "Error":
	        errorExit("list ServiceInstance failed")

	    return json.loads(retData)
	# Create serviceLB
	def createServiceLB(self, obj):
	    postUrl = self.baseUrl + '/api/serviceLBs/' + obj.serviceName + ":" + obj.tenantName  + '/'

	    jdata = json.dumps({ 
			"ipAddress": obj.ipAddress, 
			"network": obj.network, 
			"ports": obj.ports, 
			"selectors": obj.selectors, 
			"serviceName": obj.serviceName, 
			"tenantName": obj.tenantName, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("ServiceLB create failure")

	# Delete serviceLB
	def deleteServiceLB(self, serviceName, tenantName):
	    # Delete ServiceLB
	    deleteUrl = self.baseUrl + '/api/serviceLBs/' + serviceName + ":" + tenantName  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("ServiceLB create failure")

	# List all serviceLB objects
	def listServiceLB(self):
	    # Get a list of serviceLB objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/serviceLBs/')
	    if retData == "Error":
	        errorExit("list ServiceLB failed")

	    return json.loads(retData)
	# Create tenant
	def createTenant(self, obj):
	    postUrl = self.baseUrl + '/api/tenants/' + obj.tenantName  + '/'

	    jdata = json.dumps({ 
			"defaultNetwork": obj.defaultNetwork, 
			"tenantName": obj.tenantName, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("Tenant create failure")

	# Delete tenant
	def deleteTenant(self, tenantName):
	    # Delete Tenant
	    deleteUrl = self.baseUrl + '/api/tenants/' + tenantName  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("Tenant create failure")

	# List all tenant objects
	def listTenant(self):
	    # Get a list of tenant objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/tenants/')
	    if retData == "Error":
	        errorExit("list Tenant failed")

	    return json.loads(retData)
	# Create volume
	def createVolume(self, obj):
	    postUrl = self.baseUrl + '/api/volumes/' + obj.tenantName + ":" + obj.volumeName  + '/'

	    jdata = json.dumps({ 
			"datastoreType": obj.datastoreType, 
			"mountPoint": obj.mountPoint, 
			"poolName": obj.poolName, 
			"size": obj.size, 
			"tenantName": obj.tenantName, 
			"volumeName": obj.volumeName, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("Volume create failure")

	# Delete volume
	def deleteVolume(self, tenantName, volumeName):
	    # Delete Volume
	    deleteUrl = self.baseUrl + '/api/volumes/' + tenantName + ":" + volumeName  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("Volume create failure")

	# List all volume objects
	def listVolume(self):
	    # Get a list of volume objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/volumes/')
	    if retData == "Error":
	        errorExit("list Volume failed")

	    return json.loads(retData)
	# Create volumeProfile
	def createVolumeProfile(self, obj):
	    postUrl = self.baseUrl + '/api/volumeProfiles/' + obj.tenantName + ":" + obj.volumeProfileName  + '/'

	    jdata = json.dumps({ 
			"datastoreType": obj.datastoreType, 
			"mountPoint": obj.mountPoint, 
			"poolName": obj.poolName, 
			"size": obj.size, 
			"tenantName": obj.tenantName, 
			"volumeProfileName": obj.volumeProfileName, 
	    })

	    # Post the data
	    response = httpPost(postUrl, jdata)

	    if response == "Error":
	        errorExit("VolumeProfile create failure")

	# Delete volumeProfile
	def deleteVolumeProfile(self, tenantName, volumeProfileName):
	    # Delete VolumeProfile
	    deleteUrl = self.baseUrl + '/api/volumeProfiles/' + tenantName + ":" + volumeProfileName  + '/'
	    response = httpDelete(deleteUrl)

	    if response == "Error":
	        errorExit("VolumeProfile create failure")

	# List all volumeProfile objects
	def listVolumeProfile(self):
	    # Get a list of volumeProfile objects
	    retDate = urllib2.urlopen(self.baseUrl + '/api/volumeProfiles/')
	    if retData == "Error":
	        errorExit("list VolumeProfile failed")

	    return json.loads(retData)