Add the const string containing the pattern of the cpu definition filenames

[bertos.git] / wizard / bertos_utils.py

diff --git a/wizard/bertos_utils.py b/wizard/bertos_utils.py
index 0f0dc709bcd5e4041509fedd0a7d8a258479769b..99bdca23f7b248463f1dcd42452795736b4f1871 100644 (file)
--- 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")
 
@@ -27,13 +29,14 @@ def findDefinitions(ftype, path):
     L = os.walk(path)
     for element in L:
         for filename in element[2]:
-            if fnmatch.fnmatch(filename, "*." + ftype):
+            if fnmatch.fnmatch(filename, ftype):
                 yield (filename, element[0])
 
 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