python-group-based-policy-c.../gbpclient/gbp/v2_0/router.py

116 lines
3.7 KiB
Python

# 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.
#
"""
Router extension implementations
"""
import ast
from oslo_serialization import jsonutils
from cliff import hooks
from openstack.network.v2 import router as router_sdk
from openstack import resource
from openstackclient.network.v2 import router
from openstackclient.i18n import _
_get_attrs_router_new = router._get_attrs
def _get_attrs_router_extension(client_manager, parsed_args):
attrs = _get_attrs_router_new(client_manager, parsed_args)
if parsed_args.apic_distinguished_names:
attrs['apic:distinguished_names'
] = jsonutils.loads(parsed_args.apic_distinguished_names)
if parsed_args.apic_synchronization_state:
attrs['apic:synchronization_state'
] = parsed_args.apic_synchronization_state
if parsed_args.apic_external_provided_contracts:
attrs['apic:external_provided_contracts'
] = ast.literal_eval(
parsed_args.apic_external_provided_contracts)
if parsed_args.apic_external_consumed_contracts:
attrs['apic:external_consumed_contracts'
] = ast.literal_eval(
parsed_args.apic_external_consumed_contracts)
return attrs
router._get_attrs = _get_attrs_router_extension
router_sdk.Router.apic_distinguished_names = resource.Body(
'apic:distinguished_names')
router_sdk.Router.apic_synchronization_state = resource.Body(
'apic:synchronization_state')
router_sdk.Router.apic_external_provided_contracts = resource.Body(
'apic:external_provided_contracts')
router_sdk.Router.apic_external_consumed_contracts = resource.Body(
'apic:external_consumed_contracts')
class CreateAndSetRouterExtension(hooks.CommandHook):
def get_parser(self, parser):
parser.add_argument(
'--apic-distinguished-names',
metavar="<apic_distinguished_names>",
dest='apic_distinguished_names',
help=_("Apic distinguished names")
)
parser.add_argument(
'--apic-synchronization-state',
metavar="<apic_synchronization_state>",
dest='apic_synchronization_state',
help=_("Apic synchronization state")
)
parser.add_argument(
'--apic-external-provided-contracts',
metavar="<apic_external_provided_contracts>",
dest='apic_external_provided_contracts',
help=_("Apic external provided contracts")
)
parser.add_argument(
'--apic-external-consumed-contracts',
metavar="<apic_external_consumed_contracts>",
dest='apic_external_consumed_contracts',
help=_("Apic external consumed contracts")
)
return parser
def get_epilog(self):
return ''
def before(self, parsed_args):
return parsed_args
def after(self, parsed_args, return_code):
return return_code
class ShowRouterExtension(hooks.CommandHook):
def get_parser(self, parser):
return parser
def get_epilog(self):
return ''
def before(self, parsed_args):
return parsed_args
def after(self, parsed_args, return_code):
return return_code