Update setup.py

denfromufa · web-flow · commit cae7a1e1b908 · 2017-09-11T10:59:29.000-07:00
diff --git a/setup.py b/setup.py
@@ -281,7 +281,18 @@ def _find_msbuild_tool(self, tool="msbuild.exe", use_windows_sdk=False):
         path = spawn.find_executable(tool)
         if path:
             return path
-
+        
+        if tool == "msbuild.exe":
+            basePathes = subprocess.check_output(
+                    ["vswhere", "-latest",
+                     "-version", "[15.0, 16.0)", 
+                     "-requires", "Microsoft.Component.MSBuild",
+                     "-property", "InstallationPath"]).splitlines()
+            if basePathes and basePathes[0] and os.path.exists(basePathes[0]):
+                path = os.path.join(basePathes[0],
+                                    "MSBuild", "15.0", "Bin", tool)
+                return path
+        
         # Search within registry to find build tools
         try:  # PY2
             import _winreg as winreg
