From 556325cc8cb1032194c2d3739f303fe0a4cfa1a4 Mon Sep 17 00:00:00 2001 From: Thomas Herve Date: Fri, 23 Mar 2018 21:02:58 +0100 Subject: [PATCH] Add custom subclass to revert mapping To be able to revert the custom mappings down by config-download, let's add some tripleo specific subclasses which will be available in the registry. Change-Id: I6bd4107e8e1a6a9abc38d2dca7a91a6823f8b6c2 Related-Bug: #1758065 --- undercloud_heat_plugins/config.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 undercloud_heat_plugins/config.py diff --git a/undercloud_heat_plugins/config.py b/undercloud_heat_plugins/config.py new file mode 100644 index 000000000..66618c64f --- /dev/null +++ b/undercloud_heat_plugins/config.py @@ -0,0 +1,30 @@ +# +# 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 heat.engine.resources.openstack.heat import software_deployment +from heat.engine.resources.openstack.heat import structured_config + + +class SoftwareDeployment(software_deployment.SoftwareDeployment): + """A custom subclass to allow reverting replacement.""" + + +class StructuredDeployment(structured_config.StructuredDeployment): + """A custom subclass to allow reverting replacement.""" + + +def resource_mapping(): + return { + 'OS::TripleO::Heat::SoftwareDeployment': SoftwareDeployment, + 'OS::TripleO::Heat::StructuredDeployment': StructuredDeployment + }