7349dbfa5c
Fixed in all parsers (main config, profiles, variables and recommend). Signed-off-by: Jaroslav Škarvada <jskarvad@redhat.com>
74 lines
2 KiB
Python
74 lines
2 KiB
Python
import os
|
|
import re
|
|
import tuned.logs
|
|
import tuned.consts as consts
|
|
from tuned.utils.commands import commands
|
|
from configobj import ConfigObj
|
|
|
|
log = tuned.logs.get()
|
|
|
|
class Variables():
|
|
"""
|
|
Storage and processing of variables used in profiles
|
|
"""
|
|
|
|
def __init__(self):
|
|
self._cmd = commands()
|
|
self._lookup_re = {}
|
|
self._lookup_env = {}
|
|
|
|
def _add_env_prefix(self, s, prefix):
|
|
if s.find(prefix) == 0:
|
|
return s
|
|
return prefix + s
|
|
|
|
def _check_var(self, variable):
|
|
return re.match(r'\w+$',variable)
|
|
|
|
def add_variable(self, variable, value):
|
|
if value is None:
|
|
return
|
|
s = str(variable)
|
|
if not self._check_var(variable):
|
|
log.error("variable definition '%s' contains unallowed characters" % variable)
|
|
return
|
|
v = self.expand(value)
|
|
# variables referenced by ${VAR}, $ can be escaped by two $,
|
|
# i.e. the following will not expand: $${VAR}
|
|
self._lookup_re[r'(?<!\$)\${' + re.escape(s) + r'}'] = v
|
|
self._lookup_env[self._add_env_prefix(s, consts.ENV_PREFIX)] = v
|
|
|
|
def add_dict(self, d):
|
|
for item in d:
|
|
self.add_variable(item, d[item])
|
|
|
|
def add_from_file(self, filename):
|
|
if not os.path.exists(filename):
|
|
log.error("unable to find variables_file: '%s'" % filename)
|
|
return
|
|
try:
|
|
config = ConfigObj(filename, raise_errors = True, file_error = True, list_values = False, interpolation = False)
|
|
except ConfigObjError:
|
|
log.error("error parsing variables_file: '%s'" % filename)
|
|
return
|
|
for item in config:
|
|
if isinstance(config[item], dict):
|
|
self.add_dict(config[item])
|
|
else:
|
|
self.add_variable(item, config[item])
|
|
|
|
def add_from_cfg(self, cfg, dir_name):
|
|
for item in cfg:
|
|
if str(item) == "include":
|
|
self.add_from_file(os.path.normpath(os.path.join(dir_name, cfg[item])))
|
|
else:
|
|
self.add_variable(item, cfg[item])
|
|
|
|
def expand(self, value):
|
|
if value is None:
|
|
return None
|
|
# expand variables and finally convert all $${VAR} to ${VAR} (unescape)
|
|
return re.sub(r'\$(\${\w+})', r'\1', self._cmd.multiple_re_replace(self._lookup_re, str(value)))
|
|
|
|
def get_env(self):
|
|
return self._lookup_env
|