Try to retrieve proxies from all users

pupy/network/lib/proxies.py

@@ -36,45 +36,41 @@ def parse_win_proxy(val):
     return l
 
 def get_win_proxies():
-    #TODO retrieve all users proxy settings, not only HKCU
     try:
-        from _winreg import OpenKey, CloseKey, QueryValueEx, HKEY_LOCAL_MACHINE, HKEY_CURRENT_USER, KEY_QUERY_VALUE
+        from _winreg import EnumKey, OpenKey, CloseKey, QueryValueEx
+        from _winreg import HKEY_LOCAL_MACHINE, HKEY_USERS, KEY_QUERY_VALUE
     except:
         return
 
-    aKey = OpenKey(
+    duplicates = set()
-        HKEY_CURRENT_USER, r'SOFTWARE\Microsoft\Windows\CurrentVersion\Internet Settings',
-        0, KEY_QUERY_VALUE
-    )
 
-    try:
+    i = 0
-        value = QueryValueEx(aKey, 'ProxyServer')[0]
+    while True:
-        if value:
+        try:
-            for p in parse_win_proxy(value):
+            user = EnumKey(HKEY_USERS, i)
-                yield p
+            i += 1
+        except:
+            break
 
-    except Exception:
+        if user.endswith('_classes'):
-        pass
+            continue
 
-    finally:
+        try:
-        CloseKey(aKey)
+            key = '{}\\SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Internet Settings'.format(user)
+            aKey = OpenKey(HKEY_USERS, key, 0, KEY_QUERY_VALUE)
+            value = QueryValueEx(aKey, 'ProxyServer')[0]
+            if value:
+                for p in parse_win_proxy(value):
+                    if p not in duplicates:
+                        yield p
+                        duplicates.add(p)
 
-    aKey = OpenKey(
+        except Exception:
-        HKEY_LOCAL_MACHINE, r'SOFTWARE\Microsoft\Windows\CurrentVersion\Internet Settings',
+            pass
-        0, KEY_QUERY_VALUE
-    )
 
-    try:
+        finally:
-        value=QueryValueEx(aKey,'ProxyServer')[0]
+            CloseKey(aKey)
-        if value:
-            for p in parse_win_proxy(value):
-                yield p
 
-    except Exception:
-        pass
-
-    finally:
-        CloseKey(aKey)
 
 def get_python_proxies():
     global PROXY_MATCHER