python: Fix yaml.load warnings

Since pyyaml 5.1 yaml.load without specifing loader is deprecated
Details: https://msg.pyyaml.org/load

To keep code compatible with older versions of pyyaml
and keep best perfomance CLoader with fallback to Loader is used.
This commit is contained in:
Sergei Silnov
2019-06-28 16:36:20 +02:00
committed by bot
parent c27fd32fbe
commit c57dfbc0b8
7 changed files with 45 additions and 9 deletions

View File

@@ -34,6 +34,11 @@ This will prevent test cases from getting configs from other env when there're c
import yaml
try:
from yaml import CLoader as Loader
except ImportError:
from yaml import Loader as Loader
class Config(object):
""" Test Env Config """
@@ -52,7 +57,7 @@ class Config(object):
"""
try:
with open(config_file) as f:
configs = yaml.load(f)[env_name]
configs = yaml.load(f, Loader=Loader)[env_name]
except (OSError, TypeError, IOError):
configs = dict()
return configs