12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- import os
- from distutils.dist import Distribution
- __metaclass__ = type
- class EnvironmentConfig:
- def __init__(self, distutils_section='ALL', **kw):
- self._distutils_section = distutils_section
- self._conf_keys = kw
- self._conf = None
- self._hook_handler = None
- def dump_variable(self, name):
- conf_desc = self._conf_keys[name]
- hook, envvar, confvar, convert, append = conf_desc
- if not convert:
- convert = lambda x : x
- print('%s.%s:' % (self._distutils_section, name))
- v = self._hook_handler(name, hook)
- print(' hook : %s' % (convert(v),))
- if envvar:
- v = os.environ.get(envvar, None)
- print(' environ: %s' % (convert(v),))
- if confvar and self._conf:
- v = self._conf.get(confvar, (None, None))[1]
- print(' config : %s' % (convert(v),))
- def dump_variables(self):
- for name in self._conf_keys:
- self.dump_variable(name)
- def __getattr__(self, name):
- try:
- conf_desc = self._conf_keys[name]
- except KeyError:
- raise AttributeError(
- f"'EnvironmentConfig' object has no attribute '{name}'"
- ) from None
- return self._get_var(name, conf_desc)
- def get(self, name, default=None):
- try:
- conf_desc = self._conf_keys[name]
- except KeyError:
- return default
- var = self._get_var(name, conf_desc)
- if var is None:
- var = default
- return var
- def _get_var(self, name, conf_desc):
- hook, envvar, confvar, convert, append = conf_desc
- if convert is None:
- convert = lambda x: x
- var = self._hook_handler(name, hook)
- if envvar is not None:
- envvar_contents = os.environ.get(envvar)
- if envvar_contents is not None:
- envvar_contents = convert(envvar_contents)
- if var and append:
- if os.environ.get('NPY_DISTUTILS_APPEND_FLAGS', '1') == '1':
- var.extend(envvar_contents)
- else:
- # NPY_DISTUTILS_APPEND_FLAGS was explicitly set to 0
- # to keep old (overwrite flags rather than append to
- # them) behavior
- var = envvar_contents
- else:
- var = envvar_contents
- if confvar is not None and self._conf:
- if confvar in self._conf:
- source, confvar_contents = self._conf[confvar]
- var = convert(confvar_contents)
- return var
- def clone(self, hook_handler):
- ec = self.__class__(distutils_section=self._distutils_section,
- **self._conf_keys)
- ec._hook_handler = hook_handler
- return ec
- def use_distribution(self, dist):
- if isinstance(dist, Distribution):
- self._conf = dist.get_option_dict(self._distutils_section)
- else:
- self._conf = dist
|