File size: 1,413 Bytes
64094d4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
from ruamel.yaml import YAML
import logging
class YParams():
""" Yaml file parser """
def __init__(self, yaml_filename, config_name, print_params=False):
self._yaml_filename = yaml_filename
self._config_name = config_name
self.params = {}
if print_params:
print("------------------ Configuration ------------------")
with open(yaml_filename) as _file:
for key, val in YAML().load(_file)[config_name].items():
if print_params: print(key, val)
if val =='None': val = None
self.params[key] = val
self.__setattr__(key, val)
if print_params:
print("---------------------------------------------------")
def __getitem__(self, key):
return self.params[key]
def __setitem__(self, key, val):
self.params[key] = val
self.__setattr__(key, val)
def __contains__(self, key):
return (key in self.params)
def update_params(self, config):
for key, val in config.items():
self.params[key] = val
self.__setattr__(key, val)
def log(self):
logging.info("------------------ Configuration ------------------")
logging.info("Configuration file: "+str(self._yaml_filename))
logging.info("Configuration name: "+str(self._config_name))
for key, val in self.params.items():
logging.info(str(key) + ' ' + str(val))
logging.info("---------------------------------------------------") |