diff --git a/src/Mod/Arch/ArchComponent.py b/src/Mod/Arch/ArchComponent.py index ec2cb81354..2c26760ac1 100644 --- a/src/Mod/Arch/ArchComponent.py +++ b/src/Mod/Arch/ArchComponent.py @@ -215,13 +215,13 @@ class ComponentTaskPanel: def getIcon(self,obj): if hasattr(obj.ViewObject,"Proxy"): - return QtGui.QIcon(obj.ViewObject.Proxy.getIcon()) - elif obj.isDerivedFrom("Sketcher::SketchObject"): + if hasattr(obj.ViewObject.Proxy,"getIcon"): + return QtGui.QIcon(obj.ViewObject.Proxy.getIcon()) + if obj.isDerivedFrom("Sketcher::SketchObject"): return QtGui.QIcon(":/icons/Sketcher_Sketch.svg") - elif obj.isDerivedFrom("App::DocumentObjectGroup"): + if obj.isDerivedFrom("App::DocumentObjectGroup"): return QtGui.QApplication.style().standardIcon(QtGui.QStyle.SP_DirIcon) - else: - return QtGui.QIcon(":/icons/Tree_Part.svg") + return QtGui.QIcon(":/icons/Tree_Part.svg") def update(self): 'fills the treewidget'