def replace_environment_variables() -> None:
"""Enable yaml loader to process the environment variables in the yaml.
"""
import re
import os
# eg. ${USER_NAME}, ${PASSWORD}
env_var_pattern = re.compile(r"^(.*)\$\{(.*)\}(.*)$")
yaml.add_implicit_resolver("!env_var", env_var_pattern)
def env_var_constructor(loader, node):
"""Process environment variables found in the YAML.
"""
value = loader.construct_scalar(node)
expanded_vars = os.path.expandvars(value)
if "$" in expanded_vars:
not_expanded = [w for w in expanded_vars.split() if "$" in w]
raise ValueError(
"Error when trying to expand the environment variables"
" in '{}'. Please make sure to also set these environment"
" variables: '{}'.".format(value, not_expanded)
)
return expanded_vars
yaml.SafeConstructor.add_constructor("!env_var", env_var_constructor)
def replace_environment_variables():
"""Enable yaml loader to process the environment variables in the yaml."""
import re
import os
env_var_pattern = re.compile(r"^(.*)\$\{(.*)\}(.*)$")
yaml.add_implicit_resolver('!env_var', env_var_pattern)
def env_var_constructor(loader, node):
"""Process environment variables found in the YAML."""
value = loader.construct_scalar(node)
prefix, env_var, postfix = env_var_pattern.match(value).groups()
return prefix + os.environ[env_var] + postfix
yaml.SafeConstructor.add_constructor(u'!env_var', env_var_constructor)
def replace_environment_variables():
"""Enable yaml loader to process the environment variables in the yaml."""
import ruamel.yaml as yaml
import re
import os
# eg. ${USER_NAME}, ${PASSWORD}
env_var_pattern = re.compile(r'^(.*)\$\{(.*)\}(.*)$')
yaml.add_implicit_resolver('!env_var', env_var_pattern)
def env_var_constructor(loader, node):
"""Process environment variables found in the YAML."""
value = loader.construct_scalar(node)
prefix, env_var, remaining_path = env_var_pattern.match(value).groups()
return prefix + os.environ[env_var] + remaining_path
yaml.SafeConstructor.add_constructor(u'!env_var', env_var_constructor)
def get_cfg(ruta_base=""):
"""Parse the YAML config"""
pattern = re.compile(r"\$\{(.*)\}(.*)$")
yaml.add_implicit_resolver("!env", pattern)
def env_constructor(loader, node):
"""Constructor for environment variables"""
value = loader.construct_scalar(node)
env_var, remaining_path = pattern.match(value).groups()
return os.environ[env_var] + remaining_path
yaml.add_constructor('!env', env_constructor)
ruta_cfg = ruta_base + "git_creds.yml"
with open(ruta_cfg) as config:
try:
cfg = yaml.load(config, Loader=yaml.Loader)
except yaml.YAMLError:
logging.error("Error while loading config file.")
raise
return cfg
return v.reduced()
def __lt__(self, other):
return float(self) < float(other)
def __gt__(self, other):
return float(self) > float(other)
def __eq__(self, other):
try:
return float(self) == float(other)
except TypeError:
return False
def __le__(self, other):
return float(self) <= float(other)
def __ge__(self, other):
return float(self) >= float(other)
def __ne__(self, other):
try:
return float(self) != float(other)
except TypeError:
return True
# Make this tag automatic
yaml.add_implicit_resolver(PrefixedUnit.yaml_tag,
PrefixedUnit.yaml_implicit_pattern)
v = self.__class__(float(self) % float(other), self.unit)
return v.reduced()
def __lt__(self, other):
return float(self) < float(other)
def __gt__(self, other):
return float(self) > float(other)
def __eq__(self, other):
try:
return float(self) == float(other)
except TypeError:
return False
def __le__(self, other):
return float(self) <= float(other)
def __ge__(self, other):
return float(self) >= float(other)
def __ne__(self, other):
try:
return float(self) != float(other)
except TypeError:
return True
# Make this tag automatic
yaml.add_implicit_resolver(PrefixedUnit.yaml_tag, PrefixedUnit.yaml_implicit_pattern)