192 lines
7.2 KiB
Python
192 lines
7.2 KiB
Python
# Copyright (c) 2017 Intel Corporation.
|
|
#
|
|
# 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 osc_lib.command import command
|
|
from osc_lib import utils
|
|
|
|
object_path = "/classifications"
|
|
resource = 'classification'
|
|
|
|
|
|
class CreateUDPClassification(command.ShowOne):
|
|
"""Create an UDP Classification."""
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(CreateUDPClassification, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'name', metavar='NAME',
|
|
help=('Name of the UDP Classification.'))
|
|
parser.add_argument(
|
|
'--description',
|
|
help=('Description for the UDP Classification.'))
|
|
parser.add_argument(
|
|
'--negated',
|
|
help=('Whether the complement of the UDP '
|
|
'Classification should be matched.'))
|
|
parser.add_argument(
|
|
'--shared',
|
|
help=('Whether the UDP Classification should be '
|
|
'shared with other projects.'))
|
|
parser.add_argument(
|
|
'--src-port-min',
|
|
help=('Source port UDP Classification minimum value.'))
|
|
parser.add_argument(
|
|
'--src-port-max',
|
|
help=('Source port UDP Classification maximum value.'))
|
|
parser.add_argument(
|
|
'--dst-port-min',
|
|
help=('Destination port UDP Classification minimum value.'))
|
|
parser.add_argument(
|
|
'--dst-port-max',
|
|
help=('Destination port UDP Classification maximum value.'))
|
|
parser.add_argument(
|
|
'--length-min',
|
|
help=('Minimum length of the UDP header and payload data.'))
|
|
parser.add_argument(
|
|
'--length-max',
|
|
help=('Maximum length of the UDP header and payload data.'))
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
client = self.app.client_manager.neutronclient
|
|
attrs = _get_attrs(self.app.client_manager,
|
|
parsed_args, is_create=True)
|
|
obj = client.create_ext(object_path, {resource: attrs})
|
|
columns = _get_columns(obj[resource])
|
|
data = utils.get_dict_properties(obj[resource], columns)
|
|
return columns, data
|
|
|
|
|
|
class DeleteUDPClassification(command.Command):
|
|
"""Delete a given UDP Classification."""
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(DeleteUDPClassification, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'udp_classification',
|
|
metavar="UDP_CLASSIFICATION",
|
|
help=('ID of the UDP Classification to delete.'))
|
|
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
id = parsed_args.udp_classification
|
|
client = self.app.client_manager.neutronclient
|
|
client.delete_ext(object_path + '/%s', id)
|
|
|
|
|
|
class ListUDPClassification(command.Lister):
|
|
"""List the UDP Classifications that belong to a given tenant."""
|
|
|
|
def take_action(self, parsed_args):
|
|
data = self.app.client_manager.neutronclient.list(
|
|
collection='classifications',
|
|
path=object_path, retrieve_all=True, c_type='udp')
|
|
headers = ('C_Type', 'ID', 'Name', 'Description', 'Negated', 'Shared')
|
|
columns = ('c_type', 'id', 'name', 'description', 'negated', 'shared')
|
|
|
|
return (headers, (utils.get_dict_properties(
|
|
s, columns) for s in data['classifications']))
|
|
|
|
|
|
class ShowUDPClassification(command.ShowOne):
|
|
"""Show information of a given UDP Classification."""
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(ShowUDPClassification, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'udp_classification',
|
|
metavar="UDP_CLASSIFICATION",
|
|
help=('ID of the UDP Classification to display.'))
|
|
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
client = self.app.client_manager.neutronclient
|
|
cl = client.show_ext(object_path + '/%s',
|
|
parsed_args.udp_classification, c_type='udp')
|
|
columns = _get_columns(cl[resource])
|
|
data = utils.get_dict_properties(cl[resource], columns)
|
|
return columns, data
|
|
|
|
|
|
class UpdateUDPClassification(command.ShowOne):
|
|
"""Update name and description of a given UDP Classification."""
|
|
|
|
def get_parser(self, prog_name):
|
|
parser = super(UpdateUDPClassification, self).get_parser(prog_name)
|
|
parser.add_argument(
|
|
'--name', default='',
|
|
metavar='NAME',
|
|
help=('Name of the UDP Classification.'))
|
|
parser.add_argument(
|
|
'--description', default='',
|
|
help=('Description of the UDP Classification.'))
|
|
parser.add_argument(
|
|
'udp_classification',
|
|
metavar="UDP_CLASSIFICATION",
|
|
help=('ID of the UDP Classification to update.'))
|
|
|
|
return parser
|
|
|
|
def take_action(self, parsed_args):
|
|
id = parsed_args.udp_classification
|
|
client = self.app.client_manager.neutronclient
|
|
attrs = _get_attrs(self.app.client_manager,
|
|
parsed_args, is_create=False)
|
|
cl = client.update_ext(object_path + '/%s', id, {resource: attrs})
|
|
columns = _get_columns(cl[resource])
|
|
data = utils.get_dict_properties(cl[resource], columns)
|
|
return columns, data
|
|
|
|
|
|
def _get_attrs(client_manager, parsed_args, is_create=False):
|
|
attrs = {}
|
|
definition = {}
|
|
|
|
if parsed_args.name is not None:
|
|
attrs['name'] = str(parsed_args.name)
|
|
if parsed_args.description is not None:
|
|
attrs['description'] = str(parsed_args.description)
|
|
if is_create:
|
|
attrs['c_type'] = 'udp'
|
|
if parsed_args.negated is not None:
|
|
attrs['negated'] = str(parsed_args.negated)
|
|
if parsed_args.shared is not None:
|
|
attrs['shared'] = str(parsed_args.shared)
|
|
if parsed_args.src_port_min is not None:
|
|
definition['src_port_min'] = parsed_args.src_port_min
|
|
if parsed_args.src_port_max is not None:
|
|
definition['src_port_max'] = parsed_args.src_port_max
|
|
if parsed_args.dst_port_min is not None:
|
|
definition['dst_port_min'] = parsed_args.dst_port_min
|
|
if parsed_args.dst_port_max is not None:
|
|
definition['dst_port_max'] = parsed_args.dst_port_max
|
|
if parsed_args.length_min is not None:
|
|
definition['length_min'] = parsed_args.length_min
|
|
if parsed_args.length_max is not None:
|
|
definition['length_max'] = parsed_args.length_max
|
|
attrs['definition'] = definition
|
|
|
|
return attrs
|
|
|
|
|
|
def _get_columns(resource):
|
|
columns = list(resource.keys())
|
|
if 'tenant_id' in columns:
|
|
columns.remove('tenant_id')
|
|
if 'project_id' in columns:
|
|
columns.append('project_id')
|
|
return tuple(sorted(columns))
|