summaryrefslogtreecommitdiff
path: root/test/lib/ansible_test/_internal/compat/yaml.py
blob: daa5ef0ed49854f0c0b3c41db6bc7d6e52847abe (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
"""PyYAML compatibility."""
from __future__ import annotations

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)