diff --git a/storops/lib/metric.py b/storops/lib/metric.py index 27c86343..289709e9 100644 --- a/storops/lib/metric.py +++ b/storops/lib/metric.py @@ -18,6 +18,7 @@ import os import yaml +from yaml import loader from storops.lib.common import RepeatedTimer, cache from storops.lib.resource import ResourceList @@ -196,7 +197,7 @@ def get_config(cls, name): def _read_configs(cls): filename = os.path.join(cls.get_folder(), cls.config_filename) with open(filename, 'r') as stream: - ret = yaml.load(stream) + ret = yaml.load(stream, Loader=loader.SafeLoader) return ret @classmethod diff --git a/storops/lib/parser.py b/storops/lib/parser.py index a6e5680f..a683614a 100644 --- a/storops/lib/parser.py +++ b/storops/lib/parser.py @@ -24,6 +24,7 @@ import six import yaml +from yaml import loader from storops.lib.common import cache, instance_cache, Enum, \ get_clz_from_module, EnumList @@ -357,7 +358,7 @@ def get_config(self, name): def _read_configs(self): filename = os.path.join(self.get_folder(), self.config_filename) with open(filename, 'r') as stream: - ret = yaml.load(stream) + ret = yaml.load(stream, Loader=loader.SafeLoader) return ret @instance_cache