diff --git a/build.py b/build.py index 8bb77fde..3fc4c2b5 100755 --- a/build.py +++ b/build.py @@ -792,12 +792,7 @@ def checkCompiler(quiet=False): # about time to do away with this...) arch = 'x64' if PYTHON_ARCH == '64bit' else 'x86' - info = getMSVCInfo(PYTHON, arch) - - os.environ['PATH'] = info.path - os.environ['INCLUDE'] = info.include - os.environ['LIB'] = info.lib - os.environ['LIBPATH'] = info.libpath + info = getMSVCInfo(PYTHON, arch, set_env=True) # Make sure there is now a cl.exe on the PATH CL = 'NOT FOUND' diff --git a/buildtools/config.py b/buildtools/config.py index 3f593893..cbfd0ca4 100644 --- a/buildtools/config.py +++ b/buildtools/config.py @@ -981,7 +981,7 @@ def getExpectedVisCVersion(): MSVCinfo = None -def getMSVCInfo(PYTHON, arch): +def getMSVCInfo(PYTHON, arch, set_env=False): """ Fetch info from the system about MSVC, such as versions, paths, etc. """ @@ -1005,6 +1005,12 @@ def getMSVCInfo(PYTHON, arch): env = eval(runcmd('"%s" -c "%s"' % (PYTHON, cmd), getOutput=True, echoCmd=False)) info = AttrDict(env) + if set_env: + os.environ['PATH'] = info.path + os.environ['INCLUDE'] = info.include + os.environ['LIB'] = info.lib + os.environ['LIBPATH'] = info.libpath + MSVCinfo = info return info