Add construction plugin without tests.

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

diff --git a/src/PythonAPI/model/tools.py b/src/PythonAPI/model/tools.py
index 35bc3195bae573e32968bd073fcfb8773983d677..eb6243e8214cb5d711c7c11c648dcd49ac4bb959 100644 (file)
--- a/src/PythonAPI/model/tools.py
+++ b/src/PythonAPI/model/tools.py
@@ -4,16 +4,23 @@ Copyright (C) 2014-20xx CEA/DEN, EDF R&D
 """
 
 import re
+import collections
+
 import ModelAPI
+import GeomAPI
+
+# from .sketcher.sketch import Sketch
+
 
 def convert_to_underscore(name):
     """Convert CamelCase to underscore_case."""
     s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
     return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1).lower()
 
+
 def get_value(attribute):
     """Return simple type value from ModelAPI attribute if possible.
-    
+
     ModelAPI_AttributeDouble -> double
     ModelAPI_AttributeString -> str
     """
@@ -23,3 +30,62 @@ def get_value(attribute):
 
     print type(attribute)
     return attribute
+
+
+class Selection:
+    """Class for storing selection."""
+
+    def __init__(self, *args):
+        """Initialize selection."""
+        if not args:
+            self.args = (None, None)
+            return
+
+        if len(args) == 1 and isinstance(args[0], basestring):
+            self.args = args
+            return
+
+        assert(len(args) > 1 and len(args) < 4)
+        assert(isinstance(args[0], basestring) or
+               isinstance(args[0], ModelAPI.ModelAPI_Result))
+        if isinstance(args[0], basestring):
+            assert(isinstance(args[1], basestring))
+        elif isinstance(args[0], ModelAPI.ModelAPI_Result) or args[0] is None:
+            assert(isinstance(args[1], GeomAPI.GeomAPI_Shape))
+        self.args = args
+
+
+def fill_attribute(attribute, value):
+    """Set value to attribure.
+
+    This function processes complex cases.
+    ModelAPI_AttributeSelectionList can accept string, result + shape, list of
+    strings and [result + shape]...
+    ModelAPI_AttributeDouble can accept float and string
+    """
+    if isinstance(attribute, ModelAPI.ModelAPI_AttributeSelectionList):
+        attribute.clear()
+        if not value:
+            return
+
+        assert(isinstance(value, collections.Iterable))
+        for item in value:
+            assert(isinstance(item, Selection))
+            attribute.append(*item.args)
+
+    elif isinstance(attribute, ModelAPI.ModelAPI_AttributeSelection):
+        if value is None:
+            attribute.setValue(None, None)
+            return
+
+        assert(isinstance(value, Selection))
+        attribute.setValue(*value.args)
+
+    elif isinstance(attribute, ModelAPI.ModelAPI_AttributeDouble):
+        if isinstance(value, basestring):
+            attribute.setText(value)
+        else:
+            attribute.setValue(value)
+
+    else:
+        raise AssertionError("Wrong attribute type: %s" % type(attribute))