Gecko [ unitednationsplay.com ]

Name	Size	Permission	Date
functions	[ DIR ]	drwxr-xr-x	2021-11-01 09:21
__init__.py	281 B	-rw-r--r--	2019-03-21 22:10
__init__.pyc	483 B	-rw-r--r--	2021-02-02 16:30
__init__.pyo	483 B	-rw-r--r--	2021-02-02 16:30
exceptions.py	95 B	-rw-r--r--	2019-03-21 22:10
exceptions.pyc	458 B	-rw-r--r--	2021-02-02 16:30
exceptions.pyo	458 B	-rw-r--r--	2021-02-02 16:30
factory.py	141 B	-rw-r--r--	2019-03-21 22:10
factory.pyc	653 B	-rw-r--r--	2021-02-02 16:30
factory.pyo	653 B	-rw-r--r--	2021-02-02 16:30
loader.py	4.13 KB	-rw-r--r--	2019-03-21 22:10
loader.pyc	4.99 KB	-rw-r--r--	2021-02-02 16:30
loader.pyo	4.99 KB	-rw-r--r--	2021-02-02 16:30
locator.py	3.23 KB	-rw-r--r--	2019-03-21 22:10
locator.pyc	4.05 KB	-rw-r--r--	2021-02-02 16:30
locator.pyo	4.05 KB	-rw-r--r--	2021-02-02 16:30
merger.py	1.79 KB	-rw-r--r--	2019-03-21 22:10
merger.pyc	2.12 KB	-rw-r--r--	2021-02-02 16:30
merger.pyo	2.12 KB	-rw-r--r--	2021-02-02 16:30
profile.py	1.11 KB	-rw-r--r--	2019-03-21 22:10
profile.pyc	2.43 KB	-rw-r--r--	2021-02-02 16:30
profile.pyo	2.43 KB	-rw-r--r--	2021-02-02 16:30
unit.py	1.73 KB	-rw-r--r--	2019-03-21 22:10
unit.pyc	3.86 KB	-rw-r--r--	2021-02-02 16:30
unit.pyo	3.86 KB	-rw-r--r--	2021-02-02 16:30
variables.py	2.26 KB	-rw-r--r--	2019-03-21 22:10
variables.pyc	3.86 KB	-rw-r--r--	2021-02-02 16:30
variables.pyo	3.86 KB	-rw-r--r--	2021-02-02 16:30

Rename

import os
import re
import tuned.logs
from .functions import functions as functions
import tuned.consts as consts
from tuned.utils.commands import commands
from configobj import ConfigObj, ConfigObjError

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 = {}
		self._functions = functions.Functions()

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):
		for item in cfg:
			if str(item) == "include":
				self.add_from_file(os.path.normpath(cfg[item]))
			else:
				self.add_variable(item, cfg[item])

# expand static variables (no functions)
	def expand_static(self, value):
		return re.sub(r'\\(\${\w+})', r'\1', self._cmd.multiple_re_replace(self._lookup_re, value))

def expand(self, value):
		if value is None:
			return None
		# expand variables and convert all \${VAR} to ${VAR} (unescape)
		s = self.expand_static(str(value))
		# expand built-in functions
		return self._functions.expand(s)

def get_env(self):
		return self._lookup_env