X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=osm_policy_module%2Fcore%2Fconfig.py;h=1bc9b2b3454b30cd179ae16086baa3eee6886786;hb=d37c54c64eec65c9a3c490a31eef3a02a76cb474;hp=e482ec8e2847f0ae602ac2e85610f505694392c7;hpb=a14cf16181c8b39f12c872c486e0b292c0068944;p=osm%2FPOL.git

diff --git a/osm_policy_module/core/config.py b/osm_policy_module/core/config.py
index e482ec8..1bc9b2b 100644
--- a/osm_policy_module/core/config.py
+++ b/osm_policy_module/core/config.py
@@ -33,17 +33,17 @@ logger = logging.getLogger(__name__)
 
 
 class Config:
-    def __init__(self, config_file: str = ''):
+    def __init__(self, config_file: str = ""):
         self.conf = {}
         self._read_config_file(config_file)
         self._read_env()
 
     def _read_config_file(self, config_file):
         if not config_file:
-            path = 'pol.yaml'
+            path = "pol.yaml"
             config_file = pkg_resources.resource_filename(__name__, path)
         with open(config_file) as f:
-            self.conf = yaml.load(f)
+            self.conf = yaml.safe_load(f)
 
     def get(self, section, field=None):
         if not field:
@@ -63,10 +63,12 @@ class Config:
             if len(elements) < 3:
                 logger.warning(
                     "Environment variable %s=%s does not comply with required format. Section and/or field missing.",
-                    env, os.getenv(env))
+                    env,
+                    os.getenv(env),
+                )
                 continue
             section = elements[1]
-            field = '_'.join(elements[2:])
+            field = "_".join(elements[2:])
             value = os.getenv(env)
             if section not in self.conf:
                 self.conf[section] = {}