put gears into container if a container is active

looooo · Jun 20, 2020 · 0a19329 · 0a19329
1 parent 23bd463
commit 0a19329
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/freecad/gears/commands.py b/freecad/gears/commands.py
@@ -52,6 +52,16 @@ def create(cls):
         obj = FreeCAD.ActiveDocument.addObject("Part::FeaturePython", cls.NAME)
         cls.GEAR_FUNCTION(obj)
         ViewProviderGear(obj.ViewObject)
+
+        # borrowed from threaded profiles
+        # puts the gear into an active container
+        body = Gui.ActiveDocument.ActiveView.getActiveObject("pdbody")
+        part = Gui.ActiveDocument.ActiveView.getActiveObject("part")
+
+        if body:
+            body.Group += [obj]
+        elif part:
+            part.Group += [obj]
         return obj
 
     def GetResources(self):