131 lines
4.9 KiB
Python
131 lines
4.9 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.
|
|
|
|
import six
|
|
|
|
from heat.common import exception
|
|
from heat.common.i18n import _
|
|
from heat.engine import constraints
|
|
from heat.engine import properties
|
|
from heat.engine.resources.openstack.neutron import neutron
|
|
from heat.engine.resources.openstack.neutron import router
|
|
from heat.engine import support
|
|
|
|
|
|
class ExtraRoute(neutron.NeutronResource):
|
|
"""Resource for specifying extra routes for Neutron router.
|
|
|
|
Resource allows to specify nexthop IP and destination network for router.
|
|
"""
|
|
|
|
required_service_extension = 'extraroute'
|
|
|
|
support_status = support.SupportStatus(
|
|
status=support.UNSUPPORTED,
|
|
message=_('Use this resource at your own risk.'))
|
|
|
|
PROPERTIES = (
|
|
ROUTER_ID, DESTINATION, NEXTHOP,
|
|
) = (
|
|
'router_id', 'destination', 'nexthop',
|
|
)
|
|
|
|
properties_schema = {
|
|
ROUTER_ID: properties.Schema(
|
|
properties.Schema.STRING,
|
|
description=_('The router id.'),
|
|
required=True,
|
|
constraints=[
|
|
constraints.CustomConstraint('neutron.router')
|
|
]
|
|
),
|
|
DESTINATION: properties.Schema(
|
|
properties.Schema.STRING,
|
|
description=_('Network in CIDR notation.'),
|
|
required=True),
|
|
NEXTHOP: properties.Schema(
|
|
properties.Schema.STRING,
|
|
description=_('Nexthop IP address.'),
|
|
required=True)
|
|
}
|
|
|
|
def add_dependencies(self, deps):
|
|
super(ExtraRoute, self).add_dependencies(deps)
|
|
for resource in six.itervalues(self.stack):
|
|
# depend on any RouterInterface in this template with the same
|
|
# router_id as this router_id
|
|
if resource.has_interface('OS::Neutron::RouterInterface'):
|
|
try:
|
|
router_id = self.properties[self.ROUTER_ID]
|
|
dep_router_id = resource.properties.get(
|
|
router.RouterInterface.ROUTER)
|
|
except (ValueError, TypeError):
|
|
# Properties errors will be caught later in validation,
|
|
# where we can report them in their proper context.
|
|
continue
|
|
if dep_router_id == router_id:
|
|
deps += (self, resource)
|
|
# depend on any RouterGateway in this template with the same
|
|
# router_id as this router_id
|
|
elif resource.has_interface('OS::Neutron::RouterGateway'):
|
|
try:
|
|
router_id = self.properties[self.ROUTER_ID]
|
|
dep_router_id = resource.properties.get(
|
|
router.RouterGateway.ROUTER_ID)
|
|
except (ValueError, TypeError):
|
|
# Properties errors will be caught later in validation,
|
|
# where we can report them in their proper context.
|
|
continue
|
|
if dep_router_id == router_id:
|
|
deps += (self, resource)
|
|
|
|
def handle_create(self):
|
|
router_id = self.properties.get(self.ROUTER_ID)
|
|
routes = self.client().show_router(
|
|
router_id).get('router').get('routes')
|
|
if not routes:
|
|
routes = []
|
|
new_route = {'destination': self.properties[self.DESTINATION],
|
|
'nexthop': self.properties[self.NEXTHOP]}
|
|
if new_route in routes:
|
|
msg = _('Route duplicates an existing route.')
|
|
raise exception.Error(msg)
|
|
routes.append(new_route)
|
|
self.client().update_router(router_id,
|
|
{'router': {'routes': routes}})
|
|
new_route['router_id'] = router_id
|
|
self.resource_id_set(
|
|
'%(router_id)s:%(destination)s:%(nexthop)s' % new_route)
|
|
|
|
def handle_delete(self):
|
|
if not self.resource_id:
|
|
return
|
|
(router_id, destination, nexthop) = self.resource_id.split(':')
|
|
with self.client_plugin().ignore_not_found:
|
|
routes = self.client().show_router(
|
|
router_id).get('router').get('routes', [])
|
|
try:
|
|
routes.remove({'destination': destination,
|
|
'nexthop': nexthop})
|
|
except ValueError:
|
|
return
|
|
self.client().update_router(router_id,
|
|
{'router': {'routes': routes}})
|
|
|
|
|
|
def resource_mapping():
|
|
return {
|
|
'OS::Neutron::ExtraRoute': ExtraRoute,
|
|
}
|