175 lines
6.4 KiB
Python
175 lines
6.4 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
|
|
# Copyright 2011 Nicira Networks, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
# @author: Dan Wendlandt, Nicira Networks, Inc.
|
|
|
|
import httplib
|
|
import logging as LOG
|
|
import json
|
|
import socket
|
|
import sys
|
|
import urllib
|
|
|
|
from quantum.manager import QuantumManager
|
|
from optparse import OptionParser
|
|
from quantum.common.wsgi import Serializer
|
|
from quantum.cli import MiniClient
|
|
|
|
FORMAT = "json"
|
|
CONTENT_TYPE = "application/" + FORMAT
|
|
|
|
|
|
def delete_all_nets(client, tenant_id):
|
|
res = client.do_request(tenant_id, 'GET', "/networks." + FORMAT)
|
|
resdict = json.loads(res.read())
|
|
LOG.debug(resdict)
|
|
for n in resdict["networks"]:
|
|
nid = n["id"]
|
|
|
|
res = client.do_request(tenant_id, 'GET',
|
|
"/networks/%s/ports.%s" % (nid, FORMAT))
|
|
output = res.read()
|
|
if res.status != 200:
|
|
LOG.error("Failed to list ports: %s" % output)
|
|
continue
|
|
rd = json.loads(output)
|
|
LOG.debug(rd)
|
|
for port in rd["ports"]:
|
|
pid = port["id"]
|
|
|
|
data = {'port': {'attachment-id': ''}}
|
|
body = Serializer().serialize(data, CONTENT_TYPE)
|
|
res = client.do_request(tenant_id, 'DELETE',
|
|
"/networks/%s/ports/%s/attachment.%s" % \
|
|
(nid, pid, FORMAT), body=body)
|
|
output = res.read()
|
|
LOG.debug(output)
|
|
if res.status != 202:
|
|
LOG.error("Failed to unplug iface from port \"%s\": %s" % (vid,
|
|
pid, output))
|
|
continue
|
|
LOG.info("Unplugged interface from port:%s on network:%s" % (pid,
|
|
nid))
|
|
|
|
res = client.do_request(tenant_id, 'DELETE',
|
|
"/networks/%s/ports/%s.%s" % (nid, pid, FORMAT))
|
|
output = res.read()
|
|
if res.status != 202:
|
|
LOG.error("Failed to delete port: %s" % output)
|
|
continue
|
|
print "Deleted Virtual Port:%s " \
|
|
"on Virtual Network:%s" % (pid, nid)
|
|
|
|
res = client.do_request(tenant_id, 'DELETE',
|
|
"/networks/" + nid + "." + FORMAT)
|
|
status = res.status
|
|
if status != 202:
|
|
Log.error("Failed to delete network: %s" % nid)
|
|
output = res.read()
|
|
print output
|
|
else:
|
|
print "Deleted Virtual Network with ID:%s" % nid
|
|
|
|
|
|
def create_net_with_attachments(net_name, iface_ids):
|
|
data = {'network': {'net-name': '%s' % net_name}}
|
|
body = Serializer().serialize(data, CONTENT_TYPE)
|
|
res = client.do_request(tenant_id, 'POST',
|
|
"/networks." + FORMAT, body=body)
|
|
rd = json.loads(res.read())
|
|
LOG.debug(rd)
|
|
nid = rd["networks"]["network"]["id"]
|
|
print "Created a new Virtual Network %s with ID:%s" % (net_name, nid)
|
|
|
|
for iface_id in iface_ids:
|
|
res = client.do_request(tenant_id, 'POST',
|
|
"/networks/%s/ports.%s" % (nid, FORMAT))
|
|
output = res.read()
|
|
if res.status != 200:
|
|
LOG.error("Failed to create port: %s" % output)
|
|
continue
|
|
rd = json.loads(output)
|
|
new_port_id = rd["ports"]["port"]["id"]
|
|
print "Created Virtual Port:%s " \
|
|
"on Virtual Network:%s" % (new_port_id, nid)
|
|
data = {'port': {'attachment-id': '%s' % iface_id}}
|
|
body = Serializer().serialize(data, CONTENT_TYPE)
|
|
res = client.do_request(tenant_id, 'PUT',
|
|
"/networks/%s/ports/%s/attachment.%s" %\
|
|
(nid, new_port_id, FORMAT), body=body)
|
|
output = res.read()
|
|
LOG.debug(output)
|
|
if res.status != 202:
|
|
LOG.error("Failed to plug iface \"%s\" to port \"%s\": %s" % \
|
|
(iface_id, new_port_id, output))
|
|
continue
|
|
print "Plugged interface \"%s\" to port:%s on network:%s" % \
|
|
(iface_id, new_port_id, nid)
|
|
|
|
if __name__ == "__main__":
|
|
usagestr = "Usage: %prog [OPTIONS] <tenant-id> <config-string> [args]\n" \
|
|
"Example config-string: net1=instance-1,instance-2"\
|
|
":net2=instance-3,instance-4\n" \
|
|
"This string would create two networks: \n" \
|
|
"'net1' would have two ports, with iface-ids "\
|
|
"instance-1 and instance-2 attached\n" \
|
|
"'net2' would have two ports, with iface-ids"\
|
|
" instance-3 and instance-4 attached\n"
|
|
parser = OptionParser(usage=usagestr)
|
|
parser.add_option("-H", "--host", dest="host",
|
|
type="string", default="127.0.0.1", help="ip address of api host")
|
|
parser.add_option("-p", "--port", dest="port",
|
|
type="int", default=9696, help="api poort")
|
|
parser.add_option("-s", "--ssl", dest="ssl",
|
|
action="store_true", default=False, help="use ssl")
|
|
parser.add_option("-v", "--verbose", dest="verbose",
|
|
action="store_true", default=False, help="turn on verbose logging")
|
|
parser.add_option("-d", "--delete", dest="delete",
|
|
action="store_true", default=False, \
|
|
help="delete existing tenants networks")
|
|
|
|
options, args = parser.parse_args()
|
|
|
|
if options.verbose:
|
|
LOG.basicConfig(level=LOG.DEBUG)
|
|
else:
|
|
LOG.basicConfig(level=LOG.WARN)
|
|
|
|
if len(args) < 1:
|
|
parser.print_help()
|
|
help()
|
|
sys.exit(1)
|
|
|
|
nets = {}
|
|
tenant_id = args[0]
|
|
if len(args) > 1:
|
|
config_str = args[1]
|
|
for net_str in config_str.split(":"):
|
|
arr = net_str.split("=")
|
|
net_name = arr[0]
|
|
nets[net_name] = arr[1].split(",")
|
|
|
|
print "nets: %s" % str(nets)
|
|
|
|
client = MiniClient(options.host, options.port, options.ssl)
|
|
|
|
if options.delete:
|
|
delete_all_nets(client, tenant_id)
|
|
|
|
for net_name, iface_ids in nets.items():
|
|
create_net_with_attachments(net_name, iface_ids)
|
|
|
|
sys.exit(0)
|