X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=wizard%2Fwinreg_importer.py;h=2030934e9f163efb3df191e56f4776f816706a21;hb=440450f9f44c7bfc5bf1813adab87716e0a422fd;hp=4e5131a9e1a120cb18194d1dec812d476fa650f0;hpb=a8fcfb6e93a2bd82ece8953a282c7e28914a646d;p=bertos.git

diff --git a/wizard/winreg_importer.py b/wizard/winreg_importer.py
index 4e5131a9..2030934e 100644
--- a/wizard/winreg_importer.py
+++ b/wizard/winreg_importer.py
@@ -11,8 +11,16 @@
 
 from _winreg import *
 
-DIR_KEY = OpenKey(HKEY_LOCAL_MACHINE, "SOFTWARE\Develer\BeRTOS SDK\BeRTOS Dirs")
-TOOLCHAIN_KEY = OpenKey(HKEY_LOCAL_MACHINE, "SOFTWARE\Develer\BeRTOS SDK\Toolchain Executables")
+# Open the registry keys. When the keys don't exist it do nothing
+try:
+    DIR_KEY = OpenKey(HKEY_LOCAL_MACHINE, "SOFTWARE\Develer\BeRTOS SDK\BeRTOS Dirs")
+except WindowsError:
+    DIR_KEY = None
+
+try:
+    TOOLCHAIN_KEY = OpenKey(HKEY_LOCAL_MACHINE, "SOFTWARE\Develer\BeRTOS SDK\Toolchain Executables")
+except WindowsError:
+    TOOLCHAIN_KEY = None
 
 def getBertosDirs():
     """
@@ -20,7 +28,7 @@ def getBertosDirs():
     """
     return getFromRegistry(DIR_KEY)
 
-def getBertosToolchain():
+def getBertosToolchains():
     """
     Returns the path of the executables of the toolchains installed by the BeRTOS
     SDK installer.
@@ -33,10 +41,12 @@ def getFromRegistry(key):
     """
     index = 0
     items = []
-    while True:
-        try:
-            item = EnumValue(TOOLCHAIN_KEY, index)[1]
-            items.append(item)
-        except WindowsError:
-            break
+    if key:
+        while True:
+            try:
+                item = EnumValue(key, index)[1]
+                items.append(item)
+                index += 1
+            except WindowsError:
+                break
     return items
\ No newline at end of file