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()
