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

import typing as t

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: t.Union[t.Type[_yaml.CSafeLoader], t.Type[_yaml.SafeLoader]] = _yaml.CSafeLoader
    except AttributeError:
        _SafeLoader = _yaml.SafeLoader

    yaml_load = partial(_yaml.load, Loader=_SafeLoader)