Add generic function for retrieve information from the registry

git-svn-id: https://src.develer.com/svnoss/bertos/trunk@2591 38d2e660-2303-0410-9eaa-f027e97ec537

diff --git a/wizard/winreg_importer.py b/wizard/winreg_importer.py
index 714d0a7893aa1fa6177b56db3ce693ae1ba24426..4e5131a9e1a120cb18194d1dec812d476fa650f0 100644 (file)
--- a/wizard/winreg_importer.py
+++ b/wizard/winreg_importer.py
@@ -12,16 +12,31 @@
 from _winreg import *\r
 \r
 DIR_KEY = OpenKey(HKEY_LOCAL_MACHINE, "SOFTWARE\Develer\BeRTOS SDK\BeRTOS Dirs")\r
+TOOLCHAIN_KEY = OpenKey(HKEY_LOCAL_MACHINE, "SOFTWARE\Develer\BeRTOS SDK\Toolchain Executables")\r
 \r
 def getBertosDirs():\r
+    """\r
+    Returns the path of the BeRTOS versions installed by the BeRTOS SDK installer.\r
+    """\r
+    return getFromRegistry(DIR_KEY)\r
+\r
+def getBertosToolchain():\r
+    """\r
+    Returns the path of the executables of the toolchains installed by the BeRTOS\r
+    SDK installer.\r
+    """\r
+    return getFromRegistry(TOOLCHAIN_KEY)\r
+\r
+def getFromRegistry(key):\r
+    """\r
+    Returns the value of all the named values of the given key.\r
+    """\r
     index = 0\r
-    dirs = []\r
+    items = []\r
     while True:\r
         try:\r
-            dir = EnumValue(DIR_KEY, index)[1]\r
-            dirs.append(dir)\r
-            index += 1\r
+            item = EnumValue(TOOLCHAIN_KEY, index)[1]\r
+            items.append(item)\r
         except WindowsError:\r
             break\r
-    return dirs\r
-       
\ No newline at end of file
+    return items
\ No newline at end of file