ansible/test/lib/ansible_test/_internal/compat/yaml.py

22 lines
500 B
Python

"""PyYAML compatibility."""
from __future__ import (absolute_import, division, print_function)
__metaclass__ = type
from functools import (
partial,
)
try:
import yaml as _yaml
YAML_IMPORT_ERROR = None
except ImportError as ex:
yaml_load = None # pylint: disable=invalid-name
YAML_IMPORT_ERROR = ex
else:
try:
_SafeLoader = _yaml.CSafeLoader
except AttributeError:
_SafeLoader = _yaml.SafeLoader
yaml_load = partial(_yaml.load, Loader=_SafeLoader)