List installed Win10 SDK's when looking for windows sdk tool

OneBlue · OneBlue · commit a2733369ae81 · 2019-04-03T00:22:01.000+02:00
diff --git a/setup.py b/setup.py
@@ -39,7 +39,7 @@
 kits_root = "SOFTWARE\\Microsoft\\Windows Kits\\Installed Roots"
 kits_suffix = os.path.join("bin", ARCH)
 
-WIN_SDK_KEYS = (
+WIN_SDK_KEYS = [
     RegKey(sdk_name="Windows Kit 10.0", key=kits_root,
            value_name="KitsRoot10", suffix=os.path.join("bin", "10.0.16299.0", ARCH)),
 
@@ -69,7 +69,7 @@
 
     RegKey(sdk_name="Windows SDK 6.0A", key=sdks_root.format("6.0A\\WinSDK"),
            value_name="InstallationFolder", suffix=""),
-)
+]
 
 VS_KEYS = (
     RegKey(sdk_name="MSBuild 15", key=vs_root.format("15.0"),
@@ -145,6 +145,29 @@ def _update_xlat_devtools():
     elif DEVTOOLS == "Mono":
         DEVTOOLS = "dotnet"
 
+def _collect_installed_windows_kits_v10(winreg):
+    """Adds the installed Windows 10 kits to WIN_SDK_KEYS """
+    global WIN_SDK_KEYS
+    installed_kits = []
+
+    with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, kits_root, 0, winreg.KEY_READ) as key:
+        i = 0
+        while True:
+            try:
+                installed_kits.append(winreg.EnumKey(key, i))
+                i += 1
+            except WindowsError:
+                break
+
+    def make_reg_key(version):
+        return RegKey(sdk_name="Windows Kit 10.0", key=kits_root, 
+                      value_name="KitsRoot10", suffix=os.path.join("bin", version, ARCH))
+
+    WIN_SDK_KEYS += [make_reg_key(e) for e in installed_kits if e.startswith('10.')]
+
+    # Make sure this function won't be called again
+    _collect_installed_windows_kits_v10 = (lambda:None)
+
 class BuildExtPythonnet(build_ext.build_ext):
     user_options = build_ext.build_ext.user_options + [
             ('xplat', None, None)
@@ -367,6 +390,8 @@ def _find_msbuild_tool(self, tool="msbuild.exe", use_windows_sdk=False):
         except ImportError:  # PY3
             import winreg
 
+        _collect_installed_windows_kits_v10(winreg)
+
         keys_to_check = WIN_SDK_KEYS if use_windows_sdk else VS_KEYS
         hklm = winreg.HKEY_LOCAL_MACHINE
         for rkey in keys_to_check:
