import ModelAPI
-from .tools import get_value, convert_to_underscore
+from model import tools
class Feature(ModelAPI.ModelAPI_Feature):
"""Base class of user-defined Python features."""
def __init__(self):
+ """x.__init__(...) initializes x; see x.__class__.__doc__ for signature"""
ModelAPI.ModelAPI_Feature.__init__(self)
- def addRealInput (self, inputid):
+ def addRealInput(self, inputid):
+ """F.addRealInput(str) -- add real attribute"""
self.data().addAttribute(inputid,
ModelAPI.ModelAPI_AttributeDouble_typeId())
- def getRealInput (self, inputid):
+ def getRealInput(self, inputid):
+ """F.getRealInput(str) -- get real value of the attribute"""
return self.data().real(inputid).value()
- def addResult (self, result):
+ def addResult(self, result):
+ """F.addResult(ModelAPI_Result) -- add ModelAPI_Result shape as a result"""
shape = result.shape()
body = self.document().createBody(self.data())
body.store(shape)
class Interface():
- """Base class of hight level Python interfaces to features."""
+ """Base class of high level Python interfaces to features."""
def __init__(self, feature):
+ """x.__init__(...) initializes x; see x.__class__.__doc__ for signature"""
self._feature = feature
def __getattr__(self, name):
"""Process missing attributes.
Add get*() methods for access feature attributes.
- Redirect 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)
+ raise AttributeError()
- def setRealInput (self, inputid, value):
+ def _fillAttribute(self, attribute, value):
+ """Fill ModelAPI_Attribute* with value."""
+ tools.fill_attribute(attribute, value)
+
+ def feature(self):
+ """Return ModelAPI_Feature."""
+ return self._feature
+
+ def getKind(self):
+ """Return the unique kind of the feature"""
+ return self._feature.getKind()
+
+ def results(self):
+ """Return current results of the feature"""
+ return self._feature.results()
+
+ def firstResult(self):
+ """Return the first result in the list of results"""
+ return self._feature.firstResult()
+
+ def lastResult(self):
+ """Return the last result in the list of results"""
+ return self._feature.lastResult()
+
+ def setRealInput(self, inputid, value):
+ """I.setRealInput(str, float) -- set real value to the attribute"""
self._feature.data().real(inputid).setValue(value)
- def areInputValid (self):
+ def areInputValid(self):
+ """I.areInputValid() -> True or False validation result"""
validators = ModelAPI.ModelAPI_Session.get().validators()
return validators.validate(self._feature)
- def execute (self):
- self._feature.execute()
+ def execute(self):
+ """I.execute() -- validate and execute the feature.
+
+ Raises RuntimeError if validation fails.
+ """
+ if self.areInputValid():
+ self._feature.execute()
+ else:
+ raise RuntimeError("Can not execute %s: %s" %
+ (self._feature.getKind(), self._feature.error())
+ )