def populateClassItem(self, parentItem, repopulate): " Populates a class item " parentItem.populated = True # Count the number of items count = len( parentItem.sourceObj.decorators ) + \ len( parentItem.sourceObj.functions ) if parentItem.sourceObj.classes: count += 1 if parentItem.sourceObj.classAttributes: count += 1 if parentItem.sourceObj.instanceAttributes: count += 1 if count == 0: return # Insert rows if repopulate: self.beginInsertRows( self.createIndex(parentItem.row(), 0, parentItem), 0, count - 1) for item in parentItem.sourceObj.decorators: node = TreeViewDecoratorItem(parentItem, item) if parentItem.columnCount() > 1: node.appendData([parentItem.data(1), item.line]) node.setPath(self.findParentPath(parentItem)) self._addItem(node, parentItem) for item in parentItem.sourceObj.functions: node = TreeViewFunctionItem(parentItem, item) if parentItem.columnCount() > 1: node.appendData([parentItem.data(1), item.line]) node.setPath(self.findParentPath(parentItem)) self._addItem(node, parentItem) if parentItem.sourceObj.classes: node = TreeViewClassesItem(parentItem, parentItem.sourceObj) if parentItem.columnCount() > 1: node.appendData(["", ""]) self._addItem(node, parentItem) if parentItem.sourceObj.classAttributes: node = TreeViewStaticAttributesItem(parentItem) if parentItem.columnCount() > 1: node.appendData(["", ""]) self._addItem(node, parentItem) if parentItem.sourceObj.instanceAttributes: node = TreeViewInstanceAttributesItem(parentItem) if parentItem.columnCount() > 1: node.appendData(["", ""]) self._addItem(node, parentItem) if repopulate: self.endInsertRows() return
def updateSingleFuncItem(self, treeItem, funcObj): " Updates single function item " if not treeItem.populated: return # It may have decor, classes and other functions existingDecors = [] hadFunctions = False hadClasses = False itemsToRemove = [] for funcChildItem in treeItem.childItems: if funcChildItem.itemType == DecoratorItemType: name = funcChildItem.sourceObj.name found = False for decor in funcObj.decorators: if decor.name == name: found = True existingDecors.append(name) if cmpDecoratorDisplayName(funcChildItem.sourceObj, decor): funcChildItem.updateData(decor) funcChildItem.setData(2, decor.line) else: # Appearence changed funcChildItem.updateData(decor) funcChildItem.setData(2, decor.line) self.signalItemUpdated(funcChildItem) break if not found: itemsToRemove.append(funcChildItem) continue elif funcChildItem.itemType == FunctionsItemType: hadFunctions = True if not funcObj.functions: itemsToRemove.append(funcChildItem) else: funcChildItem.updateData(funcObj) self.updateFunctionsItem(funcChildItem, funcObj.functions) continue elif funcChildItem.itemType == ClassesItemType: hadClasses = True if not funcObj.classes: itemsToRemove.append(funcChildItem) else: funcChildItem.updateData(funcObj) self.updateClassesItem(funcChildItem, funcObj.classes) continue for item in itemsToRemove: self.removeTreeItem(item) # Add those which have been introduced for decor in funcObj.decorators: if decor.name not in existingDecors: newItem = TreeViewDecoratorItem(treeItem, decor) if treeItem.columnCount() > 1: newItem.appendData([treeItem.data(1), decor.line]) newItem.setPath(self.findParentPath(treeItem)) self.addTreeItem(treeItem, newItem) if not hadFunctions and funcObj.functions: newItem = TreeViewFunctionsItem(treeItem, funcObj) if treeItem.columnCount() > 1: newItem.appendData(["", ""]) self.addTreeItem(treeItem, newItem) if not hadClasses and funcObj.classes: newItem = TreeViewClassesItem(treeItem, funcObj) if treeItem.columnCount() > 1: newItem.appendData(["", ""]) self.addTreeItem(treeItem, newItem) return
def updateSingleClassItem(self, treeItem, classObj): " Updates single class item " if not treeItem.populated: return # There might be decorators, classes, methods, static attributes and # instance attributes existingDecors = [] existingMethods = [] hadStaticAttributes = False hadInstanceAttributes = False hadClasses = False itemsToRemove = [] for classChildItem in treeItem.childItems: if classChildItem.itemType == DecoratorItemType: name = classChildItem.sourceObj.name found = False for decor in classObj.decorators: if decor.name == name: found = True existingDecors.append(name) if cmpDecoratorDisplayName(classChildItem.sourceObj, decor): classChildItem.updateData(decor) classChildItem.setData(2, decor.line) else: # Appearence changed classChildItem.updateData(decor) classChildItem.setData(2, decor.line) self.signalItemUpdated(classChildItem) break if not found: itemsToRemove.append(classChildItem) continue elif classChildItem.itemType == ClassesItemType: hadClasses = True if not classObj.classes: itemsToRemove.append(classChildItem) else: classChildItem.updateData(classObj) self.updateClassesItem(classChildItem, classObj.classes) continue elif classChildItem.itemType == FunctionItemType: name = classChildItem.sourceObj.name found = False for method in classObj.functions: if method.name == name: found = True existingMethods.append(name) if cmpFunctionDisplayName(classChildItem.sourceObj, method): classChildItem.updateData(method) classChildItem.setData(2, method.line) else: # Appearence changed classChildItem.updateData(method) classChildItem.setData(2, method.line) self.signalItemUpdated(classChildItem) self.updateSingleFuncItem(classChildItem, method) break if not found: itemsToRemove.append(classChildItem) continue elif classChildItem.itemType == StaticAttributesItemType: hadStaticAttributes = True if not classObj.classAttributes: itemsToRemove.append(classChildItem) else: self.updateAttrItem(classChildItem, classObj.classAttributes) continue elif classChildItem.itemType == InstanceAttributesItemType: hadInstanceAttributes = True if not classObj.instanceAttributes: itemsToRemove.append(classChildItem) else: self.updateAttrItem(classChildItem, classObj.instanceAttributes) continue for item in itemsToRemove: self.removeTreeItem(item) # Add those which have been introduced for decor in classObj.decorators: if decor.name not in existingDecors: newItem = TreeViewDecoratorItem(treeItem, decor) if treeItem.columnCount() > 1: newItem.appendData([treeItem.data(1), decor.line]) newItem.setPath(self.findParentPath(treeItem)) self.addTreeItem(treeItem, newItem) for method in classObj.functions: if method.name not in existingMethods: newItem = TreeViewFunctionItem(treeItem, method) if treeItem.columnCount() > 1: newItem.appendData([treeItem.data(1), method.line]) newItem.setPath(self.findParentPath(treeItem)) self.addTreeItem(treeItem, newItem) if not hadClasses and classObj.classes: newItem = TreeViewClassesItem(treeItem, classObj) if treeItem.columnCount() > 1: newItem.appendData(["", ""]) self.addTreeItem(treeItem, newItem) if not hadStaticAttributes and \ classObj.classAttributes: newItem = TreeViewStaticAttributesItem(treeItem) if treeItem.columnCount() > 1: newItem.appendData(["", ""]) self.addTreeItem(treeItem, newItem) if not hadInstanceAttributes and \ classObj.instanceAttributes: newItem = TreeViewInstanceAttributesItem(treeItem) if treeItem.columnCount() > 1: newItem.appendData(["", ""]) self.addTreeItem(treeItem, newItem) return