def updateTree(self , directory): imageFiles = glob.glob(directory+'/*.png') imageFiles += glob.glob(directory+'/*.gif') imageFiles += glob.glob(directory+'/*.bmp') imageFiles += glob.glob(directory+'/*.tif') imageFiles += glob.glob(directory+'/*.jpg') # Check if the image files have a message in them trueDic = {} trueList = [] falseList = [] for file in imageFiles: s = NewSteganography(imagePath=file ,direction='horizontal') returnValue = s.checkIfMessageExists()[0] fileList = file.split('/') fileName = fileList[len(fileList) - 1] if returnValue is True: trueList.append(fileName) trueDic[fileName] = s.checkIfMessageExists()[1] else: falseList.append(fileName) # print(file + ' -> ' + str(returnValue)) if debug&0: print(trueDic) # Disable the unnecessary groups self.grpMedium.setDisabled(True) self.grpMessage.setDisabled(True) # IMPLEMENTING/ POPULATING THE TREE WIDGET for file in falseList: item = QTreeWidgetItem([file]) item.setForeground(0,QColor('blue')) self.fileTreeWidget.addTopLevelItem(item) for file in trueDic: parentItem = QTreeWidgetItem([file]) parentItem.setForeground(0,QColor('red')) font = QFont() font.setBold(True) parentItem.setFont(0,font) self.fileTreeWidget.addTopLevelItem(parentItem) childItem = QTreeWidgetItem([trueDic[file]]) childItem.setForeground(0,QColor('green')) parentItem.insertChild(0, childItem) self.trueDic = trueDic self.falseList = falseList self.fileTreeWidget.expandAll() self.fileTreeWidget.show()