342 lines
12 KiB
Python
342 lines
12 KiB
Python
|
# Copyright 2013 OpenStack Foundation.
|
||
|
# All Rights Reserved
|
||
|
#
|
||
|
# 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.
|
||
|
#
|
||
|
|
||
|
from neutronclient._i18n import _
|
||
|
from neutronclient.neutron import v2_0 as neutronV20
|
||
|
from neutronclient.neutron.v2_0 import network
|
||
|
from neutronclient.neutron.v2_0 import router
|
||
|
|
||
|
|
||
|
PERFECT_TIME_FORMAT = "%Y-%m-%dT%H:%M:%S.%f"
|
||
|
|
||
|
|
||
|
class AddNetworkToDhcpAgent(neutronV20.NeutronCommand):
|
||
|
"""Add a network to a DHCP agent."""
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(AddNetworkToDhcpAgent, self).get_parser(prog_name)
|
||
|
parser.add_argument(
|
||
|
'dhcp_agent',
|
||
|
metavar='DHCP_AGENT',
|
||
|
help=_('ID of the DHCP agent.'))
|
||
|
parser.add_argument(
|
||
|
'network',
|
||
|
metavar='NETWORK',
|
||
|
help=_('Network to add.'))
|
||
|
return parser
|
||
|
|
||
|
def take_action(self, parsed_args):
|
||
|
neutron_client = self.get_client()
|
||
|
_net_id = neutronV20.find_resourceid_by_name_or_id(
|
||
|
neutron_client, 'network', parsed_args.network)
|
||
|
neutron_client.add_network_to_dhcp_agent(parsed_args.dhcp_agent,
|
||
|
{'network_id': _net_id})
|
||
|
print(_('Added network %s to DHCP agent') % parsed_args.network,
|
||
|
file=self.app.stdout)
|
||
|
|
||
|
|
||
|
class RemoveNetworkFromDhcpAgent(neutronV20.NeutronCommand):
|
||
|
"""Remove a network from a DHCP agent."""
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(RemoveNetworkFromDhcpAgent, self).get_parser(prog_name)
|
||
|
parser.add_argument(
|
||
|
'dhcp_agent',
|
||
|
metavar='DHCP_AGENT',
|
||
|
help=_('ID of the DHCP agent.'))
|
||
|
parser.add_argument(
|
||
|
'network',
|
||
|
metavar='NETWORK',
|
||
|
help=_('Network to remove.'))
|
||
|
return parser
|
||
|
|
||
|
def take_action(self, parsed_args):
|
||
|
neutron_client = self.get_client()
|
||
|
_net_id = neutronV20.find_resourceid_by_name_or_id(
|
||
|
neutron_client, 'network', parsed_args.network)
|
||
|
neutron_client.remove_network_from_dhcp_agent(
|
||
|
parsed_args.dhcp_agent, _net_id)
|
||
|
print(_('Removed network %s from DHCP agent') % parsed_args.network,
|
||
|
file=self.app.stdout)
|
||
|
|
||
|
|
||
|
class ListNetworksOnDhcpAgent(network.ListNetwork):
|
||
|
"""List the networks on a DHCP agent."""
|
||
|
|
||
|
unknown_parts_flag = False
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(ListNetworksOnDhcpAgent,
|
||
|
self).get_parser(prog_name)
|
||
|
parser.add_argument(
|
||
|
'dhcp_agent',
|
||
|
metavar='DHCP_AGENT',
|
||
|
help=_('ID of the DHCP agent.'))
|
||
|
return parser
|
||
|
|
||
|
def call_server(self, neutron_client, search_opts, parsed_args):
|
||
|
data = neutron_client.list_networks_on_dhcp_agent(
|
||
|
parsed_args.dhcp_agent, **search_opts)
|
||
|
return data
|
||
|
|
||
|
|
||
|
class ListDhcpAgentsHostingNetwork(neutronV20.ListCommand):
|
||
|
"""List DHCP agents hosting a network."""
|
||
|
|
||
|
resource = 'agent'
|
||
|
_formatters = {}
|
||
|
list_columns = ['id', 'host', 'admin_state_up', 'alive']
|
||
|
unknown_parts_flag = False
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(ListDhcpAgentsHostingNetwork,
|
||
|
self).get_parser(prog_name)
|
||
|
parser.add_argument(
|
||
|
'network',
|
||
|
metavar='NETWORK',
|
||
|
help=_('Network to query.'))
|
||
|
return parser
|
||
|
|
||
|
def extend_list(self, data, parsed_args):
|
||
|
for agent in data:
|
||
|
agent['alive'] = ":-)" if agent['alive'] else 'xxx'
|
||
|
|
||
|
def call_server(self, neutron_client, search_opts, parsed_args):
|
||
|
_id = neutronV20.find_resourceid_by_name_or_id(neutron_client,
|
||
|
'network',
|
||
|
parsed_args.network)
|
||
|
search_opts['network'] = _id
|
||
|
data = neutron_client.list_dhcp_agent_hosting_networks(**search_opts)
|
||
|
return data
|
||
|
|
||
|
|
||
|
class AddRouterToL3Agent(neutronV20.NeutronCommand):
|
||
|
"""Add a router to a L3 agent."""
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(AddRouterToL3Agent, self).get_parser(prog_name)
|
||
|
parser.add_argument(
|
||
|
'l3_agent',
|
||
|
metavar='L3_AGENT',
|
||
|
help=_('ID of the L3 agent.'))
|
||
|
parser.add_argument(
|
||
|
'router',
|
||
|
metavar='ROUTER',
|
||
|
help=_('Router to add.'))
|
||
|
return parser
|
||
|
|
||
|
def take_action(self, parsed_args):
|
||
|
neutron_client = self.get_client()
|
||
|
_id = neutronV20.find_resourceid_by_name_or_id(
|
||
|
neutron_client, 'router', parsed_args.router)
|
||
|
neutron_client.add_router_to_l3_agent(parsed_args.l3_agent,
|
||
|
{'router_id': _id})
|
||
|
print(_('Added router %s to L3 agent') % parsed_args.router,
|
||
|
file=self.app.stdout)
|
||
|
|
||
|
|
||
|
class RemoveRouterFromL3Agent(neutronV20.NeutronCommand):
|
||
|
"""Remove a router from a L3 agent."""
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(RemoveRouterFromL3Agent, self).get_parser(prog_name)
|
||
|
parser.add_argument(
|
||
|
'l3_agent',
|
||
|
metavar='L3_AGENT',
|
||
|
help=_('ID of the L3 agent.'))
|
||
|
parser.add_argument(
|
||
|
'router',
|
||
|
metavar='ROUTER',
|
||
|
help=_('Router to remove.'))
|
||
|
return parser
|
||
|
|
||
|
def take_action(self, parsed_args):
|
||
|
neutron_client = self.get_client()
|
||
|
_id = neutronV20.find_resourceid_by_name_or_id(
|
||
|
neutron_client, 'router', parsed_args.router)
|
||
|
neutron_client.remove_router_from_l3_agent(
|
||
|
parsed_args.l3_agent, _id)
|
||
|
print(_('Removed router %s from L3 agent') % parsed_args.router,
|
||
|
file=self.app.stdout)
|
||
|
|
||
|
|
||
|
class ListRoutersOnL3Agent(neutronV20.ListCommand):
|
||
|
"""List the routers on a L3 agent."""
|
||
|
|
||
|
_formatters = {'external_gateway_info':
|
||
|
router._format_external_gateway_info}
|
||
|
list_columns = ['id', 'name', 'external_gateway_info']
|
||
|
resource = 'router'
|
||
|
unknown_parts_flag = False
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(ListRoutersOnL3Agent,
|
||
|
self).get_parser(prog_name)
|
||
|
parser.add_argument(
|
||
|
'l3_agent',
|
||
|
metavar='L3_AGENT',
|
||
|
help=_('ID of the L3 agent to query.'))
|
||
|
return parser
|
||
|
|
||
|
def call_server(self, neutron_client, search_opts, parsed_args):
|
||
|
data = neutron_client.list_routers_on_l3_agent(
|
||
|
parsed_args.l3_agent, **search_opts)
|
||
|
return data
|
||
|
|
||
|
|
||
|
class ListL3AgentsHostingRouter(neutronV20.ListCommand):
|
||
|
"""List L3 agents hosting a router."""
|
||
|
|
||
|
resource = 'agent'
|
||
|
_formatters = {}
|
||
|
list_columns = ['id', 'host', 'admin_state_up', 'alive']
|
||
|
unknown_parts_flag = False
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(ListL3AgentsHostingRouter,
|
||
|
self).get_parser(prog_name)
|
||
|
parser.add_argument('router',
|
||
|
metavar='ROUTER',
|
||
|
help=_('Router to query.'))
|
||
|
return parser
|
||
|
|
||
|
def extend_list(self, data, parsed_args):
|
||
|
# Show the ha_state column only if the server responds with it,
|
||
|
# as some plugins do not support HA routers.
|
||
|
if any('ha_state' in agent for agent in data):
|
||
|
if 'ha_state' not in self.list_columns:
|
||
|
self.list_columns.append('ha_state')
|
||
|
for agent in data:
|
||
|
agent['alive'] = ":-)" if agent['alive'] else 'xxx'
|
||
|
|
||
|
def call_server(self, neutron_client, search_opts, parsed_args):
|
||
|
_id = neutronV20.find_resourceid_by_name_or_id(neutron_client,
|
||
|
'router',
|
||
|
parsed_args.router)
|
||
|
search_opts['router'] = _id
|
||
|
data = neutron_client.list_l3_agent_hosting_routers(**search_opts)
|
||
|
return data
|
||
|
|
||
|
|
||
|
class ListPoolsOnLbaasAgent(neutronV20.ListCommand):
|
||
|
"""List the pools on a loadbalancer agent."""
|
||
|
|
||
|
list_columns = ['id', 'name', 'lb_method', 'protocol',
|
||
|
'admin_state_up', 'status']
|
||
|
resource = 'pool'
|
||
|
unknown_parts_flag = False
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(ListPoolsOnLbaasAgent, self).get_parser(prog_name)
|
||
|
parser.add_argument(
|
||
|
'lbaas_agent',
|
||
|
metavar='LBAAS_AGENT',
|
||
|
help=_('ID of the loadbalancer agent to query.'))
|
||
|
return parser
|
||
|
|
||
|
def call_server(self, neutron_client, search_opts, parsed_args):
|
||
|
data = neutron_client.list_pools_on_lbaas_agent(
|
||
|
parsed_args.lbaas_agent, **search_opts)
|
||
|
return data
|
||
|
|
||
|
|
||
|
class GetLbaasAgentHostingPool(neutronV20.ListCommand):
|
||
|
"""Get loadbalancer agent hosting a pool.
|
||
|
|
||
|
Deriving from ListCommand though server will return only one agent
|
||
|
to keep common output format for all agent schedulers
|
||
|
"""
|
||
|
|
||
|
resource = 'agent'
|
||
|
list_columns = ['id', 'host', 'admin_state_up', 'alive']
|
||
|
unknown_parts_flag = False
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(GetLbaasAgentHostingPool,
|
||
|
self).get_parser(prog_name)
|
||
|
parser.add_argument('pool',
|
||
|
metavar='POOL',
|
||
|
help=_('Pool to query.'))
|
||
|
return parser
|
||
|
|
||
|
def extend_list(self, data, parsed_args):
|
||
|
for agent in data:
|
||
|
agent['alive'] = ":-)" if agent['alive'] else 'xxx'
|
||
|
|
||
|
def call_server(self, neutron_client, search_opts, parsed_args):
|
||
|
_id = neutronV20.find_resourceid_by_name_or_id(neutron_client,
|
||
|
'pool',
|
||
|
parsed_args.pool)
|
||
|
search_opts['pool'] = _id
|
||
|
agent = neutron_client.get_lbaas_agent_hosting_pool(**search_opts)
|
||
|
data = {'agents': [agent['agent']]}
|
||
|
return data
|
||
|
|
||
|
|
||
|
class ListLoadBalancersOnLbaasAgent(neutronV20.ListCommand):
|
||
|
"""List the loadbalancers on a loadbalancer v2 agent."""
|
||
|
|
||
|
list_columns = ['id', 'name', 'admin_state_up', 'provisioning_status']
|
||
|
resource = 'loadbalancer'
|
||
|
unknown_parts_flag = False
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(ListLoadBalancersOnLbaasAgent, self).get_parser(
|
||
|
prog_name)
|
||
|
parser.add_argument(
|
||
|
'lbaas_agent',
|
||
|
metavar='LBAAS_AGENT',
|
||
|
help=_('ID of the loadbalancer agent to query.'))
|
||
|
return parser
|
||
|
|
||
|
def call_server(self, neutron_client, search_opts, parsed_args):
|
||
|
data = neutron_client.list_loadbalancers_on_lbaas_agent(
|
||
|
parsed_args.lbaas_agent, **search_opts)
|
||
|
return data
|
||
|
|
||
|
|
||
|
class GetLbaasAgentHostingLoadBalancer(neutronV20.ListCommand):
|
||
|
"""Get lbaas v2 agent hosting a loadbalancer.
|
||
|
|
||
|
Deriving from ListCommand though server will return only one agent
|
||
|
to keep common output format for all agent schedulers
|
||
|
"""
|
||
|
|
||
|
resource = 'agent'
|
||
|
list_columns = ['id', 'host', 'admin_state_up', 'alive']
|
||
|
unknown_parts_flag = False
|
||
|
|
||
|
def get_parser(self, prog_name):
|
||
|
parser = super(GetLbaasAgentHostingLoadBalancer,
|
||
|
self).get_parser(prog_name)
|
||
|
parser.add_argument('loadbalancer',
|
||
|
metavar='LOADBALANCER',
|
||
|
help=_('LoadBalancer to query.'))
|
||
|
return parser
|
||
|
|
||
|
def extend_list(self, data, parsed_args):
|
||
|
for agent in data:
|
||
|
agent['alive'] = ":-)" if agent['alive'] else 'xxx'
|
||
|
|
||
|
def call_server(self, neutron_client, search_opts, parsed_args):
|
||
|
_id = neutronV20.find_resourceid_by_name_or_id(
|
||
|
neutron_client, 'loadbalancer', parsed_args.loadbalancer)
|
||
|
search_opts['loadbalancer'] = _id
|
||
|
agent = neutron_client.get_lbaas_agent_hosting_loadbalancer(
|
||
|
**search_opts)
|
||
|
data = {'agents': [agent['agent']]}
|
||
|
return data
|