def loadControllers(self): self.treeWidget.clear() sels = pymel.core.ls( sl=1 ) currentParent = self.treeWidget self.items = [] for sel in sels: widgetItem = QTreeWidgetItem( currentParent ) widgetItem.realObject = sel widgetItem.setText( 0, sel.name() ) if isinstance( currentParent, QTreeWidgetItem ): currentParent.setExpanded( True ) currentParent = widgetItem self.items.append( sel.name() ) self.saveInfo()
def loadInfo(self): data = BaseCommands.readData( self.uiInfoPath ) if not data.has_key( 'controllers' ): return None controllerList = data[ 'controllers' ] controllers = [] for controller in controllerList: if not pymel.core.ls( controller ): continue controllers.append( pymel.core.ls( controller )[0] ) self.treeWidget.clear() currentParent = self.treeWidget self.items = [] for sel in controllers: widgetItem = QTreeWidgetItem( currentParent ) widgetItem.realObject = sel widgetItem.setText( 0, sel.name() ) if isinstance( currentParent, QTreeWidgetItem ): currentParent.setExpanded( True ) currentParent = widgetItem self.items.append( sel.name() )