aboutsummaryrefslogtreecommitdiff
path: root/vland-admin
diff options
context:
space:
mode:
authorSteve McIntyre <steve.mcintyre@linaro.org>2018-02-02 17:08:04 +0000
committerSteve McIntyre <steve.mcintyre@linaro.org>2018-02-02 17:08:04 +0000
commit53787d2f62557558a50f59c98c5db90b0eceae94 (patch)
tree938c07af05228cc22adb0d145745e9ea71f8e8ed /vland-admin
parenta6b485d55a1a9023950c71dbb06b44705d556e5c (diff)
downloadvland-53787d2f62557558a50f59c98c5db90b0eceae94.tar.gz
Renamed the admin program from "admin.py" to "vland-admin"
This will make packaging easier Change-Id: I840aa8e73911bcb4b02903a0202178005e2c58b3
Diffstat (limited to 'vland-admin')
-rwxr-xr-xvland-admin882
1 files changed, 882 insertions, 0 deletions
diff --git a/vland-admin b/vland-admin
new file mode 100755
index 0000000..96a7cfc
--- /dev/null
+++ b/vland-admin
@@ -0,0 +1,882 @@
+#! /usr/bin/python
+
+# Copyright 2014-2018 Linaro Limited
+#
+# 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 2 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, write to the Free Software
+# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
+# MA 02110-1301, USA.
+#
+# VLANd admin interface
+#
+
+import os, sys
+import argparse
+import datetime, time
+
+vlandpath = os.path.abspath(os.path.normpath(os.path.dirname(sys.argv[0])))
+sys.path.insert(0, vlandpath)
+
+from errors import InputError, SocketError, NotFoundError, Error
+from config.config import VlanConfig
+from ipc.ipc import VlanIpc
+
+prog = "vland-admin"
+version = "0.7"
+banner = "Linaro VLANd admin interface, version %s" % version
+exitcode = Error.OK
+TRUNK_ID_NONE = -1
+
+def is_positive(text):
+ if text in ('1', 'y', 'Y', 't', 'T', 'True', 'true'):
+ return True
+ elif text in ('0', 'n', 'N', 'f', 'F', 'False', 'false'):
+ return False
+ else:
+ raise InputError("Cannot parse \"%s\" as True or False" % text)
+
+def dump_switch(switch):
+ print "switch_id:%d name:%s" % (
+ int(switch['switch_id']),
+ switch['name'])
+
+def dump_port(port):
+ print "port_id:%d name:%s switch_id:%d locked:%s mode:%s base_vlan_id:%d current_vlan_id:%d number:%d trunk_id:%s lock_reason:%s" % (
+ int(port['port_id']),
+ port['name'],
+ int(port['switch_id']),
+ ("yes" if port['is_locked'] is True else "no"),
+ ("trunk" if port['is_trunk'] is True else "access"),
+ int(port['base_vlan_id']),
+ int(port['current_vlan_id']),
+ int(port['number']),
+ 'None' if (TRUNK_ID_NONE == port['trunk_id']) else port['trunk_id'],
+ port['lock_reason'])
+
+def dump_vlan(vlan):
+ print "vlan_id:%d name:%s tag:%d is_base_vlan:%s, creation_time:%s" % (
+ int(vlan['vlan_id']),
+ vlan['name'],
+ int(vlan['tag']),
+ ("yes" if vlan['is_base_vlan'] is True else "no"),
+ vlan['creation_time'])
+
+def dump_trunk(trunk):
+ print "trunk_id:%d creation_time:%s" % (
+ int(trunk['trunk_id']),
+ trunk['creation_time'])
+
+def call_vland(msgtype, msg):
+ ipc = VlanIpc()
+ ipc.client_connect('localhost', config.vland.port)
+ msg['client_name'] = 'vland-admin'
+ msg['type'] = msgtype
+ ipc.client_send(msg)
+ ret = ipc.client_recv_and_close()
+ if 'response' not in ret:
+ raise SocketError("Badly-formed response from VLANd server")
+ if ret['response'] == "ERROR":
+ raise InputError("VLANd server said \"%s\"" % ret['error'])
+ if ret['response'] == "NOTFOUND":
+ raise NotFoundError("VLANd server said \"%s\"" % ret['error'])
+ return ret['data']
+
+config = VlanConfig(filenames=('./vland.cfg',))
+
+parser = argparse.ArgumentParser()
+
+####################
+# System commands
+####################
+sp = parser.add_subparsers(title='Sub-commands',
+ help="Commands",
+ dest='subparser_name')
+p_status = sp.add_parser("status",
+ help="Describe current system status")
+p_status.set_defaults(which = "status")
+p_statistics = sp.add_parser("statistics",
+ help="Print some system statistics")
+p_statistics.set_defaults(which = "statistics")
+p_version = sp.add_parser("version",
+ help="Describe the version of this admin interface")
+p_version.set_defaults(which = "version")
+p_vland_version = sp.add_parser("vland_version",
+ help="Describe the version of the running VLANd")
+p_vland_version.set_defaults(which = "vland_version")
+p_probe_switches = sp.add_parser("probe_switches",
+ help="Probe all the configured switches")
+p_probe_switches.set_defaults(which = "probe_switches")
+p_auto_import = sp.add_parser("auto_import_switch",
+ help="Attempt to import a switch's configuration into the VLANd system")
+p_auto_import.add_argument('--name',
+ required = True,
+ help="Import the switch named SWITCH_NAME in vland.cfg")
+p_auto_import.set_defaults(which = "auto_import_switch")
+p_shutdown = sp.add_parser("shutdown",
+ help="Shut down a running VLANd")
+p_shutdown.set_defaults(which = "shutdown")
+
+####################
+# Switch commands
+####################
+p_list_all_switches = sp.add_parser("list_all_switches",
+ help="List all the existing switches in the system")
+p_list_all_switches.set_defaults(which = "list_all_switches")
+p_create_switch = sp.add_parser("create_switch",
+ help = "Add a new switch to the system")
+p_create_switch.set_defaults(which = "create_switch")
+p_create_switch.add_argument('--name',
+ required = True,
+ help="The name of the new switch, as described in vland.cfg")
+p_delete_switch = sp.add_parser("delete_switch",
+ help = "Remove an existing switch from the system")
+p_delete_switch.set_defaults(which = "delete_switch")
+p_delete_switch.add_argument('--switch_id',
+ required = True,
+ help = "The ID of the switch to remove")
+p_show_switch = sp.add_parser("show_switch",
+ help = "Show the details of an existing switch in the system")
+p_show_switch.set_defaults(which = "show_switch")
+p_show_switch.add_argument('--switch_id',
+ required = True,
+ help = "The ID of the switch to show")
+p_lookup_switch_by_name = sp.add_parser("lookup_switch_by_name",
+ help = "Lookup a switch ID by name")
+p_lookup_switch_by_name.set_defaults(which = "lookup_switch_by_name")
+p_lookup_switch_by_name.add_argument('--name',
+ required = True,
+ help = "The switch name to search for")
+
+####################
+# Ports commands
+####################
+p_list_all_ports = sp.add_parser("list_all_ports",
+ help="List all the existing ports in the system")
+p_list_all_ports.set_defaults(which = "list_all_ports")
+p_create_port = sp.add_parser("create_port",
+ help = "Add a new port to the system")
+p_create_port.set_defaults(which = "create_port")
+p_create_port.add_argument('--switch_id',
+ required = True,
+ help="The ID of the switch containing the new port")
+p_create_port.add_argument('--name',
+ required = True,
+ help="The name of the new port")
+p_create_port.add_argument('--number',
+ required = True,
+ help="The human-friendly number for the new port")
+p_delete_port = sp.add_parser("delete_port",
+ help = "Remove an existing port from the system")
+p_delete_port.set_defaults(which = "delete_port")
+p_delete_port.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to remove")
+p_show_port = sp.add_parser("show_port",
+ help = "Show the details of an existing port in the system")
+p_show_port.set_defaults(which = "show_port")
+p_show_port.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to show")
+p_lookup_port_by_switch_and_name = sp.add_parser("lookup_port_by_switch_and_name",
+ help = "Lookup a port ID by switch ID and port name")
+p_lookup_port_by_switch_and_name.set_defaults(which = "lookup_port_by_switch_and_name")
+p_lookup_port_by_switch_and_name.add_argument('--switch_id',
+ required = True,
+ help = "The switch ID to search for")
+p_lookup_port_by_switch_and_name.add_argument('--name',
+ required = True,
+ help = "The port name to search for")
+p_lookup_port_by_switch_and_number = sp.add_parser("lookup_port_by_switch_and_number",
+ help = "Lookup a port ID by switch ID and port number")
+p_lookup_port_by_switch_and_number.set_defaults(which = "lookup_port_by_switch_and_number")
+p_lookup_port_by_switch_and_number.add_argument('--switch_id',
+ required = True,
+ help = "The switch ID to search for")
+p_lookup_port_by_switch_and_number.add_argument('--number',
+ required = True,
+ help = "The port number to search for")
+p_lookup_ports_by_switch = sp.add_parser("lookup_ports_by_switch",
+ help = "Lookup port ID(s) by switch ID")
+p_lookup_ports_by_switch.set_defaults(which = "lookup_ports_by_switch")
+p_lookup_ports_by_switch.add_argument('--switch_id',
+ required = True,
+ help = "The switch ID to search for")
+p_lookup_ports_by_current_vlan = sp.add_parser("lookup_ports_by_current_vlan",
+ help = "Lookup port ID(s) by current VLAN ID")
+p_lookup_ports_by_current_vlan.set_defaults(which = "lookup_ports_by_current_vlan")
+p_lookup_ports_by_current_vlan.add_argument('--vlan_id',
+ required = True,
+ help = "The VLAN ID to search for")
+p_lookup_ports_by_base_vlan = sp.add_parser("lookup_ports_by_base_vlan",
+ help = "Lookup port ID(s) by base vlan ID")
+p_lookup_ports_by_base_vlan.set_defaults(which = "lookup_ports_by_base_vlan")
+p_lookup_ports_by_base_vlan.add_argument('--vlan_id',
+ required = True,
+ help = "The VLAN ID to search for")
+p_lookup_ports_by_trunk = sp.add_parser("lookup_ports_by_trunk",
+ help = "Lookup port ID(s) by trunk ID")
+p_lookup_ports_by_trunk.set_defaults(which = "lookup_ports_by_trunk")
+p_lookup_ports_by_trunk.add_argument('--trunk_id',
+ required = True,
+ help = "The trunk ID to search for")
+p_set_port_mode = sp.add_parser("set_port_mode",
+ help = "Set the mode of a port to 'trunk' or 'access'")
+p_set_port_mode.set_defaults(which = "set_port_mode")
+p_set_port_mode.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to modify")
+p_set_port_mode.add_argument('--mode',
+ required = True,
+ help = "The mode to select ('trunk' or 'access')")
+p_get_port_mode = sp.add_parser("get_port_mode",
+ help = "Get the mode of a port ('trunk' or 'access')")
+p_get_port_mode.set_defaults(which = "get_port_mode")
+p_get_port_mode.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to query")
+p_lock_port = sp.add_parser("lock_port",
+ help = "Lock the settings on a port")
+p_lock_port.set_defaults(which = "lock_port")
+p_lock_port.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to lock")
+p_lock_port.add_argument('--reason',
+ required = False,
+ help = "(Optional) The reason for locking the port")
+p_unlock_port = sp.add_parser("unlock_port",
+ help = "Unlock the settings on a port")
+p_unlock_port.set_defaults(which = "unlock_port")
+p_unlock_port.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to unlock")
+p_set_port_current_vlan = sp.add_parser("set_port_current_vlan",
+ help = "Set the current VLAN assignment for a port")
+p_set_port_current_vlan.set_defaults(which = "set_port_current_vlan")
+p_set_port_current_vlan.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to modify")
+p_set_port_current_vlan.add_argument('--vlan_id',
+ required = True,
+ help = "The VLAN ID to be used")
+p_get_port_current_vlan = sp.add_parser("get_port_current_vlan",
+ help = "Get the current VLAN assignment for a port")
+p_get_port_current_vlan.set_defaults(which = "get_port_current_vlan")
+p_get_port_current_vlan.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to query")
+p_set_port_base_vlan = sp.add_parser("set_port_base_vlan",
+ help = "Set the base VLAN assignment for a port")
+p_set_port_base_vlan.set_defaults(which = "set_port_base_vlan")
+p_set_port_base_vlan.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to modify")
+p_set_port_base_vlan.add_argument('--vlan_id',
+ required = True,
+ help = "The VLAN ID to be used")
+p_get_port_base_vlan = sp.add_parser("get_port_base_vlan",
+ help = "Get the base VLAN assignment for a port")
+p_get_port_base_vlan.set_defaults(which = "get_port_base_vlan")
+p_get_port_base_vlan.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to query")
+p_restore_port_to_base_vlan = sp.add_parser("restore_port_to_base_vlan",
+ help = "Reset a port back to its base VLAN")
+p_restore_port_to_base_vlan.set_defaults(which = "restore_port_to_base_vlan")
+p_restore_port_to_base_vlan.add_argument('--port_id',
+ required = True,
+ help = "The ID of the port to modify")
+
+####################
+# VLAN commands
+####################
+p_list_all_vlans = sp.add_parser("list_all_vlans",
+ help="List all the existing VLANs in the system")
+p_list_all_vlans.set_defaults(which = "list_all_vlans")
+p_create_vlan = sp.add_parser("create_vlan",
+ help = "Add a new VLAN to the system")
+p_create_vlan.set_defaults(which = "create_vlan")
+p_create_vlan.add_argument('--name',
+ required = True,
+ help="The name of the new VLAN")
+p_create_vlan.add_argument('--tag',
+ required = True,
+ help="The tag for the new VLAN (-1 to automatically select)")
+p_create_vlan.add_argument('--is_base_vlan',
+ required = True,
+ help="Is the new VLAN a base VLAN (true/false)")
+p_delete_vlan = sp.add_parser("delete_vlan",
+ help = "Remove an existing VLAN from the system")
+p_delete_vlan.set_defaults(which = "delete_vlan")
+p_delete_vlan.add_argument('--vlan_id',
+ required = True,
+ help = "The ID of the VLAN to remove")
+p_show_vlan = sp.add_parser("show_vlan",
+ help = "Show the details of an existing VLAN in the system")
+p_show_vlan.set_defaults(which = "show_vlan")
+p_show_vlan.add_argument('--vlan_id',
+ required = True,
+ help = "The ID of the VLAN to show")
+p_lookup_vlan_by_tag = sp.add_parser("lookup_vlan_by_tag",
+ help = "Find the VLAN ID of an existing VLAN in the system")
+p_lookup_vlan_by_tag.set_defaults(which = "lookup_vlan_by_tag")
+p_lookup_vlan_by_tag.add_argument('--tag',
+ required = True,
+ help = "The VLAN tag to search for")
+p_show_vlan_tag = sp.add_parser("show_vlan_tag",
+ help = "Print the VLAN tag of an existing VLAN in the system")
+p_show_vlan_tag.set_defaults(which = "show_vlan_tag")
+p_show_vlan_tag.add_argument('--vlan_id',
+ required = True,
+ help = "The VLAN ID to search for")
+
+####################
+# Trunk commands
+####################
+p_list_all_trunks = sp.add_parser("list_all_trunks",
+ help="List all the existing trunks in the system")
+p_list_all_trunks.set_defaults(which = "list_all_trunks")
+p_create_trunk = sp.add_parser("create_trunk",
+ help = "Add a new trunk to the system, linking two ports")
+p_create_trunk.set_defaults(which = "create_trunk")
+p_create_trunk.add_argument('--port_id1',
+ required = True,
+ help="The ID of the first port to be used")
+p_create_trunk.add_argument('--port_id2',
+ required = True,
+ help="The ID of the second port to be used")
+p_delete_trunk = sp.add_parser("delete_trunk",
+ help = "Remove an existing trunk from the system")
+p_delete_trunk.set_defaults(which = "delete_trunk")
+p_delete_trunk.add_argument('--trunk_id',
+ required = True,
+ help = "The ID of the trunk to remove")
+p_show_trunk = sp.add_parser("show_trunk",
+ help = "Show the details of an existing trunk in the system")
+p_show_trunk.set_defaults(which = "show_trunk")
+p_show_trunk.add_argument('--trunk_id',
+ required = True,
+ help = "The ID of the trunk to show")
+
+args = parser.parse_args()
+
+# Now work out what to do
+if args.which == 'status':
+ try:
+ print 'Config:'
+ print ' knows about %d switch(es)' % len(config.switches)
+ default_vlan_id = call_vland('db_query',
+ {'command':'db.get_vlan_id_by_tag',
+ 'data':
+ {'tag': config.vland.default_vlan_tag}})
+ print 'The default vlan tag (%d) is vlan ID %d' % (config.vland.default_vlan_tag, default_vlan_id)
+ stat = call_vland('daemon_query', {'command':'daemon.status', 'data': None})
+ print 'VLANd is running %s' % stat['running']
+ lastmod = datetime.datetime.strptime(stat['last_modified'], '%Y-%m-%dT%H:%M:%S.%f')
+ print 'DB Last modified %s' % lastmod.strftime('%Y-%m-%d %H:%M:%S %Z')
+ print 'DB via VLANd:'
+ switches = call_vland('db_query', {'command':'db.all_switches', 'data':None})
+ print ' knows about %d switch(es)' % len(switches)
+ ports = call_vland('db_query', {'command':'db.all_ports', 'data':None})
+ print ' knows about %d port(s)' % len(ports)
+ vlans = call_vland('db_query', {'command':'db.all_vlans', 'data':None})
+ print ' DB knows about %d vlan(s)' % len(vlans)
+ except (InputError, NotFoundError):
+ exitcode = Error.FAILED
+elif args.which == 'shutdown':
+ try:
+ print 'Asking VLANd to shutdown'
+ shutdown = call_vland('daemon_query',
+ {'command':'daemon.shutdown',
+ 'data': None})
+ for field in shutdown:
+ print '%s: %s' % (field, shutdown[field])
+ except (InputError, NotFoundError):
+ exitcode = Error.FAILED
+elif args.which == 'vland_version':
+ try:
+ ver = call_vland('daemon_query', {'command':'daemon.version', 'data': None})
+ print 'VLANd version %s' % ver['version']
+ except (InputError, NotFoundError):
+ exitcode = Error.FAILED
+elif args.which == 'statistics':
+ try:
+ stats = call_vland('daemon_query', {'command':'daemon.statistics', 'data': None})
+ print 'VLANd uptime: %d seconds' % stats['uptime']
+ except (InputError, NotFoundError):
+ exitcode = Error.FAILED
+elif args.which == 'version':
+ print 'VLANd admin interface version %s' % version
+elif args.which == 'auto_import_switch':
+ print 'Attempting to import switch %s' % args.name
+ if args.name not in config.switches:
+ raise InputError("Can't find switch %s in config" % args.name)
+ try:
+ imp = call_vland('vlan_update',
+ {'command':'api.auto_import_switch',
+ 'data':
+ {'switch': args.name}})
+ print 'VLANd imported switch %s successfully: new switch_id %d, %d new ports, %d new VLANs' % (args.name, imp['switch_id'], imp['num_ports_added'], imp['num_vlans_added'])
+ except (InputError, NotFoundError):
+ print 'Import failed - see log for details'
+ exitcode = Error.FAILED
+elif args.which == 'probe_switches':
+ print 'Asking VLANd to probe all the configured switches'
+ try:
+ probe = call_vland('daemon_query',
+ {'command':'daemon.probe_switches',
+ 'data': None})
+ for field in probe:
+ print '%s: %s' % (field, probe[field])
+ except (InputError, NotFoundError):
+ print 'Probe failed - see log for details'
+ exitcode = Error.FAILED
+elif args.which == 'list_all_switches':
+ try:
+ result = call_vland('db_query', {'command':'db.all_switches', 'data':None})
+ for line in result:
+ dump_switch(line)
+ except (InputError, NotFoundError):
+ exitcode = Error.FAILED
+elif args.which == 'list_all_ports':
+ try:
+ result = call_vland('db_query', {'command':'db.all_ports', 'data':None})
+ for line in result:
+ dump_port(line)
+ except (InputError, NotFoundError):
+ exitcode = Error.FAILED
+elif args.which == 'list_all_vlans':
+ try:
+ result = call_vland('db_query', {'command':'db.all_vlans', 'data':None})
+ for line in result:
+ dump_vlan(line)
+ except (InputError, NotFoundError):
+ exitcode = Error.FAILED
+elif args.which == 'list_all_trunks':
+ try:
+ result = call_vland('db_query', {'command':'db.all_trunks', 'data':None})
+ for line in result:
+ dump_trunk(line)
+ except (InputError, NotFoundError):
+ exitcode = Error.FAILED
+elif args.which == 'create_switch':
+ try:
+ switch_id = call_vland('db_update',
+ {'command':'db.create_switch',
+ 'data':
+ {'name':args.name}})
+ print 'Created switch_id %d' % switch_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'create_port':
+ try:
+ port_id = call_vland('db_update',
+ {'command':'db.create_port',
+ 'data':
+ {'switch_id': args.switch_id,
+ 'name': args.name,
+ 'number': args.number}})
+ print 'Created port_id %d' % port_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'create_vlan':
+ try:
+ (vlan_id, vlan_tag) = call_vland('vlan_update',
+ {'command':'api.create_vlan',
+ 'data':
+ {'name': args.name,
+ 'tag': args.tag,
+ 'is_base_vlan': is_positive(args.is_base_vlan)}})
+ print 'Created VLAN tag %d as vlan_id %d' % (vlan_tag, vlan_id)
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'create_trunk':
+ try:
+ trunk_id = call_vland('db_update',
+ {'command':'db.create_trunk',
+ 'data':
+ {'port_id1': args.port_id1,
+ 'port_id2': args.port_id2}})
+ print 'Created trunk_id %d' % trunk_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'delete_switch':
+ try:
+ switch_id = call_vland('db_update',
+ {'command':'db.delete_switch',
+ 'data': {'switch_id': args.switch_id}})
+ print 'Deleted switch_id %s' % switch_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'delete_port':
+ try:
+ port_id = call_vland('db_update',
+ {'command':'db.delete_port',
+ 'data': {'port_id': args.port_id}})
+ print 'Deleted port_id %s' % port_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'delete_vlan':
+ try:
+ vlan_id = call_vland('vlan_update',
+ {'command':'api.delete_vlan',
+ 'data': {'vlan_id': args.vlan_id}})
+ print 'Deleted vlan_id %d' % vlan_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'delete_trunk':
+ try:
+ trunk_id = call_vland('db_update',
+ {'command':'db.delete_trunk',
+ 'data': {'trunk_id': args.trunk_id}})
+ print 'Deleted trunk_id %s' % trunk_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'lookup_switch_by_name':
+ try:
+ switch_id = call_vland('db_query',
+ {'command':'db.get_switch_id_by_name',
+ 'data':{'name':args.name}})
+ if switch_id is not None:
+ print '%d' % switch_id
+ else:
+ print 'No switch found for name %s' % args.name
+ exitcode = Error.NOTFOUND
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'show_switch':
+ try:
+ this_switch = call_vland('db_query',
+ {'command':'db.get_switch_by_id',
+ 'data':
+ {'switch_id': args.switch_id}})
+ if this_switch is not None:
+ dump_switch(this_switch)
+ else:
+ print 'No switch found for switch_id %s' % args.switch_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'show_port':
+ try:
+ this_port = call_vland('db_query',
+ {'command':'db.get_port_by_id',
+ 'data':
+ {'port_id': args.port_id}})
+ if this_port is not None:
+ dump_port(this_port)
+ else:
+ print 'No port found for port_id %s' % args.port_id
+ exitcode = Error.NOTFOUND
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'lookup_port_by_switch_and_name':
+ try:
+ p = call_vland('db_query',
+ {'command':'db.get_port_by_switch_and_name',
+ 'data':
+ {'switch_id': args.switch_id,
+ 'name': args.name}})
+ if p is not None:
+ print p
+ else:
+ print 'No port found for switch_id %s, name %s' % (args.switch_id, args.name)
+ exitcode = Error.NOTFOUND
+ except InputError as inst:
+ print 'Failed: %s' % inst
+elif args.which == 'lookup_port_by_switch_and_number':
+ try:
+ p = call_vland('db_query',
+ {'command':'db.get_port_by_switch_and_number',
+ 'data':
+ {'switch_id': args.switch_id,
+ 'number': args.number}})
+ if p is not None:
+ print p
+ else:
+ print 'No port found for switch_id %s, port number %s' % (args.switch_id, args.number)
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'lookup_ports_by_switch':
+ try:
+ p = call_vland('db_query',
+ {'command':'db.get_ports_by_switch',
+ 'data':
+ {'switch_id': args.switch_id}})
+ if p is not None:
+ for port_id in p:
+ print port_id
+ else:
+ print 'No ports found for switch_id %s' % args.switch_id
+ exitcode = Error.NOTFOUND
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'lookup_ports_by_current_vlan':
+ try:
+ p = call_vland('db_query',
+ {'command':'db.get_ports_by_current_vlan',
+ 'data':
+ {'vlan_id': args.vlan_id}})
+ if p is not None:
+ for port_id in p:
+ print port_id
+ else:
+ print 'No ports found for current vlan_id %s' % args.vlan_id
+ exitcode = Error.NOTFOUND
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'lookup_ports_by_base_vlan':
+ try:
+ p = call_vland('db_query',
+ {'command':'db.get_ports_by_base_vlan',
+ 'data':
+ {'vlan_id': args.vlan_id}})
+ if p is not None:
+ for port_id in p:
+ print port_id
+ else:
+ print 'No ports found for base vlan_id %s' % args.vlan_id
+ exitcode = Error.NOTFOUND
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'lookup_ports_by_trunk':
+ try:
+ p = call_vland('db_query',
+ {'command':'db.get_ports_by_trunk',
+ 'data':
+ {'trunk_id': args.trunk_id}})
+ if p is not None:
+ for port_id in p:
+ print port_id
+ else:
+ print 'No ports found for trunk_id %s' % args.trunk_id
+ exitcode = Error.NOTFOUND
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'set_port_mode':
+ try:
+ port_id = call_vland('vlan_update',
+ {'command':'api.set_port_mode',
+ 'data':
+ {'port_id': args.port_id,
+ 'mode': args.mode}})
+ print "Updated mode for port_id %d" % port_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'lock_port':
+ try:
+ port_id = call_vland('db_update',
+ {'command':'db.set_port_is_locked',
+ 'data':
+ {'port_id': args.port_id,
+ 'is_locked': True,
+ 'lock_reason': args.reason}})
+ print "Locked port_id %d" % port_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'unlock_port':
+ try:
+ port_id = call_vland('db_update',
+ {'command':'db.set_port_is_locked',
+ 'data':
+ {'port_id': args.port_id,
+ 'is_locked': False,
+ 'lock_reason': ""}})
+ print "Unlocked port_id %d" % port_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'set_port_current_vlan':
+ try:
+ port_id = call_vland('vlan_update',
+ {'command':'api.set_current_vlan',
+ 'data':
+ {'port_id': args.port_id,
+ 'vlan_id': args.vlan_id}})
+ print "Set current VLAN on port_id %d" % port_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'get_port_current_vlan':
+ try:
+ vlan_id = call_vland('db_query',
+ {'command':'db.get_current_vlan_id_by_port',
+ 'data':
+ {'port_id': args.port_id}})
+ if vlan_id is not None:
+ print vlan_id
+ else:
+ print "No current_vlan_id found for port_id %s" % args.port_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'set_port_base_vlan':
+ try:
+ port_id = call_vland('db_update',
+ {'command':'db.set_base_vlan',
+ 'data':
+ {'port_id': args.port_id,
+ 'base_vlan_id': args.vlan_id}})
+ print "Set base VLAN on port_id %d" % port_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'get_port_base_vlan':
+ try:
+ vlan_id = call_vland('db_query',
+ {'command':'db.get_base_vlan_id_by_port',
+ 'data':
+ {'port_id': args.port_id}})
+ if vlan_id is not None:
+ print vlan_id
+ else:
+ print "No base_vlan_id found for port_id %d" % port_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'restore_port_to_base_vlan':
+ try:
+ port_id = call_vland('vlan_update',
+ {'command': 'api.restore_base_vlan',
+ 'data':
+ {'port_id': args.port_id}})
+ print "Restored port_id %d back to base VLAN" % port_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'show_vlan':
+ try:
+ v = call_vland('db_query',
+ {'command':'db.get_vlan_by_id',
+ 'data':
+ {'vlan_id': args.vlan_id}})
+ if v is not None:
+ dump_vlan(v)
+ else:
+ print 'No VLAN found for vlan_id %s' % args.vlan_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+elif args.which == 'lookup_vlan_by_tag':
+ try:
+ vlan_id = call_vland('db_query',
+ {'command':'db.get_vlan_id_by_tag',
+ 'data':
+ {'tag': args.tag}})
+ if vlan_id is not None:
+ print vlan_id
+ else:
+ print 'No VLAN found for vlan tag %s' % args.tag
+ exitcode = Error.NOTFOUND
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'show_vlan_tag':
+ try:
+ vlan_tag = call_vland('db_query',
+ {'command':'db.get_vlan_tag_by_id',
+ 'data':
+ {'vlan_id': args.vlan_id}})
+ if vlan_tag is not None:
+ print vlan_tag
+ else:
+ print 'No VLAN found for vlan id %s' % args.vlan_id
+ exitcode = Error.NOTFOUND
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+elif args.which == 'show_trunk':
+ try:
+ this_trunk = call_vland('db_query',
+ {'command':'db.get_trunk_by_id',
+ 'data':
+ {'trunk_id': args.trunk_id}})
+ if this_trunk is not None:
+ dump_trunk(this_trunk)
+ else:
+ print 'No port found for port_id %s' % args.trunk_id
+ except InputError as inst:
+ print 'Failed: %s' % inst
+ print 'Failed: %s' % inst
+ exitcode = Error.FAILED
+ except NotFoundError as inst:
+ print 'Failed: %s' % inst
+ exitcode = Error.NOTFOUND
+else:
+ print 'No recognised command given. Try -h for help'
+
+sys.exit(exitcode)