Add construction plugin without tests.

[modules/shaper.git] / src / PythonAPI / model / roots.py

diff --git a/src/PythonAPI/model/roots.py b/src/PythonAPI/model/roots.py
index 0cf8ae2740fe8425cbb6d61a533a61e820f7b795..a5ac3c2014234874c733336619b588704301eca1 100644 (file)
--- a/src/PythonAPI/model/roots.py
+++ b/src/PythonAPI/model/roots.py
@@ -5,7 +5,7 @@ Copyright (C) 2014-20xx CEA/DEN, EDF R&D
 
 import ModelAPI
 
-from .tools import get_value, convert_to_underscore
+from model import tools
 
 
 class Feature(ModelAPI.ModelAPI_Feature):
@@ -33,25 +33,39 @@ class Interface():
 
     def __init__(self, feature):
         self._feature = feature
+        self._attribute_white_list = [
+            "getKind", "results", "firstResult", "lastResult",
+            ]
 
     def __getattr__(self, name):
         """Process missing attributes.
 
         Add get*() methods for access feature attributes.
-        Redirect missing attributes to the feature.
+        Redirect some missing attributes to the feature.
         """
         if name.startswith("get"):
             possible_names = [
                 "_" + name[3:],
-                "_" + convert_to_underscore(name[3:]),
+                "_" + tools.convert_to_underscore(name[3:]),
                 ]
             for possible_name in possible_names:
                 if hasattr(self, possible_name):
                     def getter():
-                        return get_value(getattr(self, possible_name))
+                        return getattr(self, possible_name)
                     return getter
 
-        return self._feature.__getattribute__(name)
+        if name in self._attribute_white_list:
+            return getattr(self._feature, name)
+
+        raise AttributeError()
+
+    def feature(self):
+        """Return ModelAPI_Feature."""
+        return self._feature
+
+    def _fill_attribute(self, attribute, value):
+        """Fill ModelAPI_Attribute* with value."""
+        tools.fill_attribute(attribute, value)
 
     def setRealInput(self, inputid, value):
         self._feature.data().real(inputid).setValue(value)