python loaders: fallback to yaml package
lgz committed
Jul 9, 2018 at 23:41 UTC
e751407bec348e095fc3b901ecf4064c6cd75660
1 file changed
+10
-2
python.d/python_modules/bases/loaders.py
+10
-2
@@ -4,16 +4,24 @@
4
# SPDX-License-Identifier: GPL-3.0+
5
6
import types
7
+
8
from sys import version_info
9
10
PY_VERSION = version_info[:2]
11
12
+try:
13
+ if PY_VERSION > (3, 1):
14
+ from pyyaml3 import SafeLoader as YamlSafeLoader
15
+ else:
16
+ from pyyaml2 import SafeLoader as YamlSafeLoader
17
+except ImportError:
18
+ from yaml import SafeLoader as YamlSafeLoader
19
+
20
+
21
if PY_VERSION > (3, 1):
12
- from pyyaml3 import SafeLoader as YamlSafeLoader
22
from importlib.machinery import SourceFileLoader
23
DEFAULT_MAPPING_TAG = 'tag:yaml.org,2002:map'
24
else:
16
- from pyyaml2 import SafeLoader as YamlSafeLoader
25
from imp import load_source as SourceFileLoader
26
DEFAULT_MAPPING_TAG = u'tag:yaml.org,2002:map'
27