fixed bug when a TechDraw view contains a variant link

2 years ago · bae89072be
1 changed files with 3 additions and 0 deletions
--- a/ahb_techdraw_extensions.py
+++ b/ahb_techdraw_extensions.py
@ -419,6 +419,9 @@ class TechDrawExtensions:
 		if obj.TypeId == 'App::Link':
 			partLink = obj
 			objectCenterWorld = partLink.LinkPlacement.Matrix.multiply(partLink.LinkedObject.Shape.CenterOfGravity)
+		elif obj.TypeId == 'Part::FeaturePython' and hasattr(obj, 'LinkedObject'): # variant link
+			partLink = obj
+			objectCenterWorld = partLink.LinkedObject.Shape.CenterOfGravity
 		else:
 			objectCenterWorld = obj.Shape.CenterOfGravity