environment.py 2.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485
  1. import os
  2. from distutils.dist import Distribution
  3. __metaclass__ = type
  4. class EnvironmentConfig:
  5. def __init__(self, distutils_section='ALL', **kw):
  6. self._distutils_section = distutils_section
  7. self._conf_keys = kw
  8. self._conf = None
  9. self._hook_handler = None
  10. def dump_variable(self, name):
  11. conf_desc = self._conf_keys[name]
  12. hook, envvar, confvar, convert, append = conf_desc
  13. if not convert:
  14. convert = lambda x : x
  15. print('%s.%s:' % (self._distutils_section, name))
  16. v = self._hook_handler(name, hook)
  17. print(' hook : %s' % (convert(v),))
  18. if envvar:
  19. v = os.environ.get(envvar, None)
  20. print(' environ: %s' % (convert(v),))
  21. if confvar and self._conf:
  22. v = self._conf.get(confvar, (None, None))[1]
  23. print(' config : %s' % (convert(v),))
  24. def dump_variables(self):
  25. for name in self._conf_keys:
  26. self.dump_variable(name)
  27. def __getattr__(self, name):
  28. try:
  29. conf_desc = self._conf_keys[name]
  30. except KeyError:
  31. raise AttributeError(name)
  32. return self._get_var(name, conf_desc)
  33. def get(self, name, default=None):
  34. try:
  35. conf_desc = self._conf_keys[name]
  36. except KeyError:
  37. return default
  38. var = self._get_var(name, conf_desc)
  39. if var is None:
  40. var = default
  41. return var
  42. def _get_var(self, name, conf_desc):
  43. hook, envvar, confvar, convert, append = conf_desc
  44. if convert is None:
  45. convert = lambda x: x
  46. var = self._hook_handler(name, hook)
  47. if envvar is not None:
  48. envvar_contents = os.environ.get(envvar)
  49. if envvar_contents is not None:
  50. envvar_contents = convert(envvar_contents)
  51. if var and append:
  52. if os.environ.get('NPY_DISTUTILS_APPEND_FLAGS', '1') == '1':
  53. var.extend(envvar_contents)
  54. else:
  55. # NPY_DISTUTILS_APPEND_FLAGS was explicitly set to 0
  56. # to keep old (overwrite flags rather than append to
  57. # them) behavior
  58. var = envvar_contents
  59. else:
  60. var = envvar_contents
  61. if confvar is not None and self._conf:
  62. if confvar in self._conf:
  63. source, confvar_contents = self._conf[confvar]
  64. var = convert(confvar_contents)
  65. return var
  66. def clone(self, hook_handler):
  67. ec = self.__class__(distutils_section=self._distutils_section,
  68. **self._conf_keys)
  69. ec._hook_handler = hook_handler
  70. return ec
  71. def use_distribution(self, dist):
  72. if isinstance(dist, Distribution):
  73. self._conf = dist.get_option_dict(self._distutils_section)
  74. else:
  75. self._conf = dist