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 # yardstick: this file is copied from python-heatclient and slightly modified
  14
  15 from __future__ import absolute_import
  16
  17 import yaml
  18
  19
  20 if hasattr(yaml, 'CSafeLoader'):
  21     # make a dynamic subclass so we don't override global yaml Loader
  22     yaml_loader = type('CustomLoader', (yaml.CSafeLoader,), {})
  23 else:
  24     yaml_loader = type('CustomLoader', (yaml.SafeLoader,), {})
  25
  26 if hasattr(yaml, 'CSafeDumper'):
  27     yaml_dumper = yaml.CSafeDumper
  28 else:
  29     yaml_dumper = yaml.SafeDumper
  30
  31
  32 def yaml_load(tmpl_str):
  33     return yaml.load(tmpl_str, Loader=yaml_loader)