Fix __getitem__ for DataSmart

Ensure it raises KeyError for a missing key, this is required to use this as a
mapping in various places, e.g. as locals in an eval.

Signed-off-by: Chris Larson <chris_larson@mentor.com>

1 files changed, 6 insertions, 2 deletions

diff --git a/lib/bb/data_smart.py b/lib/bb/data_smart.py
index aac13490d..64fae2d7d 100644
--- a/lib/bb/data_smart.py
+++ b/lib/bb/data_smart.py
@@ -125,7 +125,7 @@ class DataSmart(MutableMapping):
             for var in vars:
                 name = var[:-l]
                 try:
-                    self[name] = self[var]
+                    self.setVar(name, self.getVar(var, False))
                 except Exception:
                     logger.info("Untracked delVar")
 
@@ -343,7 +343,11 @@ class DataSmart(MutableMapping):
         return len(frozenset(self))
 
     def __getitem__(self, item):
-        return self.getVar(item, False)
+        value = self.getVar(item, False)
+        if value is None:
+            raise KeyError(item)
+        else:
+            return value
 
     def __setitem__(self, var, value):
         self.setVar(var, value)