From 1144555eca38a0f8a0e3ec6eacb5b32239828289 Mon Sep 17 00:00:00 2001
From: sledgehammer999 <hammered999@gmail.com>
Date: Sun, 14 Sep 2014 21:49:52 +0300
Subject: [PATCH] WINDOWS: Can now correctly detect 64bit and 32bit python
 installations on 64bit Windows. Closes #1148 #445 #795 #1708.

---
 src/preferences/preferences.cpp | 51 +++++++++++++++++++++++++++++----
 1 file changed, 46 insertions(+), 5 deletions(-)

diff --git a/src/preferences/preferences.cpp b/src/preferences/preferences.cpp
index d89f4e46d..fcd7d1978 100644
--- a/src/preferences/preferences.cpp
+++ b/src/preferences/preferences.cpp
@@ -1340,6 +1340,8 @@ void Preferences::disableRecursiveDownload(bool disable) {
 
 #ifdef Q_OS_WIN
 namespace {
+enum REG_SEARCH_TYPE {USER, SYSTEM_32BIT, SYSTEM_64BIT};
+
 QStringList getRegSubkeys(const HKEY &handle) {
   QStringList keys;
   DWORD subkeys_count = 0;
@@ -1395,11 +1397,22 @@ QString getRegValue(const HKEY &handle, const QString &name = QString()) {
   return end_result;
 }
 
-}
-
-QString Preferences::getPythonPath() {
+QString pythonSearchReg(const REG_SEARCH_TYPE type) {
   HKEY key_handle1;
-  long res = ::RegOpenKeyEx(HKEY_LOCAL_MACHINE, TEXT("SOFTWARE\\Python\\PythonCore"), 0, KEY_READ, &key_handle1);
+  long res = 0;
+
+  switch (type) {
+  case USER:
+    res = ::RegOpenKeyEx(HKEY_CURRENT_USER, TEXT("SOFTWARE\\Python\\PythonCore"), 0, KEY_READ, &key_handle1);
+    break;
+  case SYSTEM_32BIT:
+    res = ::RegOpenKeyEx(HKEY_LOCAL_MACHINE, TEXT("SOFTWARE\\Python\\PythonCore"), 0, KEY_READ|KEY_WOW64_32KEY, &key_handle1);
+    break;
+  case SYSTEM_64BIT:
+    res = ::RegOpenKeyEx(HKEY_LOCAL_MACHINE, TEXT("SOFTWARE\\Python\\PythonCore"), 0, KEY_READ|KEY_WOW64_64KEY, &key_handle1);
+    break;
+  }
+
   if (res == ERROR_SUCCESS) {
     QStringList versions = getRegSubkeys(key_handle1);
     qDebug("Python versions nb: %d", versions.size());
@@ -1412,7 +1425,18 @@ QString Preferences::getPythonPath() {
       version.toWCharArray(subkey);
       subkey[version.size()] = '\0';
 
-      res = ::RegOpenKeyEx(key_handle1, subkey, 0, KEY_READ, &key_handle2);
+      switch (type) {
+      case USER:
+        res = ::RegOpenKeyEx(key_handle1, subkey, 0, KEY_READ, &key_handle2);
+        break;
+      case SYSTEM_32BIT:
+        res = ::RegOpenKeyEx(key_handle1, subkey, 0, KEY_READ|KEY_WOW64_32KEY, &key_handle2);
+        break;
+      case SYSTEM_64BIT:
+        res = ::RegOpenKeyEx(key_handle1, subkey, 0, KEY_READ|KEY_WOW64_64KEY, &key_handle2);
+        break;
+      }
+
       delete[] subkey;
       if (res == ERROR_SUCCESS) {
         qDebug("Detected possible Python v%s location", qPrintable(version));
@@ -1429,6 +1453,23 @@ QString Preferences::getPythonPath() {
     }
   }
   ::RegCloseKey(key_handle1);
+  return QString::null;
+}
+
+}
+
+QString Preferences::getPythonPath() {
+  QString path = pythonSearchReg(USER);
+  if (path.isEmpty())
+    path = pythonSearchReg(SYSTEM_32BIT);
+  else return path;
+
+  if (path.isEmpty())
+    path = pythonSearchReg(SYSTEM_64BIT);
+  else return path;
+
+  if (!path.isEmpty())
+    return path;
 
   // Fallback: Detect python from default locations
   QStringList supported_versions;