1 # Licensed under the Apache License, Version 2.0 (the "License"); you may
2 # not use this file except in compliance with the License. You may obtain
3 # a copy of the License at
5 # http://www.apache.org/licenses/LICENSE-2.0
7 # Unless required by applicable law or agreed to in writing, software
8 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
9 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
10 # License for the specific language governing permissions and limitations
13 # yardstick: this file is copied from python-heatclient and slightly modified
18 if hasattr(yaml, 'CSafeLoader'):
19 yaml_loader = yaml.CSafeLoader
21 yaml_loader = yaml.SafeLoader
23 if hasattr(yaml, 'CSafeDumper'):
24 yaml_dumper = yaml.CSafeDumper
26 yaml_dumper = yaml.SafeDumper
29 def _construct_yaml_str(self, node):
30 # Override the default string handling function
31 # to always return unicode objects
32 return self.construct_scalar(node)
33 yaml_loader.add_constructor(u'tag:yaml.org,2002:str', _construct_yaml_str)
34 # Unquoted dates like 2013-05-23 in yaml files get loaded as objects of type
35 # datetime.data which causes problems in API layer when being processed by
36 # openstack.common.jsonutils. Therefore, make unicode string out of timestamps
37 # until jsonutils can handle dates.
38 yaml_loader.add_constructor(u'tag:yaml.org,2002:timestamp',
43 '''Takes a string and returns a dict containing the parsed structure.
45 This includes determination of whether the string is using the
48 if tmpl_str.startswith('{'):
49 tpl = json.loads(tmpl_str)
52 tpl = yaml.load(tmpl_str, Loader=yaml_loader)
53 except yaml.YAMLError as yea:
58 # Looking for supported version keys in the loaded template
59 if not ('HeatTemplateFormatVersion' in tpl or
60 'heat_template_version' in tpl or
61 'AWSTemplateFormatVersion' in tpl):
62 raise ValueError("Template format version not found.")