X-Git-Url: https://codewiz.org/gitweb?a=blobdiff_plain;f=wizard%2Fbertos_utils.py;h=0a1076911acf06ed8f470093c3c8c977b10f59f5;hb=29094c2d235a65c3edadd2ec3398b8d71c91a998;hp=b927ec48e1e46906c77b730a0f2e7f5fafef638d;hpb=0676a7451f0fd727af3c17350f6abbde825b4e3b;p=bertos.git

diff --git a/wizard/bertos_utils.py b/wizard/bertos_utils.py
index b927ec48..0a107691 100644
--- a/wizard/bertos_utils.py
+++ b/wizard/bertos_utils.py
@@ -12,6 +12,8 @@
 import os
 import fnmatch
 
+import const
+
 def isBertosDir(directory):
    return os.path.exists(directory + "/VERSION")
 
@@ -32,11 +34,13 @@ def findDefinitions(ftype, path):
 
 def loadCpuInfos(path):
     cpuInfos = []
-    for definition in findDefinitions("cdef", path):
+    for definition in findDefinitions(const.CPU_DEFINITION, path):
         D = {}
+        D.update(const.CPU_DEF)
         def include(filename, dict = D, directory=definition[1]):
             execfile(directory + "/" + filename, {}, D)
         D["include"] = include
         include(definition[0], D)
+        D["CPU_NAME"] = definition[0].split(".")[0]
         cpuInfos.append(D)
     return cpuInfos