Salome HOME
Issue #1774: Can't select edge for tangent constraint creation
[modules/shaper.git] / src / PythonAddons / addons_Features.py
index 00c72b0274ebaacd5c10dffc49b7726d7517f28d..9499f21134c5aefef059bad27012e07f1a15ea45 100644 (file)
@@ -2,32 +2,38 @@
 """
 
 import ModelAPI
-from macros.box.feature      import BoxFeature
-from macros.cylinder.feature import CylinderFeature
+from macros.box.feature       import BoxFeature
+from macros.rectangle.feature import SketchPlugin_Rectangle
 
 
 class PythonFeaturesPlugin(ModelAPI.ModelAPI_Plugin):
+    """Implementation of features plugin.
+
+    PythonFeaturesPlugin() -> plugin object
+    """
 
     def __init__(self):
+        """x.__init__(...) initializes x; see x.__class__.__doc__ for signature"""
         ModelAPI.ModelAPI_Plugin.__init__(self)
         aSession = ModelAPI.ModelAPI_Session.get()
         aSession.registerPlugin(self)
         pass
 
     def createFeature(self, theFeatureID):
+        """Override ModelAPI_Plugin.createFeature()"""
         aFeature = None
 
         if theFeatureID == BoxFeature.ID():
             aFeature = BoxFeature().__disown__()
 
-        elif theFeatureID == CylinderFeature.ID():
-            aFeature = CylinderFeature().__disown__()
+        elif theFeatureID == SketchPlugin_Rectangle.ID():
+            aFeature = SketchPlugin_Rectangle().__disown__()
 
         else:
             raise StandardError("No such feature %s" % theFeatureID)
 
         return aFeature
 
-
+# The plugin object
 plugin = PythonFeaturesPlugin()
 plugin.__disown__()