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