Create Dockerfile to create a yardstick-image of docker
[yardstick.git] / yardstick / common / yaml_loader.py
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
4 #
5 #         http://www.apache.org/licenses/LICENSE-2.0
6 #
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
11 #    under the License.
12
13 import yaml
14
15
16 if hasattr(yaml, 'CSafeLoader'):
17     # make a dynamic subclass so we don't override global yaml Loader
18     yaml_loader = type('CustomLoader', (yaml.CSafeLoader,), {})
19 else:
20     yaml_loader = type('CustomLoader', (yaml.SafeLoader,), {})
21
22
23 if hasattr(yaml, 'CSafeDumper'):
24     yaml_dumper = yaml.CSafeDumper
25 else:
26     yaml_dumper = yaml.SafeDumper
27
28
29 def yaml_load(tmpl_str):
30     return yaml.load(tmpl_str, Loader=yaml_loader)
31
32
33 def read_yaml_file(path):
34     """Read yaml file"""
35     with open(path) as stream:
36         data = yaml_load(stream)
37         return data